import wflixu/Orion.{schema, pk_auto_inc, required, unique_required}

let user_schema = schema("users")
  |> add_int_field("id", pk_auto_inc())
  |> add_string_field("name", required())
  |> add_string_field("email", unique_required())
  |> add_int_field("age", [])
  |> build_schema()

// generated/users.mbt
// @generated by Orion v0.2.0

pub struct Users {
  id: Int,
  name: String,
  email: String,
  age: Int
} derive(Eq, Hash, FromJson, ToJson, Show)

pub struct UsersMapper { db: @runtime.Db }

pub fn create_users(self: UsersMapper, name: String, email: String, age: Int) -> Result[Int, @runtime.DbError]
pub fn find_users_by_id(self: UsersMapper, id: Int) -> Result[Option[Users], @runtime.DbError]
pub fn find_all_users(self: UsersMapper) -> Result[List[Users], @runtime.DbError]

// Query Builder
pub fn new_users_query(db: @runtime.Db) -> UsersQuery
pub fn UsersQuery::where_name(self, value: String) -> Self
pub fn UsersQuery::execute(self) -> Result[List[Users], @runtime.DbError]

import wflixu/Orion

fn main {
  // 连接数据库
  let db = Orion.connect({
    url: "sqlite://app.db",
    pool_size: 10
  })
  
  // 创建 Mapper
  let user_mapper = UserMapper::new(db)
  
  // 查询用户
  match user_mapper.get_user_by_id(1) {
    Some(user) => println("User: \(user.name)")
    None => println("User not found")
  }
  
  // 创建用户
  let user_id = user_mapper.create_user("Alice", 25)?
  
  // 启用日志
  Orion.enable_log(db, Debug)
}

// 基本连接
let db = Orion.connect({
  url: "sqlite://app.db"
})

// 连接池配置
let db = Orion.connect({
  url: "sqlite://app.db",
  pool_size: 10,
  query_timeout: 5000  // 毫秒
})

// 启用日志
Orion.enable_log(db, level: Debug)

// 关闭连接
Orion.close(db)

Orion.transaction(db, fn(tx) {
  let user_id = UserMapper.create(tx, "Alice", 25)?
  OrderMapper.create(tx, user_id, 100)?
  Ok(unit)
})

///|
enum DbError {
  NotFound // 未找到记录
  UniqueViolation // 唯一约束违反
  ConnectionError // 连接错误
  QueryError // 查询错误
  Timeout // 超时
}