feat: lodge server, client and admin

This commit is contained in:
hh committed 2026-09-19 23:33:24 +02:00
commit 95fc50e2c2
84 files changed
+7625

No files matched your search

+24
View File
@@ -0,0 +1,24 @@
#!/bin/sh
set -eu
os=$(uname -s | tr '[:upper:]' '[:lower:]')
arch=$(uname -m)
case "$arch" in
x86_64 | amd64) arch=amd64 ;;
aarch64 | arm64) arch=arm64 ;;
*) echo "lodge: unsupported architecture $arch" >&2; exit 1 ;;
esac
bin="$HOME/.local/bin"
mkdir -p "$bin"
curl -fsSL "{{URL}}/dl/lodge-$os-$arch" -o "$bin/lodge.new"
chmod +x "$bin/lodge.new"
mv "$bin/lodge.new" "$bin/lodge"
"$bin/lodge" login "{{URL}}" "{{CODE}}"
"$bin/lodge" install
case ":$PATH:" in
*":$bin:"*) ;;
*) echo "lodge: add $bin to your PATH to use the lodge command" ;;
esac
+409
View File
@@ -0,0 +1,409 @@
package server
import (
"context"
"crypto/sha256"
"crypto/subtle"
_ "embed"
"encoding/hex"
"encoding/json"
"errors"
"fmt"
"io"
"net/http"
"os"
"path/filepath"
"regexp"
"strconv"
"strings"
"sync"
"time"
"lodge/internal/protocol"
)
const (
heartbeat = 30 * time.Second
sessionCookie = "lodge_session"
loginDelay = time.Second
)
var (
//go:embed install.sh
installScript string
repoKeyPattern = regexp.MustCompile(`^(tmp-)?[0-9a-f]{40,64}$`)
hashPattern = regexp.MustCompile(`^[0-9a-f]{64}$`)
codePattern = regexp.MustCompile(`^[0-9a-f]+$`)
)
type Config struct {
AdminToken string
PublicURL string
StaticDir string
ClientsDir string
}
type Server struct {
config Config
store *Store
mu sync.Mutex
subscribers map[chan protocol.Event]struct{}
}
type machineKey struct{}
func New(store *Store, config Config) http.Handler {
s := &Server{config: config, store: store, subscribers: map[chan protocol.Event]struct{}{}}
mux := http.NewServeMux()
mux.HandleFunc("GET /i/{code}", s.install)
mux.Handle("GET /dl/", http.StripPrefix("/dl/", http.FileServer(http.Dir(config.ClientsDir))))
mux.HandleFunc("POST /api/claim", s.claim)
mux.HandleFunc("POST /api/session", s.login)
mux.HandleFunc("GET /api/session", s.admin(noContent))
mux.HandleFunc("DELETE /api/session", s.logout)
mux.HandleFunc("POST /api/codes", s.admin(s.createCode))
mux.HandleFunc("GET /api/machines", s.admin(s.machines))
mux.HandleFunc("DELETE /api/machines/{id}", s.admin(s.deleteMachine))
mux.HandleFunc("DELETE /api/repos/{key}", s.admin(s.deleteRepo))
mux.HandleFunc("GET /api/repos", s.auth(s.repos))
mux.HandleFunc("GET /api/repos/{key}/state", s.auth(s.state))
mux.HandleFunc("GET /api/repos/{key}/revisions", s.auth(s.revisions))
mux.HandleFunc("POST /api/repos/{key}/revisions", s.auth(s.push))
mux.HandleFunc("GET /api/blobs/{hash}", s.auth(s.blob))
mux.HandleFunc("PUT /api/blobs/{hash}", s.auth(s.putBlob))
mux.HandleFunc("GET /api/events", s.auth(s.events))
mux.HandleFunc("GET /api/", http.NotFound)
mux.HandleFunc("GET /", s.static)
return mux
}
func writeJSON(w http.ResponseWriter, status int, value any) {
w.Header().Set("Content-Type", "application/json")
w.WriteHeader(status)
_ = json.NewEncoder(w).Encode(value)
}
func writeError(w http.ResponseWriter, err error) {
status := http.StatusInternalServerError
switch {
case errors.Is(err, ErrConflict):
status = http.StatusConflict
case errors.Is(err, ErrNotFound):
status = http.StatusNotFound
}
http.Error(w, err.Error(), status)
}
func (s *Server) isAdminToken(token string) bool {
return s.config.AdminToken != "" && subtle.ConstantTimeCompare([]byte(token), []byte(s.config.AdminToken)) == 1
}
func (s *Server) isAdmin(r *http.Request) bool {
if cookie, err := r.Cookie(sessionCookie); err == nil && s.store.SessionValid(r.Context(), cookie.Value) {
return true
}
return s.isAdminToken(bearer(r))
}
func noContent(w http.ResponseWriter, _ *http.Request) {
w.WriteHeader(http.StatusNoContent)
}
func bearer(r *http.Request) string {
token, _ := strings.CutPrefix(r.Header.Get("Authorization"), "Bearer ")
return token
}
func (s *Server) admin(next http.HandlerFunc) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
if !s.isAdmin(r) {
http.Error(w, "unauthorized", http.StatusUnauthorized)
return
}
next(w, r)
}
}
func (s *Server) auth(next http.HandlerFunc) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
var machineID int64
if !s.isAdmin(r) {
id, err := s.store.MachineID(r.Context(), bearer(r))
if err != nil {
http.Error(w, "unauthorized", http.StatusUnauthorized)
return
}
machineID = id
}
if key := r.PathValue("key"); key != "" && !repoKeyPattern.MatchString(key) {
http.Error(w, "bad repository key", http.StatusBadRequest)
return
}
next(w, r.WithContext(context.WithValue(r.Context(), machineKey{}, machineID)))
}
}
func (s *Server) publicURL(r *http.Request) string {
if s.config.PublicURL != "" {
return strings.TrimSuffix(s.config.PublicURL, "/")
}
scheme := r.Header.Get("X-Forwarded-Proto")
if scheme == "" {
scheme = "http"
}
return scheme + "://" + r.Host
}
func (s *Server) install(w http.ResponseWriter, r *http.Request) {
code := r.PathValue("code")
if !codePattern.MatchString(code) || !s.store.CodeValid(r.Context(), code) {
http.Error(w, "echo 'lodge: this install link has expired, create a new one' >&2; exit 1", http.StatusNotFound)
return
}
w.Header().Set("Content-Type", "text/x-shellscript")
_, _ = strings.NewReplacer("{{URL}}", s.publicURL(r), "{{CODE}}", code).WriteString(w, installScript)
}
func (s *Server) setSession(w http.ResponseWriter, r *http.Request, token string, expires time.Time) {
http.SetCookie(w, &http.Cookie{
Name: sessionCookie,
Value: token,
Path: "/",
Expires: expires,
HttpOnly: true,
Secure: strings.HasPrefix(s.publicURL(r), "https://"),
SameSite: http.SameSiteStrictMode,
})
}
func (s *Server) login(w http.ResponseWriter, r *http.Request) {
var req protocol.Login
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
http.Error(w, err.Error(), http.StatusBadRequest)
return
}
if !s.isAdminToken(req.Password) {
time.Sleep(loginDelay)
http.Error(w, "wrong password", http.StatusUnauthorized)
return
}
token, expires, err := s.store.CreateSession(r.Context())
if err != nil {
writeError(w, err)
return
}
s.setSession(w, r, token, expires)
w.WriteHeader(http.StatusNoContent)
}
func (s *Server) logout(w http.ResponseWriter, r *http.Request) {
if cookie, err := r.Cookie(sessionCookie); err == nil {
_ = s.store.DeleteSession(r.Context(), cookie.Value)
}
s.setSession(w, r, "", time.Unix(0, 0))
w.WriteHeader(http.StatusNoContent)
}
func (s *Server) machines(w http.ResponseWriter, r *http.Request) {
machines, err := s.store.Machines(r.Context())
if err != nil {
writeError(w, err)
return
}
writeJSON(w, http.StatusOK, machines)
}
func (s *Server) deleteMachine(w http.ResponseWriter, r *http.Request) {
id, err := strconv.ParseInt(r.PathValue("id"), 10, 64)
if err == nil {
err = s.store.DeleteMachine(r.Context(), id)
}
if err != nil {
writeError(w, ErrNotFound)
return
}
w.WriteHeader(http.StatusNoContent)
}
func (s *Server) deleteRepo(w http.ResponseWriter, r *http.Request) {
key := r.PathValue("key")
if !repoKeyPattern.MatchString(key) {
http.Error(w, "bad repository key", http.StatusBadRequest)
return
}
if err := s.store.DeleteRepo(r.Context(), key); err != nil {
writeError(w, err)
return
}
s.publish(protocol.Event{Repo: key})
w.WriteHeader(http.StatusNoContent)
}
func (s *Server) createCode(w http.ResponseWriter, r *http.Request) {
code, expires, err := s.store.CreateCode(r.Context())
if err != nil {
writeError(w, err)
return
}
writeJSON(w, http.StatusCreated, protocol.CodeResponse{
Code: code,
Command: fmt.Sprintf("curl -fsSL %s/i/%s | sh", s.publicURL(r), code),
ExpiresAt: expires.UnixMilli(),
})
}
func (s *Server) claim(w http.ResponseWriter, r *http.Request) {
var req protocol.ClaimRequest
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
http.Error(w, err.Error(), http.StatusBadRequest)
return
}
token, err := s.store.Claim(r.Context(), req.Code, req.Name)
if err != nil {
writeError(w, err)
return
}
writeJSON(w, http.StatusCreated, protocol.ClaimResponse{Token: token})
}
func (s *Server) repos(w http.ResponseWriter, r *http.Request) {
repos, err := s.store.Repos(r.Context())
if err != nil {
writeError(w, err)
return
}
writeJSON(w, http.StatusOK, repos)
}
func (s *Server) state(w http.ResponseWriter, r *http.Request) {
state, err := s.store.State(r.Context(), r.PathValue("key"), r.URL.Query().Get("at"))
if err != nil {
writeError(w, err)
return
}
writeJSON(w, http.StatusOK, state)
}
func (s *Server) revisions(w http.ResponseWriter, r *http.Request) {
revisions, err := s.store.Revisions(r.Context(), r.PathValue("key"), r.URL.Query().Get("path"))
if err != nil {
writeError(w, err)
return
}
writeJSON(w, http.StatusOK, revisions)
}
func (s *Server) push(w http.ResponseWriter, r *http.Request) {
var push protocol.Push
if err := json.NewDecoder(r.Body).Decode(&push); err != nil {
http.Error(w, err.Error(), http.StatusBadRequest)
return
}
if len(push.Changes) == 0 {
http.Error(w, "no changes", http.StatusBadRequest)
return
}
for _, change := range push.Changes {
validHash := change.Hash == "" || hashPattern.MatchString(change.Hash)
if !protocol.ValidSpace(change.Space) || !protocol.SafePath(change.Path) || !validHash || change.Hash == change.Base {
http.Error(w, "bad change for "+change.Path, http.StatusBadRequest)
return
}
}
key := r.PathValue("key")
machineID, _ := r.Context().Value(machineKey{}).(int64)
seq, err := s.store.Push(r.Context(), key, machineID, push)
if err != nil {
writeError(w, err)
return
}
s.publish(protocol.Event{Repo: key, Seq: seq})
writeJSON(w, http.StatusCreated, protocol.Event{Repo: key, Seq: seq})
}
func (s *Server) blob(w http.ResponseWriter, r *http.Request) {
data, err := s.store.Blob(r.Context(), r.PathValue("hash"))
if err != nil {
writeError(w, err)
return
}
w.Header().Set("Content-Type", "application/octet-stream")
w.Header().Set("Cache-Control", "private, max-age=31536000, immutable")
_, _ = w.Write(data)
}
func (s *Server) putBlob(w http.ResponseWriter, r *http.Request) {
data, err := io.ReadAll(http.MaxBytesReader(w, r.Body, protocol.MaxBlobSize))
if err != nil {
http.Error(w, err.Error(), http.StatusRequestEntityTooLarge)
return
}
sum := sha256.Sum256(data)
hash := hex.EncodeToString(sum[:])
if hash != r.PathValue("hash") {
http.Error(w, "content does not match hash", http.StatusBadRequest)
return
}
if err := s.store.PutBlob(r.Context(), hash, data); err != nil {
writeError(w, err)
return
}
w.WriteHeader(http.StatusNoContent)
}
func (s *Server) publish(event protocol.Event) {
s.mu.Lock()
defer s.mu.Unlock()
for subscriber := range s.subscribers {
select {
case subscriber <- event:
default:
}
}
}
func (s *Server) events(w http.ResponseWriter, r *http.Request) {
flusher, ok := w.(http.Flusher)
if !ok {
http.Error(w, "streaming unsupported", http.StatusInternalServerError)
return
}
subscriber := make(chan protocol.Event, 16)
s.mu.Lock()
s.subscribers[subscriber] = struct{}{}
s.mu.Unlock()
defer func() {
s.mu.Lock()
delete(s.subscribers, subscriber)
s.mu.Unlock()
}()
w.Header().Set("Content-Type", "text/event-stream")
w.Header().Set("Cache-Control", "no-cache")
fmt.Fprint(w, ": connected\n\n")
flusher.Flush()
ticker := time.NewTicker(heartbeat)
defer ticker.Stop()
for {
select {
case <-r.Context().Done():
return
case <-ticker.C:
fmt.Fprint(w, ": ping\n\n")
case event := <-subscriber:
data, _ := json.Marshal(event)
fmt.Fprintf(w, "data: %s\n\n", data)
}
flusher.Flush()
}
}
func (s *Server) static(w http.ResponseWriter, r *http.Request) {
file := filepath.Join(s.config.StaticDir, filepath.FromSlash(filepath.Clean("/"+r.URL.Path)))
if info, err := os.Stat(file); err != nil || info.IsDir() {
file = filepath.Join(s.config.StaticDir, "index.html")
}
http.ServeFile(w, r, file)
}
+368
View File
@@ -0,0 +1,368 @@
package server
import (
"context"
"crypto/rand"
"crypto/sha256"
"database/sql"
"encoding/hex"
"errors"
"fmt"
"strconv"
"time"
"lodge/internal/protocol"
_ "modernc.org/sqlite"
)
const (
codeLifetime = 10 * time.Minute
sessionLifetime = 30 * 24 * time.Hour
revisionLimit = 500
schema = `
create table if not exists machines (
id integer primary key,
name text not null,
token_hash text not null unique,
created_at integer not null
);
create table if not exists codes (
code text primary key,
expires_at integer not null
);
create table if not exists sessions (
token_hash text primary key,
expires_at integer not null
);
create table if not exists repos (
key text primary key,
name text not null,
seq integer not null,
time integer not null
);
create table if not exists blobs (
hash text primary key,
data blob not null
);
create table if not exists revisions (
repo text not null,
seq integer not null,
machine_id integer,
time integer not null,
main_sha text not null,
main_branch text not null,
primary key (repo, seq)
);
create table if not exists changes (
repo text not null,
seq integer not null,
space text not null,
path text not null,
base text not null,
hash text not null,
primary key (repo, seq, space, path)
);
create index if not exists changes_by_path on changes (repo, space, path, seq);
`
)
var (
ErrConflict = errors.New("state changed since base")
ErrNotFound = errors.New("not found")
)
type Store struct {
db *sql.DB
}
func OpenStore(path string) (*Store, error) {
dsn := "file:" + path + "?_pragma=journal_mode(WAL)&_pragma=busy_timeout(5000)&_pragma=synchronous(NORMAL)"
db, err := sql.Open("sqlite", dsn)
if err != nil {
return nil, err
}
db.SetMaxOpenConns(1)
if _, err := db.Exec(schema); err != nil {
return nil, err
}
return &Store{db: db}, nil
}
func (s *Store) Close() error {
return s.db.Close()
}
func randomHex(bytes int) string {
buf := make([]byte, bytes)
if _, err := rand.Read(buf); err != nil {
panic(err)
}
return hex.EncodeToString(buf)
}
func hashToken(token string) string {
sum := sha256.Sum256([]byte(token))
return hex.EncodeToString(sum[:])
}
func (s *Store) CreateCode(ctx context.Context) (string, time.Time, error) {
code, expires := randomHex(12), time.Now().Add(codeLifetime)
_, err := s.db.ExecContext(ctx, `insert into codes (code, expires_at) values (?, ?)`, code, expires.UnixMilli())
return code, expires, err
}
func (s *Store) CodeValid(ctx context.Context, code string) bool {
var found int
err := s.db.QueryRowContext(ctx, `select 1 from codes where code = ? and expires_at > ?`, code, time.Now().UnixMilli()).Scan(&found)
return err == nil
}
func (s *Store) Claim(ctx context.Context, code, name string) (string, error) {
now := time.Now().UnixMilli()
res, err := s.db.ExecContext(ctx, `delete from codes where code = ? and expires_at > ?`, code, now)
if err != nil {
return "", err
}
if deleted, _ := res.RowsAffected(); deleted == 0 {
return "", ErrNotFound
}
token := randomHex(32)
_, err = s.db.ExecContext(ctx, `insert into machines (name, token_hash, created_at) values (?, ?, ?)`, name, hashToken(token), now)
return token, err
}
func (s *Store) MachineID(ctx context.Context, token string) (int64, error) {
var id int64
err := s.db.QueryRowContext(ctx, `select id from machines where token_hash = ?`, hashToken(token)).Scan(&id)
return id, err
}
func (s *Store) CreateSession(ctx context.Context) (string, time.Time, error) {
token, expires := randomHex(32), time.Now().Add(sessionLifetime)
_, err := s.db.ExecContext(ctx, `insert into sessions (token_hash, expires_at) values (?, ?)`, hashToken(token), expires.UnixMilli())
return token, expires, err
}
func (s *Store) SessionValid(ctx context.Context, token string) bool {
var found int
err := s.db.QueryRowContext(ctx, `select 1 from sessions where token_hash = ? and expires_at > ?`, hashToken(token), time.Now().UnixMilli()).Scan(&found)
return err == nil
}
func (s *Store) DeleteSession(ctx context.Context, token string) error {
_, err := s.db.ExecContext(ctx, `delete from sessions where token_hash = ? or expires_at <= ?`, hashToken(token), time.Now().UnixMilli())
return err
}
func (s *Store) Machines(ctx context.Context) ([]protocol.Machine, error) {
rows, err := s.db.QueryContext(ctx, `
select m.id, m.name, m.created_at, coalesce(max(r.time), 0), count(r.seq)
from machines m left join revisions r on r.machine_id = m.id
group by m.id order by m.created_at`)
if err != nil {
return nil, err
}
defer rows.Close()
machines := []protocol.Machine{}
for rows.Next() {
var machine protocol.Machine
if err := rows.Scan(&machine.ID, &machine.Name, &machine.CreatedAt, &machine.LastPush, &machine.Revisions); err != nil {
return nil, err
}
machines = append(machines, machine)
}
return machines, rows.Err()
}
func (s *Store) DeleteMachine(ctx context.Context, id int64) error {
res, err := s.db.ExecContext(ctx, `delete from machines where id = ?`, id)
if err != nil {
return err
}
if deleted, _ := res.RowsAffected(); deleted == 0 {
return ErrNotFound
}
return nil
}
func (s *Store) DeleteRepo(ctx context.Context, repo string) error {
tx, err := s.db.BeginTx(ctx, nil)
if err != nil {
return err
}
defer tx.Rollback()
if _, err := tx.ExecContext(ctx, `delete from changes where repo = ?`, repo); err != nil {
return err
}
if _, err := tx.ExecContext(ctx, `delete from revisions where repo = ?`, repo); err != nil {
return err
}
res, err := tx.ExecContext(ctx, `delete from repos where key = ?`, repo)
if err != nil {
return err
}
if deleted, _ := res.RowsAffected(); deleted == 0 {
return ErrNotFound
}
if _, err := tx.ExecContext(ctx, `delete from blobs where hash not in (select hash from changes)`); err != nil {
return err
}
return tx.Commit()
}
func (s *Store) PutBlob(ctx context.Context, hash string, data []byte) error {
_, err := s.db.ExecContext(ctx, `insert or ignore into blobs (hash, data) values (?, ?)`, hash, data)
return err
}
func (s *Store) Blob(ctx context.Context, hash string) ([]byte, error) {
var data []byte
err := s.db.QueryRowContext(ctx, `select data from blobs where hash = ?`, hash).Scan(&data)
if errors.Is(err, sql.ErrNoRows) {
return nil, ErrNotFound
}
return data, err
}
func (s *Store) Repos(ctx context.Context) ([]protocol.Repo, error) {
rows, err := s.db.QueryContext(ctx, `select key, name, seq, time from repos order by time desc`)
if err != nil {
return nil, err
}
defer rows.Close()
repos := []protocol.Repo{}
for rows.Next() {
var repo protocol.Repo
if err := rows.Scan(&repo.Key, &repo.Name, &repo.Seq, &repo.Time); err != nil {
return nil, err
}
repos = append(repos, repo)
}
return repos, rows.Err()
}
func (s *Store) resolveSeq(ctx context.Context, repo, at string) (int64, error) {
var seq sql.NullInt64
var err error
if n, convErr := strconv.ParseInt(at, 10, 64); at != "" && convErr == nil {
return n, nil
}
if at == "" {
err = s.db.QueryRowContext(ctx, `select max(seq) from revisions where repo = ?`, repo).Scan(&seq)
} else {
err = s.db.QueryRowContext(ctx, `select max(seq) from revisions where repo = ? and main_sha like ? || '%'`, repo, at).Scan(&seq)
if err == nil && !seq.Valid {
return 0, ErrNotFound
}
}
return seq.Int64, err
}
func (s *Store) State(ctx context.Context, repo, at string) (protocol.State, error) {
state := protocol.State{Spaces: map[string]protocol.Manifest{}}
seq, err := s.resolveSeq(ctx, repo, at)
if err != nil {
return state, err
}
state.Seq = seq
rows, err := s.db.QueryContext(ctx, `
select c.space, c.path, c.hash, r.time
from changes c join revisions r on r.repo = c.repo and r.seq = c.seq
where c.repo = ?1 and c.hash != '' and c.seq = (
select max(seq) from changes
where repo = c.repo and space = c.space and path = c.path and seq <= ?2
)`, repo, seq)
if err != nil {
return state, err
}
defer rows.Close()
for rows.Next() {
var space, path string
var file protocol.File
if err := rows.Scan(&space, &path, &file.Hash, &file.Time); err != nil {
return state, err
}
if state.Spaces[space] == nil {
state.Spaces[space] = protocol.Manifest{}
}
state.Spaces[space][path] = file
}
return state, rows.Err()
}
func (s *Store) Revisions(ctx context.Context, repo, path string) ([]protocol.Revision, error) {
rows, err := s.db.QueryContext(ctx, `
select r.seq, coalesce(m.name, iif(r.machine_id is null, 'web', 'revoked')), r.time, r.main_sha, r.main_branch, c.space, c.path, c.base, c.hash
from revisions r
join changes c on c.repo = r.repo and c.seq = r.seq
left join machines m on m.id = r.machine_id
where r.repo = ?1 and (?2 = '' or c.path = ?2)
order by r.seq desc, c.path
limit ?3`, repo, path, revisionLimit)
if err != nil {
return nil, err
}
defer rows.Close()
revisions := []protocol.Revision{}
for rows.Next() {
var rev protocol.Revision
var change protocol.Change
if err := rows.Scan(&rev.Seq, &rev.Machine, &rev.Time, &rev.MainSHA, &rev.MainBranch, &change.Space, &change.Path, &change.Base, &change.Hash); err != nil {
return nil, err
}
if last := len(revisions) - 1; last >= 0 && revisions[last].Seq == rev.Seq {
revisions[last].Changes = append(revisions[last].Changes, change)
continue
}
rev.Changes = []protocol.Change{change}
revisions = append(revisions, rev)
}
return revisions, rows.Err()
}
func (s *Store) Push(ctx context.Context, repo string, machineID int64, push protocol.Push) (int64, error) {
tx, err := s.db.BeginTx(ctx, nil)
if err != nil {
return 0, err
}
defer tx.Rollback()
for _, change := range push.Changes {
var current string
err := tx.QueryRowContext(ctx, `select hash from changes where repo = ? and space = ? and path = ? order by seq desc limit 1`, repo, change.Space, change.Path).Scan(&current)
if err != nil && !errors.Is(err, sql.ErrNoRows) {
return 0, err
}
if current != change.Base {
return 0, ErrConflict
}
if change.Hash == "" {
continue
}
var found int
if err := tx.QueryRowContext(ctx, `select 1 from blobs where hash = ?`, change.Hash).Scan(&found); err != nil {
return 0, fmt.Errorf("blob %s was not uploaded: %w", change.Hash, ErrNotFound)
}
}
now := time.Now().UnixMilli()
var seq int64
err = tx.QueryRowContext(ctx, `
insert into repos (key, name, seq, time) values (?, ?, 1, ?)
on conflict (key) do update set seq = seq + 1, name = excluded.name, time = excluded.time
returning seq`, repo, push.Name, now).Scan(&seq)
if err != nil {
return 0, err
}
machine := sql.NullInt64{Int64: machineID, Valid: machineID != 0}
if _, err := tx.ExecContext(ctx, `insert into revisions (repo, seq, machine_id, time, main_sha, main_branch) values (?, ?, ?, ?, ?, ?)`, repo, seq, machine, now, push.MainSHA, push.MainBranch); err != nil {
return 0, err
}
for _, change := range push.Changes {
if _, err := tx.ExecContext(ctx, `insert into changes (repo, seq, space, path, base, hash) values (?, ?, ?, ?, ?, ?)`, repo, seq, change.Space, change.Path, change.Base, change.Hash); err != nil {
return 0, err
}
}
return seq, tx.Commit()
}