chore(runner): register new runner

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

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
+}