nullkey
1b20b06f5e
Output ergonomics:
- `--json` accepts a comma-separated field list (gh-parity); selects
named keys from the per-command payload. Bare `--json` keeps the
full shape.
- `--jq <expr>` evaluates a gojq expression over the JSON; pairs with
`--json field-list` so projection runs before jq.
- `--version` is a global cobra flag in addition to the `version`
subcommand; both render the same line.
- Per-command `--help` now renders the available JSON field list under
"JSON fields available via `--json id,name,...`" (field-discovery
parity with gh / kubectl `-o jsonpath`).
New commands:
- `auth token` — print the active context's credential to stdout for
shell command substitution (`WEKNORA_TOKEN=$(weknora auth token)`).
Default: raw secret, no trailing newline. `--json` emits
`{token, mode, context}`.
- `doc upload --from-url <URL>` — ingest a remote URL via the SDK
`CreateKnowledgeFromURL`. `--name` forwarded as `FileName` so the
server's known-extension heuristic upgrades crawl-mode to
file-download-mode where appropriate.
Includes the simplify post-review polish pass (field-filter unit
tests, --json/--jq compose check, agent_help copy fixes).
59 lines
1.9 KiB
Go
59 lines
1.9 KiB
Go
// Package auth holds the cobra commands for authentication
|
|
// (login / logout / list / refresh / status / token).
|
|
package auth
|
|
|
|
import (
|
|
"github.com/spf13/cobra"
|
|
|
|
"github.com/Tencent/WeKnora/cli/internal/cmdutil"
|
|
)
|
|
|
|
// Credential-mode tokens used in the JSON envelope of auth list / login /
|
|
// status / token. The string names describe the HTTP credential type rather
|
|
// than the login flow (e.g. JWT → bearer regardless of whether it was
|
|
// obtained via password or refresh) so an agent can branch directly on the
|
|
// header to construct: `Authorization: Bearer <token>` (ModeBearer) or
|
|
// `X-API-Key: <token>` (ModeAPIKey).
|
|
const (
|
|
ModeBearer = "bearer"
|
|
ModeAPIKey = "api-key"
|
|
ModeUnknown = "unknown"
|
|
)
|
|
|
|
// modeFromRefs maps the per-context TokenRef / APIKeyRef presence to a
|
|
// canonical credential-mode token. Bearer wins when both are present —
|
|
// matches the precedence in cmdutil.buildClient.
|
|
func modeFromRefs(apiKeyRef, tokenRef string) string {
|
|
switch {
|
|
case tokenRef != "":
|
|
return ModeBearer
|
|
case apiKeyRef != "":
|
|
return ModeAPIKey
|
|
default:
|
|
return ModeUnknown
|
|
}
|
|
}
|
|
|
|
// NewCmdAuth builds the `weknora auth` command tree and registers its
|
|
// subcommands. Called from cli/cmd/root.go.
|
|
func NewCmdAuth(f *cmdutil.Factory) *cobra.Command {
|
|
cmd := &cobra.Command{
|
|
Use: "auth",
|
|
Short: "Manage authentication credentials and contexts",
|
|
// NoArgs makes cobra emit its canonical `unknown command "X" for
|
|
// "weknora auth"` for any positional, which mapCobraError tags as
|
|
// FlagError → exit 2. Run (not RunE) is required: a parent with
|
|
// neither Run nor RunE short-circuits to help and skips Args
|
|
// validation entirely.
|
|
Args: cobra.NoArgs,
|
|
Run: func(c *cobra.Command, _ []string) { _ = c.Help() },
|
|
}
|
|
cmd.AddCommand(NewCmdLogin(f, nil))
|
|
cmd.AddCommand(NewCmdLogout(f))
|
|
cmd.AddCommand(NewCmdList(f))
|
|
cmd.AddCommand(NewCmdRefresh(f))
|
|
cmd.AddCommand(NewCmdStatus(f))
|
|
cmd.AddCommand(NewCmdToken(f))
|
|
return cmd
|
|
}
|