69 lines
1.6 KiB
Go
69 lines
1.6 KiB
Go
|
// SPDX-License-Identifier: MIT
|
||
|
|
|
||
|
|
package actions
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"encoding/hex"
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
auth_model "code.gitea.io/gitea/models/auth"
|
||
|
|
"code.gitea.io/gitea/models/db"
|
||
|
|
"code.gitea.io/gitea/modules/util"
|
||
|
|
|
||
|
|
gouuid "github.com/google/uuid"
|
||
|
|
)
|
||
|
|
|
||
|
|
func RegisterRunner(ctx context.Context, ownerID, repoID int64, token string, labels []string, name, version string) (*ActionRunner, error) {
|
||
|
|
uuid, err := gouuid.FromBytes([]byte(token[:16]))
|
||
|
|
if err != nil {
|
||
|
|
return nil, fmt.Errorf("gouuid.FromBytes %v", err)
|
||
|
|
}
|
||
|
|
uuidString := uuid.String()
|
||
|
|
|
||
|
|
var runner ActionRunner
|
||
|
|
|
||
|
|
has, err := db.GetEngine(ctx).Where("uuid=?", uuidString).Get(&runner)
|
||
|
|
if err != nil {
|
||
|
|
return nil, fmt.Errorf("GetRunner %v", err)
|
||
|
|
} else if !has {
|
||
|
|
//
|
||
|
|
// The runner does not exist yet, create it
|
||
|
|
//
|
||
|
|
saltBytes, err := util.CryptoRandomBytes(16)
|
||
|
|
if err != nil {
|
||
|
|
return nil, fmt.Errorf("CryptoRandomBytes %v", err)
|
||
|
|
}
|
||
|
|
salt := hex.EncodeToString(saltBytes)
|
||
|
|
|
||
|
|
hash := auth_model.HashToken(token, salt)
|
||
|
|
|
||
|
|
runner = ActionRunner{
|
||
|
|
UUID: uuidString,
|
||
|
|
TokenHash: hash,
|
||
|
|
TokenSalt: salt,
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := CreateRunner(ctx, &runner); err != nil {
|
||
|
|
return &runner, fmt.Errorf("can't create new runner %w", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// Update the existing runner
|
||
|
|
//
|
||
|
|
name, _ = util.SplitStringAtByteN(name, 255)
|
||
|
|
|
||
|
|
runner.Name = name
|
||
|
|
runner.OwnerID = ownerID
|
||
|
|
runner.RepoID = repoID
|
||
|
|
runner.Version = version
|
||
|
|
runner.AgentLabels = labels
|
||
|
|
|
||
|
|
if err := UpdateRunner(ctx, &runner, "name", "owner_id", "repo_id", "version", "agent_labels"); err != nil {
|
||
|
|
return &runner, fmt.Errorf("can't update the runner %+v %w", runner, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return &runner, nil
|
||
|
|
}
|