OneAuth/models/app.go

package models

import (
	"github.com/vyes/vigo/logv"
	"gorm.io/gorm"
)

const AUSTATUS_OK = "ok"
const AUSTATUS_NO = "no"
const AUSTATUS_APPLYING = "applying"
const AUSTATUS_REJECT = "reject"

type App struct {
	BaseModel
	Name       string  `json:"name"  parse:"json"`
	Icon       string  `json:"icon"  parse:"json"`
	Des        string  `json:"des"  parse:"json"`
	Typ        string  `json:"typ" gorm:"default:public"  parse:"json"`
	Status     string  `json:"status" gorm:"default:ok"  parse:"json"`
	InitRoleID *string `json:"init_role_id" gorm:"index;type:varchar(32);default: null"  parse:"json"`
	InitRole   *Role   `json:"init_role" gorm:"foreignKey:InitRoleID;references:ID"`
	InitUrl    string  `json:"init_url"  parse:"json"`
	UserCount  uint    `json:"user_count" default:"0"`
	Key        string  `json:"-"`
}

type AppUser struct {
	BaseModel
	AppID string `json:"app_id"  parse:"path"`
	App   *App   `json:"app" gorm:"foreignKey:AppID;references:ID"`

	UserID string `json:"user_id"  parse:"json"`
	User   *User  `json:"user" gorm:"foreignKey:UserID;references:ID"`

	Status string `json:"status"  parse:"json"`
}

func (m *AppUser) onOk(tx *gorm.DB) (err error) {
	app := &App{}
	logv.AssertError(tx.Where("id = ?", m.AppID).First(app).Error)
	if app.InitRoleID != nil {
		urList := make([]*UserRole, 0, 2)
		logv.AssertError(tx.Debug().Where("app_id = ? AND user_id = ?", m.AppID, m.UserID).Find(&urList).Error)
		if len(urList) == 0 {
			return tx.Create(&UserRole{
				AppID:  m.AppID,
				UserID: m.UserID,
				RoleID: *app.InitRoleID,
				Status: "ok",
			}).Error
		}
	}
	return nil
}

func (m *AppUser) AfterCreate(tx *gorm.DB) error {
	if m.Status == AUSTATUS_OK {
		logv.AssertError(m.onOk(tx))
	}
	return tx.Model(&App{}).Where("id = ?", m.AppID).Update("user_count", gorm.Expr("user_count + ?", 1)).Error
}

func (m *AppUser) AfterUpdate(tx *gorm.DB) error {
	if m.Status == AUSTATUS_OK {
		return m.onOk(tx)
	}
	return nil
}

type Resource struct {
	BaseModel
	AppID string `json:"app_id" parse:"path@app_id"`
	App   *App   `json:"-" gorm:"foreignKey:AppID;references:ID"`
	Name  string `json:"name" parse:"json"`
	Des   string `json:"des" parse:"json"`
}

type Role struct {
	BaseModel
	AppID     string    `json:"app_id" parse:"path@app_id"`
	App       *App      `json:"-" gorm:"foreignKey:AppID;references:ID"`
	Name      string    `json:"name" parse:"json"`
	Des       string    `json:"des" parse:"json"`
	UserCount uint      `json:"user_count" default:"0"`
	Access    []*Access `json:"-"`
}

type Access struct {
	BaseModel
	AppID string `json:"app_id" gorm:"index;type:varchar(32)" parse:"path@app_id"`
	App   *App   `json:"-" gorm:"foreignKey:AppID;references:ID"`

	UserID *string `json:"user_id" gorm:"index;type:varchar(32);default: null" parse:"json"`
	User   *User   `json:"-" gorm:"foreignKey:UserID;references:ID"`

	RoleID *string `json:"role_id" gorm:"index;type:varchar(32);default: null" parse:"json"`
	Role   *Role   `json:"-" gorm:"foreignKey:RoleID;references:ID"`

	ResourceID *string   `json:"resource_id" gorm:"index;type:varchar(32);default: null" parse:"json"`
	Resource   *Resource `json:"-" gorm:"foreignKey:ResourceID;references:ID"`

	Name  string `json:"name" parse:"json"`
	TID   string `json:"tid" parse:"json"`
	Level uint   `json:"level" parse:"json"`
}