Merge branch 'feat/agents-write-git-ssh' into feat/agents-write-git-ssh-s
This commit is contained in:
@@ -17,6 +17,7 @@ import (
|
|||||||
|
|
||||||
"golang.org/x/crypto/nacl/box"
|
"golang.org/x/crypto/nacl/box"
|
||||||
|
|
||||||
|
"gitea/internal/forgeaccount"
|
||||||
"gitea/internal/forgehttp"
|
"gitea/internal/forgehttp"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -31,7 +32,7 @@ const (
|
|||||||
// TokenName is the token's name in the agent account's list, found and deleted by it at the next mint.
|
// TokenName is the token's name in the agent account's list, found and deleted by it at the next mint.
|
||||||
TokenName = "mesh-agent-read"
|
TokenName = "mesh-agent-read"
|
||||||
// AccountPrefix names an agent's account for its node (ADR 0365 §1).
|
// AccountPrefix names an agent's account for its node (ADR 0365 §1).
|
||||||
AccountPrefix = "agent-"
|
AccountPrefix = forgeaccount.AccountPrefix
|
||||||
// vaultModule is the vault's module: its account is `<node>.mesh-vault` (ADR 0259 §8's runs-as).
|
// vaultModule is the vault's module: its account is `<node>.mesh-vault` (ADR 0259 §8's runs-as).
|
||||||
vaultModule = "mesh-vault"
|
vaultModule = "mesh-vault"
|
||||||
// deadline bounds one call, inside the vault's ask.
|
// deadline bounds one call, inside the vault's ask.
|
||||||
@@ -43,22 +44,12 @@ const (
|
|||||||
// the agents team's (a write team on every repository); main takes no push whatever the token (ADR 0398 §2).
|
// the agents team's (a write team on every repository); main takes no push whatever the token (ADR 0398 §2).
|
||||||
var TokenScopes = []string{"write:repository", "write:issue"}
|
var TokenScopes = []string{"write:repository", "write:issue"}
|
||||||
|
|
||||||
|
// The agents team and its shape are the forgeaccount package's (ADR 0398 §1).
|
||||||
const (
|
const (
|
||||||
// TeamOrg is the organisation whose repositories are the mesh's.
|
TeamOrg = forgeaccount.TeamOrg
|
||||||
TeamOrg = "novox"
|
TeamName = forgeaccount.TeamName
|
||||||
// TeamName is the one team every agent's account is in (ADR 0398 §1).
|
|
||||||
TeamName = "agents"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
// teamShape is what the agents team grants: write on code, issues and pull requests of every repository of the
|
|
||||||
// organisation and no more, the shape of the forge tools' team (tools-account.ts).
|
|
||||||
func teamShape() map[string]any {
|
|
||||||
return map[string]any{
|
|
||||||
"name": TeamName, "permission": "write", "includes_all_repositories": true, "can_create_org_repo": false,
|
|
||||||
"units_map": map[string]string{"repo.code": "write", "repo.issues": "write", "repo.pulls": "write"},
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
var nodeName = regexp.MustCompile(`^[a-z0-9][a-z0-9-]{0,31}$`)
|
var nodeName = regexp.MustCompile(`^[a-z0-9][a-z0-9-]{0,31}$`)
|
||||||
|
|
||||||
// fromTheVault refuses every caller but the vault's own account (ADR 0349 clause 10): whoever else could call this
|
// fromTheVault refuses every caller but the vault's own account (ADR 0349 clause 10): whoever else could call this
|
||||||
@@ -206,10 +197,10 @@ func (m *Minter) Answer(ask Ask) (Answer, error) {
|
|||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
account := AccountPrefix + h.Node
|
account := AccountPrefix + h.Node
|
||||||
if err := m.ensureAccount(ctx, account); err != nil {
|
if err := m.accounts().EnsureAccount(ctx, account); err != nil {
|
||||||
return Answer{}, err
|
return Answer{}, err
|
||||||
}
|
}
|
||||||
if err := m.ensureInTeam(ctx, account); err != nil {
|
if err := m.accounts().EnsureInTeam(ctx, account); err != nil {
|
||||||
return Answer{}, err
|
return Answer{}, err
|
||||||
}
|
}
|
||||||
token, err := m.mint(ctx, account)
|
token, err := m.mint(ctx, account)
|
||||||
@@ -274,91 +265,6 @@ func credentialLines(account, token string, origins []string) (string, error) {
|
|||||||
return strings.Join(lines, "\n") + "\n", nil
|
return strings.Join(lines, "\n") + "\n", nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// gitea's account as the admin reads it.
|
|
||||||
type forgeUser struct {
|
|
||||||
IsAdmin bool `json:"is_admin"`
|
|
||||||
Active bool `json:"active"`
|
|
||||||
ProhibitLogin bool `json:"prohibit_login"`
|
|
||||||
Restricted bool `json:"restricted"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// ensureAccount makes the agent's account, plain, when there is none, and puts back what was changed by hand: never
|
|
||||||
// an administrator, active, able to use its token. Its team is ensureInTeam's.
|
|
||||||
func (m *Minter) ensureAccount(ctx context.Context, account string) error {
|
|
||||||
var u forgeUser
|
|
||||||
err := m.Forge.GetJSON(ctx, "/api/v1/users/"+account, &u)
|
|
||||||
switch {
|
|
||||||
case errors.Is(err, forgehttp.ErrNotFound):
|
|
||||||
password := m.password()
|
|
||||||
err := m.Forge.SendJSON(ctx, http.MethodPost, "/api/v1/admin/users", map[string]any{
|
|
||||||
"username": account, "email": account + "@noreply.localhost", "password": password,
|
|
||||||
"must_change_password": false, "visibility": "private",
|
|
||||||
}, nil)
|
|
||||||
password = ""
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("the account %s could not be made: %w", account, err)
|
|
||||||
}
|
|
||||||
m.say("made the forge account %s for that machine's agent, in no team yet", account)
|
|
||||||
return nil
|
|
||||||
case err != nil:
|
|
||||||
return fmt.Errorf("the account %s could not be read: %w", account, err)
|
|
||||||
}
|
|
||||||
if u.IsAdmin || !u.Active || u.ProhibitLogin {
|
|
||||||
if err := m.Forge.SendJSON(ctx, http.MethodPatch, "/api/v1/admin/users/"+account, map[string]any{
|
|
||||||
"login_name": account, "source_id": 0, "admin": false, "active": true, "prohibit_login": false,
|
|
||||||
}, nil); err != nil {
|
|
||||||
return fmt.Errorf("the account %s could not be made a plain account again: %w", account, err)
|
|
||||||
}
|
|
||||||
m.say("the account %s was changed by hand (administrator %t, active %t, login prohibited %t); made a plain account again",
|
|
||||||
account, u.IsAdmin, u.Active, u.ProhibitLogin)
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// ensureInTeam makes the agents team when there is none, puts a team that is there back to its shape, and adds the
|
|
||||||
// account to it; an account already in it stays (ADR 0398 §1, issue 948).
|
|
||||||
func (m *Minter) ensureInTeam(ctx context.Context, account string) error {
|
|
||||||
var id int64
|
|
||||||
for page := 1; page <= 20 && id == 0; page++ {
|
|
||||||
var teams []struct {
|
|
||||||
ID int64 `json:"id"`
|
|
||||||
Name string `json:"name"`
|
|
||||||
}
|
|
||||||
if err := m.Forge.GetJSON(ctx, fmt.Sprintf("/api/v1/orgs/%s/teams?limit=50&page=%d", TeamOrg, page), &teams); err != nil {
|
|
||||||
return fmt.Errorf("the teams of %s could not be read: %w", TeamOrg, err)
|
|
||||||
}
|
|
||||||
for _, t := range teams {
|
|
||||||
if t.Name == TeamName {
|
|
||||||
id = t.ID
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if len(teams) < 50 {
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if id != 0 {
|
|
||||||
if err := m.Forge.SendJSON(ctx, http.MethodPatch, fmt.Sprintf("/api/v1/teams/%d", id), teamShape(), nil); err != nil {
|
|
||||||
return fmt.Errorf("the team %s could not be put back to its shape: %w", TeamName, err)
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
var made struct {
|
|
||||||
ID int64 `json:"id"`
|
|
||||||
}
|
|
||||||
if err := m.Forge.SendJSON(ctx, http.MethodPost, "/api/v1/orgs/"+TeamOrg+"/teams", teamShape(), &made); err != nil {
|
|
||||||
return fmt.Errorf("the team %s could not be made: %w", TeamName, err)
|
|
||||||
}
|
|
||||||
id = made.ID
|
|
||||||
m.say("made the forge team %s in %s: write on every repository", TeamName, TeamOrg)
|
|
||||||
}
|
|
||||||
if id == 0 {
|
|
||||||
return fmt.Errorf("the forge answered no id for the team %s", TeamName)
|
|
||||||
}
|
|
||||||
if err := m.Forge.SendJSON(ctx, http.MethodPut, fmt.Sprintf("/api/v1/teams/%d/members/%s", id, account), nil, nil); err != nil {
|
|
||||||
return fmt.Errorf("the account %s could not be added to the team %s: %w", account, TeamName, err)
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
type forgeToken struct {
|
type forgeToken struct {
|
||||||
ID int64 `json:"id"`
|
ID int64 `json:"id"`
|
||||||
Name string `json:"name"`
|
Name string `json:"name"`
|
||||||
@@ -392,13 +298,9 @@ func (m *Minter) mint(ctx context.Context, account string) (string, error) {
|
|||||||
return strings.TrimSpace(made.SHA1), nil
|
return strings.TrimSpace(made.SHA1), nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *Minter) password() string {
|
// accounts is the forge's accounts as this minter reaches them (ADR 0400 §3: one place says what an agent's account is).
|
||||||
if m.Password != nil {
|
func (m *Minter) accounts() *forgeaccount.Accounts {
|
||||||
return m.Password()
|
return &forgeaccount.Accounts{Forge: m.Forge, Password: m.Password, Log: m.Log}
|
||||||
}
|
|
||||||
b := make([]byte, 24)
|
|
||||||
_, _ = rand.Read(b)
|
|
||||||
return base64.RawURLEncoding.EncodeToString(b)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *Minter) say(format string, a ...any) {
|
func (m *Minter) say(format string, a ...any) {
|
||||||
@@ -408,3 +310,6 @@ func (m *Minter) say(format string, a ...any) {
|
|||||||
}
|
}
|
||||||
fmt.Fprintf(os.Stderr, format+"\n", a...)
|
fmt.Fprintf(os.Stderr, format+"\n", a...)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// forgeUser is gitea's account as the admin reads it (the forgeaccount package's).
|
||||||
|
type forgeUser = forgeaccount.User
|
||||||
|
|||||||
@@ -0,0 +1,145 @@
|
|||||||
|
// Package forgeaccount is the forge accounts the mesh makes for its agents (novox/hq ADR 0365 clauses 1 and 3, ADR
|
||||||
|
// 0398 §1, ADR 0399, ADR 0400; issue 955): an account of the agent's own, no administrator, in the agents write
|
||||||
|
// team. Both the token tool (cmd/agent-token) and the forge-account provider use it, so there is one place that says
|
||||||
|
// what an agent's account is.
|
||||||
|
package forgeaccount
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"crypto/rand"
|
||||||
|
"encoding/base64"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"net/http"
|
||||||
|
"os"
|
||||||
|
|
||||||
|
"gitea/internal/forgehttp"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
// AccountPrefix names an agent's account for its node (ADR 0365 §1).
|
||||||
|
AccountPrefix = "agent-"
|
||||||
|
// TeamOrg is the organisation whose repositories are the mesh's.
|
||||||
|
TeamOrg = "novox"
|
||||||
|
// TeamName is the one team every agent's account is in (ADR 0398 §1).
|
||||||
|
TeamName = "agents"
|
||||||
|
)
|
||||||
|
|
||||||
|
// teamShape is what the agents team grants: write on code, issues and pull requests of every repository of the
|
||||||
|
// organisation and no more, the shape of the forge tools' team (tools-account.ts).
|
||||||
|
func teamShape() map[string]any {
|
||||||
|
return map[string]any{
|
||||||
|
"name": TeamName, "permission": "write", "includes_all_repositories": true, "can_create_org_repo": false,
|
||||||
|
"units_map": map[string]string{"repo.code": "write", "repo.issues": "write", "repo.pulls": "write"},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// User is gitea's account as the admin reads it.
|
||||||
|
type User struct {
|
||||||
|
IsAdmin bool `json:"is_admin"`
|
||||||
|
Active bool `json:"active"`
|
||||||
|
ProhibitLogin bool `json:"prohibit_login"`
|
||||||
|
Restricted bool `json:"restricted"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// Accounts makes and keeps the agents' accounts with the administrator's login.
|
||||||
|
type Accounts struct {
|
||||||
|
Forge forgehttp.Client
|
||||||
|
// Password makes the password an account is created with: random, sent once and kept nowhere.
|
||||||
|
Password func() string
|
||||||
|
Log func(format string, a ...any)
|
||||||
|
}
|
||||||
|
|
||||||
|
// EnsureAccount makes the agent's account, plain, when there is none, and puts back what was changed by hand: never
|
||||||
|
// an administrator, active, able to use its token. Its team is EnsureInTeam's.
|
||||||
|
func (m *Accounts) EnsureAccount(ctx context.Context, account string) error {
|
||||||
|
var u User
|
||||||
|
err := m.Forge.GetJSON(ctx, "/api/v1/users/"+account, &u)
|
||||||
|
switch {
|
||||||
|
case errors.Is(err, forgehttp.ErrNotFound):
|
||||||
|
password := m.password()
|
||||||
|
err := m.Forge.SendJSON(ctx, http.MethodPost, "/api/v1/admin/users", map[string]any{
|
||||||
|
"username": account, "email": account + "@noreply.localhost", "password": password,
|
||||||
|
"must_change_password": false, "visibility": "private",
|
||||||
|
}, nil)
|
||||||
|
password = ""
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("the account %s could not be made: %w", account, err)
|
||||||
|
}
|
||||||
|
m.say("made the forge account %s for that machine's agent, in no team yet", account)
|
||||||
|
return nil
|
||||||
|
case err != nil:
|
||||||
|
return fmt.Errorf("the account %s could not be read: %w", account, err)
|
||||||
|
}
|
||||||
|
if u.IsAdmin || !u.Active || u.ProhibitLogin {
|
||||||
|
if err := m.Forge.SendJSON(ctx, http.MethodPatch, "/api/v1/admin/users/"+account, map[string]any{
|
||||||
|
"login_name": account, "source_id": 0, "admin": false, "active": true, "prohibit_login": false,
|
||||||
|
}, nil); err != nil {
|
||||||
|
return fmt.Errorf("the account %s could not be made a plain account again: %w", account, err)
|
||||||
|
}
|
||||||
|
m.say("the account %s was changed by hand (administrator %t, active %t, login prohibited %t); made a plain account again",
|
||||||
|
account, u.IsAdmin, u.Active, u.ProhibitLogin)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// EnsureInTeam makes the agents team when there is none, puts a team that is there back to its shape, and adds the
|
||||||
|
// account to it; an account already in it stays (ADR 0398 §1, issue 948).
|
||||||
|
func (m *Accounts) EnsureInTeam(ctx context.Context, account string) error {
|
||||||
|
var id int64
|
||||||
|
for page := 1; page <= 20 && id == 0; page++ {
|
||||||
|
var teams []struct {
|
||||||
|
ID int64 `json:"id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
}
|
||||||
|
if err := m.Forge.GetJSON(ctx, fmt.Sprintf("/api/v1/orgs/%s/teams?limit=50&page=%d", TeamOrg, page), &teams); err != nil {
|
||||||
|
return fmt.Errorf("the teams of %s could not be read: %w", TeamOrg, err)
|
||||||
|
}
|
||||||
|
for _, t := range teams {
|
||||||
|
if t.Name == TeamName {
|
||||||
|
id = t.ID
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if len(teams) < 50 {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if id != 0 {
|
||||||
|
if err := m.Forge.SendJSON(ctx, http.MethodPatch, fmt.Sprintf("/api/v1/teams/%d", id), teamShape(), nil); err != nil {
|
||||||
|
return fmt.Errorf("the team %s could not be put back to its shape: %w", TeamName, err)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
var made struct {
|
||||||
|
ID int64 `json:"id"`
|
||||||
|
}
|
||||||
|
if err := m.Forge.SendJSON(ctx, http.MethodPost, "/api/v1/orgs/"+TeamOrg+"/teams", teamShape(), &made); err != nil {
|
||||||
|
return fmt.Errorf("the team %s could not be made: %w", TeamName, err)
|
||||||
|
}
|
||||||
|
id = made.ID
|
||||||
|
m.say("made the forge team %s in %s: write on every repository", TeamName, TeamOrg)
|
||||||
|
}
|
||||||
|
if id == 0 {
|
||||||
|
return fmt.Errorf("the forge answered no id for the team %s", TeamName)
|
||||||
|
}
|
||||||
|
if err := m.Forge.SendJSON(ctx, http.MethodPut, fmt.Sprintf("/api/v1/teams/%d/members/%s", id, account), nil, nil); err != nil {
|
||||||
|
return fmt.Errorf("the account %s could not be added to the team %s: %w", account, TeamName, err)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *Accounts) password() string {
|
||||||
|
if m.Password != nil {
|
||||||
|
return m.Password()
|
||||||
|
}
|
||||||
|
b := make([]byte, 24)
|
||||||
|
_, _ = rand.Read(b)
|
||||||
|
return base64.RawURLEncoding.EncodeToString(b)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *Accounts) say(format string, a ...any) {
|
||||||
|
if m.Log != nil {
|
||||||
|
m.Log(format, a...)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
fmt.Fprintf(os.Stderr, format+"\n", a...)
|
||||||
|
}
|
||||||
@@ -0,0 +1,132 @@
|
|||||||
|
package forgeaccount
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"net/http"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"gitea/internal/forgehttp"
|
||||||
|
)
|
||||||
|
|
||||||
|
// TitlePrefix starts the title of every key the mesh registers; a key without it is the operator's and is never
|
||||||
|
// touched (ADR 0399 §2, ADR 0400).
|
||||||
|
const TitlePrefix = "mesh:"
|
||||||
|
|
||||||
|
// KeyTitle is the title of a machine account's key: `mesh:<node>.<account>`.
|
||||||
|
func KeyTitle(node, account string) string { return TitlePrefix + node + "." + account }
|
||||||
|
|
||||||
|
// WantedKey is a public key the mesh wants registered, under its title.
|
||||||
|
type WantedKey struct {
|
||||||
|
Title string
|
||||||
|
// Key is the public key line: type, base64 and a comment.
|
||||||
|
Key string
|
||||||
|
}
|
||||||
|
|
||||||
|
// Key is a key the forge holds for an account.
|
||||||
|
type Key struct {
|
||||||
|
ID int64 `json:"id"`
|
||||||
|
Title string `json:"title"`
|
||||||
|
Key string `json:"key"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// material is a public key line without its comment: its type and its base64, which is what makes it the same key.
|
||||||
|
func material(line string) string {
|
||||||
|
f := strings.Fields(line)
|
||||||
|
if len(f) < 2 {
|
||||||
|
return strings.TrimSpace(line)
|
||||||
|
}
|
||||||
|
return f[0] + " " + f[1]
|
||||||
|
}
|
||||||
|
|
||||||
|
// Keys lists the SSH keys the account holds.
|
||||||
|
func (m *Accounts) Keys(ctx context.Context, user string) ([]Key, error) {
|
||||||
|
var all []Key
|
||||||
|
for page := 1; page <= 20; page++ {
|
||||||
|
var got []Key
|
||||||
|
if err := m.Forge.GetJSON(ctx, fmt.Sprintf("/api/v1/users/%s/keys?limit=50&page=%d", user, page), &got); err != nil {
|
||||||
|
return nil, fmt.Errorf("the keys of %s could not be read: %w", user, err)
|
||||||
|
}
|
||||||
|
all = append(all, got...)
|
||||||
|
if len(got) < 50 {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return all, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// AddKey registers a key for the account, one it may write with.
|
||||||
|
func (m *Accounts) AddKey(ctx context.Context, user string, k WantedKey) error {
|
||||||
|
err := m.Forge.SendJSON(ctx, http.MethodPost, "/api/v1/admin/users/"+user+"/keys",
|
||||||
|
map[string]any{"title": k.Title, "key": k.Key, "read_only": false}, nil)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("the key %s could not be added for %s: %w", k.Title, user, err)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// DeleteKey removes one key of the account, by its id.
|
||||||
|
func (m *Accounts) DeleteKey(ctx context.Context, user string, id int64) error {
|
||||||
|
err := m.Forge.SendJSON(ctx, http.MethodDelete, fmt.Sprintf("/api/v1/admin/users/%s/keys/%d", user, id), nil, nil)
|
||||||
|
if err != nil && !errors.Is(err, forgehttp.ErrNotFound) {
|
||||||
|
return fmt.Errorf("the key %d of %s could not be deleted: %w", id, user, err)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// EnsureKeys registers each wanted key the account does not hold and removes the keys the mesh registered
|
||||||
|
// (title starting `mesh:`) that are no longer wanted. A key whose title does not start `mesh:` is the operator's and
|
||||||
|
// is never touched, and a wanted key the account already holds under another title is left as it is (ADR 0399 §2).
|
||||||
|
// A key is added before another title's is removed, so an account is not left without one.
|
||||||
|
func (m *Accounts) EnsureKeys(ctx context.Context, user string, wanted []WantedKey) error {
|
||||||
|
held, err := m.Keys(ctx, user)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
have := map[string]bool{}
|
||||||
|
for _, k := range held {
|
||||||
|
have[material(k.Key)] = true
|
||||||
|
}
|
||||||
|
want := map[string]bool{}
|
||||||
|
for _, w := range wanted {
|
||||||
|
want[material(w.Key)] = true
|
||||||
|
}
|
||||||
|
gone := map[int64]bool{}
|
||||||
|
remove := func(k Key) error {
|
||||||
|
if gone[k.ID] {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
gone[k.ID] = true
|
||||||
|
if err := m.DeleteKey(ctx, user, k.ID); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
m.say("removed the key %s of %s: it is no longer wanted", k.Title, user)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
for _, w := range wanted {
|
||||||
|
if have[material(w.Key)] {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
// The forge holds a title once: the mesh's key of the same title, now another key, goes first.
|
||||||
|
for _, k := range held {
|
||||||
|
if k.Title == w.Title && strings.HasPrefix(k.Title, TitlePrefix) && !want[material(k.Key)] {
|
||||||
|
if err := remove(k); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if err := m.AddKey(ctx, user, w); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
m.say("registered the key %s for %s", w.Title, user)
|
||||||
|
}
|
||||||
|
for _, k := range held {
|
||||||
|
if strings.HasPrefix(k.Title, TitlePrefix) && !want[material(k.Key)] {
|
||||||
|
if err := remove(k); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
@@ -0,0 +1,192 @@
|
|||||||
|
package forgeaccount
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"net/http"
|
||||||
|
"net/http/httptest"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"regexp"
|
||||||
|
"strings"
|
||||||
|
"sync"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"gitea/internal/forgehttp"
|
||||||
|
)
|
||||||
|
|
||||||
|
type heldKey struct {
|
||||||
|
ID int64 `json:"id"`
|
||||||
|
Title string `json:"title"`
|
||||||
|
Key string `json:"key"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// keyForge is the part of gitea's API the keys use, as the administrator's login reaches it (issue 955 F2).
|
||||||
|
type keyForge struct {
|
||||||
|
mu sync.Mutex
|
||||||
|
keys map[string][]heldKey
|
||||||
|
next int64
|
||||||
|
calls []string
|
||||||
|
posts []map[string]any
|
||||||
|
}
|
||||||
|
|
||||||
|
var keysPath = regexp.MustCompile(`^/api/v1/(?:admin/)?users/([^/]+)/keys(?:/(\d+))?$`)
|
||||||
|
|
||||||
|
func (f *keyForge) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
||||||
|
f.mu.Lock()
|
||||||
|
defer f.mu.Unlock()
|
||||||
|
f.calls = append(f.calls, r.Method+" "+r.URL.Path)
|
||||||
|
m := keysPath.FindStringSubmatch(r.URL.Path)
|
||||||
|
if m == nil {
|
||||||
|
http.NotFound(w, r)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
user := m[1]
|
||||||
|
reply := func(status int, v any) {
|
||||||
|
w.Header().Set("Content-Type", "application/json")
|
||||||
|
w.WriteHeader(status)
|
||||||
|
if v != nil {
|
||||||
|
_ = json.NewEncoder(w).Encode(v)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
switch {
|
||||||
|
case r.Method == http.MethodGet && m[2] == "" && !strings.Contains(r.URL.Path, "/admin/"):
|
||||||
|
reply(200, f.keys[user])
|
||||||
|
case r.Method == http.MethodPost && m[2] == "" && strings.Contains(r.URL.Path, "/admin/"):
|
||||||
|
body, _ := io.ReadAll(r.Body)
|
||||||
|
var in map[string]any
|
||||||
|
_ = json.Unmarshal(body, &in)
|
||||||
|
f.posts = append(f.posts, in)
|
||||||
|
key, _ := in["key"].(string)
|
||||||
|
for _, k := range f.keys[user] {
|
||||||
|
if k.Key == key {
|
||||||
|
reply(422, map[string]any{"message": "Key content has been used as non-deploy key"})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
title, _ := in["title"].(string)
|
||||||
|
for _, k := range f.keys[user] {
|
||||||
|
if k.Title == title {
|
||||||
|
reply(422, map[string]any{"message": "Key title has been used"})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
f.next++
|
||||||
|
f.keys[user] = append(f.keys[user], heldKey{ID: f.next, Title: title, Key: key})
|
||||||
|
reply(201, map[string]any{"id": f.next})
|
||||||
|
case r.Method == http.MethodDelete && m[2] != "" && strings.Contains(r.URL.Path, "/admin/"):
|
||||||
|
var kept []heldKey
|
||||||
|
for _, k := range f.keys[user] {
|
||||||
|
if fmt.Sprint(k.ID) != m[2] {
|
||||||
|
kept = append(kept, k)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
f.keys[user] = kept
|
||||||
|
reply(204, nil)
|
||||||
|
default:
|
||||||
|
http.Error(w, "unexpected "+r.Method+" "+r.URL.Path, 400)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const (
|
||||||
|
keyA = "ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAIAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA"
|
||||||
|
keyB = "ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAIBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBB"
|
||||||
|
)
|
||||||
|
|
||||||
|
func keyAccounts(t *testing.T, f *keyForge) *Accounts {
|
||||||
|
t.Helper()
|
||||||
|
srv := httptest.NewServer(f)
|
||||||
|
t.Cleanup(srv.Close)
|
||||||
|
pw := filepath.Join(t.TempDir(), "pw")
|
||||||
|
if err := os.WriteFile(pw, []byte("x\n"), 0o600); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
return &Accounts{Forge: forgehttp.Client{URL: srv.URL, User: "mesh-admin", PasswordFile: pw}, Log: func(string, ...any) {}}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestEnsureKeysAddsTheWantedKeyUnderItsTitle(t *testing.T) {
|
||||||
|
f := &keyForge{keys: map[string][]heldKey{}}
|
||||||
|
a := keyAccounts(t, f)
|
||||||
|
if err := a.EnsureKeys(t.Context(), "agent-g14", []WantedKey{{Title: "mesh:g14.nox", Key: keyA + " nox@g14 (mesh)"}}); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
if len(f.posts) != 1 || f.posts[0]["title"] != "mesh:g14.nox" || f.posts[0]["read_only"] != false ||
|
||||||
|
f.posts[0]["key"] != keyA+" nox@g14 (mesh)" {
|
||||||
|
t.Fatalf("posts = %v", f.posts)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestEnsureKeysAgainChangesNothing(t *testing.T) {
|
||||||
|
f := &keyForge{keys: map[string][]heldKey{}}
|
||||||
|
a := keyAccounts(t, f)
|
||||||
|
want := []WantedKey{{Title: "mesh:g14.nox", Key: keyA + " nox@g14 (mesh)"}}
|
||||||
|
for range 2 {
|
||||||
|
if err := a.EnsureKeys(t.Context(), "agent-g14", want); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if len(f.posts) != 1 || len(f.keys["agent-g14"]) != 1 {
|
||||||
|
t.Fatalf("posts = %v, held = %v", f.posts, f.keys)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestEnsureKeysReplacesAChangedMeshKeyOfTheSameTitle(t *testing.T) {
|
||||||
|
// The forge refuses a title held twice, so the old key goes first.
|
||||||
|
f := &keyForge{keys: map[string][]heldKey{"agent-g14": {{ID: 7, Title: "mesh:g14.nox", Key: keyA}}}, next: 7}
|
||||||
|
a := keyAccounts(t, f)
|
||||||
|
if err := a.EnsureKeys(t.Context(), "agent-g14", []WantedKey{{Title: "mesh:g14.nox", Key: keyB + " nox@g14 (mesh)"}}); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
held := f.keys["agent-g14"]
|
||||||
|
if len(held) != 1 || !strings.HasPrefix(held[0].Key, keyB) {
|
||||||
|
t.Fatalf("held = %v", held)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestEnsureKeysRemovesAMeshKeyOfAnotherTitleAfterTheNewOneIsIn(t *testing.T) {
|
||||||
|
f := &keyForge{keys: map[string][]heldKey{"agent-g14": {{ID: 7, Title: "mesh:g14.old", Key: keyA}}}, next: 7}
|
||||||
|
a := keyAccounts(t, f)
|
||||||
|
if err := a.EnsureKeys(t.Context(), "agent-g14", []WantedKey{{Title: "mesh:g14.nox", Key: keyB}}); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
var post, del int
|
||||||
|
for i, c := range f.calls {
|
||||||
|
if strings.HasPrefix(c, "POST") {
|
||||||
|
post = i
|
||||||
|
}
|
||||||
|
if strings.HasPrefix(c, "DELETE") {
|
||||||
|
del = i
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if post == 0 || del == 0 || post > del || len(f.keys["agent-g14"]) != 1 || f.keys["agent-g14"][0].Title != "mesh:g14.nox" {
|
||||||
|
t.Fatalf("calls = %v, held = %v", f.calls, f.keys)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestEnsureKeysNeverTouchesAKeyTheOperatorAdded(t *testing.T) {
|
||||||
|
f := &keyForge{keys: map[string][]heldKey{"agent-g14": {
|
||||||
|
{ID: 3, Title: "my laptop", Key: keyB},
|
||||||
|
{ID: 7, Title: "mesh:g14.nox", Key: keyA},
|
||||||
|
}}, next: 7}
|
||||||
|
a := keyAccounts(t, f)
|
||||||
|
// Nothing wanted any more: the mesh's key goes, the operator's stays.
|
||||||
|
if err := a.EnsureKeys(t.Context(), "agent-g14", nil); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
held := f.keys["agent-g14"]
|
||||||
|
if len(held) != 1 || held[0].Title != "my laptop" {
|
||||||
|
t.Fatalf("held = %v", held)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestEnsureKeysLeavesAWantedKeyTheOperatorAlreadyHoldsUnderAnotherTitle(t *testing.T) {
|
||||||
|
f := &keyForge{keys: map[string][]heldKey{"agent-g14": {{ID: 3, Title: "by hand", Key: keyA}}}, next: 3}
|
||||||
|
a := keyAccounts(t, f)
|
||||||
|
if err := a.EnsureKeys(t.Context(), "agent-g14", []WantedKey{{Title: "mesh:g14.nox", Key: keyA + " nox@g14 (mesh)"}}); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
if len(f.posts) != 0 || len(f.keys["agent-g14"]) != 1 || f.keys["agent-g14"][0].Title != "by hand" {
|
||||||
|
t.Fatalf("posts = %v, held = %v", f.posts, f.keys)
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user