Only generate tables for a single schema

SourceCodeGenerator.scala

import slick.codegen.SourceCodeGenerator
import slick.jdbc.JdbcProfile
import slick.model.Model

import scala.concurrent.{Await, ExecutionContext}
import scala.concurrent.duration.Duration

object SourceCodeGenerator {

  def run(slickDriver: String,
          jdbcDriver: String,
          url: String,
          outputDir: String,
          pkg: String,
          schema:String): Unit = {

    val driver: JdbcProfile =
      Class.forName(slickDriver + "$").getField("MODULE$").get(null).asInstanceOf[JdbcProfile]
    val dbFactory = driver.api.Database
    val db = dbFactory.forURL(url, driver = jdbcDriver, keepAliveConnection = true)
    try {
      val allSchemas = Await.result(db.run(
        driver.createModel(None, ignoreInvalidDefaults = false)(ExecutionContext.global).withPinnedSession), Duration.Inf)
      val publicSchema = new Model(allSchemas.tables.filter(x=> x.name.schema.contains(schema)), allSchemas.options)
      new SourceCodeGenerator(publicSchema).writeToFile(slickDriver, outputDir, pkg)
    } finally db.close
  }

  def main(args: Array[String]) = {
    //Handle parameter reading then call run
    ???
  }
}