// @TABLE(
//   name="User",
//   ux=[{name="UxName", fields=["Name"]}],
//   kw=["Name","Email"]
// )
type User struct {
    common.ImmutableTable
    OrgID int64  `db:"org_id" json:"org_id"`
    Name  string `db:"name" json:"name"`
    Email string `db:"email" json:"email"`
}

// @TABLE(
//   name="Order",
//   idx=[{name="IdxUserItem", fields=["UserID","ItemID"]}, {name="IdxItem", fields=["ItemID"]}],
//   ct
// )
type Order struct {
    common.MutableTable
    UserID int64 `db:"user_id"`
    ItemID int64 `db:"item_id"`
    Amount int64 `db:"amount"`
    Price  int64 `db:"price"`
    Status OrderStatus `db:"status"`
}

// @TABLE(
//   ux=[{fields=["Name"]}],
//   idx=[{name="IdxCategory", fields=["CategoryID"]}],
//   kw=["Name"]
// )
type Item struct {
    common.ImmutableTable
    CategoryID int64 `db:"category_id"`
    Name       string `db:"name,size:200"`
    Price      int64  `db:"price"`
}

// @TABLE(
//   ux=[{fields=["Name"]}],
//   kw=["Name","Description"]
// )
type Category struct {
    common.ImmutableTable
    CategoryContent
}
type CategoryContent struct {
    Type        CategoryType `db:"type" json:"type"`
    Name        string       `db:"name,size:200" json:"name"`
    Description string       `db:"description,size:4096" json:"description"`
}

// @TABLE(
//   name="Org",
//   ux=[{name="UxName", fields=["Name"]}]
// )
type Org struct {
    common.ImmutableTable
    Name string `db:"name"`
}

package main

import (
    "context"
    "database/sql"
    "encoding/json"
    "fmt"
    "os"
    "github.com/juju/errors"
    _ "github.com/mattn/go-sqlite3"
    logrus "log/slog"
    "github.com/tmoeish/tsq"
    "github.com/tmoeish/tsq/examples/database"
    "gopkg.in/gorp.v2"
)

func main() {
    logrus.SetLevel(logrus.TraceLevel)

    // 1. 连接 SQLite 内存数据库
    db, err := sql.Open("sqlite3", ":memory:")
    if err != nil {
        logrus.Fatal(errors.ErrorStack(err))
    }
    defer func() { _ = db.Close() }()

    // 2. 初始化 gorp
    dbmap := &gorp.DbMap{Db: db, Dialect: gorp.SqliteDialect{}}
    err = tsq.Init(dbmap, true, TraceDB)
    if err != nil {
        logrus.Fatal(errors.ErrorStack(err))
    }

    // 初始化数据库，执行 mock.sql 文件
    mockSQL, err := os.ReadFile("examples/database/mock.sql")
    if err != nil {
        logrus.Fatal(errors.ErrorStack(err))
    }
    _, err = db.Exec(string(mockSQL))
    if err != nil {
        logrus.Fatal(errors.ErrorStack(err))
    }

    // 3. 构造分页参数
    pageReq := &tsq.PageReq{
        Page:    1,
        Size:    10,
        Order:   "asc,desc",
        OrderBy: "user_id,order_id",
    }

    // 4. 调用 PageUserOrder，假设 user_id = 1
    ctx := context.Background()
    resp, err := database.PageUserOrder(ctx, dbmap, pageReq, 1, "图书", "视频", "杂志")
    if err != nil {
        logrus.Fatal(errors.ErrorStack(err))
    }

    // 5. 打印结果
    rs, _ := json.MarshalIndent(resp, "", "  ")
    fmt.Println(string(rs))
}

func TraceDB(next tsq.Fn) tsq.Fn {
    return func(ctx context.Context) error {
        err := next(ctx)
        return err
    }
}

// @DTO(
//   name="UserOrder"
// )
type UserOrder struct {
    UserID    int64  `json:"user_id"    tsq:"User.ID"`
    UserName  string `json:"user_name"  tsq:"User.Name"`
    UserEmail string `json:"user_email" tsq:"User.Email"`
    OrgName   string `json:"org_name"   tsq:"Org.Name"`
    OrderID     int         `json:"order_id"     tsq:"Order.ID"`
    OrderAmount float64     `json:"order_amount" tsq:"Order.Amount"`
    OrderPrice  float64     `json:"order_price"  tsq:"Order.Price"`
    OrderStatus OrderStatus `json:"order_status" tsq:"Order.Status"`
    OrderTime   string      `json:"order_time"   tsq:"Order.CT"`
    ItemID    int64  `json:"item_id"    tsq:"Item.ID"`
    ItemName  string `json:"item_name"  tsq:"Item.Name"`
    ItemPrice int64  `json:"item_price" tsq:"Item.Price"`
    ItemCategory string `json:"item_category" tsq:"Category.Name"`
}

// 复杂分页查询
resp, err := database.PageUserOrder(ctx, dbmap, pageReq, 1, "图书", "视频", "杂志")

// 条件查询
query := tsq.
    Select(database.TableUser.Cols()...).
    Where(database.User_Name.Like("%admin%")).
    OrderBy(database.User_ID.Desc()).
    MustBuild()
users, err := database.ListUserByQuery(ctx, dbmap, query)

// 分页查询
pageReq := &tsq.PageReq{
    Page:    1,
    Size:    10,
    Order:   "asc,desc",
    OrderBy: "user_id,order_id",
}
resp, err := database.PageUserOrder(ctx, dbmap, pageReq, 1, "图书", "视频", "杂志")

users := []*database.User{
    {OrgID: 1, Name: "张三", Email: "[email protected]"},
    {OrgID: 1, Name: "李四", Email: "[email protected]"},
}
err := tsq.BatchInsert(ctx, dbmap, users)

mockSQL, err := os.ReadFile("examples/database/mock.sql")
if err != nil {
    logrus.Fatal(errors.ErrorStack(err))
}
_, err = db.Exec(string(mockSQL))
if err != nil {
    logrus.Fatal(errors.ErrorStack(err))
}