OwnCord/Server/token_cli.go

package main

import (
	"context"
	"flag"
	"fmt"
	"os"
	"strconv"
	"text/tabwriter"
	"time"

	"github.com/owncord/server/auth"
	"github.com/owncord/server/config"
	"github.com/owncord/server/db"
)

// runTokenCLI implements `server token <create|list|revoke>`. It operates
// directly against the database — no HTTP, no login — so an operator can mint
// the first API token without any existing credential (the bootstrap path).
// Returns a process exit code.
func runTokenCLI(args []string) int {
	if len(args) == 0 {
		tokenUsage()
		return 2
	}

	cfg, err := config.Load("config.yaml")
	if err != nil {
		fmt.Fprintf(os.Stderr, "error: load config: %v\n", err)
		return 1
	}
	database, err := db.Open(cfg.Database.Path)
	if err != nil {
		fmt.Fprintf(os.Stderr, "error: open database: %v\n", err)
		return 1
	}
	defer database.Close() //nolint:errcheck
	// Idempotent: ensures the api_tokens table exists even if the server has
	// never started against this database.
	if err := db.Migrate(database); err != nil {
		fmt.Fprintf(os.Stderr, "error: migrate: %v\n", err)
		return 1
	}

	ctx := context.Background()
	switch args[0] {
	case "create":
		return tokenCreate(ctx, database, args[1:])
	case "list":
		return tokenList(ctx, database, args[1:])
	case "revoke":
		return tokenRevoke(ctx, database, args[1:])
	default:
		fmt.Fprintf(os.Stderr, "unknown token subcommand %q\n", args[0])
		tokenUsage()
		return 2
	}
}

func tokenUsage() {
	fmt.Fprint(os.Stderr, `usage: server token <command>

Commands:
  create --label <name> [--user <username>] [--expires <dur>]
        Mint a new API token. Prints the raw token once to stdout — store it
        now, it is never recoverable. Defaults to the owner account and no
        expiry. --expires accepts a Go duration, e.g. 720h.
  list
        List API tokens (never prints raw tokens).
  revoke <id|label>
        Revoke a token by numeric id or by label.
`)
}

func tokenCreate(ctx context.Context, database *db.DB, args []string) int {
	fs := flag.NewFlagSet("token create", flag.ContinueOnError)
	label := fs.String("label", "", "human-readable label (required)")
	username := fs.String("user", "", "username to bind the token to (default: owner)")
	expires := fs.Duration("expires", 0, "validity duration, e.g. 720h (default: never)")
	if err := fs.Parse(args); err != nil {
		return 2
	}
	if *label == "" {
		fmt.Fprintln(os.Stderr, "error: --label is required")
		return 2
	}

	var user *db.User
	var err error
	if *username != "" {
		user, err = database.GetUserByUsername(ctx, *username)
	} else {
		user, err = database.GetOwnerUser(ctx)
	}
	if err != nil {
		fmt.Fprintf(os.Stderr, "error: look up user: %v\n", err)
		return 1
	}
	if user == nil {
		if *username != "" {
			fmt.Fprintf(os.Stderr, "error: no user named %q\n", *username)
		} else {
			fmt.Fprintln(os.Stderr, "error: no users exist yet — create the owner account first")
		}
		return 1
	}

	raw, err := auth.GenerateToken()
	if err != nil {
		fmt.Fprintf(os.Stderr, "error: generate token: %v\n", err)
		return 1
	}
	var expiresAt *time.Time
	if *expires > 0 {
		t := time.Now().Add(*expires)
		expiresAt = &t
	}
	id, err := database.CreateAPIToken(ctx, user.ID, auth.HashToken(raw), *label, expiresAt)
	if err != nil {
		fmt.Fprintf(os.Stderr, "error: create token: %v\n", err)
		return 1
	}
	db.WriteAudit(ctx, database, user.ID, "api_token_create", "api_token", id, *label)

	// Metadata to stderr, raw token alone to stdout — so `... | tail -1` or a
	// capture pipe gets exactly the token.
	fmt.Fprintf(os.Stderr, "Created API token #%d for user %q (label %q).\n", id, user.Username, *label)
	fmt.Fprintln(os.Stderr, "Store this token now — it is shown only once:")
	fmt.Println(raw)
	return 0
}

func tokenList(ctx context.Context, database *db.DB, args []string) int {
	fs := flag.NewFlagSet("token list", flag.ContinueOnError)
	if err := fs.Parse(args); err != nil {
		return 2
	}
	tokens, err := database.ListAPITokens(ctx)
	if err != nil {
		fmt.Fprintf(os.Stderr, "error: list tokens: %v\n", err)
		return 1
	}
	if len(tokens) == 0 {
		fmt.Println("no API tokens")
		return 0
	}
	tw := tabwriter.NewWriter(os.Stdout, 0, 2, 2, ' ', 0)
	// Buffer-write errors surface via tw.Flush() below, which is checked.
	_, _ = fmt.Fprintln(tw, "ID\tUSER\tLABEL\tCREATED\tLAST USED\tEXPIRES\tREVOKED")
	for _, t := range tokens {
		_, _ = fmt.Fprintf(tw, "%d\t%s\t%s\t%s\t%s\t%s\t%s\n",
			t.ID, t.Username, t.Label, t.CreatedAt,
			orDash(t.LastUsed), orDash(t.ExpiresAt), orDash(t.RevokedAt))
	}
	if err := tw.Flush(); err != nil {
		fmt.Fprintf(os.Stderr, "error: %v\n", err)
		return 1
	}
	return 0
}

func tokenRevoke(ctx context.Context, database *db.DB, args []string) int {
	fs := flag.NewFlagSet("token revoke", flag.ContinueOnError)
	if err := fs.Parse(args); err != nil {
		return 2
	}
	rest := fs.Args()
	if len(rest) != 1 {
		fmt.Fprintln(os.Stderr, "error: revoke takes exactly one argument (id or label)")
		return 2
	}
	arg := rest[0]

	var affected int64
	var err error
	if id, perr := strconv.ParseInt(arg, 10, 64); perr == nil {
		affected, err = database.RevokeAPIToken(ctx, id)
		if err == nil && affected > 0 {
			db.WriteAudit(ctx, database, 0, "api_token_revoke", "api_token", id, arg)
		}
	} else {
		affected, err = database.RevokeAPITokenByLabel(ctx, arg)
		if err == nil && affected > 0 {
			db.WriteAudit(ctx, database, 0, "api_token_revoke", "api_token", 0, arg)
		}
	}
	if err != nil {
		fmt.Fprintf(os.Stderr, "error: revoke token: %v\n", err)
		return 1
	}
	if affected == 0 {
		fmt.Fprintf(os.Stderr, "no active token matched %q\n", arg)
		return 1
	}
	fmt.Printf("revoked %d token(s)\n", affected)
	return 0
}

// orDash renders a nullable timestamp column for the list table.
func orDash(s *string) string {
	if s == nil || *s == "" {
		return "-"
	}
	return *s
}