package app import ( "fmt" "math/rand" "time" "github.com/veypi/OneAuth/cfg" "github.com/veypi/OneAuth/libs/auth" "github.com/veypi/OneAuth/libs/utils" "github.com/veypi/OneAuth/models" "github.com/vyes/vigo" "gorm.io/gorm" ) var _ = Router.Get("/:app_id/key", auth.Check("app", "app_id", auth.DoDelete), appKey) func appKey(x *vigo.X) (any, error) { id := x.Params.Get("app_id") if id == "" { return nil, vigo.ErrArgMissing.WithArgs("app_id") } data := &models.App{} data.ID = id key := utils.RandSeq(32) // err := cfg.DB().Where("id = ?", x.Params.GetStr("app_id")).First(data).Error err := cfg.DB().Model(data).Update("key", key).Error return key, err } var _ = Router.Delete("/:app_id", auth.Check("app", "app_id", auth.DoDelete), appDelete) func appDelete(x *vigo.X) (any, error) { data := &models.App{} err := cfg.DB().Where("id = ?", x.Params.Get("app_id")).Delete(data).Error return data, err } var _ = Router.Get("/:app_id", auth.Check("app", "app_id", auth.DoRead), appGet) func appGet(x *vigo.X) (any, error) { data := &models.App{} err := cfg.DB().Where("id = ?", x.Params.Get("app_id")).First(data).Error return data, err } type listOpts struct { Name *string `json:"name" parse:"query"` } var _ = Router.Get("/", appList) func appList(x *vigo.X) (any, error) { opts := &listOpts{} err := x.Parse(opts) if err != nil { return nil, err } data := make([]*struct { models.App UserStatus string `json:"user_status"` }, 0, 10) tokenAny, err := auth.CheckJWT(x) if err == nil { token := tokenAny.(*auth.Claims) uid := token.UID query := cfg.DB().Table("apps").Select("apps.*,app_users.status user_status") if opts.Name != nil { query = query.Joins("LEFT JOIN app_users ON app_users.app_id = apps.id AND app_users.user_id = ? AND apps.name LIKE ?", uid, opts.Name) } else { query = query.Joins("LEFT JOIN app_users ON app_users.app_id = apps.id AND app_users.user_id = ?", uid) } err = query.Scan(&data).Error } else { err = cfg.DB().Table("apps").Select("id", "name", "icon").Find(&data).Error } return data, err } type postOpts struct { Name string `json:"name" parse:"json"` Icon string `json:"icon" parse:"json"` Des *string `json:"des" parse:"json"` Typ string `json:"typ" parse:"json"` Status string `json:"status" parse:"json"` InitUrl string `json:"init_url" parse:"json"` } var _ = Router.Post("/", auth.Check("app", "", auth.DoCreate), appPost) func appPost(x *vigo.X) (any, error) { opts := &postOpts{} err := x.Parse(opts) if err != nil { return nil, err } data := &models.App{} data.Name = opts.Name data.Icon = opts.Icon if data.Icon == "" { data.Icon = fmt.Sprintf("https://public.veypi.com/img/avatar/%04d.jpg", rand.New(rand.NewSource(time.Now().UnixNano())).Intn(220)) } data.Typ = opts.Typ if opts.Des != nil { data.Des = *opts.Des } err = cfg.DB().Transaction(func(tx *gorm.DB) error { data.Key = utils.RandSeq(32) err := tx.Create(data).Error if err != nil { return err } au := &models.AppUser{ AppID: data.ID, UserID: x.Request.Context().Value("uid").(string), Status: models.AUSTATUS_OK, } return tx.Create(au).Error }) return data, err } type patchOpts struct { ID string `json:"id" gorm:"primaryKey;type:varchar(32)" parse:"path@app_id"` Name *string `json:"name" parse:"json"` Icon *string `json:"icon" parse:"json"` Des *string `json:"des" parse:"json"` Typ *string `json:"typ" gorm:"default:auto" parse:"json"` InitRoleID *string `json:"init_role_id" gorm:"index;type:varchar(32)" parse:"json"` Status *string `json:"status" gorm:"default:ok" parse:"json"` InitUrl *string `json:"init_url" parse:"json"` } var _ = Router.Patch("/:app_id", auth.Check("app", "app_id", auth.DoUpdate), appPatch) func appPatch(x *vigo.X) (any, error) { opts := &patchOpts{} err := x.Parse(opts) if err != nil { return nil, err } data := &models.App{} err = cfg.DB().Where("id = ?", opts.ID).First(data).Error if err != nil { return nil, err } optsMap := make(map[string]interface{}) if opts.Name != nil { optsMap["name"] = opts.Name } if opts.Icon != nil { optsMap["icon"] = opts.Icon } if opts.Des != nil { optsMap["des"] = opts.Des } if opts.Typ != nil { optsMap["typ"] = opts.Typ } if opts.InitRoleID != nil { optsMap["init_role_id"] = opts.InitRoleID } if opts.Status != nil { optsMap["status"] = opts.Status } if opts.InitUrl != nil { optsMap["init_url"] = opts.InitUrl } err = cfg.DB().Model(data).Updates(optsMap).Error return data, err }