addSbtPlugin("io.github.sbt-dao-generator" % "sbt-dao-generator" % "1.3.0")

// JDBC Driver Class Name (required)
generator / driverClassName := "org.h2.Driver"

// JDBC URL (required)
generator / jdbcUrl := "jdbc:h2:file:./target/test"

// JDBC User Name (required)
generator / jdbcUser := "sa"

// JDBC Password (required)
generator / jdbcPassword := ""

// The Function that convert The Column Type Name to Property Type Name (Optional)
generator / propertyTypeNameMapper := {
  case "INTEGER" => "Int"
  case "VARCHAR" => "String"
  case "BOOLEAN" => "Boolean"
  case "DATE" | "TIMESTAMP" => "java.util.Date"
  case "DECIMAL" => "BigDecimal"
}

// More flexible `PropertyTypeNameMapper` (Optional)
// NOTE: Currently this is ignored when the `propertyTypeNameMapper` is defined since it keeps compatibility.
// NOTE: We plan to rename this setting to `propertyTypeNameMapper` in the next major version.
generator / advancedPropertyTypeNameMapper := {
  case (_, TableDesc(tableName, _, _), ColumnDesc("id", _, _, _, _, _)) => s"${tableName}Id"
  case (_, _, ColumnDesc(_, _, _, _, _, Some(remarks))) => remarks
  case ("INTEGER", _, _) => "Int"
  case ("VARCHAR", _, _) => "String"
  case ("BOOLEAN", _, _) => "Boolean"
  case ("DATE" | "TIMESTAMP", _, _) => "java.util.Date"
  case ("DECIMAL", _, _) => "BigDecimal"
}

// Schema Name (Optional, Default is None)
generator / schemaName := None,

// The Function for filtering the table to be processed (Optional, default is the following)
generator / tableNameFilter := { tableName: String => tableName.toUpperCase != "SCHEMA_VERSION"}

// The Function for converting Table Name to Class Name (Optional, default is the following)
generator / classNameMapper := { tableName: String =>
    Seq(StringUtil.camelize(tableName))
}

// e.g.) If you want to specify multiple output files, you can configure it as follows.
/*
generator / classNameMapper := {
  case "DEPT" => Seq("Dept", "DeptSpec")
  case "EMP" => Seq("Emp", "EmpSpec")
}
*/

// The Function for converting Column Name to Property Name (Optional, default is the following)
generator / propertyNameMapper := { columnName: String =>
    StringUtil.decapitalize(StringUtil.camelize(columnName))
}

// The Function that decides which Template Name for Model Name (Optional, defaults below)
generator / templateNameMapper := { className: String => "template.ftl" },

// e.g.) If you want to specify different templates for the model and spec, you can configure it as follows.
/*
generator / templateNameMapper := {
  case className if className.endsWith("Spec") => "template_spec.ftl"
  case _ => "template.ftl"
}
*/

// The Directory where template files are placed (Optional, default is the following)
generator / templateDirectory := baseDirectory.value / "templates"

// The Directory where source code is output (Optional, default is the following)
generator / outputDirectoryMapper := { className: String => (Compile / sourceManaged).value }

// e.g.) You can change the output destination directory for each class name dynamically.
/*
generator / outputDirectoryMapper := { className: String =>
  className match {
    case s if s.endsWith("Spec") => (Test / sourceManaged).value
    case s => (Compile / sourceManaged).value
  }
}
*/

Compile / sourceGenerators += (generator / generateAll).value