123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127 |
- // Copyright 2023 The Gitea Authors. All rights reserved.
- // SPDX-License-Identifier: MIT
-
- package actions
-
- import (
- "context"
- "errors"
- "strings"
-
- "code.gitea.io/gitea/models/db"
- "code.gitea.io/gitea/modules/log"
- "code.gitea.io/gitea/modules/timeutil"
-
- "xorm.io/builder"
- )
-
- type ActionVariable struct {
- ID int64 `xorm:"pk autoincr"`
- OwnerID int64 `xorm:"UNIQUE(owner_repo_name)"`
- RepoID int64 `xorm:"INDEX UNIQUE(owner_repo_name)"`
- Name string `xorm:"UNIQUE(owner_repo_name) NOT NULL"`
- Data string `xorm:"LONGTEXT NOT NULL"`
- CreatedUnix timeutil.TimeStamp `xorm:"created NOT NULL"`
- UpdatedUnix timeutil.TimeStamp `xorm:"updated"`
- }
-
- func init() {
- db.RegisterModel(new(ActionVariable))
- }
-
- func (v *ActionVariable) Validate() error {
- if v.OwnerID != 0 && v.RepoID != 0 {
- return errors.New("a variable should not be bound to an owner and a repository at the same time")
- }
- return nil
- }
-
- func InsertVariable(ctx context.Context, ownerID, repoID int64, name, data string) (*ActionVariable, error) {
- variable := &ActionVariable{
- OwnerID: ownerID,
- RepoID: repoID,
- Name: strings.ToUpper(name),
- Data: data,
- }
- if err := variable.Validate(); err != nil {
- return variable, err
- }
- return variable, db.Insert(ctx, variable)
- }
-
- type FindVariablesOpts struct {
- db.ListOptions
- OwnerID int64
- RepoID int64
- Name string
- }
-
- func (opts FindVariablesOpts) ToConds() builder.Cond {
- cond := builder.NewCond()
- // Since we now support instance-level variables,
- // there is no need to check for null values for `owner_id` and `repo_id`
- cond = cond.And(builder.Eq{"owner_id": opts.OwnerID})
- cond = cond.And(builder.Eq{"repo_id": opts.RepoID})
-
- if opts.Name != "" {
- cond = cond.And(builder.Eq{"name": strings.ToUpper(opts.Name)})
- }
- return cond
- }
-
- func FindVariables(ctx context.Context, opts FindVariablesOpts) ([]*ActionVariable, error) {
- return db.Find[ActionVariable](ctx, opts)
- }
-
- func UpdateVariable(ctx context.Context, variable *ActionVariable) (bool, error) {
- count, err := db.GetEngine(ctx).ID(variable.ID).Cols("name", "data").
- Update(&ActionVariable{
- Name: variable.Name,
- Data: variable.Data,
- })
- return count != 0, err
- }
-
- func DeleteVariable(ctx context.Context, id int64) error {
- if _, err := db.DeleteByID[ActionVariable](ctx, id); err != nil {
- return err
- }
- return nil
- }
-
- func GetVariablesOfRun(ctx context.Context, run *ActionRun) (map[string]string, error) {
- variables := map[string]string{}
-
- if err := run.LoadRepo(ctx); err != nil {
- log.Error("LoadRepo: %v", err)
- return nil, err
- }
-
- // Global
- globalVariables, err := db.Find[ActionVariable](ctx, FindVariablesOpts{})
- if err != nil {
- log.Error("find global variables: %v", err)
- return nil, err
- }
-
- // Org / User level
- ownerVariables, err := db.Find[ActionVariable](ctx, FindVariablesOpts{OwnerID: run.Repo.OwnerID})
- if err != nil {
- log.Error("find variables of org: %d, error: %v", run.Repo.OwnerID, err)
- return nil, err
- }
-
- // Repo level
- repoVariables, err := db.Find[ActionVariable](ctx, FindVariablesOpts{RepoID: run.RepoID})
- if err != nil {
- log.Error("find variables of repo: %d, error: %v", run.RepoID, err)
- return nil, err
- }
-
- // Level precedence: Repo > Org / User > Global
- for _, v := range append(globalVariables, append(ownerVariables, repoVariables...)...) {
- variables[v.Name] = v.Data
- }
-
- return variables, nil
- }
|