enki
1933306392
- fix two stale nzbstr comment refs - migration 002: drop api_keys.visibility and curation_set - remove Visibility from APIKey struct, GetAPIKey, CreateAPIKey, CLI - omit torznab size/seeders/peers attrs when data is absent - reset relay backoff on successful connection (>= 30s uptime) - use last_event from relays table as since on reconnect - fix TestSearchWithResults to actually query the test server DB
96 lines
2.2 KiB
Go
96 lines
2.2 KiB
Go
// Command kindexr-cli is the admin CLI for kindexr.
|
|
//
|
|
// TODO (Phase 3): implement the following sub-commands:
|
|
// - apikey list
|
|
// - apikey revoke <key>
|
|
// - relay add <url>
|
|
// - relay remove <url>
|
|
// - relay list
|
|
// - publisher trust <pubkey>
|
|
// - publisher block <pubkey>
|
|
// - db stats
|
|
package main
|
|
|
|
import (
|
|
"context"
|
|
"crypto/rand"
|
|
"encoding/hex"
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
|
|
"git.utn.lol/enki/kindexr/internal/config"
|
|
"git.utn.lol/enki/kindexr/internal/db"
|
|
)
|
|
|
|
func main() {
|
|
if len(os.Args) < 3 {
|
|
fmt.Fprintln(os.Stderr, "usage: kindexr-cli <command> <subcommand> [flags]")
|
|
fmt.Fprintln(os.Stderr, " kindexr-cli apikey create --label <name>")
|
|
os.Exit(1)
|
|
}
|
|
|
|
switch os.Args[1] {
|
|
case "apikey":
|
|
runAPIKey(os.Args[2:])
|
|
default:
|
|
fmt.Fprintf(os.Stderr, "unknown command: %s\n", os.Args[1])
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
func runAPIKey(args []string) {
|
|
if len(args) < 1 {
|
|
fmt.Fprintln(os.Stderr, "usage: kindexr-cli apikey create --label <name>")
|
|
os.Exit(1)
|
|
}
|
|
switch args[0] {
|
|
case "create":
|
|
runAPIKeyCreate(args[1:])
|
|
default:
|
|
fmt.Fprintf(os.Stderr, "unknown apikey subcommand: %s\n", args[0])
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
func runAPIKeyCreate(args []string) {
|
|
fs := flag.NewFlagSet("apikey create", flag.ExitOnError)
|
|
configPath := fs.String("config", "/etc/kindexr/config.yaml", "path to config file")
|
|
label := fs.String("label", "", "key label (e.g. sonarr, radarr)")
|
|
if err := fs.Parse(args); err != nil {
|
|
os.Exit(1)
|
|
}
|
|
|
|
if *label == "" {
|
|
fmt.Fprintln(os.Stderr, "error: --label is required")
|
|
os.Exit(1)
|
|
}
|
|
|
|
cfg, err := config.Load(*configPath)
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "error loading config: %v\n", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
database, err := db.Open(cfg.Database.Path)
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "error opening database: %v\n", err)
|
|
os.Exit(1)
|
|
}
|
|
defer database.Close()
|
|
|
|
keyBytes := make([]byte, 32)
|
|
if _, err := rand.Read(keyBytes); err != nil {
|
|
fmt.Fprintf(os.Stderr, "error generating key: %v\n", err)
|
|
os.Exit(1)
|
|
}
|
|
key := hex.EncodeToString(keyBytes)
|
|
|
|
if err := database.CreateAPIKey(context.Background(), key, *label); err != nil {
|
|
fmt.Fprintf(os.Stderr, "error creating api key: %v\n", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
fmt.Println(key)
|
|
}
|