chore(runner): register new runner

Signed-off-by: Bo-Yi.Wu <appleboy.tw@gmail.com>

models/bots/runner.go

@@ -43,7 +43,7 @@ type Runner struct {
 	Description string                 `xorm:"TEXT"`
 	Base        int                    // 0 native 1 docker 2 virtual machine
 	RepoRange   string                 // glob match which repositories could use this runner
-	Token       string
+	Token       string                 `xorm:"CHAR(36) UNIQUE"`
 
 	// instance status (idle)
 	Status core.RunnerStatus
@@ -148,7 +148,7 @@ func GetRunnerByToken(token string) (*Runner, error) {
 		return nil, err
 	} else if !has {
 		return nil, ErrRunnerNotExist{
-			UUID: "",
+			Token: token,
 		}
 	}
 	return &runner, nil
@@ -177,3 +177,9 @@ func FindRunnersByRepoID(repoID int64) ([]*Runner, error) {
 	err = db.GetEngine(db.DefaultContext).Join("INNER", "repository", "repository.owner_id = bot_runner.owner_id").Find(&runners)
 	return runners, err
 }
+
+// NewRunner creates new runner.
+func NewRunner(ctx context.Context, t *Runner) error {
+	_, err := db.GetEngine(ctx).Insert(t)
+	return err
+}

models/bots/runner_token.go

@@ -5,6 +5,9 @@
 package bots
 
 import (
+	"context"
+	"fmt"
+
 	"code.gitea.io/gitea/models/db"
 	repo_model "code.gitea.io/gitea/models/repo"
 	user_model "code.gitea.io/gitea/models/user"
@@ -14,14 +17,24 @@ import (
 	gouuid "github.com/google/uuid"
 )
 
+// ErrRunnerNotExist represents an error for bot runner not exist
+type ErrRunnerTokenNotExist struct {
+	Token string
+}
+
+func (err ErrRunnerTokenNotExist) Error() string {
+	return fmt.Sprintf("runner token [%s] is not exist", err.Token)
+}
+
 // RunnerToken represents runner tokens
 type RunnerToken struct {
-	ID      int64
-	Token   string                 `xorm:"CHAR(36) UNIQUE"`
-	OwnerID int64                  `xorm:"index"` // org level runner, 0 means system
-	Owner   *user_model.User       `xorm:"-"`
-	RepoID  int64                  `xorm:"index"` // repo level runner, if orgid also is zero, then it's a global
-	Repo    *repo_model.Repository `xorm:"-"`
+	ID       int64
+	Token    string                 `xorm:"CHAR(36) UNIQUE"`
+	OwnerID  int64                  `xorm:"index"` // org level runner, 0 means system
+	Owner    *user_model.User       `xorm:"-"`
+	RepoID   int64                  `xorm:"index"` // repo level runner, if orgid also is zero, then it's a global
+	Repo     *repo_model.Repository `xorm:"-"`
+	IsActive bool
 
 	CreatedUnix timeutil.TimeStamp `xorm:"INDEX created"`
 	UpdatedUnix timeutil.TimeStamp `xorm:"INDEX updated"`
@@ -35,9 +48,35 @@ func init() {
 	db.RegisterModel(new(RunnerToken))
 }
 
-// NewAccessToken creates new access token.
+// NewRunnerToken creates new runner token.
 func NewRunnerToken(t *RunnerToken) error {
 	t.Token = base.EncodeSha1(gouuid.New().String())
 	_, err := db.GetEngine(db.DefaultContext).Insert(t)
 	return err
 }
+
+// GetRunnerByToken returns a bot runner via token
+func GetRunnerToken(token string) (*RunnerToken, error) {
+	var runnerToken RunnerToken
+	has, err := db.GetEngine(db.DefaultContext).Where("token=?", token).Get(&runnerToken)
+	if err != nil {
+		return nil, err
+	} else if !has {
+		return nil, ErrRunnerTokenNotExist{
+			Token: token,
+		}
+	}
+	return &runnerToken, nil
+}
+
+// UpdateRunnerToken updates runner token information.
+func UpdateRunnerToken(ctx context.Context, r *RunnerToken, cols ...string) (err error) {
+	e := db.GetEngine(ctx)
+
+	if len(cols) == 0 {
+		_, err = e.ID(r.ID).AllCols().Update(r)
+	} else {
+		_, err = e.ID(r.ID).Cols(cols...).Update(r)
+	}
+	return err
+}