OneAuth/api/app/app.go

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
}