tamakyi
e3e59e7c69
Repo .git was damaged (refs/ missing, index truncated, object store near-empty). Repaired via git init (rebuilt refs), stale index preserved as .git/index.corrupt-20260910.bak, then full worktree committed as a rollback baseline.
62 lines
1.3 KiB
Go
62 lines
1.3 KiB
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/rs/xid"
|
|
"github.com/urfave/cli/v2"
|
|
|
|
"github.com/tamakyi/TamaBox/internal/conf"
|
|
"github.com/tamakyi/TamaBox/internal/db"
|
|
)
|
|
|
|
var Uid = &cli.Command{
|
|
Name: "uid",
|
|
Usage: "Create uid for user",
|
|
Action: runUid,
|
|
}
|
|
|
|
func runUid(ctx *cli.Context) error {
|
|
if err := conf.Init(); err != nil {
|
|
return errors.Wrap(err, "load configuration")
|
|
}
|
|
|
|
dbType := conf.Database.Type
|
|
if dbType == "postgres" {
|
|
conf.Database.DSN = conf.PostgresDsn()
|
|
} else {
|
|
conf.Database.DSN = conf.MySQLDsn()
|
|
}
|
|
|
|
database, err := db.Init(dbType, conf.Database.DSN)
|
|
if err != nil {
|
|
return errors.Wrap(err, "connect to database")
|
|
}
|
|
|
|
var users []*db.User
|
|
if err := database.WithContext(ctx.Context).Unscoped().Find(&users).Error; err != nil {
|
|
return errors.Wrap(err, "query users")
|
|
}
|
|
|
|
for idx, user := range users {
|
|
user := user
|
|
if user.UID != "" {
|
|
continue
|
|
}
|
|
|
|
if idx%1000 == 0 {
|
|
fmt.Printf("Processing user %d/%d\n", idx, len(users))
|
|
}
|
|
|
|
uid := xid.NewWithTime(user.CreatedAt).String()
|
|
if err := database.WithContext(ctx.Context).Unscoped().
|
|
Model(&db.User{}).
|
|
Omit("updated_at").
|
|
Where("id = ?", user.ID).UpdateColumn("uid", uid).Error; err != nil {
|
|
return errors.Wrap(err, "update uid")
|
|
}
|
|
}
|
|
return nil
|
|
}
|