interface User : Entity<User> {
    companion object : Entity.Factory<User>()

    val id: Int
    var username: String
    var fullName: String
    var email: String
    var passwordHash: String
}

object Users : Table<User>("users") {
    val id by int("id").primaryKey().bindTo { it.id }
    val username by varchar("username").bindTo { it.username }
    val fullName by varchar("full_name").bindTo { it.fullName }
    val email by varchar("email").bindTo { it.email }
    val passwordHash by varchar("password_hash").bindTo { it.passwordHash }
}

interface Project : Entity<Project> {
    companion object : Entity.Factory<Project>()

    val id: Int
    var name: String
    var description: String
    var owner: User
}

object Projects : Table<Project>("projects") {
    val id by int("id").primaryKey().bindTo { it.id }
    val name by varchar("name").bindTo { it.name }
    val description by text("description").bindTo { it.description }
    val projectOwnerId by int("project_owner_id").references(Users) { it.owner }
}

interface ProjectMembershipMapping : Entity<ProjectMembershipMapping> {
    companion object : Entity.Factory<ProjectMembershipMapping>()

    val id: Int
    var user: User
    var project: Project
}

object ProjectMembership : Table<ProjectMembershipMapping>("project_membership") {
    val id by int("id").primaryKey().bindTo { it.id }
    val userId by int("user_id").references(Users) { it.user }
    val projectId by int("project_id").references(Projects) { it.project }
}

interface ComponentSpecificationEntity : Entity<ComponentSpecificationEntity> {
    companion object : Entity.Factory<ComponentSpecificationEntity>()

    val id: Int
    var name: String
    var type: ComponentType
    var project: Project
}

object ComponentSpecifications : Table<ComponentSpecificationEntity>("component_specifications") {
    val id by int("id").primaryKey().bindTo { it.id }
    val name by varchar("name").bindTo { it.name }
    val type by componentType("type").bindTo { it.type }
    val projectId by int("project_id").references(Projects) { it.project }
}

interface VariableEntity : Entity<VariableEntity> {
    companion object : Entity.Factory<VariableEntity>()

    val id: Int
    var name: String
    var type: VariableType
    var unit: String
    var dimension: String
    var objective: Objective
    var exogenous: Boolean
    var systemDescriptive: Boolean
    var project: Project
}

object Variables : Table<VariableEntity>("variables") {
    val id by int("id").primaryKey().bindTo { it.id }
    val name by varchar("name").bindTo { it.name }
    val type by variableType("type").bindTo { it.type }
    val unit by varchar("unit").bindTo { it.unit }
    val dimension by varchar("dimension").bindTo { it.dimension }
    val objective by objective("objective").bindTo { it.objective }
    val exogenous by boolean("exogenous").bindTo { it.exogenous }
    val systemDescriptive by boolean("system_descriptive").bindTo { it.systemDescriptive }
    val projectId by int("project_id").references(Projects) { it.project }
}

interface ComponentSpecificationInternalVariableMapping : Entity<ComponentSpecificationInternalVariableMapping> {
    companion object : Entity.Factory<ComponentSpecificationInternalVariableMapping>()

    val id: Int
    var component: ComponentSpecificationEntity
    var variable: VariableEntity
}

object ComponentSpecificationInternalVariables :
    Table<ComponentSpecificationInternalVariableMapping>("component_specification_internal_variables") {
    val id by int("id").primaryKey().bindTo { it.id }
    val componentId by int("component_id").references(ComponentSpecifications) { it.component }
    val variableId by int("variable_id").references(Variables) { it.variable }
}

interface ComponentSpecificationAdditionalInternalVariableMapping :
    Entity<ComponentSpecificationAdditionalInternalVariableMapping> {
    companion object : Entity.Factory<ComponentSpecificationAdditionalInternalVariableMapping>()

    val id: Int
    var component: ComponentSpecificationEntity
    var variable: VariableEntity
}

object ComponentSpecificationAdditionalInternalVariables :
    Table<ComponentSpecificationAdditionalInternalVariableMapping>("component_specification_additional_internal_variables") {
    val id by int("id").primaryKey().bindTo { it.id }
    val componentId by int("component_id").references(ComponentSpecifications) { it.component }
    val variableId by int("variable_id").references(Variables) { it.variable }
}

database.useTransaction {
  database.sequenceOf(Users).add(
    User {
      username = "testuser"
      fullName = "Test Tester"
      email = "[email protected]"
      passwordHash = passwordHashForTestUsers
    }
  )

  val ownerUser: User =
    database.sequenceOf(Users).find { it.username eq "testuser" }
      ?: fail("Couldn't find recently created user")

  database.sequenceOf(Projects).add(
    Project {
      name = "Test Project"
      description = "Some description"
      owner = ownerUser
    }
  )

  val project: Project =
    database.sequenceOf(Projects).find { it.name eq "Test Project" }
      ?: fail("Couldn't find recently created project")

  database.sequenceOf(Variables).add(
    VariableEntity {
      name = "T1"
      type = VariableType.EFFORT
      unit = "T^2"
      dimension = "T.T"
      objective = Objective.NONE
      exogenous = false
      systemDescriptive = false
      this.project = project
    }
  )

  val variable: VariableEntity =
    database.sequenceOf(Variables).find { it.name eq "T1" }
      ?: fail("Couldn't find recently created variable")

  database.sequenceOf(ComponentSpecifications).add(
    ComponentSpecificationEntity {
      name = "SOE"
      type = ComponentType.SOURCE_OF_EFFORT
      this.project = project
    }
  )
  database.sequenceOf(ComponentSpecifications).add(
    ComponentSpecificationEntity {
      name = "SNK"
      type = ComponentType.SINK
      this.project = project
    }
  )

  val sourceOfEffortComponent: ComponentSpecificationEntity =
    database.sequenceOf(ComponentSpecifications).find { it.name eq "SOE" }
      ?: fail("Couldn't find recently created component specification")
  val sinkComponent: ComponentSpecificationEntity =
    database.sequenceOf(ComponentSpecifications).find { it.name eq "SNK" }
      ?: fail("Couldn't find recently created component specification")

  database.sequenceOf(ComponentSpecificationInternalVariables).add(
    ComponentSpecificationInternalVariableMapping {
      component = sourceOfEffortComponent
      this.variable = variable
    }
  )
  database.sequenceOf(ComponentSpecificationAdditionalInternalVariables).add(
    ComponentSpecificationAdditionalInternalVariableMapping {
      component = sinkComponent
      this.variable = variable
    }
  )

  val internalVariableMapping: ComponentSpecificationInternalVariableMapping =
    database
      .sequenceOf(ComponentSpecificationInternalVariables)
      .find { it.componentId eq sourceOfEffortComponent.id }
      ?: fail("Couldn't find recently created internal variable mapping")
  val additionalInternalVariableMapping: ComponentSpecificationAdditionalInternalVariableMapping =
    database
      .sequenceOf(ComponentSpecificationAdditionalInternalVariables)
      .find { it.componentId eq sinkComponent.id }
      ?: fail("Couldn't find recently created additional internal variable mapping")

  println("Internal variable name: ${internalVariableMapping.variable.name}")
  println("Additional internal variable name: ${additionalInternalVariableMapping.variable.name}")
}