Compare commits
4
Commits
631bb64109
...
main
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
d7c0205c50 | ||
|
|
196b543bee | ||
|
|
3d4e6d0939 | ||
|
|
f94fddddb1 |
@@ -23,12 +23,15 @@ auth:
|
|||||||
keys:
|
keys:
|
||||||
- id: "local-client"
|
- id: "local-client"
|
||||||
value: "replace-me"
|
value: "replace-me"
|
||||||
|
tenant_id: "local"
|
||||||
|
superadmin: true
|
||||||
description: "main local client key"
|
description: "main local client key"
|
||||||
oauth:
|
oauth:
|
||||||
clients:
|
clients:
|
||||||
- id: "oauth-client"
|
- id: "oauth-client"
|
||||||
client_id: ""
|
client_id: ""
|
||||||
client_secret: ""
|
client_secret: ""
|
||||||
|
superadmin: false
|
||||||
description: "optional OAuth client credentials"
|
description: "optional OAuth client credentials"
|
||||||
|
|
||||||
database:
|
database:
|
||||||
|
|||||||
@@ -21,6 +21,8 @@ auth:
|
|||||||
keys:
|
keys:
|
||||||
- id: "local-client"
|
- id: "local-client"
|
||||||
value: "replace-me"
|
value: "replace-me"
|
||||||
|
tenant_id: "local"
|
||||||
|
superadmin: true
|
||||||
description: "main local client key"
|
description: "main local client key"
|
||||||
oauth:
|
oauth:
|
||||||
clients:
|
clients:
|
||||||
|
|||||||
@@ -21,6 +21,8 @@ auth:
|
|||||||
keys:
|
keys:
|
||||||
- id: "local-client"
|
- id: "local-client"
|
||||||
value: "replace-me"
|
value: "replace-me"
|
||||||
|
tenant_id: "local"
|
||||||
|
superadmin: true
|
||||||
description: "main local client key"
|
description: "main local client key"
|
||||||
oauth:
|
oauth:
|
||||||
clients:
|
clients:
|
||||||
|
|||||||
@@ -0,0 +1,19 @@
|
|||||||
|
# Tenant identity management
|
||||||
|
|
||||||
|
The previous tenancy implementation used every API-key ID as its tenant key. This meant that two keys could not share data and old unscoped rows could not be reached by any key after tenancy was enabled.
|
||||||
|
|
||||||
|
Implemented a separate tenant mapping: configured and managed API keys now resolve to an assigned tenant ID, while unassigned configured keys retain their historical key-ID tenant boundary for compatibility. Added tenant, tenant-user, key-assignment, and managed-secret records, with a one-time, explicit legacy adoption endpoint that assigns only `NULL` tenant-key rows to a selected tenant.
|
||||||
|
|
||||||
|
The admin UI now has an Identity page for tenants, users, configured-key assignment, managed-key creation (secret shown once), disabling keys, and legacy adoption. Managed secrets are stored only as SHA-256 hashes. The AMCS MCP capture-thought tool was unavailable in this session, so this local log is the required fallback summary.
|
||||||
|
|
||||||
|
Configured keys also accept an optional `auth.keys[].tenant_id`. This supplies the initial tenant boundary at startup; an explicit assignment saved through the Identity UI overrides it.
|
||||||
|
|
||||||
|
Tenant-owned root records now reference `tenants(id)` through `tenant_id`; tenant scoped resources include skills, guardrails, personas, parts, traits, and character arcs. Tenant-selection in the UI is sent as `X-AMCS-Tenant-ID` for admin/ResolveSpec requests.
|
||||||
|
|
||||||
|
With explicit approval that tenancy data is disposable, the compatibility migration now renames the ownership column from `tenant_key` to `tenant_id` across the historical migration chain.
|
||||||
|
|
||||||
|
Tenant selection now centrally filters every tenant-owned ResolveSpec read, update, and delete in the admin UI, and injects the selected `tenant_id` on create requests.
|
||||||
|
|
||||||
|
Identity mutations now require a configured API key with `superadmin: true`; tenants themselves remain database-managed.
|
||||||
|
|
||||||
|
Fixed the admin sidebar tenant-scope selector flicker. Its focus handler reloaded the tenant list and disabled the native select while its picker was opening; tenants now load on sidebar mount only, so a selection remains open and usable. Validated with `pnpm check` (0 errors, 0 warnings).
|
||||||
@@ -0,0 +1,3 @@
|
|||||||
|
# ResolveSpec array scanning
|
||||||
|
|
||||||
|
Diagnosed `/api/rs/public/agent_skills` failing to scan `tags text[]`. The generated Bun model correctly uses `sqltypes.SqlStringArray`, which implements `sql.Scanner`, but ResolveSpec's read path supplied a separate scan destination instead of scanning the query's registered Bun model. Updated the vendored ResolveSpec handler to call `ScanModel`, and for single-row reads to set the single record as the query model first. This preserves Bun's model field scanner for PostgreSQL arrays.
|
||||||
@@ -0,0 +1,3 @@
|
|||||||
|
# Per-user tenancy progress review
|
||||||
|
|
||||||
|
Reviewed `docs/per-user-tenancy-plan.md` against the staged worktree. Identity, tenant mappings, schema/model migrations, and primary project/thought/file/catalog scoping are present. The plan's named hardening targets—learnings, plans, chat histories, thought-learning links, and project persona joins—still have no tenant helper predicates. Only auth/keyring tests are staged; the plan's cross-tenant store/tool and migration test matrix has not yet been added or run.
|
||||||
@@ -7,9 +7,9 @@ Gitea issue #10 asks AMCS to isolate memories by user, tenant, or workspace. The
|
|||||||
Observed code paths:
|
Observed code paths:
|
||||||
|
|
||||||
- Authentication enters through `internal/auth/middleware.go`. API-key auth uses the configured header, defaulting to `x-brain-key`; bearer tokens can resolve through OAuth `TokenStore` or API keyring; HTTP Basic resolves OAuth client credentials.
|
- Authentication enters through `internal/auth/middleware.go`. API-key auth uses the configured header, defaulting to `x-brain-key`; bearer tokens can resolve through OAuth `TokenStore` or API keyring; HTTP Basic resolves OAuth client credentials.
|
||||||
- `internal/auth/middleware.go` writes both `auth.key_id` and `tenancy.tenant_key` into the request context. The tenant key is intentionally opaque and currently equals the authenticated key id or OAuth client id.
|
- `internal/auth/middleware.go` writes both `auth.key_id` and `tenancy.tenant_id` into the request context. The tenant key is intentionally opaque and currently equals the authenticated key id or OAuth client id.
|
||||||
- `internal/tenancy/tenancy.go` exposes `WithTenantKey` and `KeyFromContext` only; callers should not infer user semantics from the tenant string.
|
- `internal/tenancy/tenancy.go` exposes `WithTenantKey` and `KeyFromContext` only; callers should not infer user semantics from the tenant string.
|
||||||
- Schema already has `tenant_key` on `projects`, `thoughts`, `stored_files`, `learnings`, `plans`, and `chat_histories` in `schema/*.dbml`.
|
- Schema already has `tenant_id` on `projects`, `thoughts`, `stored_files`, `learnings`, `plans`, and `chat_histories` in `schema/*.dbml`.
|
||||||
- `internal/store/tenancy.go` provides helper functions for appending tenant predicates to SQL.
|
- `internal/store/tenancy.go` provides helper functions for appending tenant predicates to SQL.
|
||||||
- Tenant scoping is already present in project, thought, and stored-file store paths. Some other project-owned domains still need explicit tenant enforcement.
|
- Tenant scoping is already present in project, thought, and stored-file store paths. Some other project-owned domains still need explicit tenant enforcement.
|
||||||
|
|
||||||
@@ -17,9 +17,9 @@ Observed code paths:
|
|||||||
|
|
||||||
Use the authenticated principal id as the tenant boundary:
|
Use the authenticated principal id as the tenant boundary:
|
||||||
|
|
||||||
1. API key: resolve token through `auth.Keyring.Lookup`; use returned key id as `tenant_key`.
|
1. API key: resolve token through `auth.Keyring.Lookup`; use returned key id as `tenant_id`.
|
||||||
2. OAuth bearer token: resolve token through `TokenStore.Lookup`; use returned client id/key id as `tenant_key`.
|
2. OAuth bearer token: resolve token through `TokenStore.Lookup`; use returned client id/key id as `tenant_id`.
|
||||||
3. OAuth Basic client credentials: resolve through `OAuthRegistry.Lookup`; use returned client id as `tenant_key`.
|
3. OAuth Basic client credentials: resolve through `OAuthRegistry.Lookup`; use returned client id as `tenant_id`.
|
||||||
4. Unauthenticated requests must not get tenant context and must not reach protected MCP/API handlers.
|
4. Unauthenticated requests must not get tenant context and must not reach protected MCP/API handlers.
|
||||||
|
|
||||||
Do not store raw API keys or bearer tokens in tenant columns. Store only stable configured key ids/client ids. Yes, it is less flashy than inventing an account service before breakfast, but it keeps the trust boundary small and auditable.
|
Do not store raw API keys or bearer tokens in tenant columns. Store only stable configured key ids/client ids. Yes, it is less flashy than inventing an account service before breakfast, but it keeps the trust boundary small and auditable.
|
||||||
@@ -28,7 +28,7 @@ Do not store raw API keys or bearer tokens in tenant columns. Store only stable
|
|||||||
|
|
||||||
Source-of-truth DBML changes:
|
Source-of-truth DBML changes:
|
||||||
|
|
||||||
- Add nullable `tenant_key text` to all user-owned tables:
|
- Add nullable `tenant_id text` to all user-owned tables:
|
||||||
- `projects`
|
- `projects`
|
||||||
- `thoughts`
|
- `thoughts`
|
||||||
- `stored_files`
|
- `stored_files`
|
||||||
@@ -36,9 +36,9 @@ Source-of-truth DBML changes:
|
|||||||
- `plans`
|
- `plans`
|
||||||
- `chat_histories`
|
- `chat_histories`
|
||||||
- Add tenant indexes:
|
- Add tenant indexes:
|
||||||
- single-column `tenant_key` for direct filtering
|
- single-column `tenant_id` for direct filtering
|
||||||
- `(tenant_key, name)` unique on `projects`, replacing global `projects.name` uniqueness
|
- `(tenant_id, name)` unique on `projects`, replacing global `projects.name` uniqueness
|
||||||
- `(tenant_key, project_id)` on `thoughts` for common project-scoped memory lookups
|
- `(tenant_id, project_id)` on `thoughts` for common project-scoped memory lookups
|
||||||
- Regenerate SQL migrations and generated models from DBML; do not hand-edit generated Go models except as a temporary debugging step.
|
- Regenerate SQL migrations and generated models from DBML; do not hand-edit generated Go models except as a temporary debugging step.
|
||||||
|
|
||||||
Important follow-up: `agent_skills`, `agent_guardrails`, `agent_personas`, `agent_parts`, traits, and arcs are currently global catalogs. Keep them global unless product requirements say skills/personas are private per tenant. Project join tables inherit protection through tenant-scoped project ids, but direct join queries must verify the project belongs to the tenant.
|
Important follow-up: `agent_skills`, `agent_guardrails`, `agent_personas`, `agent_parts`, traits, and arcs are currently global catalogs. Keep them global unless product requirements say skills/personas are private per tenant. Project join tables inherit protection through tenant-scoped project ids, but direct join queries must verify the project belongs to the tenant.
|
||||||
@@ -49,9 +49,9 @@ All request-facing store methods that touch tenant-owned rows must include tenan
|
|||||||
|
|
||||||
Rules:
|
Rules:
|
||||||
|
|
||||||
- Inserts must populate `tenant_key` from context.
|
- Inserts must populate `tenant_id` from context.
|
||||||
- Gets/updates/deletes by id or guid must add `and tenant_key = $n`.
|
- Gets/updates/deletes by id or guid must add `and tenant_id = $n`.
|
||||||
- Lists/searches must add `tenant_key = $n` before other filters.
|
- Lists/searches must add `tenant_id = $n` before other filters.
|
||||||
- Joins must scope the tenant-owned root table. Example: project summaries should count only thoughts that belong to the same tenant as the project, not merely thoughts with the same `project_id`.
|
- Joins must scope the tenant-owned root table. Example: project summaries should count only thoughts that belong to the same tenant as the project, not merely thoughts with the same `project_id`.
|
||||||
- Background maintenance jobs must either run per tenant, carry tenant context explicitly, or intentionally operate cross-tenant with an internal-only code path documented in the job.
|
- Background maintenance jobs must either run per tenant, carry tenant context explicitly, or intentionally operate cross-tenant with an internal-only code path documented in the job.
|
||||||
|
|
||||||
@@ -74,24 +74,24 @@ Paths needing audit/hardening:
|
|||||||
|
|
||||||
Migration approach for existing single-tenant installs:
|
Migration approach for existing single-tenant installs:
|
||||||
|
|
||||||
1. Add nullable `tenant_key` columns first; do not make them `not null` initially because existing deployments have historical rows without a principal.
|
1. Add nullable `tenant_id` columns first; do not make them `not null` initially because existing deployments have historical rows without a principal.
|
||||||
2. Backfill existing rows to a configured default tenant only if the instance enables multi-tenant mode or defines `auth.default_tenant_key`. Otherwise leave `NULL` rows visible only to unauthenticated/internal single-tenant contexts.
|
2. Backfill existing rows to a configured default tenant only if the instance enables multi-tenant mode or defines `auth.default_tenant_id`. Otherwise leave `NULL` rows visible only to unauthenticated/internal single-tenant contexts.
|
||||||
3. Replace global project-name uniqueness with `(tenant_key, name)` uniqueness. For PostgreSQL, preserve legacy `NULL` semantics carefully: multiple null-tenant projects with the same name may be possible unless a partial unique index is added for null tenant rows.
|
3. Replace global project-name uniqueness with `(tenant_id, name)` uniqueness. For PostgreSQL, preserve legacy `NULL` semantics carefully: multiple null-tenant projects with the same name may be possible unless a partial unique index is added for null tenant rows.
|
||||||
4. Add indexes concurrently where practical for production-sized tables.
|
4. Add indexes concurrently where practical for production-sized tables.
|
||||||
5. Document that after enabling auth-backed tenancy, legacy null-tenant data is not visible to authenticated tenants unless backfilled.
|
5. Document that after enabling auth-backed tenancy, legacy null-tenant data is not visible to authenticated tenants unless backfilled.
|
||||||
|
|
||||||
Recommended config addition:
|
Recommended config addition:
|
||||||
|
|
||||||
- `auth.default_tenant_key` or `tenancy.default_key` for one-time/self-hosted backfill and development.
|
- `auth.default_tenant_id` or `tenancy.default_key` for one-time/self-hosted backfill and development.
|
||||||
- Optional `tenancy.mode: single|authenticated` so operators can keep current single-user behavior deliberately instead of discovering isolation by accident. An accident in auth is just a breach with better branding.
|
- Optional `tenancy.mode: single|authenticated` so operators can keep current single-user behavior deliberately instead of discovering isolation by accident. An accident in auth is just a breach with better branding.
|
||||||
|
|
||||||
## Authorization checks
|
## Authorization checks
|
||||||
|
|
||||||
Authorization is row ownership by tenant key:
|
Authorization is row ownership by tenant key:
|
||||||
|
|
||||||
- A request may only see or mutate rows whose `tenant_key` equals the authenticated tenant key.
|
- A request may only see or mutate rows whose `tenant_id` equals the authenticated tenant key.
|
||||||
- Cross-tenant ids/gids should behave as not found, not forbidden, to avoid existence leaks.
|
- Cross-tenant ids/gids should behave as not found, not forbidden, to avoid existence leaks.
|
||||||
- Tenant key is server-derived only. Ignore any client-provided `tenant_key` fields on tool/API inputs.
|
- Tenant key is server-derived only. Ignore any client-provided `tenant_id` fields on tool/API inputs.
|
||||||
- Project id references in create/update operations must be validated against the same tenant before use. This prevents attaching a new thought/file/plan to another tenant's project id.
|
- Project id references in create/update operations must be validated against the same tenant before use. This prevents attaching a new thought/file/plan to another tenant's project id.
|
||||||
- Relationship operations must verify both endpoints are in the same tenant before creating links.
|
- Relationship operations must verify both endpoints are in the same tenant before creating links.
|
||||||
- Global catalogs can be read across tenants only if intentionally shared; project-specific associations must be tenant-checked through the project.
|
- Global catalogs can be read across tenants only if intentionally shared; project-specific associations must be tenant-checked through the project.
|
||||||
@@ -117,7 +117,7 @@ HTTP/API boundaries:
|
|||||||
UI:
|
UI:
|
||||||
|
|
||||||
- Project selector/list should naturally show tenant-scoped projects.
|
- Project selector/list should naturally show tenant-scoped projects.
|
||||||
- Admin tables for projects/thoughts/files/learnings/plans/chat histories must not show `tenant_key` as an editable field.
|
- Admin tables for projects/thoughts/files/learnings/plans/chat histories must not show `tenant_id` as an editable field.
|
||||||
- If a tenant switcher is ever added, it must map to a server-side authenticated principal or admin impersonation path, not a client-side query parameter. Obviously.
|
- If a tenant switcher is ever added, it must map to a server-side authenticated principal or admin impersonation path, not a client-side query parameter. Obviously.
|
||||||
|
|
||||||
## Test cases
|
## Test cases
|
||||||
@@ -143,7 +143,7 @@ Minimum test matrix:
|
|||||||
Assumptions:
|
Assumptions:
|
||||||
|
|
||||||
- The first tenancy boundary is authenticated principal id, not human account, org, or workspace. This is consistent with the current auth system and avoids adding an account model prematurely.
|
- The first tenancy boundary is authenticated principal id, not human account, org, or workspace. This is consistent with the current auth system and avoids adding an account model prematurely.
|
||||||
- Null `tenant_key` remains the compatibility path for existing single-tenant/internal flows.
|
- Null `tenant_id` remains the compatibility path for existing single-tenant/internal flows.
|
||||||
- Global skills/personas/guardrails remain shared catalogs until a separate product decision makes them tenant-private.
|
- Global skills/personas/guardrails remain shared catalogs until a separate product decision makes them tenant-private.
|
||||||
|
|
||||||
Blockers/decisions needed:
|
Blockers/decisions needed:
|
||||||
|
|||||||
@@ -3,7 +3,8 @@ module git.warky.dev/wdevs/amcs
|
|||||||
go 1.26.1
|
go 1.26.1
|
||||||
|
|
||||||
require (
|
require (
|
||||||
github.com/bitechdev/ResolveSpec v1.1.24
|
git.warky.dev/wdevs/relspecgo v1.0.62
|
||||||
|
github.com/bitechdev/ResolveSpec v1.1.27
|
||||||
github.com/google/jsonschema-go v0.4.3
|
github.com/google/jsonschema-go v0.4.3
|
||||||
github.com/google/uuid v1.6.0
|
github.com/google/uuid v1.6.0
|
||||||
github.com/jackc/pgx/v5 v5.9.2
|
github.com/jackc/pgx/v5 v5.9.2
|
||||||
@@ -46,7 +47,6 @@ require (
|
|||||||
github.com/prometheus/procfs v0.20.1 // indirect
|
github.com/prometheus/procfs v0.20.1 // indirect
|
||||||
github.com/puzpuzpuz/xsync/v3 v3.5.1 // indirect
|
github.com/puzpuzpuz/xsync/v3 v3.5.1 // indirect
|
||||||
github.com/redis/go-redis/v9 v9.19.0 // indirect
|
github.com/redis/go-redis/v9 v9.19.0 // indirect
|
||||||
github.com/rogpeppe/go-internal v1.14.1 // indirect
|
|
||||||
github.com/sagikazarmark/locafero v0.12.0 // indirect
|
github.com/sagikazarmark/locafero v0.12.0 // indirect
|
||||||
github.com/segmentio/asm v1.1.3 // indirect
|
github.com/segmentio/asm v1.1.3 // indirect
|
||||||
github.com/segmentio/encoding v0.5.4 // indirect
|
github.com/segmentio/encoding v0.5.4 // indirect
|
||||||
|
|||||||
@@ -2,6 +2,8 @@ dario.cat/mergo v1.0.2 h1:85+piFYR1tMbRrLcDwR18y4UKJ3aH1Tbzi24VRW1TK8=
|
|||||||
dario.cat/mergo v1.0.2/go.mod h1:E/hbnu0NxMFBjpMIE34DRGLWqDy0g5FuKDhCb31ngxA=
|
dario.cat/mergo v1.0.2/go.mod h1:E/hbnu0NxMFBjpMIE34DRGLWqDy0g5FuKDhCb31ngxA=
|
||||||
entgo.io/ent v0.14.3 h1:wokAV/kIlH9TeklJWGGS7AYJdVckr0DloWjIcO9iIIQ=
|
entgo.io/ent v0.14.3 h1:wokAV/kIlH9TeklJWGGS7AYJdVckr0DloWjIcO9iIIQ=
|
||||||
entgo.io/ent v0.14.3/go.mod h1:aDPE/OziPEu8+OWbzy4UlvWmD2/kbRuWfK2A40hcxJM=
|
entgo.io/ent v0.14.3/go.mod h1:aDPE/OziPEu8+OWbzy4UlvWmD2/kbRuWfK2A40hcxJM=
|
||||||
|
git.warky.dev/wdevs/relspecgo v1.0.62 h1:byBe2IlcwQRKsV5qXHfGITtfvGZewHlNstP+O8BGnns=
|
||||||
|
git.warky.dev/wdevs/relspecgo v1.0.62/go.mod h1:JpZBvui9dYc/QcD5TZ1wKab+4bUvPw5/rOLOA/h2F6A=
|
||||||
github.com/Azure/azure-sdk-for-go/sdk/azcore v1.7.0/go.mod h1:bjGvMhVMb+EEm3VRNQawDMUyMMjo+S5ewNjflkep/0Q=
|
github.com/Azure/azure-sdk-for-go/sdk/azcore v1.7.0/go.mod h1:bjGvMhVMb+EEm3VRNQawDMUyMMjo+S5ewNjflkep/0Q=
|
||||||
github.com/Azure/azure-sdk-for-go/sdk/azcore v1.7.1/go.mod h1:bjGvMhVMb+EEm3VRNQawDMUyMMjo+S5ewNjflkep/0Q=
|
github.com/Azure/azure-sdk-for-go/sdk/azcore v1.7.1/go.mod h1:bjGvMhVMb+EEm3VRNQawDMUyMMjo+S5ewNjflkep/0Q=
|
||||||
github.com/Azure/azure-sdk-for-go/sdk/azcore v1.11.1/go.mod h1:a6xsAQUZg+VsS3TJ05SRp524Hs4pZ/AeFSr5ENf0Yjo=
|
github.com/Azure/azure-sdk-for-go/sdk/azcore v1.11.1/go.mod h1:a6xsAQUZg+VsS3TJ05SRp524Hs4pZ/AeFSr5ENf0Yjo=
|
||||||
@@ -34,8 +36,8 @@ github.com/Microsoft/go-winio v0.6.2 h1:F2VQgta7ecxGYO8k3ZZz3RS8fVIXVxONVUPlNERo
|
|||||||
github.com/Microsoft/go-winio v0.6.2/go.mod h1:yd8OoFMLzJbo9gZq8j5qaps8bJ9aShtEA8Ipt1oGCvU=
|
github.com/Microsoft/go-winio v0.6.2/go.mod h1:yd8OoFMLzJbo9gZq8j5qaps8bJ9aShtEA8Ipt1oGCvU=
|
||||||
github.com/beorn7/perks v1.0.1 h1:VlbKKnNfV8bJzeqoa4cOKqO6bYr3WgKZxO8Z16+hsOM=
|
github.com/beorn7/perks v1.0.1 h1:VlbKKnNfV8bJzeqoa4cOKqO6bYr3WgKZxO8Z16+hsOM=
|
||||||
github.com/beorn7/perks v1.0.1/go.mod h1:G2ZrVWU2WbWT9wwq4/hrbKbnv/1ERSJQ0ibhJ6rlkpw=
|
github.com/beorn7/perks v1.0.1/go.mod h1:G2ZrVWU2WbWT9wwq4/hrbKbnv/1ERSJQ0ibhJ6rlkpw=
|
||||||
github.com/bitechdev/ResolveSpec v1.1.24 h1:+Ku3jE8ZSQ2c6IdVyqYp9CdCh4wSasCd1yLDF8GXy5U=
|
github.com/bitechdev/ResolveSpec v1.1.27 h1:qugBwR3Qoy4tNKhAyJsFszZE+kRR0gzl5w42XE3/scY=
|
||||||
github.com/bitechdev/ResolveSpec v1.1.24/go.mod h1:GF51sMRCWbAyri2WNae3IZAFM/2s6DG6i3eTTrobbVs=
|
github.com/bitechdev/ResolveSpec v1.1.27/go.mod h1:GF51sMRCWbAyri2WNae3IZAFM/2s6DG6i3eTTrobbVs=
|
||||||
github.com/bradfitz/gomemcache v0.0.0-20260422231931-4d751bb6e37c h1:6Gpm9YYUEQx2T9zMsYolQhr6sjwwGtFitSA0pQsa7a8=
|
github.com/bradfitz/gomemcache v0.0.0-20260422231931-4d751bb6e37c h1:6Gpm9YYUEQx2T9zMsYolQhr6sjwwGtFitSA0pQsa7a8=
|
||||||
github.com/bradfitz/gomemcache v0.0.0-20260422231931-4d751bb6e37c/go.mod h1:r5xuitiExdLAJ09PR7vBVENGvp4ZuTBeWTGtxuX3K+c=
|
github.com/bradfitz/gomemcache v0.0.0-20260422231931-4d751bb6e37c/go.mod h1:r5xuitiExdLAJ09PR7vBVENGvp4ZuTBeWTGtxuX3K+c=
|
||||||
github.com/bsm/ginkgo/v2 v2.12.0 h1:Ny8MWAHyOepLGlLKYmXG4IEkioBysk6GpaRTLC8zwWs=
|
github.com/bsm/ginkgo/v2 v2.12.0 h1:Ny8MWAHyOepLGlLKYmXG4IEkioBysk6GpaRTLC8zwWs=
|
||||||
@@ -452,8 +454,8 @@ golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc
|
|||||||
golang.org/x/tools v0.6.0/go.mod h1:Xwgl3UAJ/d3gWutnCtw505GrjyAbvKui8lOU390QaIU=
|
golang.org/x/tools v0.6.0/go.mod h1:Xwgl3UAJ/d3gWutnCtw505GrjyAbvKui8lOU390QaIU=
|
||||||
golang.org/x/tools v0.13.0/go.mod h1:HvlwmtVNQAhOuCjW7xxvovg8wbNq7LwfXh/k7wXUl58=
|
golang.org/x/tools v0.13.0/go.mod h1:HvlwmtVNQAhOuCjW7xxvovg8wbNq7LwfXh/k7wXUl58=
|
||||||
golang.org/x/tools v0.21.1-0.20240508182429-e35e4ccd0d2d/go.mod h1:aiJjzUbINMkxbQROHiO6hDPo2LHcIPhhQsa9DLh0yGk=
|
golang.org/x/tools v0.21.1-0.20240508182429-e35e4ccd0d2d/go.mod h1:aiJjzUbINMkxbQROHiO6hDPo2LHcIPhhQsa9DLh0yGk=
|
||||||
golang.org/x/tools v0.44.0 h1:UP4ajHPIcuMjT1GqzDWRlalUEoY+uzoZKnhOjbIPD2c=
|
golang.org/x/tools v0.45.0 h1:18qN3FAooORvApf5XjCXgsuayZOEtXf6JK18I3+ONa8=
|
||||||
golang.org/x/tools v0.44.0/go.mod h1:KA0AfVErSdxRZIsOVipbv3rQhVXTnlU6UhKxHd1seDI=
|
golang.org/x/tools v0.45.0/go.mod h1:LuUGqqaXcXMEFEruIVJVm5mgDD8vww/z/SR1gQ4uE/0=
|
||||||
golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||||
google.golang.org/protobuf v1.36.11 h1:fV6ZwhNocDyBLK0dj+fg8ektcVegBBuEolpbTQyBNVE=
|
google.golang.org/protobuf v1.36.11 h1:fV6ZwhNocDyBLK0dj+fg8ektcVegBBuEolpbTQyBNVE=
|
||||||
google.golang.org/protobuf v1.36.11/go.mod h1:HTf+CrKn2C3g5S8VImy6tdcUvCska2kB7j23XfzDpco=
|
google.golang.org/protobuf v1.36.11/go.mod h1:HTf+CrKn2C3g5S8VImy6tdcUvCska2kB7j23XfzDpco=
|
||||||
|
|||||||
@@ -0,0 +1,372 @@
|
|||||||
|
package app
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"crypto/rand"
|
||||||
|
"encoding/hex"
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"log/slog"
|
||||||
|
"net/http"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/jackc/pgx/v5/pgxpool"
|
||||||
|
|
||||||
|
"git.warky.dev/wdevs/amcs/internal/auth"
|
||||||
|
)
|
||||||
|
|
||||||
|
type identityAdmin struct {
|
||||||
|
pool *pgxpool.Pool
|
||||||
|
keyring *auth.Keyring
|
||||||
|
oauthRegistry *auth.OAuthRegistry
|
||||||
|
logger *slog.Logger
|
||||||
|
}
|
||||||
|
|
||||||
|
func newIdentityAdmin(pool *pgxpool.Pool, keyring *auth.Keyring, oauthRegistry *auth.OAuthRegistry, logger *slog.Logger) *identityAdmin {
|
||||||
|
return &identityAdmin{pool: pool, keyring: keyring, oauthRegistry: oauthRegistry, logger: logger}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *identityAdmin) isSuperadmin(keyID string) bool {
|
||||||
|
if a.keyring != nil && a.keyring.IsSuperadmin(keyID) {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
return a.oauthRegistry != nil && a.oauthRegistry.IsSuperadmin(keyID)
|
||||||
|
}
|
||||||
|
|
||||||
|
func loadIdentityKeyring(ctx context.Context, pool *pgxpool.Pool, keyring *auth.Keyring) error {
|
||||||
|
if keyring == nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
rows, err := pool.Query(ctx, `select a.key_id, a.tenant_id, a.enabled, coalesce(m.secret_hash, '') from api_key_assignments a left join managed_api_keys m on m.key_id = a.key_id`)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
for rows.Next() {
|
||||||
|
var keyID, tenantID, hash string
|
||||||
|
var enabled bool
|
||||||
|
if err := rows.Scan(&keyID, &tenantID, &enabled, &hash); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
keyring.AssignTenant(keyID, tenantID)
|
||||||
|
if hash != "" {
|
||||||
|
keyring.AddManaged(keyID, hash, enabled)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return rows.Err()
|
||||||
|
}
|
||||||
|
|
||||||
|
// ensureConfiguredTenants makes a tenant referenced in static YAML visible to
|
||||||
|
// the admin UI as well as to the authentication middleware.
|
||||||
|
func ensureConfiguredTenants(ctx context.Context, pool *pgxpool.Pool, keyring *auth.Keyring) error {
|
||||||
|
if keyring == nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
for _, key := range keyring.ConfiguredKeys() {
|
||||||
|
tenantID := strings.TrimSpace(key.TenantID)
|
||||||
|
if tenantID == "" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if _, err := pool.Exec(ctx, `insert into tenants (id, name) values ($1, $1) on conflict (id) do nothing`, tenantID); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *identityAdmin) handler() http.Handler {
|
||||||
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
if r.Method != http.MethodGet {
|
||||||
|
keyID, ok := auth.KeyIDFromContext(r.Context())
|
||||||
|
if !ok || !a.isSuperadmin(keyID) {
|
||||||
|
writeJSON(w, http.StatusForbidden, map[string]string{"error": "superadmin API key required"})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
path := strings.TrimPrefix(r.URL.Path, "/api/admin/identity")
|
||||||
|
switch {
|
||||||
|
case r.Method == http.MethodGet && path == "":
|
||||||
|
a.list(w, r)
|
||||||
|
case r.Method == http.MethodPost && path == "/tenants":
|
||||||
|
a.createTenant(w, r)
|
||||||
|
case r.Method == http.MethodPost && path == "/users":
|
||||||
|
a.createUser(w, r)
|
||||||
|
case r.Method == http.MethodPost && path == "/keys":
|
||||||
|
a.createKey(w, r)
|
||||||
|
case r.Method == http.MethodPatch && strings.HasPrefix(path, "/keys/"):
|
||||||
|
a.updateKey(w, r, strings.TrimPrefix(path, "/keys/"))
|
||||||
|
case r.Method == http.MethodPost && strings.HasPrefix(path, "/tenants/") && strings.HasSuffix(path, "/adopt-legacy"):
|
||||||
|
a.adoptLegacy(w, r, strings.TrimSuffix(strings.TrimPrefix(path, "/tenants/"), "/adopt-legacy"))
|
||||||
|
default:
|
||||||
|
http.NotFound(w, r)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
type tenantDTO struct {
|
||||||
|
ID string `json:"id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
CreatedAt time.Time `json:"created_at"`
|
||||||
|
}
|
||||||
|
type userDTO struct {
|
||||||
|
ID string `json:"id"`
|
||||||
|
TenantID string `json:"tenant_id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
Email *string `json:"email,omitempty"`
|
||||||
|
CreatedAt time.Time `json:"created_at"`
|
||||||
|
}
|
||||||
|
type keyDTO struct {
|
||||||
|
ID string `json:"id"`
|
||||||
|
TenantID string `json:"tenant_id"`
|
||||||
|
UserID *string `json:"user_id,omitempty"`
|
||||||
|
Description string `json:"description"`
|
||||||
|
Source string `json:"source"`
|
||||||
|
Enabled bool `json:"enabled"`
|
||||||
|
CreatedAt time.Time `json:"created_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *identityAdmin) list(w http.ResponseWriter, r *http.Request) {
|
||||||
|
result := struct {
|
||||||
|
Tenants []tenantDTO `json:"tenants"`
|
||||||
|
Users []userDTO `json:"users"`
|
||||||
|
Keys []keyDTO `json:"keys"`
|
||||||
|
}{Tenants: []tenantDTO{}, Users: []userDTO{}, Keys: []keyDTO{}}
|
||||||
|
rows, err := a.pool.Query(r.Context(), `select id, name, created_at from tenants order by name`)
|
||||||
|
if err != nil {
|
||||||
|
identityError(w, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
for rows.Next() {
|
||||||
|
var x tenantDTO
|
||||||
|
if err := rows.Scan(&x.ID, &x.Name, &x.CreatedAt); err != nil {
|
||||||
|
identityError(w, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
result.Tenants = append(result.Tenants, x)
|
||||||
|
}
|
||||||
|
rows, err = a.pool.Query(r.Context(), `select id, tenant_id, name, email, created_at from tenant_users order by name`)
|
||||||
|
if err != nil {
|
||||||
|
identityError(w, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
for rows.Next() {
|
||||||
|
var x userDTO
|
||||||
|
if err := rows.Scan(&x.ID, &x.TenantID, &x.Name, &x.Email, &x.CreatedAt); err != nil {
|
||||||
|
identityError(w, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
result.Users = append(result.Users, x)
|
||||||
|
}
|
||||||
|
configured := make(map[string]authKey)
|
||||||
|
if a.keyring != nil {
|
||||||
|
for _, key := range a.keyring.ConfiguredKeys() {
|
||||||
|
configured[key.ID] = authKey{description: key.Description}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
rows, err = a.pool.Query(r.Context(), `select key_id, tenant_id, user_id, description, source, enabled, created_at from api_key_assignments order by key_id`)
|
||||||
|
if err != nil {
|
||||||
|
identityError(w, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
for rows.Next() {
|
||||||
|
var x keyDTO
|
||||||
|
if err := rows.Scan(&x.ID, &x.TenantID, &x.UserID, &x.Description, &x.Source, &x.Enabled, &x.CreatedAt); err != nil {
|
||||||
|
identityError(w, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
result.Keys = append(result.Keys, x)
|
||||||
|
delete(configured, x.ID)
|
||||||
|
}
|
||||||
|
for id, key := range configured {
|
||||||
|
result.Keys = append(result.Keys, keyDTO{ID: id, Description: key.description, Source: "configured", Enabled: true})
|
||||||
|
}
|
||||||
|
writeJSON(w, http.StatusOK, result)
|
||||||
|
}
|
||||||
|
|
||||||
|
type authKey struct{ description string }
|
||||||
|
|
||||||
|
func (a *identityAdmin) createTenant(w http.ResponseWriter, r *http.Request) {
|
||||||
|
var body struct {
|
||||||
|
Name string `json:"name"`
|
||||||
|
}
|
||||||
|
if !decodeJSON(w, r, &body) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
body.Name = strings.TrimSpace(body.Name)
|
||||||
|
if body.Name == "" {
|
||||||
|
badRequest(w, "name is required")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
x := tenantDTO{ID: newIdentityID(), Name: body.Name}
|
||||||
|
err := a.pool.QueryRow(r.Context(), `insert into tenants (id,name) values ($1,$2) returning created_at`, x.ID, x.Name).Scan(&x.CreatedAt)
|
||||||
|
if err != nil {
|
||||||
|
identityError(w, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
writeJSON(w, http.StatusCreated, x)
|
||||||
|
}
|
||||||
|
func (a *identityAdmin) createUser(w http.ResponseWriter, r *http.Request) {
|
||||||
|
var body struct {
|
||||||
|
TenantID, Name string
|
||||||
|
Email *string `json:"email"`
|
||||||
|
}
|
||||||
|
if !decodeJSON(w, r, &body) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
body.TenantID = strings.TrimSpace(body.TenantID)
|
||||||
|
body.Name = strings.TrimSpace(body.Name)
|
||||||
|
if body.TenantID == "" || body.Name == "" {
|
||||||
|
badRequest(w, "tenant_id and name are required")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
x := userDTO{ID: newIdentityID(), TenantID: body.TenantID, Name: body.Name, Email: body.Email}
|
||||||
|
err := a.pool.QueryRow(r.Context(), `insert into tenant_users (id,tenant_id,name,email) values ($1,$2,$3,$4) returning created_at`, x.ID, x.TenantID, x.Name, x.Email).Scan(&x.CreatedAt)
|
||||||
|
if err != nil {
|
||||||
|
identityError(w, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
writeJSON(w, http.StatusCreated, x)
|
||||||
|
}
|
||||||
|
func (a *identityAdmin) createKey(w http.ResponseWriter, r *http.Request) {
|
||||||
|
var body struct {
|
||||||
|
TenantID string `json:"tenant_id"`
|
||||||
|
UserID *string `json:"user_id"`
|
||||||
|
Description string `json:"description"`
|
||||||
|
}
|
||||||
|
if !decodeJSON(w, r, &body) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
body.TenantID = strings.TrimSpace(body.TenantID)
|
||||||
|
if body.TenantID == "" {
|
||||||
|
badRequest(w, "tenant_id is required")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
secret, hash, err := auth.GenerateSecret()
|
||||||
|
if err != nil {
|
||||||
|
identityError(w, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
x := keyDTO{ID: newIdentityID(), TenantID: body.TenantID, UserID: body.UserID, Description: strings.TrimSpace(body.Description), Source: "managed", Enabled: true}
|
||||||
|
tx, err := a.pool.Begin(r.Context())
|
||||||
|
if err != nil {
|
||||||
|
identityError(w, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer tx.Rollback(r.Context())
|
||||||
|
if err = tx.QueryRow(r.Context(), `insert into api_key_assignments (key_id,tenant_id,user_id,description,source,enabled) values ($1,$2,$3,$4,'managed',true) returning created_at`, x.ID, x.TenantID, x.UserID, x.Description).Scan(&x.CreatedAt); err == nil {
|
||||||
|
_, err = tx.Exec(r.Context(), `insert into managed_api_keys (key_id,secret_hash) values ($1,$2)`, x.ID, hash)
|
||||||
|
}
|
||||||
|
if err == nil {
|
||||||
|
err = tx.Commit(r.Context())
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
identityError(w, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
a.keyring.AddManaged(x.ID, hash, true)
|
||||||
|
a.keyring.AssignTenant(x.ID, x.TenantID)
|
||||||
|
writeJSON(w, http.StatusCreated, struct {
|
||||||
|
Key keyDTO `json:"key"`
|
||||||
|
Secret string `json:"secret"`
|
||||||
|
}{x, secret})
|
||||||
|
}
|
||||||
|
func (a *identityAdmin) updateKey(w http.ResponseWriter, r *http.Request, keyID string) {
|
||||||
|
var body struct {
|
||||||
|
TenantID string `json:"tenant_id"`
|
||||||
|
UserID *string `json:"user_id"`
|
||||||
|
Description *string `json:"description"`
|
||||||
|
Enabled *bool `json:"enabled"`
|
||||||
|
}
|
||||||
|
if !decodeJSON(w, r, &body) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
body.TenantID = strings.TrimSpace(body.TenantID)
|
||||||
|
if body.TenantID == "" {
|
||||||
|
badRequest(w, "tenant_id is required")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if !a.keyring.IsConfigured(keyID) {
|
||||||
|
var exists bool
|
||||||
|
if err := a.pool.QueryRow(r.Context(), `select exists(select 1 from managed_api_keys where key_id=$1)`, keyID).Scan(&exists); err != nil {
|
||||||
|
identityError(w, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if !exists {
|
||||||
|
badRequest(w, "unknown key id")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
var x keyDTO
|
||||||
|
err := a.pool.QueryRow(r.Context(), `insert into api_key_assignments (key_id,tenant_id,user_id,description,source,enabled) values ($1,$2,$3,coalesce($4,''),case when $6 then 'configured' else 'managed' end,coalesce($5,true)) on conflict (key_id) do update set tenant_id=excluded.tenant_id,user_id=excluded.user_id,description=coalesce($4,api_key_assignments.description),enabled=coalesce($5,api_key_assignments.enabled) returning key_id,tenant_id,user_id,description,source,enabled,created_at`, keyID, body.TenantID, body.UserID, body.Description, body.Enabled, a.keyring.IsConfigured(keyID)).Scan(&x.ID, &x.TenantID, &x.UserID, &x.Description, &x.Source, &x.Enabled, &x.CreatedAt)
|
||||||
|
if err != nil {
|
||||||
|
identityError(w, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
a.keyring.AssignTenant(x.ID, x.TenantID)
|
||||||
|
a.keyring.SetManagedEnabled(x.ID, x.Enabled)
|
||||||
|
writeJSON(w, http.StatusOK, x)
|
||||||
|
}
|
||||||
|
func (a *identityAdmin) adoptLegacy(w http.ResponseWriter, r *http.Request, tenantID string) {
|
||||||
|
tenantID = strings.TrimSpace(tenantID)
|
||||||
|
if tenantID == "" {
|
||||||
|
badRequest(w, "tenant id is required")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
tables := []string{"projects", "thoughts", "stored_files", "learnings", "plans", "chat_histories"}
|
||||||
|
tx, err := a.pool.Begin(r.Context())
|
||||||
|
if err != nil {
|
||||||
|
identityError(w, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer tx.Rollback(r.Context())
|
||||||
|
var exists bool
|
||||||
|
if err = tx.QueryRow(r.Context(), `select exists(select 1 from tenants where id=$1)`, tenantID).Scan(&exists); err == nil && !exists {
|
||||||
|
badRequest(w, "tenant does not exist")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
for _, table := range tables {
|
||||||
|
if _, err = tx.Exec(r.Context(), fmt.Sprintf("update %s set tenant_id=$1 where tenant_id is null", table), tenantID); err != nil {
|
||||||
|
identityError(w, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if err = tx.Commit(r.Context()); err != nil {
|
||||||
|
identityError(w, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
w.WriteHeader(http.StatusNoContent)
|
||||||
|
}
|
||||||
|
func newIdentityID() string {
|
||||||
|
b := make([]byte, 16)
|
||||||
|
if _, err := rand.Read(b); err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
return hex.EncodeToString(b)
|
||||||
|
}
|
||||||
|
func decodeJSON(w http.ResponseWriter, r *http.Request, v any) bool {
|
||||||
|
defer r.Body.Close()
|
||||||
|
if err := json.NewDecoder(r.Body).Decode(v); err != nil {
|
||||||
|
badRequest(w, "invalid JSON")
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
func writeJSON(w http.ResponseWriter, status int, v any) {
|
||||||
|
w.Header().Set("Content-Type", "application/json")
|
||||||
|
w.WriteHeader(status)
|
||||||
|
_ = json.NewEncoder(w).Encode(v)
|
||||||
|
}
|
||||||
|
func badRequest(w http.ResponseWriter, message string) {
|
||||||
|
writeJSON(w, http.StatusBadRequest, map[string]string{"error": message})
|
||||||
|
}
|
||||||
|
func identityError(w http.ResponseWriter, err error) {
|
||||||
|
if a, ok := err.(interface{ SQLState() string }); ok && a.SQLState() == "23505" {
|
||||||
|
badRequest(w, "that value already exists")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
writeJSON(w, http.StatusInternalServerError, map[string]string{"error": "identity operation failed"})
|
||||||
|
}
|
||||||
@@ -91,6 +91,14 @@ func Run(ctx context.Context, configPath string) error {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
} else {
|
||||||
|
keyring = auth.NewManagedKeyring()
|
||||||
|
}
|
||||||
|
if err := ensureConfiguredTenants(ctx, db.Pool(), keyring); err != nil {
|
||||||
|
return fmt.Errorf("create configured tenants: %w", err)
|
||||||
|
}
|
||||||
|
if err := loadIdentityKeyring(ctx, db.Pool(), keyring); err != nil {
|
||||||
|
return fmt.Errorf("load identity key assignments: %w", err)
|
||||||
}
|
}
|
||||||
tokenStore = auth.NewTokenStore(0)
|
tokenStore = auth.NewTokenStore(0)
|
||||||
if len(cfg.Auth.OAuth.Clients) > 0 {
|
if len(cfg.Auth.OAuth.Clients) > 0 {
|
||||||
@@ -192,6 +200,7 @@ func routes(logger *slog.Logger, cfg *config.Config, info buildinfo.Info, db *st
|
|||||||
enrichmentRetryer := tools.NewEnrichmentRetryer(context.Background(), db, bgMetadata, cfg.Capture, cfg.AI.Metadata.Timeout, activeProjects, logger)
|
enrichmentRetryer := tools.NewEnrichmentRetryer(context.Background(), db, bgMetadata, cfg.Capture, cfg.AI.Metadata.Timeout, activeProjects, logger)
|
||||||
backfillTool := tools.NewBackfillTool(db, bgEmbeddings, activeProjects, logger)
|
backfillTool := tools.NewBackfillTool(db, bgEmbeddings, activeProjects, logger)
|
||||||
adminActions := newAdminActions(backfillTool, enrichmentRetryer, logger)
|
adminActions := newAdminActions(backfillTool, enrichmentRetryer, logger)
|
||||||
|
identityAdmin := newIdentityAdmin(db.Pool(), keyring, oauthRegistry, logger)
|
||||||
|
|
||||||
toolSet := mcpserver.ToolSet{
|
toolSet := mcpserver.ToolSet{
|
||||||
Capture: tools.NewCaptureTool(db, embeddings, cfg.Capture, activeProjects, enrichmentRetryer, backfillTool),
|
Capture: tools.NewCaptureTool(db, embeddings, cfg.Capture, activeProjects, enrichmentRetryer, backfillTool),
|
||||||
@@ -246,6 +255,8 @@ func routes(logger *slog.Logger, cfg *config.Config, info buildinfo.Info, db *st
|
|||||||
mux.HandleFunc("/api/oauth/token", oauthTokenHandler(oauthRegistry, tokenStore, authCodes, logger))
|
mux.HandleFunc("/api/oauth/token", oauthTokenHandler(oauthRegistry, tokenStore, authCodes, logger))
|
||||||
mux.Handle("/api/admin/actions/backfill", authMiddleware(adminActions.backfillHandler()))
|
mux.Handle("/api/admin/actions/backfill", authMiddleware(adminActions.backfillHandler()))
|
||||||
mux.Handle("/api/admin/actions/retry-metadata", authMiddleware(adminActions.retryMetadataHandler()))
|
mux.Handle("/api/admin/actions/retry-metadata", authMiddleware(adminActions.retryMetadataHandler()))
|
||||||
|
mux.Handle("/api/admin/identity", authMiddleware(identityAdmin.handler()))
|
||||||
|
mux.Handle("/api/admin/identity/", authMiddleware(identityAdmin.handler()))
|
||||||
mux.HandleFunc("/favicon.ico", serveFavicon)
|
mux.HandleFunc("/favicon.ico", serveFavicon)
|
||||||
mux.HandleFunc("/images/project.jpg", serveHomeImage)
|
mux.HandleFunc("/images/project.jpg", serveHomeImage)
|
||||||
mux.HandleFunc("/images/icon.png", serveIcon)
|
mux.HandleFunc("/images/icon.png", serveIcon)
|
||||||
|
|||||||
@@ -10,6 +10,7 @@ import (
|
|||||||
"github.com/uptrace/bunrouter"
|
"github.com/uptrace/bunrouter"
|
||||||
|
|
||||||
"git.warky.dev/wdevs/amcs/internal/store"
|
"git.warky.dev/wdevs/amcs/internal/store"
|
||||||
|
"git.warky.dev/wdevs/amcs/internal/tenancy"
|
||||||
)
|
)
|
||||||
|
|
||||||
func registerResolveSpecAdminRoutes(mux *http.ServeMux, db *store.DB, middleware func(http.Handler) http.Handler, logger *slog.Logger) error {
|
func registerResolveSpecAdminRoutes(mux *http.ServeMux, db *store.DB, middleware func(http.Handler) http.Handler, logger *slog.Logger) error {
|
||||||
@@ -45,7 +46,12 @@ func registerResolveSpecAdminRoutes(mux *http.ServeMux, db *store.DB, middleware
|
|||||||
rsMount.ServeHTTP(w, r)
|
rsMount.ServeHTTP(w, r)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
middleware(rsMount).ServeHTTP(w, r)
|
middleware(http.HandlerFunc(func(w http.ResponseWriter, authenticated *http.Request) {
|
||||||
|
if tenantID := strings.TrimSpace(authenticated.Header.Get("X-AMCS-Tenant-ID")); tenantID != "" {
|
||||||
|
authenticated = authenticated.WithContext(tenancy.WithTenantKey(authenticated.Context(), tenantID))
|
||||||
|
}
|
||||||
|
rsMount.ServeHTTP(w, authenticated)
|
||||||
|
})).ServeHTTP(w, r)
|
||||||
})
|
})
|
||||||
|
|
||||||
mux.Handle("/api/rs/", protectedRSMount)
|
mux.Handle("/api/rs/", protectedRSMount)
|
||||||
|
|||||||
@@ -14,12 +14,14 @@ func resolveSpecModels() []resolveSpecModel {
|
|||||||
{schema: "public", entity: "agent_personas", model: generatedmodels.ModelPublicAgentPersonas{}},
|
{schema: "public", entity: "agent_personas", model: generatedmodels.ModelPublicAgentPersonas{}},
|
||||||
{schema: "public", entity: "agent_skills", model: generatedmodels.ModelPublicAgentSkills{}},
|
{schema: "public", entity: "agent_skills", model: generatedmodels.ModelPublicAgentSkills{}},
|
||||||
{schema: "public", entity: "agent_traits", model: generatedmodels.ModelPublicAgentTraits{}},
|
{schema: "public", entity: "agent_traits", model: generatedmodels.ModelPublicAgentTraits{}},
|
||||||
|
{schema: "public", entity: "api_key_assignments", model: generatedmodels.ModelPublicAPIKeyAssignments{}},
|
||||||
{schema: "public", entity: "arc_stage_parts", model: generatedmodels.ModelPublicArcStageParts{}},
|
{schema: "public", entity: "arc_stage_parts", model: generatedmodels.ModelPublicArcStageParts{}},
|
||||||
{schema: "public", entity: "arc_stages", model: generatedmodels.ModelPublicArcStages{}},
|
{schema: "public", entity: "arc_stages", model: generatedmodels.ModelPublicArcStages{}},
|
||||||
{schema: "public", entity: "character_arcs", model: generatedmodels.ModelPublicCharacterArcs{}},
|
{schema: "public", entity: "character_arcs", model: generatedmodels.ModelPublicCharacterArcs{}},
|
||||||
{schema: "public", entity: "chat_histories", model: generatedmodels.ModelPublicChatHistories{}},
|
{schema: "public", entity: "chat_histories", model: generatedmodels.ModelPublicChatHistories{}},
|
||||||
{schema: "public", entity: "embeddings", model: generatedmodels.ModelPublicEmbeddings{}},
|
{schema: "public", entity: "embeddings", model: generatedmodels.ModelPublicEmbeddings{}},
|
||||||
{schema: "public", entity: "learnings", model: generatedmodels.ModelPublicLearnings{}},
|
{schema: "public", entity: "learnings", model: generatedmodels.ModelPublicLearnings{}},
|
||||||
|
{schema: "public", entity: "managed_api_keys", model: generatedmodels.ModelPublicManagedAPIKeys{}},
|
||||||
{schema: "public", entity: "oauth_clients", model: generatedmodels.ModelPublicOauthClients{}},
|
{schema: "public", entity: "oauth_clients", model: generatedmodels.ModelPublicOauthClients{}},
|
||||||
{schema: "public", entity: "persona_arc", model: generatedmodels.ModelPublicPersonaArc{}},
|
{schema: "public", entity: "persona_arc", model: generatedmodels.ModelPublicPersonaArc{}},
|
||||||
{schema: "public", entity: "plan_dependencies", model: generatedmodels.ModelPublicPlanDependencies{}},
|
{schema: "public", entity: "plan_dependencies", model: generatedmodels.ModelPublicPlanDependencies{}},
|
||||||
@@ -32,6 +34,9 @@ func resolveSpecModels() []resolveSpecModel {
|
|||||||
{schema: "public", entity: "project_skills", model: generatedmodels.ModelPublicProjectSkills{}},
|
{schema: "public", entity: "project_skills", model: generatedmodels.ModelPublicProjectSkills{}},
|
||||||
{schema: "public", entity: "projects", model: generatedmodels.ModelPublicProjects{}},
|
{schema: "public", entity: "projects", model: generatedmodels.ModelPublicProjects{}},
|
||||||
{schema: "public", entity: "stored_files", model: generatedmodels.ModelPublicStoredFiles{}},
|
{schema: "public", entity: "stored_files", model: generatedmodels.ModelPublicStoredFiles{}},
|
||||||
|
{schema: "public", entity: "tenant_users", model: generatedmodels.ModelPublicTenantUsers{}},
|
||||||
|
{schema: "public", entity: "tenants", model: generatedmodels.ModelPublicTenants{}},
|
||||||
|
{schema: "public", entity: "thought_learning_links", model: generatedmodels.ModelPublicThoughtLearningLinks{}},
|
||||||
{schema: "public", entity: "thought_links", model: generatedmodels.ModelPublicThoughtLinks{}},
|
{schema: "public", entity: "thought_links", model: generatedmodels.ModelPublicThoughtLinks{}},
|
||||||
{schema: "public", entity: "thoughts", model: generatedmodels.ModelPublicThoughts{}},
|
{schema: "public", entity: "thoughts", model: generatedmodels.ModelPublicThoughts{}},
|
||||||
{schema: "public", entity: "tool_annotations", model: generatedmodels.ModelPublicToolAnnotations{}},
|
{schema: "public", entity: "tool_annotations", model: generatedmodels.ModelPublicToolAnnotations{}},
|
||||||
|
|||||||
@@ -7,8 +7,8 @@ import (
|
|||||||
|
|
||||||
var (
|
var (
|
||||||
//go:embed ui/dist
|
//go:embed ui/dist
|
||||||
uiFiles embed.FS
|
uiFiles embed.FS
|
||||||
uiDistFS fs.FS
|
uiDistFS fs.FS
|
||||||
indexHTML []byte
|
indexHTML []byte
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
+113
-2
@@ -1,14 +1,27 @@
|
|||||||
package auth
|
package auth
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"crypto/rand"
|
||||||
|
"crypto/sha256"
|
||||||
"crypto/subtle"
|
"crypto/subtle"
|
||||||
|
"encoding/hex"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"strings"
|
||||||
|
"sync"
|
||||||
|
|
||||||
"git.warky.dev/wdevs/amcs/internal/config"
|
"git.warky.dev/wdevs/amcs/internal/config"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Keyring struct {
|
type Keyring struct {
|
||||||
keys []config.APIKey
|
mu sync.RWMutex
|
||||||
|
keys []config.APIKey
|
||||||
|
tenantsByKeyID map[string]string
|
||||||
|
managed map[string]managedKey
|
||||||
|
}
|
||||||
|
|
||||||
|
type managedKey struct {
|
||||||
|
secretHash string
|
||||||
|
enabled bool
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewKeyring(keys []config.APIKey) (*Keyring, error) {
|
func NewKeyring(keys []config.APIKey) (*Keyring, error) {
|
||||||
@@ -16,14 +29,112 @@ func NewKeyring(keys []config.APIKey) (*Keyring, error) {
|
|||||||
return nil, fmt.Errorf("keyring requires at least one key")
|
return nil, fmt.Errorf("keyring requires at least one key")
|
||||||
}
|
}
|
||||||
|
|
||||||
return &Keyring{keys: append([]config.APIKey(nil), keys...)}, nil
|
tenantsByKeyID := make(map[string]string)
|
||||||
|
for _, key := range keys {
|
||||||
|
if tenantID := strings.TrimSpace(key.TenantID); tenantID != "" {
|
||||||
|
tenantsByKeyID[key.ID] = tenantID
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return &Keyring{keys: append([]config.APIKey(nil), keys...), tenantsByKeyID: tenantsByKeyID, managed: make(map[string]managedKey)}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewManagedKeyring is used when API credentials are administered in the
|
||||||
|
// database rather than supplied through static configuration.
|
||||||
|
func NewManagedKeyring() *Keyring {
|
||||||
|
return &Keyring{tenantsByKeyID: make(map[string]string), managed: make(map[string]managedKey)}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (k *Keyring) Lookup(value string) (string, bool) {
|
func (k *Keyring) Lookup(value string) (string, bool) {
|
||||||
|
k.mu.RLock()
|
||||||
|
defer k.mu.RUnlock()
|
||||||
for _, key := range k.keys {
|
for _, key := range k.keys {
|
||||||
if subtle.ConstantTimeCompare([]byte(key.Value), []byte(value)) == 1 {
|
if subtle.ConstantTimeCompare([]byte(key.Value), []byte(value)) == 1 {
|
||||||
return key.ID, true
|
return key.ID, true
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
hash := secretHash(value)
|
||||||
|
for keyID, key := range k.managed {
|
||||||
|
if key.enabled && subtle.ConstantTimeCompare([]byte(key.secretHash), []byte(hash)) == 1 {
|
||||||
|
return keyID, true
|
||||||
|
}
|
||||||
|
}
|
||||||
return "", false
|
return "", false
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// TenantForKey returns the tenant boundary assigned to keyID. Unassigned
|
||||||
|
// configured keys retain the historical key-ID boundary for compatibility.
|
||||||
|
func (k *Keyring) TenantForKey(keyID string) string {
|
||||||
|
k.mu.RLock()
|
||||||
|
defer k.mu.RUnlock()
|
||||||
|
if tenantID := k.tenantsByKeyID[keyID]; tenantID != "" {
|
||||||
|
return tenantID
|
||||||
|
}
|
||||||
|
return keyID
|
||||||
|
}
|
||||||
|
|
||||||
|
func (k *Keyring) AssignTenant(keyID, tenantID string) {
|
||||||
|
k.mu.Lock()
|
||||||
|
defer k.mu.Unlock()
|
||||||
|
if tenantID == "" {
|
||||||
|
delete(k.tenantsByKeyID, keyID)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
k.tenantsByKeyID[keyID] = tenantID
|
||||||
|
}
|
||||||
|
|
||||||
|
func (k *Keyring) AddManaged(keyID, secretHash string, enabled bool) {
|
||||||
|
k.mu.Lock()
|
||||||
|
defer k.mu.Unlock()
|
||||||
|
k.managed[keyID] = managedKey{secretHash: secretHash, enabled: enabled}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (k *Keyring) ConfiguredKeys() []config.APIKey {
|
||||||
|
k.mu.RLock()
|
||||||
|
defer k.mu.RUnlock()
|
||||||
|
return append([]config.APIKey(nil), k.keys...)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (k *Keyring) IsConfigured(keyID string) bool {
|
||||||
|
k.mu.RLock()
|
||||||
|
defer k.mu.RUnlock()
|
||||||
|
for _, key := range k.keys {
|
||||||
|
if key.ID == keyID {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
func (k *Keyring) IsSuperadmin(keyID string) bool {
|
||||||
|
k.mu.RLock()
|
||||||
|
defer k.mu.RUnlock()
|
||||||
|
for _, key := range k.keys {
|
||||||
|
if key.ID == keyID {
|
||||||
|
return key.Superadmin
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
func (k *Keyring) SetManagedEnabled(keyID string, enabled bool) {
|
||||||
|
k.mu.Lock()
|
||||||
|
defer k.mu.Unlock()
|
||||||
|
if key, ok := k.managed[keyID]; ok {
|
||||||
|
key.enabled = enabled
|
||||||
|
k.managed[keyID] = key
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func GenerateSecret() (string, string, error) {
|
||||||
|
b := make([]byte, 32)
|
||||||
|
if _, err := rand.Read(b); err != nil {
|
||||||
|
return "", "", err
|
||||||
|
}
|
||||||
|
secret := "amcs_" + hex.EncodeToString(b)
|
||||||
|
return secret, secretHash(secret), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func secretHash(secret string) string {
|
||||||
|
sum := sha256.Sum256([]byte(secret))
|
||||||
|
return hex.EncodeToString(sum[:])
|
||||||
|
}
|
||||||
|
|||||||
@@ -36,6 +36,26 @@ func TestNewKeyringAndLookup(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestConfiguredKeyUsesTenantID(t *testing.T) {
|
||||||
|
keyring, err := NewKeyring([]config.APIKey{{ID: "agent-key", Value: "secret", TenantID: "acme"}})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("NewKeyring() error = %v", err)
|
||||||
|
}
|
||||||
|
if got := keyring.TenantForKey("agent-key"); got != "acme" {
|
||||||
|
t.Fatalf("TenantForKey() = %q, want acme", got)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestConfiguredKeySuperadmin(t *testing.T) {
|
||||||
|
keyring, err := NewKeyring([]config.APIKey{{ID: "operator", Value: "secret", Superadmin: true}})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("NewKeyring() error = %v", err)
|
||||||
|
}
|
||||||
|
if !keyring.IsSuperadmin("operator") || keyring.IsSuperadmin("missing") {
|
||||||
|
t.Fatal("IsSuperadmin() did not return the configured role")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func TestMiddlewareAllowsHeaderAuthAndSetsContext(t *testing.T) {
|
func TestMiddlewareAllowsHeaderAuthAndSetsContext(t *testing.T) {
|
||||||
keyring, err := NewKeyring([]config.APIKey{{ID: "client-a", Value: "secret"}})
|
keyring, err := NewKeyring([]config.APIKey{{ID: "client-a", Value: "secret"}})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
|||||||
@@ -53,7 +53,11 @@ func Middleware(cfg config.AuthConfig, keyring *Keyring, oauthRegistry *OAuthReg
|
|||||||
}
|
}
|
||||||
withTenant := func(ctx context.Context, keyID string) context.Context {
|
withTenant := func(ctx context.Context, keyID string) context.Context {
|
||||||
ctx = context.WithValue(ctx, keyIDContextKey, keyID)
|
ctx = context.WithValue(ctx, keyIDContextKey, keyID)
|
||||||
return tenancy.WithTenantKey(ctx, keyID)
|
tenantID := keyID
|
||||||
|
if keyring != nil {
|
||||||
|
tenantID = keyring.TenantForKey(keyID)
|
||||||
|
}
|
||||||
|
return tenancy.WithTenantKey(ctx, tenantID)
|
||||||
}
|
}
|
||||||
return func(next http.Handler) http.Handler {
|
return func(next http.Handler) http.Handler {
|
||||||
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
|||||||
@@ -31,3 +31,18 @@ func (o *OAuthRegistry) Lookup(clientID string, clientSecret string) (string, bo
|
|||||||
}
|
}
|
||||||
return "", false
|
return "", false
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// IsSuperadmin reports whether keyID (as returned by Lookup) belongs to an
|
||||||
|
// OAuth client configured with superadmin: true.
|
||||||
|
func (o *OAuthRegistry) IsSuperadmin(keyID string) bool {
|
||||||
|
for _, client := range o.clients {
|
||||||
|
id := client.ID
|
||||||
|
if id == "" {
|
||||||
|
id = client.ClientID
|
||||||
|
}
|
||||||
|
if id == keyID {
|
||||||
|
return client.Superadmin
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|||||||
@@ -32,6 +32,26 @@ func TestNewOAuthRegistryAndLookup(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestOAuthRegistryIsSuperadmin(t *testing.T) {
|
||||||
|
registry, err := NewOAuthRegistry([]config.OAuthClient{
|
||||||
|
{ID: "oauth-admin", ClientID: "admin-id", ClientSecret: "admin-secret", Superadmin: true},
|
||||||
|
{ID: "oauth-client", ClientID: "client-id", ClientSecret: "client-secret"},
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("NewOAuthRegistry() error = %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if !registry.IsSuperadmin("oauth-admin") {
|
||||||
|
t.Fatal("IsSuperadmin(oauth-admin) = false, want true")
|
||||||
|
}
|
||||||
|
if registry.IsSuperadmin("oauth-client") {
|
||||||
|
t.Fatal("IsSuperadmin(oauth-client) = true, want false")
|
||||||
|
}
|
||||||
|
if registry.IsSuperadmin("unknown") {
|
||||||
|
t.Fatal("IsSuperadmin(unknown) = true, want false")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func TestMiddlewareAllowsOAuthBasicAuthAndSetsContext(t *testing.T) {
|
func TestMiddlewareAllowsOAuthBasicAuthAndSetsContext(t *testing.T) {
|
||||||
oauthRegistry, err := NewOAuthRegistry([]config.OAuthClient{{
|
oauthRegistry, err := NewOAuthRegistry([]config.OAuthClient{{
|
||||||
ID: "oauth-client",
|
ID: "oauth-client",
|
||||||
|
|||||||
@@ -53,6 +53,8 @@ type AuthConfig struct {
|
|||||||
type APIKey struct {
|
type APIKey struct {
|
||||||
ID string `yaml:"id"`
|
ID string `yaml:"id"`
|
||||||
Value string `yaml:"value"`
|
Value string `yaml:"value"`
|
||||||
|
TenantID string `yaml:"tenant_id"`
|
||||||
|
Superadmin bool `yaml:"superadmin"`
|
||||||
Description string `yaml:"description"`
|
Description string `yaml:"description"`
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -64,6 +66,7 @@ type OAuthClient struct {
|
|||||||
ID string `yaml:"id"`
|
ID string `yaml:"id"`
|
||||||
ClientID string `yaml:"client_id"`
|
ClientID string `yaml:"client_id"`
|
||||||
ClientSecret string `yaml:"client_secret"`
|
ClientSecret string `yaml:"client_secret"`
|
||||||
|
Superadmin bool `yaml:"superadmin"`
|
||||||
Description string `yaml:"description"`
|
Description string `yaml:"description"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -3,21 +3,23 @@ package generatedmodels
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
resolvespec_common "github.com/bitechdev/ResolveSpec/pkg/spectypes"
|
sql_types "git.warky.dev/wdevs/relspecgo/pkg/sqltypes"
|
||||||
"github.com/uptrace/bun"
|
"github.com/uptrace/bun"
|
||||||
)
|
)
|
||||||
|
|
||||||
type ModelPublicAgentGuardrails struct {
|
type ModelPublicAgentGuardrails struct {
|
||||||
bun.BaseModel `bun:"table:public.agent_guardrails,alias:agent_guardrails"`
|
bun.BaseModel `bun:"table:public.agent_guardrails,alias:agent_guardrails"`
|
||||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
ID sql_types.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||||
Content resolvespec_common.SqlString `bun:"content,type:text,notnull," json:"content"`
|
Content sql_types.SqlString `bun:"content,type:text,notnull," json:"content"`
|
||||||
CreatedAt resolvespec_common.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
CreatedAt sql_types.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
||||||
Description resolvespec_common.SqlString `bun:"description,type:text,default:'',notnull," json:"description"`
|
Description sql_types.SqlString `bun:"description,type:text,default:'',notnull," json:"description"`
|
||||||
GUID resolvespec_common.SqlUUID `bun:"guid,type:uuid,default:gen_random_uuid(),notnull," json:"guid"`
|
GUID sql_types.SqlUUID `bun:"guid,type:uuid,default:gen_random_uuid(),notnull," json:"guid"`
|
||||||
Name resolvespec_common.SqlString `bun:"name,type:text,notnull," json:"name"`
|
Name sql_types.SqlString `bun:"name,type:text,notnull," json:"name"`
|
||||||
Severity resolvespec_common.SqlString `bun:"severity,type:text,default:'medium',notnull," json:"severity"`
|
Severity sql_types.SqlString `bun:"severity,type:text,default:'medium',notnull," json:"severity"`
|
||||||
Tags resolvespec_common.SqlStringArray `bun:"tags,type:text[],default:'{}',notnull," json:"tags"`
|
Tags sql_types.SqlStringArray `bun:"tags,type:text[],default:'{}',notnull," json:"tags"`
|
||||||
UpdatedAt resolvespec_common.SqlTimeStamp `bun:"updated_at,type:timestamptz,default:now(),notnull," json:"updated_at"`
|
TenantID sql_types.SqlString `bun:"tenant_id,type:text,nullzero," json:"tenant_id"`
|
||||||
|
UpdatedAt sql_types.SqlTimeStamp `bun:"updated_at,type:timestamptz,default:now(),notnull," json:"updated_at"`
|
||||||
|
RelTenantID *ModelPublicTenants `bun:"rel:has-one,join:tenant_id=id" json:"reltenantid,omitempty"` // Has one ModelPublicTenants
|
||||||
RelGuardrailIDPublicAgentPersonaGuardrails []*ModelPublicAgentPersonaGuardrails `bun:"rel:has-many,join:id=guardrail_id" json:"relguardrailidpublicagentpersonaguardrails,omitempty"` // Has many ModelPublicAgentPersonaGuardrails
|
RelGuardrailIDPublicAgentPersonaGuardrails []*ModelPublicAgentPersonaGuardrails `bun:"rel:has-many,join:id=guardrail_id" json:"relguardrailidpublicagentpersonaguardrails,omitempty"` // Has many ModelPublicAgentPersonaGuardrails
|
||||||
RelGuardrailIDPublicPlanGuardrails []*ModelPublicPlanGuardrails `bun:"rel:has-many,join:id=guardrail_id" json:"relguardrailidpublicplanguardrails,omitempty"` // Has many ModelPublicPlanGuardrails
|
RelGuardrailIDPublicPlanGuardrails []*ModelPublicPlanGuardrails `bun:"rel:has-many,join:id=guardrail_id" json:"relguardrailidpublicplanguardrails,omitempty"` // Has many ModelPublicPlanGuardrails
|
||||||
RelGuardrailIDPublicProjectGuardrails []*ModelPublicProjectGuardrails `bun:"rel:has-many,join:id=guardrail_id" json:"relguardrailidpublicprojectguardrails,omitempty"` // Has many ModelPublicProjectGuardrails
|
RelGuardrailIDPublicProjectGuardrails []*ModelPublicProjectGuardrails `bun:"rel:has-many,join:id=guardrail_id" json:"relguardrailidpublicprojectguardrails,omitempty"` // Has many ModelPublicProjectGuardrails
|
||||||
|
|||||||
@@ -3,24 +3,26 @@ package generatedmodels
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
resolvespec_common "github.com/bitechdev/ResolveSpec/pkg/spectypes"
|
sql_types "git.warky.dev/wdevs/relspecgo/pkg/sqltypes"
|
||||||
"github.com/uptrace/bun"
|
"github.com/uptrace/bun"
|
||||||
)
|
)
|
||||||
|
|
||||||
type ModelPublicAgentParts struct {
|
type ModelPublicAgentParts struct {
|
||||||
bun.BaseModel `bun:"table:public.agent_parts,alias:agent_parts"`
|
bun.BaseModel `bun:"table:public.agent_parts,alias:agent_parts"`
|
||||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
ID sql_types.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||||
Content resolvespec_common.SqlString `bun:"content,type:text,default:'',notnull," json:"content"`
|
Content sql_types.SqlString `bun:"content,type:text,default:'',notnull," json:"content"`
|
||||||
CreatedAt resolvespec_common.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
CreatedAt sql_types.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
||||||
Description resolvespec_common.SqlString `bun:"description,type:text,default:'',notnull," json:"description"`
|
Description sql_types.SqlString `bun:"description,type:text,default:'',notnull," json:"description"`
|
||||||
GUID resolvespec_common.SqlUUID `bun:"guid,type:uuid,default:gen_random_uuid(),notnull," json:"guid"`
|
GUID sql_types.SqlUUID `bun:"guid,type:uuid,default:gen_random_uuid(),notnull," json:"guid"`
|
||||||
Name resolvespec_common.SqlString `bun:"name,type:text,notnull," json:"name"`
|
Name sql_types.SqlString `bun:"name,type:text,notnull," json:"name"`
|
||||||
PartType resolvespec_common.SqlString `bun:"part_type,type:text,notnull," json:"part_type"`
|
PartType sql_types.SqlString `bun:"part_type,type:text,notnull," json:"part_type"`
|
||||||
Summary resolvespec_common.SqlString `bun:"summary,type:text,notnull," json:"summary"`
|
Summary sql_types.SqlString `bun:"summary,type:text,notnull," json:"summary"`
|
||||||
Tags resolvespec_common.SqlStringArray `bun:"tags,type:text[],default:'{}',notnull," json:"tags"`
|
Tags sql_types.SqlStringArray `bun:"tags,type:text[],default:'{}',notnull," json:"tags"`
|
||||||
UpdatedAt resolvespec_common.SqlTimeStamp `bun:"updated_at,type:timestamptz,default:now(),notnull," json:"updated_at"`
|
TenantID sql_types.SqlString `bun:"tenant_id,type:text,nullzero," json:"tenant_id"`
|
||||||
RelPartIDPublicAgentPersonaParts []*ModelPublicAgentPersonaParts `bun:"rel:has-many,join:id=part_id" json:"relpartidpublicagentpersonaparts,omitempty"` // Has many ModelPublicAgentPersonaParts
|
UpdatedAt sql_types.SqlTimeStamp `bun:"updated_at,type:timestamptz,default:now(),notnull," json:"updated_at"`
|
||||||
RelPartIDPublicArcStageParts []*ModelPublicArcStageParts `bun:"rel:has-many,join:id=part_id" json:"relpartidpublicarcstageparts,omitempty"` // Has many ModelPublicArcStageParts
|
RelTenantID *ModelPublicTenants `bun:"rel:has-one,join:tenant_id=id" json:"reltenantid,omitempty"` // Has one ModelPublicTenants
|
||||||
|
RelPartIDPublicAgentPersonaParts []*ModelPublicAgentPersonaParts `bun:"rel:has-many,join:id=part_id" json:"relpartidpublicagentpersonaparts,omitempty"` // Has many ModelPublicAgentPersonaParts
|
||||||
|
RelPartIDPublicArcStageParts []*ModelPublicArcStageParts `bun:"rel:has-many,join:id=part_id" json:"relpartidpublicarcstageparts,omitempty"` // Has many ModelPublicArcStageParts
|
||||||
}
|
}
|
||||||
|
|
||||||
// TableName returns the table name for ModelPublicAgentParts
|
// TableName returns the table name for ModelPublicAgentParts
|
||||||
|
|||||||
@@ -3,13 +3,13 @@ package generatedmodels
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
resolvespec_common "github.com/bitechdev/ResolveSpec/pkg/spectypes"
|
sql_types "git.warky.dev/wdevs/relspecgo/pkg/sqltypes"
|
||||||
"github.com/uptrace/bun"
|
"github.com/uptrace/bun"
|
||||||
)
|
)
|
||||||
|
|
||||||
type ModelPublicAgentPersonaGuardrails struct {
|
type ModelPublicAgentPersonaGuardrails struct {
|
||||||
bun.BaseModel `bun:"table:public.agent_persona_guardrails,alias:agent_persona_guardrails"`
|
bun.BaseModel `bun:"table:public.agent_persona_guardrails,alias:agent_persona_guardrails"`
|
||||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
ID sql_types.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||||
GuardrailID int64 `bun:"guardrail_id,type:bigint,notnull," json:"guardrail_id"`
|
GuardrailID int64 `bun:"guardrail_id,type:bigint,notnull," json:"guardrail_id"`
|
||||||
PersonaID int64 `bun:"persona_id,type:bigint,notnull," json:"persona_id"`
|
PersonaID int64 `bun:"persona_id,type:bigint,notnull," json:"persona_id"`
|
||||||
RelGuardrailID *ModelPublicAgentGuardrails `bun:"rel:has-one,join:guardrail_id=id" json:"relguardrailid,omitempty"` // Has one ModelPublicAgentGuardrails
|
RelGuardrailID *ModelPublicAgentGuardrails `bun:"rel:has-one,join:guardrail_id=id" json:"relguardrailid,omitempty"` // Has one ModelPublicAgentGuardrails
|
||||||
|
|||||||
@@ -3,19 +3,19 @@ package generatedmodels
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
resolvespec_common "github.com/bitechdev/ResolveSpec/pkg/spectypes"
|
sql_types "git.warky.dev/wdevs/relspecgo/pkg/sqltypes"
|
||||||
"github.com/uptrace/bun"
|
"github.com/uptrace/bun"
|
||||||
)
|
)
|
||||||
|
|
||||||
type ModelPublicAgentPersonaParts struct {
|
type ModelPublicAgentPersonaParts struct {
|
||||||
bun.BaseModel `bun:"table:public.agent_persona_parts,alias:agent_persona_parts"`
|
bun.BaseModel `bun:"table:public.agent_persona_parts,alias:agent_persona_parts"`
|
||||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
ID sql_types.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||||
PartID int64 `bun:"part_id,type:bigint,notnull," json:"part_id"`
|
PartID int64 `bun:"part_id,type:bigint,notnull," json:"part_id"`
|
||||||
PartOrder int32 `bun:"part_order,type:int,default:0,notnull," json:"part_order"`
|
PartOrder int32 `bun:"part_order,type:int,default:0,notnull," json:"part_order"`
|
||||||
PersonaID int64 `bun:"persona_id,type:bigint,notnull," json:"persona_id"`
|
PersonaID int64 `bun:"persona_id,type:bigint,notnull," json:"persona_id"`
|
||||||
Priority int32 `bun:"priority,type:int,default:0,notnull," json:"priority"`
|
Priority int32 `bun:"priority,type:int,default:0,notnull," json:"priority"`
|
||||||
RelPartID *ModelPublicAgentParts `bun:"rel:has-one,join:part_id=id" json:"relpartid,omitempty"` // Has one ModelPublicAgentParts
|
RelPartID *ModelPublicAgentParts `bun:"rel:has-one,join:part_id=id" json:"relpartid,omitempty"` // Has one ModelPublicAgentParts
|
||||||
RelPersonaID *ModelPublicAgentPersonas `bun:"rel:has-one,join:persona_id=id" json:"relpersonaid,omitempty"` // Has one ModelPublicAgentPersonas
|
RelPersonaID *ModelPublicAgentPersonas `bun:"rel:has-one,join:persona_id=id" json:"relpersonaid,omitempty"` // Has one ModelPublicAgentPersonas
|
||||||
}
|
}
|
||||||
|
|
||||||
// TableName returns the table name for ModelPublicAgentPersonaParts
|
// TableName returns the table name for ModelPublicAgentPersonaParts
|
||||||
|
|||||||
@@ -3,18 +3,18 @@ package generatedmodels
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
resolvespec_common "github.com/bitechdev/ResolveSpec/pkg/spectypes"
|
sql_types "git.warky.dev/wdevs/relspecgo/pkg/sqltypes"
|
||||||
"github.com/uptrace/bun"
|
"github.com/uptrace/bun"
|
||||||
)
|
)
|
||||||
|
|
||||||
type ModelPublicAgentPersonaSkills struct {
|
type ModelPublicAgentPersonaSkills struct {
|
||||||
bun.BaseModel `bun:"table:public.agent_persona_skills,alias:agent_persona_skills"`
|
bun.BaseModel `bun:"table:public.agent_persona_skills,alias:agent_persona_skills"`
|
||||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
ID sql_types.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||||
Override bool `bun:"override,type:boolean,default:false,notnull," json:"override"`
|
Override bool `bun:"override,type:boolean,default:false,notnull," json:"override"`
|
||||||
PersonaID int64 `bun:"persona_id,type:bigint,notnull," json:"persona_id"`
|
PersonaID int64 `bun:"persona_id,type:bigint,notnull," json:"persona_id"`
|
||||||
SkillID int64 `bun:"skill_id,type:bigint,notnull," json:"skill_id"`
|
SkillID int64 `bun:"skill_id,type:bigint,notnull," json:"skill_id"`
|
||||||
RelPersonaID *ModelPublicAgentPersonas `bun:"rel:has-one,join:persona_id=id" json:"relpersonaid,omitempty"` // Has one ModelPublicAgentPersonas
|
RelPersonaID *ModelPublicAgentPersonas `bun:"rel:has-one,join:persona_id=id" json:"relpersonaid,omitempty"` // Has one ModelPublicAgentPersonas
|
||||||
RelSkillID *ModelPublicAgentSkills `bun:"rel:has-one,join:skill_id=id" json:"relskillid,omitempty"` // Has one ModelPublicAgentSkills
|
RelSkillID *ModelPublicAgentSkills `bun:"rel:has-one,join:skill_id=id" json:"relskillid,omitempty"` // Has one ModelPublicAgentSkills
|
||||||
}
|
}
|
||||||
|
|
||||||
// TableName returns the table name for ModelPublicAgentPersonaSkills
|
// TableName returns the table name for ModelPublicAgentPersonaSkills
|
||||||
|
|||||||
@@ -3,17 +3,17 @@ package generatedmodels
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
resolvespec_common "github.com/bitechdev/ResolveSpec/pkg/spectypes"
|
sql_types "git.warky.dev/wdevs/relspecgo/pkg/sqltypes"
|
||||||
"github.com/uptrace/bun"
|
"github.com/uptrace/bun"
|
||||||
)
|
)
|
||||||
|
|
||||||
type ModelPublicAgentPersonaTraits struct {
|
type ModelPublicAgentPersonaTraits struct {
|
||||||
bun.BaseModel `bun:"table:public.agent_persona_traits,alias:agent_persona_traits"`
|
bun.BaseModel `bun:"table:public.agent_persona_traits,alias:agent_persona_traits"`
|
||||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
ID sql_types.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||||
PersonaID int64 `bun:"persona_id,type:bigint,notnull," json:"persona_id"`
|
PersonaID int64 `bun:"persona_id,type:bigint,notnull," json:"persona_id"`
|
||||||
TraitID int64 `bun:"trait_id,type:bigint,notnull," json:"trait_id"`
|
TraitID int64 `bun:"trait_id,type:bigint,notnull," json:"trait_id"`
|
||||||
RelPersonaID *ModelPublicAgentPersonas `bun:"rel:has-one,join:persona_id=id" json:"relpersonaid,omitempty"` // Has one ModelPublicAgentPersonas
|
RelPersonaID *ModelPublicAgentPersonas `bun:"rel:has-one,join:persona_id=id" json:"relpersonaid,omitempty"` // Has one ModelPublicAgentPersonas
|
||||||
RelTraitID *ModelPublicAgentTraits `bun:"rel:has-one,join:trait_id=id" json:"reltraitid,omitempty"` // Has one ModelPublicAgentTraits
|
RelTraitID *ModelPublicAgentTraits `bun:"rel:has-one,join:trait_id=id" json:"reltraitid,omitempty"` // Has one ModelPublicAgentTraits
|
||||||
}
|
}
|
||||||
|
|
||||||
// TableName returns the table name for ModelPublicAgentPersonaTraits
|
// TableName returns the table name for ModelPublicAgentPersonaTraits
|
||||||
|
|||||||
@@ -3,24 +3,26 @@ package generatedmodels
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
resolvespec_common "github.com/bitechdev/ResolveSpec/pkg/spectypes"
|
sql_types "git.warky.dev/wdevs/relspecgo/pkg/sqltypes"
|
||||||
"github.com/uptrace/bun"
|
"github.com/uptrace/bun"
|
||||||
)
|
)
|
||||||
|
|
||||||
type ModelPublicAgentPersonas struct {
|
type ModelPublicAgentPersonas struct {
|
||||||
bun.BaseModel `bun:"table:public.agent_personas,alias:agent_personas"`
|
bun.BaseModel `bun:"table:public.agent_personas,alias:agent_personas"`
|
||||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
ID sql_types.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||||
CompiledAt resolvespec_common.SqlTimeStamp `bun:"compiled_at,type:timestamptz,nullzero," json:"compiled_at"`
|
CompiledAt sql_types.SqlTimeStamp `bun:"compiled_at,type:timestamptz,nullzero," json:"compiled_at"`
|
||||||
CompiledDetail resolvespec_common.SqlString `bun:"compiled_detail,type:text,default:'',notnull," json:"compiled_detail"`
|
CompiledDetail sql_types.SqlString `bun:"compiled_detail,type:text,default:'',notnull," json:"compiled_detail"`
|
||||||
CompiledSummary resolvespec_common.SqlString `bun:"compiled_summary,type:text,default:'',notnull," json:"compiled_summary"`
|
CompiledSummary sql_types.SqlString `bun:"compiled_summary,type:text,default:'',notnull," json:"compiled_summary"`
|
||||||
CreatedAt resolvespec_common.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
CreatedAt sql_types.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
||||||
Description resolvespec_common.SqlString `bun:"description,type:text,default:'',notnull," json:"description"`
|
Description sql_types.SqlString `bun:"description,type:text,default:'',notnull," json:"description"`
|
||||||
Detail resolvespec_common.SqlString `bun:"detail,type:text,default:'',notnull," json:"detail"`
|
Detail sql_types.SqlString `bun:"detail,type:text,default:'',notnull," json:"detail"`
|
||||||
GUID resolvespec_common.SqlUUID `bun:"guid,type:uuid,default:gen_random_uuid(),notnull," json:"guid"`
|
GUID sql_types.SqlUUID `bun:"guid,type:uuid,default:gen_random_uuid(),notnull," json:"guid"`
|
||||||
Name resolvespec_common.SqlString `bun:"name,type:text,notnull," json:"name"`
|
Name sql_types.SqlString `bun:"name,type:text,notnull," json:"name"`
|
||||||
Summary resolvespec_common.SqlString `bun:"summary,type:text,notnull," json:"summary"`
|
Summary sql_types.SqlString `bun:"summary,type:text,notnull," json:"summary"`
|
||||||
Tags resolvespec_common.SqlStringArray `bun:"tags,type:text[],default:'{}',notnull," json:"tags"`
|
Tags sql_types.SqlStringArray `bun:"tags,type:text[],default:'{}',notnull," json:"tags"`
|
||||||
UpdatedAt resolvespec_common.SqlTimeStamp `bun:"updated_at,type:timestamptz,default:now(),notnull," json:"updated_at"`
|
TenantID sql_types.SqlString `bun:"tenant_id,type:text,nullzero," json:"tenant_id"`
|
||||||
|
UpdatedAt sql_types.SqlTimeStamp `bun:"updated_at,type:timestamptz,default:now(),notnull," json:"updated_at"`
|
||||||
|
RelTenantID *ModelPublicTenants `bun:"rel:has-one,join:tenant_id=id" json:"reltenantid,omitempty"` // Has one ModelPublicTenants
|
||||||
RelPersonaIDPublicAgentPersonaParts []*ModelPublicAgentPersonaParts `bun:"rel:has-many,join:id=persona_id" json:"relpersonaidpublicagentpersonaparts,omitempty"` // Has many ModelPublicAgentPersonaParts
|
RelPersonaIDPublicAgentPersonaParts []*ModelPublicAgentPersonaParts `bun:"rel:has-many,join:id=persona_id" json:"relpersonaidpublicagentpersonaparts,omitempty"` // Has many ModelPublicAgentPersonaParts
|
||||||
RelPersonaIDPublicAgentPersonaSkills []*ModelPublicAgentPersonaSkills `bun:"rel:has-many,join:id=persona_id" json:"relpersonaidpublicagentpersonaskills,omitempty"` // Has many ModelPublicAgentPersonaSkills
|
RelPersonaIDPublicAgentPersonaSkills []*ModelPublicAgentPersonaSkills `bun:"rel:has-many,join:id=persona_id" json:"relpersonaidpublicagentpersonaskills,omitempty"` // Has many ModelPublicAgentPersonaSkills
|
||||||
RelPersonaIDPublicProjectPersonas []*ModelPublicProjectPersonas `bun:"rel:has-many,join:id=persona_id" json:"relpersonaidpublicprojectpersonas,omitempty"` // Has many ModelPublicProjectPersonas
|
RelPersonaIDPublicProjectPersonas []*ModelPublicProjectPersonas `bun:"rel:has-many,join:id=persona_id" json:"relpersonaidpublicprojectpersonas,omitempty"` // Has many ModelPublicProjectPersonas
|
||||||
|
|||||||
@@ -3,28 +3,31 @@ package generatedmodels
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
resolvespec_common "github.com/bitechdev/ResolveSpec/pkg/spectypes"
|
|
||||||
|
sql_types "git.warky.dev/wdevs/relspecgo/pkg/sqltypes"
|
||||||
"github.com/uptrace/bun"
|
"github.com/uptrace/bun"
|
||||||
)
|
)
|
||||||
|
|
||||||
type ModelPublicAgentSkills struct {
|
type ModelPublicAgentSkills struct {
|
||||||
bun.BaseModel `bun:"table:public.agent_skills,alias:agent_skills"`
|
bun.BaseModel `bun:"table:public.agent_skills,alias:agent_skills"`
|
||||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
ID sql_types.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||||
Content resolvespec_common.SqlString `bun:"content,type:text,notnull," json:"content"`
|
Content sql_types.SqlString `bun:"content,type:text,notnull," json:"content"`
|
||||||
CreatedAt resolvespec_common.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
CreatedAt sql_types.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
||||||
Description resolvespec_common.SqlString `bun:"description,type:text,default:'',notnull," json:"description"`
|
Description sql_types.SqlString `bun:"description,type:text,default:'',notnull," json:"description"`
|
||||||
DomainTags resolvespec_common.SqlStringArray `bun:"domain_tags,type:text[],default:'{}',notnull," json:"domain_tags"`
|
DomainTags sql_types.SqlStringArray `bun:"domain_tags,type:text[],default:'{}',notnull," json:"domain_tags"`
|
||||||
FrameworkTags resolvespec_common.SqlStringArray `bun:"framework_tags,type:text[],default:'{}',notnull," json:"framework_tags"`
|
FrameworkTags sql_types.SqlStringArray `bun:"framework_tags,type:text[],default:'{}',notnull," json:"framework_tags"`
|
||||||
GUID resolvespec_common.SqlUUID `bun:"guid,type:uuid,default:gen_random_uuid(),notnull," json:"guid"`
|
GUID sql_types.SqlUUID `bun:"guid,type:uuid,default:gen_random_uuid(),notnull," json:"guid"`
|
||||||
LanguageTags resolvespec_common.SqlStringArray `bun:"language_tags,type:text[],default:'{}',notnull," json:"language_tags"`
|
LanguageTags sql_types.SqlStringArray `bun:"language_tags,type:text[],default:'{}',notnull," json:"language_tags"`
|
||||||
LibraryTags resolvespec_common.SqlStringArray `bun:"library_tags,type:text[],default:'{}',notnull," json:"library_tags"`
|
LibraryTags sql_types.SqlStringArray `bun:"library_tags,type:text[],default:'{}',notnull," json:"library_tags"`
|
||||||
Name resolvespec_common.SqlString `bun:"name,type:text,notnull," json:"name"`
|
Name sql_types.SqlString `bun:"name,type:text,notnull," json:"name"`
|
||||||
Tags resolvespec_common.SqlStringArray `bun:"tags,type:text[],default:'{}',notnull," json:"tags"`
|
Tags sql_types.SqlStringArray `bun:"tags,array,type:text[],default:'{}',notnull," json:"tags"`
|
||||||
UpdatedAt resolvespec_common.SqlTimeStamp `bun:"updated_at,type:timestamptz,default:now(),notnull," json:"updated_at"`
|
TenantID sql_types.SqlString `bun:"tenant_id,type:text,nullzero," json:"tenant_id"`
|
||||||
RelSkillIDPublicAgentPersonaSkills []*ModelPublicAgentPersonaSkills `bun:"rel:has-many,join:id=skill_id" json:"relskillidpublicagentpersonaskills,omitempty"` // Has many ModelPublicAgentPersonaSkills
|
UpdatedAt sql_types.SqlTimeStamp `bun:"updated_at,type:timestamptz,default:now(),notnull," json:"updated_at"`
|
||||||
RelRelatedSkillIDPublicLearnings []*ModelPublicLearnings `bun:"rel:has-many,join:id=related_skill_id" json:"relrelatedskillidpubliclearnings,omitempty"` // Has many ModelPublicLearnings
|
RelTenantID *ModelPublicTenants `bun:"rel:has-one,join:tenant_id=id" json:"reltenantid,omitempty"` // Has one ModelPublicTenants
|
||||||
RelSkillIDPublicPlanSkills []*ModelPublicPlanSkills `bun:"rel:has-many,join:id=skill_id" json:"relskillidpublicplanskills,omitempty"` // Has many ModelPublicPlanSkills
|
RelSkillIDPublicAgentPersonaSkills []*ModelPublicAgentPersonaSkills `bun:"rel:has-many,join:id=skill_id" json:"relskillidpublicagentpersonaskills,omitempty"` // Has many ModelPublicAgentPersonaSkills
|
||||||
RelSkillIDPublicProjectSkills []*ModelPublicProjectSkills `bun:"rel:has-many,join:id=skill_id" json:"relskillidpublicprojectskills,omitempty"` // Has many ModelPublicProjectSkills
|
RelRelatedSkillIDPublicLearnings []*ModelPublicLearnings `bun:"rel:has-many,join:id=related_skill_id" json:"relrelatedskillidpubliclearnings,omitempty"` // Has many ModelPublicLearnings
|
||||||
|
RelSkillIDPublicPlanSkills []*ModelPublicPlanSkills `bun:"rel:has-many,join:id=skill_id" json:"relskillidpublicplanskills,omitempty"` // Has many ModelPublicPlanSkills
|
||||||
|
RelSkillIDPublicProjectSkills []*ModelPublicProjectSkills `bun:"rel:has-many,join:id=skill_id" json:"relskillidpublicprojectskills,omitempty"` // Has many ModelPublicProjectSkills
|
||||||
}
|
}
|
||||||
|
|
||||||
// TableName returns the table name for ModelPublicAgentSkills
|
// TableName returns the table name for ModelPublicAgentSkills
|
||||||
|
|||||||
@@ -3,22 +3,24 @@ package generatedmodels
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
resolvespec_common "github.com/bitechdev/ResolveSpec/pkg/spectypes"
|
sql_types "git.warky.dev/wdevs/relspecgo/pkg/sqltypes"
|
||||||
"github.com/uptrace/bun"
|
"github.com/uptrace/bun"
|
||||||
)
|
)
|
||||||
|
|
||||||
type ModelPublicAgentTraits struct {
|
type ModelPublicAgentTraits struct {
|
||||||
bun.BaseModel `bun:"table:public.agent_traits,alias:agent_traits"`
|
bun.BaseModel `bun:"table:public.agent_traits,alias:agent_traits"`
|
||||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
ID sql_types.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||||
CreatedAt resolvespec_common.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
CreatedAt sql_types.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
||||||
Description resolvespec_common.SqlString `bun:"description,type:text,default:'',notnull," json:"description"`
|
Description sql_types.SqlString `bun:"description,type:text,default:'',notnull," json:"description"`
|
||||||
GUID resolvespec_common.SqlUUID `bun:"guid,type:uuid,default:gen_random_uuid(),notnull," json:"guid"`
|
GUID sql_types.SqlUUID `bun:"guid,type:uuid,default:gen_random_uuid(),notnull," json:"guid"`
|
||||||
Instruction resolvespec_common.SqlString `bun:"instruction,type:text,default:'',notnull," json:"instruction"`
|
Instruction sql_types.SqlString `bun:"instruction,type:text,default:'',notnull," json:"instruction"`
|
||||||
Name resolvespec_common.SqlString `bun:"name,type:text,notnull," json:"name"`
|
Name sql_types.SqlString `bun:"name,type:text,notnull," json:"name"`
|
||||||
Tags resolvespec_common.SqlStringArray `bun:"tags,type:text[],default:'{}',notnull," json:"tags"`
|
Tags sql_types.SqlStringArray `bun:"tags,type:text[],default:'{}',notnull," json:"tags"`
|
||||||
TraitType resolvespec_common.SqlString `bun:"trait_type,type:text,notnull," json:"trait_type"`
|
TenantID sql_types.SqlString `bun:"tenant_id,type:text,nullzero," json:"tenant_id"`
|
||||||
UpdatedAt resolvespec_common.SqlTimeStamp `bun:"updated_at,type:timestamptz,default:now(),notnull," json:"updated_at"`
|
TraitType sql_types.SqlString `bun:"trait_type,type:text,notnull," json:"trait_type"`
|
||||||
RelTraitIDPublicAgentPersonaTraits []*ModelPublicAgentPersonaTraits `bun:"rel:has-many,join:id=trait_id" json:"reltraitidpublicagentpersonatraits,omitempty"` // Has many ModelPublicAgentPersonaTraits
|
UpdatedAt sql_types.SqlTimeStamp `bun:"updated_at,type:timestamptz,default:now(),notnull," json:"updated_at"`
|
||||||
|
RelTenantID *ModelPublicTenants `bun:"rel:has-one,join:tenant_id=id" json:"reltenantid,omitempty"` // Has one ModelPublicTenants
|
||||||
|
RelTraitIDPublicAgentPersonaTraits []*ModelPublicAgentPersonaTraits `bun:"rel:has-many,join:id=trait_id" json:"reltraitidpublicagentpersonatraits,omitempty"` // Has many ModelPublicAgentPersonaTraits
|
||||||
}
|
}
|
||||||
|
|
||||||
// TableName returns the table name for ModelPublicAgentTraits
|
// TableName returns the table name for ModelPublicAgentTraits
|
||||||
|
|||||||
@@ -0,0 +1,66 @@
|
|||||||
|
// Code generated by relspecgo. DO NOT EDIT.
|
||||||
|
package generatedmodels
|
||||||
|
|
||||||
|
import (
|
||||||
|
sql_types "git.warky.dev/wdevs/relspecgo/pkg/sqltypes"
|
||||||
|
"github.com/uptrace/bun"
|
||||||
|
)
|
||||||
|
|
||||||
|
type ModelPublicAPIKeyAssignments struct {
|
||||||
|
bun.BaseModel `bun:"table:public.api_key_assignments,alias:api_key_assignments"`
|
||||||
|
KeyID sql_types.SqlString `bun:"key_id,type:text,pk," json:"key_id"`
|
||||||
|
CreatedAt sql_types.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
||||||
|
Description sql_types.SqlString `bun:"description,type:text,default:'',notnull," json:"description"`
|
||||||
|
Enabled bool `bun:"enabled,type:boolean,default:true,notnull," json:"enabled"`
|
||||||
|
Source sql_types.SqlString `bun:"source,type:text,notnull," json:"source"`
|
||||||
|
TenantID sql_types.SqlString `bun:"tenant_id,type:text,notnull," json:"tenant_id"`
|
||||||
|
UserID sql_types.SqlString `bun:"user_id,type:text,nullzero," json:"user_id"`
|
||||||
|
RelTenantID *ModelPublicTenants `bun:"rel:has-one,join:tenant_id=id" json:"reltenantid,omitempty"` // Has one ModelPublicTenants
|
||||||
|
RelUserID *ModelPublicTenantUsers `bun:"rel:has-one,join:user_id=id" json:"reluserid,omitempty"` // Has one ModelPublicTenantUsers
|
||||||
|
RelKeyIDPublicManagedAPIKeys []*ModelPublicManagedAPIKeys `bun:"rel:has-many,join:key_id=key_id" json:"relkeyidpublicmanagedapikeys,omitempty"` // Has many ModelPublicManagedAPIKeys
|
||||||
|
}
|
||||||
|
|
||||||
|
// TableName returns the table name for ModelPublicAPIKeyAssignments
|
||||||
|
func (m ModelPublicAPIKeyAssignments) TableName() string {
|
||||||
|
return "public.api_key_assignments"
|
||||||
|
}
|
||||||
|
|
||||||
|
// TableNameOnly returns the table name without schema for ModelPublicAPIKeyAssignments
|
||||||
|
func (m ModelPublicAPIKeyAssignments) TableNameOnly() string {
|
||||||
|
return "api_key_assignments"
|
||||||
|
}
|
||||||
|
|
||||||
|
// SchemaName returns the schema name for ModelPublicAPIKeyAssignments
|
||||||
|
func (m ModelPublicAPIKeyAssignments) SchemaName() string {
|
||||||
|
return "public"
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetID returns the primary key value
|
||||||
|
func (m ModelPublicAPIKeyAssignments) GetID() string {
|
||||||
|
return m.KeyID.String()
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetIDStr returns the primary key as a string
|
||||||
|
func (m ModelPublicAPIKeyAssignments) GetIDStr() string {
|
||||||
|
return m.KeyID.String()
|
||||||
|
}
|
||||||
|
|
||||||
|
// SetID sets the primary key value
|
||||||
|
func (m ModelPublicAPIKeyAssignments) SetID(newid string) {
|
||||||
|
m.UpdateID(newid)
|
||||||
|
}
|
||||||
|
|
||||||
|
// UpdateID updates the primary key value
|
||||||
|
func (m *ModelPublicAPIKeyAssignments) UpdateID(newid string) {
|
||||||
|
m.KeyID.FromString(newid)
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetIDName returns the name of the primary key column
|
||||||
|
func (m ModelPublicAPIKeyAssignments) GetIDName() string {
|
||||||
|
return "key_id"
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetPrefix returns the table prefix
|
||||||
|
func (m ModelPublicAPIKeyAssignments) GetPrefix() string {
|
||||||
|
return "AKA"
|
||||||
|
}
|
||||||
@@ -3,17 +3,17 @@ package generatedmodels
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
resolvespec_common "github.com/bitechdev/ResolveSpec/pkg/spectypes"
|
sql_types "git.warky.dev/wdevs/relspecgo/pkg/sqltypes"
|
||||||
"github.com/uptrace/bun"
|
"github.com/uptrace/bun"
|
||||||
)
|
)
|
||||||
|
|
||||||
type ModelPublicArcStageParts struct {
|
type ModelPublicArcStageParts struct {
|
||||||
bun.BaseModel `bun:"table:public.arc_stage_parts,alias:arc_stage_parts"`
|
bun.BaseModel `bun:"table:public.arc_stage_parts,alias:arc_stage_parts"`
|
||||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
ID sql_types.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||||
PartID int64 `bun:"part_id,type:bigint,notnull," json:"part_id"`
|
PartID int64 `bun:"part_id,type:bigint,notnull," json:"part_id"`
|
||||||
StageID int64 `bun:"stage_id,type:bigint,notnull," json:"stage_id"`
|
StageID int64 `bun:"stage_id,type:bigint,notnull," json:"stage_id"`
|
||||||
RelPartID *ModelPublicAgentParts `bun:"rel:has-one,join:part_id=id" json:"relpartid,omitempty"` // Has one ModelPublicAgentParts
|
RelPartID *ModelPublicAgentParts `bun:"rel:has-one,join:part_id=id" json:"relpartid,omitempty"` // Has one ModelPublicAgentParts
|
||||||
RelStageID *ModelPublicArcStages `bun:"rel:has-one,join:stage_id=id" json:"relstageid,omitempty"` // Has one ModelPublicArcStages
|
RelStageID *ModelPublicArcStages `bun:"rel:has-one,join:stage_id=id" json:"relstageid,omitempty"` // Has one ModelPublicArcStages
|
||||||
}
|
}
|
||||||
|
|
||||||
// TableName returns the table name for ModelPublicArcStageParts
|
// TableName returns the table name for ModelPublicArcStageParts
|
||||||
|
|||||||
@@ -3,22 +3,22 @@ package generatedmodels
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
resolvespec_common "github.com/bitechdev/ResolveSpec/pkg/spectypes"
|
sql_types "git.warky.dev/wdevs/relspecgo/pkg/sqltypes"
|
||||||
"github.com/uptrace/bun"
|
"github.com/uptrace/bun"
|
||||||
)
|
)
|
||||||
|
|
||||||
type ModelPublicArcStages struct {
|
type ModelPublicArcStages struct {
|
||||||
bun.BaseModel `bun:"table:public.arc_stages,alias:arc_stages"`
|
bun.BaseModel `bun:"table:public.arc_stages,alias:arc_stages"`
|
||||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
ID sql_types.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||||
ArcID int64 `bun:"arc_id,type:bigint,notnull," json:"arc_id"`
|
ArcID int64 `bun:"arc_id,type:bigint,notnull," json:"arc_id"`
|
||||||
Condition resolvespec_common.SqlString `bun:"condition,type:text,default:'',notnull," json:"condition"`
|
Condition sql_types.SqlString `bun:"condition,type:text,default:'',notnull," json:"condition"`
|
||||||
CreatedAt resolvespec_common.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
CreatedAt sql_types.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
||||||
Description resolvespec_common.SqlString `bun:"description,type:text,default:'',notnull," json:"description"`
|
Description sql_types.SqlString `bun:"description,type:text,default:'',notnull," json:"description"`
|
||||||
Name resolvespec_common.SqlString `bun:"name,type:text,notnull," json:"name"`
|
Name sql_types.SqlString `bun:"name,type:text,notnull," json:"name"`
|
||||||
StageOrder int32 `bun:"stage_order,type:int,default:0,notnull," json:"stage_order"`
|
StageOrder int32 `bun:"stage_order,type:int,default:0,notnull," json:"stage_order"`
|
||||||
RelArcID *ModelPublicCharacterArcs `bun:"rel:has-one,join:arc_id=id" json:"relarcid,omitempty"` // Has one ModelPublicCharacterArcs
|
RelArcID *ModelPublicCharacterArcs `bun:"rel:has-one,join:arc_id=id" json:"relarcid,omitempty"` // Has one ModelPublicCharacterArcs
|
||||||
RelStageIDPublicArcStageParts []*ModelPublicArcStageParts `bun:"rel:has-many,join:id=stage_id" json:"relstageidpublicarcstageparts,omitempty"` // Has many ModelPublicArcStageParts
|
RelStageIDPublicArcStageParts []*ModelPublicArcStageParts `bun:"rel:has-many,join:id=stage_id" json:"relstageidpublicarcstageparts,omitempty"` // Has many ModelPublicArcStageParts
|
||||||
RelCurrentStageIDPublicPersonaArcs []*ModelPublicPersonaArc `bun:"rel:has-many,join:id=current_stage_id" json:"relcurrentstageidpublicpersonaarcs,omitempty"` // Has many ModelPublicPersonaArc
|
RelCurrentStageIDPublicPersonaArcs []*ModelPublicPersonaArc `bun:"rel:has-many,join:id=current_stage_id" json:"relcurrentstageidpublicpersonaarcs,omitempty"` // Has many ModelPublicPersonaArc
|
||||||
}
|
}
|
||||||
|
|
||||||
// TableName returns the table name for ModelPublicArcStages
|
// TableName returns the table name for ModelPublicArcStages
|
||||||
|
|||||||
@@ -3,20 +3,22 @@ package generatedmodels
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
resolvespec_common "github.com/bitechdev/ResolveSpec/pkg/spectypes"
|
sql_types "git.warky.dev/wdevs/relspecgo/pkg/sqltypes"
|
||||||
"github.com/uptrace/bun"
|
"github.com/uptrace/bun"
|
||||||
)
|
)
|
||||||
|
|
||||||
type ModelPublicCharacterArcs struct {
|
type ModelPublicCharacterArcs struct {
|
||||||
bun.BaseModel `bun:"table:public.character_arcs,alias:character_arcs"`
|
bun.BaseModel `bun:"table:public.character_arcs,alias:character_arcs"`
|
||||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
ID sql_types.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||||
CreatedAt resolvespec_common.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
CreatedAt sql_types.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
||||||
Description resolvespec_common.SqlString `bun:"description,type:text,default:'',notnull," json:"description"`
|
Description sql_types.SqlString `bun:"description,type:text,default:'',notnull," json:"description"`
|
||||||
Name resolvespec_common.SqlString `bun:"name,type:text,notnull," json:"name"`
|
Name sql_types.SqlString `bun:"name,type:text,notnull," json:"name"`
|
||||||
Summary resolvespec_common.SqlString `bun:"summary,type:text,default:'',notnull," json:"summary"`
|
Summary sql_types.SqlString `bun:"summary,type:text,default:'',notnull," json:"summary"`
|
||||||
UpdatedAt resolvespec_common.SqlTimeStamp `bun:"updated_at,type:timestamptz,default:now(),notnull," json:"updated_at"`
|
TenantID sql_types.SqlString `bun:"tenant_id,type:text,nullzero," json:"tenant_id"`
|
||||||
RelArcIDPublicArcStages []*ModelPublicArcStages `bun:"rel:has-many,join:id=arc_id" json:"relarcidpublicarcstages,omitempty"` // Has many ModelPublicArcStages
|
UpdatedAt sql_types.SqlTimeStamp `bun:"updated_at,type:timestamptz,default:now(),notnull," json:"updated_at"`
|
||||||
RelArcIDPublicPersonaArcs []*ModelPublicPersonaArc `bun:"rel:has-many,join:id=arc_id" json:"relarcidpublicpersonaarcs,omitempty"` // Has many ModelPublicPersonaArc
|
RelTenantID *ModelPublicTenants `bun:"rel:has-one,join:tenant_id=id" json:"reltenantid,omitempty"` // Has one ModelPublicTenants
|
||||||
|
RelArcIDPublicArcStages []*ModelPublicArcStages `bun:"rel:has-many,join:id=arc_id" json:"relarcidpublicarcstages,omitempty"` // Has many ModelPublicArcStages
|
||||||
|
RelArcIDPublicPersonaArcs []*ModelPublicPersonaArc `bun:"rel:has-many,join:id=arc_id" json:"relarcidpublicpersonaarcs,omitempty"` // Has many ModelPublicPersonaArc
|
||||||
}
|
}
|
||||||
|
|
||||||
// TableName returns the table name for ModelPublicCharacterArcs
|
// TableName returns the table name for ModelPublicCharacterArcs
|
||||||
|
|||||||
@@ -3,25 +3,27 @@ package generatedmodels
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
resolvespec_common "github.com/bitechdev/ResolveSpec/pkg/spectypes"
|
sql_types "git.warky.dev/wdevs/relspecgo/pkg/sqltypes"
|
||||||
"github.com/uptrace/bun"
|
"github.com/uptrace/bun"
|
||||||
)
|
)
|
||||||
|
|
||||||
type ModelPublicChatHistories struct {
|
type ModelPublicChatHistories struct {
|
||||||
bun.BaseModel `bun:"table:public.chat_histories,alias:chat_histories"`
|
bun.BaseModel `bun:"table:public.chat_histories,alias:chat_histories"`
|
||||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
ID sql_types.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||||
AgentID resolvespec_common.SqlString `bun:"agent_id,type:text,nullzero," json:"agent_id"`
|
AgentID sql_types.SqlString `bun:"agent_id,type:text,nullzero," json:"agent_id"`
|
||||||
Channel resolvespec_common.SqlString `bun:"channel,type:text,nullzero," json:"channel"`
|
Channel sql_types.SqlString `bun:"channel,type:text,nullzero," json:"channel"`
|
||||||
CreatedAt resolvespec_common.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
CreatedAt sql_types.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
||||||
GUID resolvespec_common.SqlUUID `bun:"guid,type:uuid,default:gen_random_uuid(),notnull," json:"guid"`
|
GUID sql_types.SqlUUID `bun:"guid,type:uuid,default:gen_random_uuid(),notnull," json:"guid"`
|
||||||
Messages resolvespec_common.SqlJSONB `bun:"messages,type:jsonb,default:'[]',notnull," json:"messages"`
|
Messages sql_types.SqlJSONB `bun:"messages,type:jsonb,default:'[]',notnull," json:"messages"`
|
||||||
Metadata resolvespec_common.SqlJSONB `bun:"metadata,type:jsonb,default:'{}',notnull," json:"metadata"`
|
Metadata sql_types.SqlJSONB `bun:"metadata,type:jsonb,default:'{}',notnull," json:"metadata"`
|
||||||
ProjectID resolvespec_common.SqlInt64 `bun:"project_id,type:bigint,nullzero," json:"project_id"`
|
ProjectID sql_types.SqlInt64 `bun:"project_id,type:bigint,nullzero," json:"project_id"`
|
||||||
SessionID resolvespec_common.SqlString `bun:"session_id,type:text,notnull," json:"session_id"`
|
SessionID sql_types.SqlString `bun:"session_id,type:text,notnull," json:"session_id"`
|
||||||
Summary resolvespec_common.SqlString `bun:"summary,type:text,nullzero," json:"summary"`
|
Summary sql_types.SqlString `bun:"summary,type:text,nullzero," json:"summary"`
|
||||||
Title resolvespec_common.SqlString `bun:"title,type:text,nullzero," json:"title"`
|
TenantID sql_types.SqlString `bun:"tenant_id,type:text,nullzero," json:"tenant_id"`
|
||||||
UpdatedAt resolvespec_common.SqlTimeStamp `bun:"updated_at,type:timestamptz,default:now(),notnull," json:"updated_at"`
|
Title sql_types.SqlString `bun:"title,type:text,nullzero," json:"title"`
|
||||||
RelProjectID *ModelPublicProjects `bun:"rel:has-one,join:project_id=id" json:"relprojectid,omitempty"` // Has one ModelPublicProjects
|
UpdatedAt sql_types.SqlTimeStamp `bun:"updated_at,type:timestamptz,default:now(),notnull," json:"updated_at"`
|
||||||
|
RelProjectID *ModelPublicProjects `bun:"rel:has-one,join:project_id=id" json:"relprojectid,omitempty"` // Has one ModelPublicProjects
|
||||||
|
RelTenantID *ModelPublicTenants `bun:"rel:has-one,join:tenant_id=id" json:"reltenantid,omitempty"` // Has one ModelPublicTenants
|
||||||
}
|
}
|
||||||
|
|
||||||
// TableName returns the table name for ModelPublicChatHistories
|
// TableName returns the table name for ModelPublicChatHistories
|
||||||
|
|||||||
@@ -3,21 +3,21 @@ package generatedmodels
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
resolvespec_common "github.com/bitechdev/ResolveSpec/pkg/spectypes"
|
sql_types "git.warky.dev/wdevs/relspecgo/pkg/sqltypes"
|
||||||
"github.com/uptrace/bun"
|
"github.com/uptrace/bun"
|
||||||
)
|
)
|
||||||
|
|
||||||
type ModelPublicEmbeddings struct {
|
type ModelPublicEmbeddings struct {
|
||||||
bun.BaseModel `bun:"table:public.embeddings,alias:embeddings"`
|
bun.BaseModel `bun:"table:public.embeddings,alias:embeddings"`
|
||||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
ID sql_types.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||||
CreatedAt resolvespec_common.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),nullzero," json:"created_at"`
|
CreatedAt sql_types.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),nullzero," json:"created_at"`
|
||||||
Dim int32 `bun:"dim,type:int,notnull," json:"dim"`
|
Dim int32 `bun:"dim,type:int,notnull," json:"dim"`
|
||||||
Embedding resolvespec_common.SqlVector `bun:"embedding,type:vector,notnull," json:"embedding"`
|
Embedding sql_types.SqlVector `bun:"embedding,type:vector,notnull," json:"embedding"`
|
||||||
GUID resolvespec_common.SqlUUID `bun:"guid,type:uuid,default:gen_random_uuid(),notnull," json:"guid"`
|
GUID sql_types.SqlUUID `bun:"guid,type:uuid,default:gen_random_uuid(),notnull," json:"guid"`
|
||||||
Model resolvespec_common.SqlString `bun:"model,type:text,notnull,unique:uidx_embeddings_thought_id_model," json:"model"`
|
Model sql_types.SqlString `bun:"model,type:text,notnull,unique:uidx_embeddings_thought_id_model," json:"model"`
|
||||||
ThoughtID int64 `bun:"thought_id,type:bigint,notnull,unique:uidx_embeddings_thought_id_model," json:"thought_id"`
|
ThoughtID int64 `bun:"thought_id,type:bigint,notnull,unique:uidx_embeddings_thought_id_model," json:"thought_id"`
|
||||||
UpdatedAt resolvespec_common.SqlTimeStamp `bun:"updated_at,type:timestamptz,default:now(),nullzero," json:"updated_at"`
|
UpdatedAt sql_types.SqlTimeStamp `bun:"updated_at,type:timestamptz,default:now(),nullzero," json:"updated_at"`
|
||||||
RelThoughtID *ModelPublicThoughts `bun:"rel:has-one,join:thought_id=id" json:"relthoughtid,omitempty"` // Has one ModelPublicThoughts
|
RelThoughtID *ModelPublicThoughts `bun:"rel:has-one,join:thought_id=id" json:"relthoughtid,omitempty"` // Has one ModelPublicThoughts
|
||||||
}
|
}
|
||||||
|
|
||||||
// TableName returns the table name for ModelPublicEmbeddings
|
// TableName returns the table name for ModelPublicEmbeddings
|
||||||
|
|||||||
@@ -3,39 +3,42 @@ package generatedmodels
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
resolvespec_common "github.com/bitechdev/ResolveSpec/pkg/spectypes"
|
sql_types "git.warky.dev/wdevs/relspecgo/pkg/sqltypes"
|
||||||
"github.com/uptrace/bun"
|
"github.com/uptrace/bun"
|
||||||
)
|
)
|
||||||
|
|
||||||
type ModelPublicLearnings struct {
|
type ModelPublicLearnings struct {
|
||||||
bun.BaseModel `bun:"table:public.learnings,alias:learnings"`
|
bun.BaseModel `bun:"table:public.learnings,alias:learnings"`
|
||||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
ID sql_types.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||||
ActionRequired bool `bun:"action_required,type:boolean,default:false,notnull," json:"action_required"`
|
ActionRequired bool `bun:"action_required,type:boolean,default:false,notnull," json:"action_required"`
|
||||||
Area resolvespec_common.SqlString `bun:"area,type:text,default:'other',notnull," json:"area"`
|
Area sql_types.SqlString `bun:"area,type:text,default:'other',notnull," json:"area"`
|
||||||
Category resolvespec_common.SqlString `bun:"category,type:text,default:'insight',notnull," json:"category"`
|
Category sql_types.SqlString `bun:"category,type:text,default:'insight',notnull," json:"category"`
|
||||||
Confidence resolvespec_common.SqlString `bun:"confidence,type:text,default:'hypothesis',notnull," json:"confidence"`
|
Confidence sql_types.SqlString `bun:"confidence,type:text,default:'hypothesis',notnull," json:"confidence"`
|
||||||
CreatedAt resolvespec_common.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
CreatedAt sql_types.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
||||||
Details resolvespec_common.SqlString `bun:"details,type:text,default:'',notnull," json:"details"`
|
Details sql_types.SqlString `bun:"details,type:text,default:'',notnull," json:"details"`
|
||||||
DuplicateOfLearningID resolvespec_common.SqlInt64 `bun:"duplicate_of_learning_id,type:bigint,nullzero," json:"duplicate_of_learning_id"`
|
DuplicateOfLearningID sql_types.SqlInt64 `bun:"duplicate_of_learning_id,type:bigint,nullzero," json:"duplicate_of_learning_id"`
|
||||||
GUID resolvespec_common.SqlUUID `bun:"guid,type:uuid,default:gen_random_uuid(),notnull," json:"guid"`
|
GUID sql_types.SqlUUID `bun:"guid,type:uuid,default:gen_random_uuid(),notnull," json:"guid"`
|
||||||
Priority resolvespec_common.SqlString `bun:"priority,type:text,default:'medium',notnull," json:"priority"`
|
Priority sql_types.SqlString `bun:"priority,type:text,default:'medium',notnull," json:"priority"`
|
||||||
ProjectID resolvespec_common.SqlInt64 `bun:"project_id,type:bigint,nullzero," json:"project_id"`
|
ProjectID sql_types.SqlInt64 `bun:"project_id,type:bigint,nullzero," json:"project_id"`
|
||||||
RelatedSkillID resolvespec_common.SqlInt64 `bun:"related_skill_id,type:bigint,nullzero," json:"related_skill_id"`
|
RelatedSkillID sql_types.SqlInt64 `bun:"related_skill_id,type:bigint,nullzero," json:"related_skill_id"`
|
||||||
RelatedThoughtID resolvespec_common.SqlInt64 `bun:"related_thought_id,type:bigint,nullzero," json:"related_thought_id"`
|
RelatedThoughtID sql_types.SqlInt64 `bun:"related_thought_id,type:bigint,nullzero," json:"related_thought_id"`
|
||||||
ReviewedAt resolvespec_common.SqlTimeStamp `bun:"reviewed_at,type:timestamptz,nullzero," json:"reviewed_at"`
|
ReviewedAt sql_types.SqlTimeStamp `bun:"reviewed_at,type:timestamptz,nullzero," json:"reviewed_at"`
|
||||||
ReviewedBy resolvespec_common.SqlString `bun:"reviewed_by,type:text,nullzero," json:"reviewed_by"`
|
ReviewedBy sql_types.SqlString `bun:"reviewed_by,type:text,nullzero," json:"reviewed_by"`
|
||||||
SourceRef resolvespec_common.SqlString `bun:"source_ref,type:text,nullzero," json:"source_ref"`
|
SourceRef sql_types.SqlString `bun:"source_ref,type:text,nullzero," json:"source_ref"`
|
||||||
SourceType resolvespec_common.SqlString `bun:"source_type,type:text,nullzero," json:"source_type"`
|
SourceType sql_types.SqlString `bun:"source_type,type:text,nullzero," json:"source_type"`
|
||||||
Status resolvespec_common.SqlString `bun:"status,type:text,default:'pending',notnull," json:"status"`
|
Status sql_types.SqlString `bun:"status,type:text,default:'pending',notnull," json:"status"`
|
||||||
Summary resolvespec_common.SqlString `bun:"summary,type:text,notnull," json:"summary"`
|
Summary sql_types.SqlString `bun:"summary,type:text,notnull," json:"summary"`
|
||||||
SupersedesLearningID resolvespec_common.SqlInt64 `bun:"supersedes_learning_id,type:bigint,nullzero," json:"supersedes_learning_id"`
|
SupersedesLearningID sql_types.SqlInt64 `bun:"supersedes_learning_id,type:bigint,nullzero," json:"supersedes_learning_id"`
|
||||||
Tags resolvespec_common.SqlStringArray `bun:"tags,type:text[],default:'{}',notnull," json:"tags"`
|
Tags sql_types.SqlStringArray `bun:"tags,type:text[],default:'{}',notnull," json:"tags"`
|
||||||
UpdatedAt resolvespec_common.SqlTimeStamp `bun:"updated_at,type:timestamptz,default:now(),notnull," json:"updated_at"`
|
TenantID sql_types.SqlString `bun:"tenant_id,type:text,nullzero," json:"tenant_id"`
|
||||||
RelDuplicateOfLearningID *ModelPublicLearnings `bun:"rel:has-one,join:duplicate_of_learning_id=id" json:"relduplicateoflearningid,omitempty"` // Has one ModelPublicLearnings
|
UpdatedAt sql_types.SqlTimeStamp `bun:"updated_at,type:timestamptz,default:now(),notnull," json:"updated_at"`
|
||||||
RelProjectID *ModelPublicProjects `bun:"rel:has-one,join:project_id=id" json:"relprojectid,omitempty"` // Has one ModelPublicProjects
|
RelDuplicateOfLearningID *ModelPublicLearnings `bun:"rel:has-one,join:duplicate_of_learning_id=id" json:"relduplicateoflearningid,omitempty"` // Has one ModelPublicLearnings
|
||||||
RelRelatedSkillID *ModelPublicAgentSkills `bun:"rel:has-one,join:related_skill_id=id" json:"relrelatedskillid,omitempty"` // Has one ModelPublicAgentSkills
|
RelProjectID *ModelPublicProjects `bun:"rel:has-one,join:project_id=id" json:"relprojectid,omitempty"` // Has one ModelPublicProjects
|
||||||
RelRelatedThoughtID *ModelPublicThoughts `bun:"rel:has-one,join:related_thought_id=id" json:"relrelatedthoughtid,omitempty"` // Has one ModelPublicThoughts
|
RelRelatedSkillID *ModelPublicAgentSkills `bun:"rel:has-one,join:related_skill_id=id" json:"relrelatedskillid,omitempty"` // Has one ModelPublicAgentSkills
|
||||||
RelSupersedesLearningID *ModelPublicLearnings `bun:"rel:has-one,join:supersedes_learning_id=id" json:"relsupersedeslearningid,omitempty"` // Has one ModelPublicLearnings
|
RelRelatedThoughtID *ModelPublicThoughts `bun:"rel:has-one,join:related_thought_id=id" json:"relrelatedthoughtid,omitempty"` // Has one ModelPublicThoughts
|
||||||
|
RelSupersedesLearningID *ModelPublicLearnings `bun:"rel:has-one,join:supersedes_learning_id=id" json:"relsupersedeslearningid,omitempty"` // Has one ModelPublicLearnings
|
||||||
|
RelTenantID *ModelPublicTenants `bun:"rel:has-one,join:tenant_id=id" json:"reltenantid,omitempty"` // Has one ModelPublicTenants
|
||||||
|
RelLearningIDPublicThoughtLearningLinks []*ModelPublicThoughtLearningLinks `bun:"rel:has-many,join:id=learning_id" json:"rellearningidpublicthoughtlearninglinks,omitempty"` // Has many ModelPublicThoughtLearningLinks
|
||||||
}
|
}
|
||||||
|
|
||||||
// TableName returns the table name for ModelPublicLearnings
|
// TableName returns the table name for ModelPublicLearnings
|
||||||
|
|||||||
@@ -0,0 +1,59 @@
|
|||||||
|
// Code generated by relspecgo. DO NOT EDIT.
|
||||||
|
package generatedmodels
|
||||||
|
|
||||||
|
import (
|
||||||
|
sql_types "git.warky.dev/wdevs/relspecgo/pkg/sqltypes"
|
||||||
|
"github.com/uptrace/bun"
|
||||||
|
)
|
||||||
|
|
||||||
|
type ModelPublicManagedAPIKeys struct {
|
||||||
|
bun.BaseModel `bun:"table:public.managed_api_keys,alias:managed_api_keys"`
|
||||||
|
KeyID sql_types.SqlString `bun:"key_id,type:text,pk," json:"key_id"`
|
||||||
|
SecretHash sql_types.SqlString `bun:"secret_hash,type:text,notnull," json:"secret_hash"`
|
||||||
|
RelKeyID *ModelPublicAPIKeyAssignments `bun:"rel:has-one,join:key_id=key_id" json:"relkeyid,omitempty"` // Has one ModelPublicAPIKeyAssignments
|
||||||
|
}
|
||||||
|
|
||||||
|
// TableName returns the table name for ModelPublicManagedAPIKeys
|
||||||
|
func (m ModelPublicManagedAPIKeys) TableName() string {
|
||||||
|
return "public.managed_api_keys"
|
||||||
|
}
|
||||||
|
|
||||||
|
// TableNameOnly returns the table name without schema for ModelPublicManagedAPIKeys
|
||||||
|
func (m ModelPublicManagedAPIKeys) TableNameOnly() string {
|
||||||
|
return "managed_api_keys"
|
||||||
|
}
|
||||||
|
|
||||||
|
// SchemaName returns the schema name for ModelPublicManagedAPIKeys
|
||||||
|
func (m ModelPublicManagedAPIKeys) SchemaName() string {
|
||||||
|
return "public"
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetID returns the primary key value
|
||||||
|
func (m ModelPublicManagedAPIKeys) GetID() string {
|
||||||
|
return m.KeyID.String()
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetIDStr returns the primary key as a string
|
||||||
|
func (m ModelPublicManagedAPIKeys) GetIDStr() string {
|
||||||
|
return m.KeyID.String()
|
||||||
|
}
|
||||||
|
|
||||||
|
// SetID sets the primary key value
|
||||||
|
func (m ModelPublicManagedAPIKeys) SetID(newid string) {
|
||||||
|
m.UpdateID(newid)
|
||||||
|
}
|
||||||
|
|
||||||
|
// UpdateID updates the primary key value
|
||||||
|
func (m *ModelPublicManagedAPIKeys) UpdateID(newid string) {
|
||||||
|
m.KeyID.FromString(newid)
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetIDName returns the name of the primary key column
|
||||||
|
func (m ModelPublicManagedAPIKeys) GetIDName() string {
|
||||||
|
return "key_id"
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetPrefix returns the table prefix
|
||||||
|
func (m ModelPublicManagedAPIKeys) GetPrefix() string {
|
||||||
|
return "MAK"
|
||||||
|
}
|
||||||
@@ -3,17 +3,17 @@ package generatedmodels
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
resolvespec_common "github.com/bitechdev/ResolveSpec/pkg/spectypes"
|
sql_types "git.warky.dev/wdevs/relspecgo/pkg/sqltypes"
|
||||||
"github.com/uptrace/bun"
|
"github.com/uptrace/bun"
|
||||||
)
|
)
|
||||||
|
|
||||||
type ModelPublicOauthClients struct {
|
type ModelPublicOauthClients struct {
|
||||||
bun.BaseModel `bun:"table:public.oauth_clients,alias:oauth_clients"`
|
bun.BaseModel `bun:"table:public.oauth_clients,alias:oauth_clients"`
|
||||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
ID sql_types.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||||
ClientID resolvespec_common.SqlString `bun:"client_id,type:text,notnull," json:"client_id"`
|
ClientID sql_types.SqlString `bun:"client_id,type:text,notnull," json:"client_id"`
|
||||||
ClientName resolvespec_common.SqlString `bun:"client_name,type:text,default:'',notnull," json:"client_name"`
|
ClientName sql_types.SqlString `bun:"client_name,type:text,default:'',notnull," json:"client_name"`
|
||||||
CreatedAt resolvespec_common.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
CreatedAt sql_types.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
||||||
RedirectUris resolvespec_common.SqlStringArray `bun:"redirect_uris,type:text[],default:'{}',notnull," json:"redirect_uris"`
|
RedirectUris sql_types.SqlStringArray `bun:"redirect_uris,type:text[],default:'{}',notnull," json:"redirect_uris"`
|
||||||
}
|
}
|
||||||
|
|
||||||
// TableName returns the table name for ModelPublicOauthClients
|
// TableName returns the table name for ModelPublicOauthClients
|
||||||
|
|||||||
@@ -3,20 +3,20 @@ package generatedmodels
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
resolvespec_common "github.com/bitechdev/ResolveSpec/pkg/spectypes"
|
sql_types "git.warky.dev/wdevs/relspecgo/pkg/sqltypes"
|
||||||
"github.com/uptrace/bun"
|
"github.com/uptrace/bun"
|
||||||
)
|
)
|
||||||
|
|
||||||
type ModelPublicPersonaArc struct {
|
type ModelPublicPersonaArc struct {
|
||||||
bun.BaseModel `bun:"table:public.persona_arc,alias:persona_arc"`
|
bun.BaseModel `bun:"table:public.persona_arc,alias:persona_arc"`
|
||||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
ID sql_types.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||||
PersonaID int64 `bun:"persona_id,type:bigint,pk," json:"persona_id"`
|
ArcID int64 `bun:"arc_id,type:bigint,notnull," json:"arc_id"`
|
||||||
ArcID int64 `bun:"arc_id,type:bigint,notnull," json:"arc_id"`
|
CurrentStageID int64 `bun:"current_stage_id,type:bigint,notnull," json:"current_stage_id"`
|
||||||
CurrentStageID int64 `bun:"current_stage_id,type:bigint,notnull," json:"current_stage_id"`
|
PersonaID int64 `bun:"persona_id,type:bigint,notnull," json:"persona_id"`
|
||||||
UpdatedAt resolvespec_common.SqlTimeStamp `bun:"updated_at,type:timestamptz,default:now(),notnull," json:"updated_at"`
|
UpdatedAt sql_types.SqlTimeStamp `bun:"updated_at,type:timestamptz,default:now(),notnull," json:"updated_at"`
|
||||||
RelArcID *ModelPublicCharacterArcs `bun:"rel:has-one,join:arc_id=id" json:"relarcid,omitempty"` // Has one ModelPublicCharacterArcs
|
RelArcID *ModelPublicCharacterArcs `bun:"rel:has-one,join:arc_id=id" json:"relarcid,omitempty"` // Has one ModelPublicCharacterArcs
|
||||||
RelCurrentStageID *ModelPublicArcStages `bun:"rel:has-one,join:current_stage_id=id" json:"relcurrentstageid,omitempty"` // Has one ModelPublicArcStages
|
RelCurrentStageID *ModelPublicArcStages `bun:"rel:has-one,join:current_stage_id=id" json:"relcurrentstageid,omitempty"` // Has one ModelPublicArcStages
|
||||||
RelPersonaID *ModelPublicAgentPersonas `bun:"rel:has-one,join:persona_id=id" json:"relpersonaid,omitempty"` // Has one ModelPublicAgentPersonas
|
RelPersonaID *ModelPublicAgentPersonas `bun:"rel:has-one,join:persona_id=id" json:"relpersonaid,omitempty"` // Has one ModelPublicAgentPersonas
|
||||||
}
|
}
|
||||||
|
|
||||||
// TableName returns the table name for ModelPublicPersonaArc
|
// TableName returns the table name for ModelPublicPersonaArc
|
||||||
|
|||||||
@@ -3,18 +3,18 @@ package generatedmodels
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
resolvespec_common "github.com/bitechdev/ResolveSpec/pkg/spectypes"
|
sql_types "git.warky.dev/wdevs/relspecgo/pkg/sqltypes"
|
||||||
"github.com/uptrace/bun"
|
"github.com/uptrace/bun"
|
||||||
)
|
)
|
||||||
|
|
||||||
type ModelPublicPlanDependencies struct {
|
type ModelPublicPlanDependencies struct {
|
||||||
bun.BaseModel `bun:"table:public.plan_dependencies,alias:plan_dependencies"`
|
bun.BaseModel `bun:"table:public.plan_dependencies,alias:plan_dependencies"`
|
||||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
ID sql_types.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||||
CreatedAt resolvespec_common.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
CreatedAt sql_types.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
||||||
DependsOnPlanID int64 `bun:"depends_on_plan_id,type:bigint,notnull,unique:uidx_plan_dependencies_plan_id_depends_on_plan_id," json:"depends_on_plan_id"`
|
DependsOnPlanID int64 `bun:"depends_on_plan_id,type:bigint,notnull,unique:uidx_plan_dependencies_plan_id_depends_on_plan_id," json:"depends_on_plan_id"`
|
||||||
PlanID int64 `bun:"plan_id,type:bigint,notnull,unique:uidx_plan_dependencies_plan_id_depends_on_plan_id," json:"plan_id"`
|
PlanID int64 `bun:"plan_id,type:bigint,notnull,unique:uidx_plan_dependencies_plan_id_depends_on_plan_id," json:"plan_id"`
|
||||||
RelDependsOnPlanID *ModelPublicPlans `bun:"rel:has-one,join:depends_on_plan_id=id" json:"reldependsonplanid,omitempty"` // Has one ModelPublicPlans
|
RelDependsOnPlanID *ModelPublicPlans `bun:"rel:has-one,join:depends_on_plan_id=id" json:"reldependsonplanid,omitempty"` // Has one ModelPublicPlans
|
||||||
RelPlanID *ModelPublicPlans `bun:"rel:has-one,join:plan_id=id" json:"relplanid,omitempty"` // Has one ModelPublicPlans
|
RelPlanID *ModelPublicPlans `bun:"rel:has-one,join:plan_id=id" json:"relplanid,omitempty"` // Has one ModelPublicPlans
|
||||||
}
|
}
|
||||||
|
|
||||||
// TableName returns the table name for ModelPublicPlanDependencies
|
// TableName returns the table name for ModelPublicPlanDependencies
|
||||||
|
|||||||
@@ -3,18 +3,18 @@ package generatedmodels
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
resolvespec_common "github.com/bitechdev/ResolveSpec/pkg/spectypes"
|
sql_types "git.warky.dev/wdevs/relspecgo/pkg/sqltypes"
|
||||||
"github.com/uptrace/bun"
|
"github.com/uptrace/bun"
|
||||||
)
|
)
|
||||||
|
|
||||||
type ModelPublicPlanGuardrails struct {
|
type ModelPublicPlanGuardrails struct {
|
||||||
bun.BaseModel `bun:"table:public.plan_guardrails,alias:plan_guardrails"`
|
bun.BaseModel `bun:"table:public.plan_guardrails,alias:plan_guardrails"`
|
||||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
ID sql_types.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||||
CreatedAt resolvespec_common.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
CreatedAt sql_types.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
||||||
GuardrailID int64 `bun:"guardrail_id,type:bigint,notnull,unique:uidx_plan_guardrails_plan_id_guardrail_id," json:"guardrail_id"`
|
GuardrailID int64 `bun:"guardrail_id,type:bigint,notnull,unique:uidx_plan_guardrails_plan_id_guardrail_id," json:"guardrail_id"`
|
||||||
PlanID int64 `bun:"plan_id,type:bigint,notnull,unique:uidx_plan_guardrails_plan_id_guardrail_id," json:"plan_id"`
|
PlanID int64 `bun:"plan_id,type:bigint,notnull,unique:uidx_plan_guardrails_plan_id_guardrail_id," json:"plan_id"`
|
||||||
RelGuardrailID *ModelPublicAgentGuardrails `bun:"rel:has-one,join:guardrail_id=id" json:"relguardrailid,omitempty"` // Has one ModelPublicAgentGuardrails
|
RelGuardrailID *ModelPublicAgentGuardrails `bun:"rel:has-one,join:guardrail_id=id" json:"relguardrailid,omitempty"` // Has one ModelPublicAgentGuardrails
|
||||||
RelPlanID *ModelPublicPlans `bun:"rel:has-one,join:plan_id=id" json:"relplanid,omitempty"` // Has one ModelPublicPlans
|
RelPlanID *ModelPublicPlans `bun:"rel:has-one,join:plan_id=id" json:"relplanid,omitempty"` // Has one ModelPublicPlans
|
||||||
}
|
}
|
||||||
|
|
||||||
// TableName returns the table name for ModelPublicPlanGuardrails
|
// TableName returns the table name for ModelPublicPlanGuardrails
|
||||||
|
|||||||
@@ -3,18 +3,18 @@ package generatedmodels
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
resolvespec_common "github.com/bitechdev/ResolveSpec/pkg/spectypes"
|
sql_types "git.warky.dev/wdevs/relspecgo/pkg/sqltypes"
|
||||||
"github.com/uptrace/bun"
|
"github.com/uptrace/bun"
|
||||||
)
|
)
|
||||||
|
|
||||||
type ModelPublicPlanRelatedPlans struct {
|
type ModelPublicPlanRelatedPlans struct {
|
||||||
bun.BaseModel `bun:"table:public.plan_related_plans,alias:plan_related_plans"`
|
bun.BaseModel `bun:"table:public.plan_related_plans,alias:plan_related_plans"`
|
||||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
ID sql_types.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||||
CreatedAt resolvespec_common.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
CreatedAt sql_types.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
||||||
PlanAID int64 `bun:"plan_a_id,type:bigint,notnull,unique:uidx_plan_related_plans_plan_a_id_plan_b_id," json:"plan_a_id"`
|
PlanAID int64 `bun:"plan_a_id,type:bigint,notnull,unique:uidx_plan_related_plans_plan_a_id_plan_b_id," json:"plan_a_id"`
|
||||||
PlanBID int64 `bun:"plan_b_id,type:bigint,notnull,unique:uidx_plan_related_plans_plan_a_id_plan_b_id," json:"plan_b_id"`
|
PlanBID int64 `bun:"plan_b_id,type:bigint,notnull,unique:uidx_plan_related_plans_plan_a_id_plan_b_id," json:"plan_b_id"`
|
||||||
RelPlanAID *ModelPublicPlans `bun:"rel:has-one,join:plan_a_id=id" json:"relplanaid,omitempty"` // Has one ModelPublicPlans
|
RelPlanAID *ModelPublicPlans `bun:"rel:has-one,join:plan_a_id=id" json:"relplanaid,omitempty"` // Has one ModelPublicPlans
|
||||||
RelPlanBID *ModelPublicPlans `bun:"rel:has-one,join:plan_b_id=id" json:"relplanbid,omitempty"` // Has one ModelPublicPlans
|
RelPlanBID *ModelPublicPlans `bun:"rel:has-one,join:plan_b_id=id" json:"relplanbid,omitempty"` // Has one ModelPublicPlans
|
||||||
}
|
}
|
||||||
|
|
||||||
// TableName returns the table name for ModelPublicPlanRelatedPlans
|
// TableName returns the table name for ModelPublicPlanRelatedPlans
|
||||||
|
|||||||
@@ -3,18 +3,18 @@ package generatedmodels
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
resolvespec_common "github.com/bitechdev/ResolveSpec/pkg/spectypes"
|
sql_types "git.warky.dev/wdevs/relspecgo/pkg/sqltypes"
|
||||||
"github.com/uptrace/bun"
|
"github.com/uptrace/bun"
|
||||||
)
|
)
|
||||||
|
|
||||||
type ModelPublicPlanSkills struct {
|
type ModelPublicPlanSkills struct {
|
||||||
bun.BaseModel `bun:"table:public.plan_skills,alias:plan_skills"`
|
bun.BaseModel `bun:"table:public.plan_skills,alias:plan_skills"`
|
||||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
ID sql_types.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||||
CreatedAt resolvespec_common.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
CreatedAt sql_types.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
||||||
PlanID int64 `bun:"plan_id,type:bigint,notnull,unique:uidx_plan_skills_plan_id_skill_id," json:"plan_id"`
|
PlanID int64 `bun:"plan_id,type:bigint,notnull,unique:uidx_plan_skills_plan_id_skill_id," json:"plan_id"`
|
||||||
SkillID int64 `bun:"skill_id,type:bigint,notnull,unique:uidx_plan_skills_plan_id_skill_id," json:"skill_id"`
|
SkillID int64 `bun:"skill_id,type:bigint,notnull,unique:uidx_plan_skills_plan_id_skill_id," json:"skill_id"`
|
||||||
RelPlanID *ModelPublicPlans `bun:"rel:has-one,join:plan_id=id" json:"relplanid,omitempty"` // Has one ModelPublicPlans
|
RelPlanID *ModelPublicPlans `bun:"rel:has-one,join:plan_id=id" json:"relplanid,omitempty"` // Has one ModelPublicPlans
|
||||||
RelSkillID *ModelPublicAgentSkills `bun:"rel:has-one,join:skill_id=id" json:"relskillid,omitempty"` // Has one ModelPublicAgentSkills
|
RelSkillID *ModelPublicAgentSkills `bun:"rel:has-one,join:skill_id=id" json:"relskillid,omitempty"` // Has one ModelPublicAgentSkills
|
||||||
}
|
}
|
||||||
|
|
||||||
// TableName returns the table name for ModelPublicPlanSkills
|
// TableName returns the table name for ModelPublicPlanSkills
|
||||||
|
|||||||
@@ -3,36 +3,38 @@ package generatedmodels
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
resolvespec_common "github.com/bitechdev/ResolveSpec/pkg/spectypes"
|
sql_types "git.warky.dev/wdevs/relspecgo/pkg/sqltypes"
|
||||||
"github.com/uptrace/bun"
|
"github.com/uptrace/bun"
|
||||||
)
|
)
|
||||||
|
|
||||||
type ModelPublicPlans struct {
|
type ModelPublicPlans struct {
|
||||||
bun.BaseModel `bun:"table:public.plans,alias:plans"`
|
bun.BaseModel `bun:"table:public.plans,alias:plans"`
|
||||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
ID sql_types.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||||
CompletedAt resolvespec_common.SqlTimeStamp `bun:"completed_at,type:timestamptz,nullzero," json:"completed_at"`
|
CompletedAt sql_types.SqlTimeStamp `bun:"completed_at,type:timestamptz,nullzero," json:"completed_at"`
|
||||||
CreatedAt resolvespec_common.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
CreatedAt sql_types.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
||||||
Description resolvespec_common.SqlString `bun:"description,type:text,default:'',notnull," json:"description"`
|
Description sql_types.SqlString `bun:"description,type:text,default:'',notnull," json:"description"`
|
||||||
DueDate resolvespec_common.SqlTimeStamp `bun:"due_date,type:timestamptz,nullzero," json:"due_date"`
|
DueDate sql_types.SqlTimeStamp `bun:"due_date,type:timestamptz,nullzero," json:"due_date"`
|
||||||
GUID resolvespec_common.SqlUUID `bun:"guid,type:uuid,default:gen_random_uuid(),notnull," json:"guid"`
|
GUID sql_types.SqlUUID `bun:"guid,type:uuid,default:gen_random_uuid(),notnull," json:"guid"`
|
||||||
LastReviewedAt resolvespec_common.SqlTimeStamp `bun:"last_reviewed_at,type:timestamptz,nullzero," json:"last_reviewed_at"`
|
LastReviewedAt sql_types.SqlTimeStamp `bun:"last_reviewed_at,type:timestamptz,nullzero," json:"last_reviewed_at"`
|
||||||
Owner resolvespec_common.SqlString `bun:"owner,type:text,nullzero," json:"owner"`
|
Owner sql_types.SqlString `bun:"owner,type:text,nullzero," json:"owner"`
|
||||||
Priority resolvespec_common.SqlString `bun:"priority,type:text,default:'medium',notnull," json:"priority"` // low, medium, high, critical
|
Priority sql_types.SqlString `bun:"priority,type:text,default:'medium',notnull," json:"priority"` // low, medium, high, critical
|
||||||
ProjectID resolvespec_common.SqlInt64 `bun:"project_id,type:bigint,nullzero," json:"project_id"`
|
ProjectID sql_types.SqlInt64 `bun:"project_id,type:bigint,nullzero," json:"project_id"`
|
||||||
ReviewedBy resolvespec_common.SqlString `bun:"reviewed_by,type:text,nullzero," json:"reviewed_by"`
|
ReviewedBy sql_types.SqlString `bun:"reviewed_by,type:text,nullzero," json:"reviewed_by"`
|
||||||
Status resolvespec_common.SqlString `bun:"status,type:text,default:'draft',notnull," json:"status"` // draft, active, blocked, completed, cancelled, superseded
|
Status sql_types.SqlString `bun:"status,type:text,default:'draft',notnull," json:"status"` // draft, active, blocked, completed, cancelled, superseded
|
||||||
SupersedesPlanID resolvespec_common.SqlInt64 `bun:"supersedes_plan_id,type:bigint,nullzero," json:"supersedes_plan_id"`
|
SupersedesPlanID sql_types.SqlInt64 `bun:"supersedes_plan_id,type:bigint,nullzero," json:"supersedes_plan_id"`
|
||||||
Tags resolvespec_common.SqlStringArray `bun:"tags,type:text[],default:'{}',notnull," json:"tags"`
|
Tags sql_types.SqlStringArray `bun:"tags,type:text[],default:'{}',notnull," json:"tags"`
|
||||||
Title resolvespec_common.SqlString `bun:"title,type:text,notnull," json:"title"`
|
TenantID sql_types.SqlString `bun:"tenant_id,type:text,nullzero," json:"tenant_id"`
|
||||||
UpdatedAt resolvespec_common.SqlTimeStamp `bun:"updated_at,type:timestamptz,default:now(),notnull," json:"updated_at"`
|
Title sql_types.SqlString `bun:"title,type:text,notnull," json:"title"`
|
||||||
RelProjectID *ModelPublicProjects `bun:"rel:has-one,join:project_id=id" json:"relprojectid,omitempty"` // Has one ModelPublicProjects
|
UpdatedAt sql_types.SqlTimeStamp `bun:"updated_at,type:timestamptz,default:now(),notnull," json:"updated_at"`
|
||||||
RelSupersedesPlanID *ModelPublicPlans `bun:"rel:has-one,join:supersedes_plan_id=id" json:"relsupersedesplanid,omitempty"` // Has one ModelPublicPlans
|
RelProjectID *ModelPublicProjects `bun:"rel:has-one,join:project_id=id" json:"relprojectid,omitempty"` // Has one ModelPublicProjects
|
||||||
RelDependsOnPlanIDPublicPlanDependencies []*ModelPublicPlanDependencies `bun:"rel:has-many,join:id=depends_on_plan_id" json:"reldependsonplanidpublicplandependencies,omitempty"` // Has many ModelPublicPlanDependencies
|
RelSupersedesPlanID *ModelPublicPlans `bun:"rel:has-one,join:supersedes_plan_id=id" json:"relsupersedesplanid,omitempty"` // Has one ModelPublicPlans
|
||||||
RelPlanIDPublicPlanDependencies []*ModelPublicPlanDependencies `bun:"rel:has-many,join:id=plan_id" json:"relplanidpublicplandependencies,omitempty"` // Has many ModelPublicPlanDependencies
|
RelTenantID *ModelPublicTenants `bun:"rel:has-one,join:tenant_id=id" json:"reltenantid,omitempty"` // Has one ModelPublicTenants
|
||||||
RelPlanAIDPublicPlanRelatedPlans []*ModelPublicPlanRelatedPlans `bun:"rel:has-many,join:id=plan_a_id" json:"relplanaidpublicplanrelatedplans,omitempty"` // Has many ModelPublicPlanRelatedPlans
|
RelDependsOnPlanIDPublicPlanDependencies []*ModelPublicPlanDependencies `bun:"rel:has-many,join:id=depends_on_plan_id" json:"reldependsonplanidpublicplandependencies,omitempty"` // Has many ModelPublicPlanDependencies
|
||||||
RelPlanBIDPublicPlanRelatedPlans []*ModelPublicPlanRelatedPlans `bun:"rel:has-many,join:id=plan_b_id" json:"relplanbidpublicplanrelatedplans,omitempty"` // Has many ModelPublicPlanRelatedPlans
|
RelPlanIDPublicPlanDependencies []*ModelPublicPlanDependencies `bun:"rel:has-many,join:id=plan_id" json:"relplanidpublicplandependencies,omitempty"` // Has many ModelPublicPlanDependencies
|
||||||
RelPlanIDPublicPlanSkills []*ModelPublicPlanSkills `bun:"rel:has-many,join:id=plan_id" json:"relplanidpublicplanskills,omitempty"` // Has many ModelPublicPlanSkills
|
RelPlanAIDPublicPlanRelatedPlans []*ModelPublicPlanRelatedPlans `bun:"rel:has-many,join:id=plan_a_id" json:"relplanaidpublicplanrelatedplans,omitempty"` // Has many ModelPublicPlanRelatedPlans
|
||||||
RelPlanIDPublicPlanGuardrails []*ModelPublicPlanGuardrails `bun:"rel:has-many,join:id=plan_id" json:"relplanidpublicplanguardrails,omitempty"` // Has many ModelPublicPlanGuardrails
|
RelPlanBIDPublicPlanRelatedPlans []*ModelPublicPlanRelatedPlans `bun:"rel:has-many,join:id=plan_b_id" json:"relplanbidpublicplanrelatedplans,omitempty"` // Has many ModelPublicPlanRelatedPlans
|
||||||
|
RelPlanIDPublicPlanSkills []*ModelPublicPlanSkills `bun:"rel:has-many,join:id=plan_id" json:"relplanidpublicplanskills,omitempty"` // Has many ModelPublicPlanSkills
|
||||||
|
RelPlanIDPublicPlanGuardrails []*ModelPublicPlanGuardrails `bun:"rel:has-many,join:id=plan_id" json:"relplanidpublicplanguardrails,omitempty"` // Has many ModelPublicPlanGuardrails
|
||||||
}
|
}
|
||||||
|
|
||||||
// TableName returns the table name for ModelPublicPlans
|
// TableName returns the table name for ModelPublicPlans
|
||||||
|
|||||||
@@ -3,18 +3,18 @@ package generatedmodels
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
resolvespec_common "github.com/bitechdev/ResolveSpec/pkg/spectypes"
|
sql_types "git.warky.dev/wdevs/relspecgo/pkg/sqltypes"
|
||||||
"github.com/uptrace/bun"
|
"github.com/uptrace/bun"
|
||||||
)
|
)
|
||||||
|
|
||||||
type ModelPublicProjectGuardrails struct {
|
type ModelPublicProjectGuardrails struct {
|
||||||
bun.BaseModel `bun:"table:public.project_guardrails,alias:project_guardrails"`
|
bun.BaseModel `bun:"table:public.project_guardrails,alias:project_guardrails"`
|
||||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
ID sql_types.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||||
CreatedAt resolvespec_common.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
CreatedAt sql_types.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
||||||
GuardrailID int64 `bun:"guardrail_id,type:bigint,notnull," json:"guardrail_id"`
|
GuardrailID int64 `bun:"guardrail_id,type:bigint,notnull," json:"guardrail_id"`
|
||||||
ProjectID int64 `bun:"project_id,type:bigint,notnull," json:"project_id"`
|
ProjectID int64 `bun:"project_id,type:bigint,notnull," json:"project_id"`
|
||||||
RelGuardrailID *ModelPublicAgentGuardrails `bun:"rel:has-one,join:guardrail_id=id" json:"relguardrailid,omitempty"` // Has one ModelPublicAgentGuardrails
|
RelGuardrailID *ModelPublicAgentGuardrails `bun:"rel:has-one,join:guardrail_id=id" json:"relguardrailid,omitempty"` // Has one ModelPublicAgentGuardrails
|
||||||
RelProjectID *ModelPublicProjects `bun:"rel:has-one,join:project_id=id" json:"relprojectid,omitempty"` // Has one ModelPublicProjects
|
RelProjectID *ModelPublicProjects `bun:"rel:has-one,join:project_id=id" json:"relprojectid,omitempty"` // Has one ModelPublicProjects
|
||||||
}
|
}
|
||||||
|
|
||||||
// TableName returns the table name for ModelPublicProjectGuardrails
|
// TableName returns the table name for ModelPublicProjectGuardrails
|
||||||
|
|||||||
@@ -3,19 +3,19 @@ package generatedmodels
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
resolvespec_common "github.com/bitechdev/ResolveSpec/pkg/spectypes"
|
sql_types "git.warky.dev/wdevs/relspecgo/pkg/sqltypes"
|
||||||
"github.com/uptrace/bun"
|
"github.com/uptrace/bun"
|
||||||
)
|
)
|
||||||
|
|
||||||
type ModelPublicProjectPersonas struct {
|
type ModelPublicProjectPersonas struct {
|
||||||
bun.BaseModel `bun:"table:public.project_personas,alias:project_personas"`
|
bun.BaseModel `bun:"table:public.project_personas,alias:project_personas"`
|
||||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
ID sql_types.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||||
CreatedAt resolvespec_common.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
CreatedAt sql_types.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
||||||
IsDefault bool `bun:"is_default,type:boolean,default:false,notnull," json:"is_default"`
|
IsDefault bool `bun:"is_default,type:boolean,default:false,notnull," json:"is_default"`
|
||||||
PersonaID int64 `bun:"persona_id,type:bigint,notnull," json:"persona_id"`
|
PersonaID int64 `bun:"persona_id,type:bigint,notnull," json:"persona_id"`
|
||||||
ProjectID int64 `bun:"project_id,type:bigint,notnull," json:"project_id"`
|
ProjectID int64 `bun:"project_id,type:bigint,notnull," json:"project_id"`
|
||||||
RelPersonaID *ModelPublicAgentPersonas `bun:"rel:has-one,join:persona_id=id" json:"relpersonaid,omitempty"` // Has one ModelPublicAgentPersonas
|
RelPersonaID *ModelPublicAgentPersonas `bun:"rel:has-one,join:persona_id=id" json:"relpersonaid,omitempty"` // Has one ModelPublicAgentPersonas
|
||||||
RelProjectID *ModelPublicProjects `bun:"rel:has-one,join:project_id=id" json:"relprojectid,omitempty"` // Has one ModelPublicProjects
|
RelProjectID *ModelPublicProjects `bun:"rel:has-one,join:project_id=id" json:"relprojectid,omitempty"` // Has one ModelPublicProjects
|
||||||
}
|
}
|
||||||
|
|
||||||
// TableName returns the table name for ModelPublicProjectPersonas
|
// TableName returns the table name for ModelPublicProjectPersonas
|
||||||
|
|||||||
@@ -3,19 +3,19 @@ package generatedmodels
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
resolvespec_common "github.com/bitechdev/ResolveSpec/pkg/spectypes"
|
sql_types "git.warky.dev/wdevs/relspecgo/pkg/sqltypes"
|
||||||
"github.com/uptrace/bun"
|
"github.com/uptrace/bun"
|
||||||
)
|
)
|
||||||
|
|
||||||
type ModelPublicProjectSkills struct {
|
type ModelPublicProjectSkills struct {
|
||||||
bun.BaseModel `bun:"table:public.project_skills,alias:project_skills"`
|
bun.BaseModel `bun:"table:public.project_skills,alias:project_skills"`
|
||||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
ID sql_types.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||||
CreatedAt resolvespec_common.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
CreatedAt sql_types.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
||||||
Override bool `bun:"override,type:boolean,default:false,notnull," json:"override"`
|
Override bool `bun:"override,type:boolean,default:false,notnull," json:"override"`
|
||||||
ProjectID int64 `bun:"project_id,type:bigint,notnull," json:"project_id"`
|
ProjectID int64 `bun:"project_id,type:bigint,notnull," json:"project_id"`
|
||||||
SkillID int64 `bun:"skill_id,type:bigint,notnull," json:"skill_id"`
|
SkillID int64 `bun:"skill_id,type:bigint,notnull," json:"skill_id"`
|
||||||
RelProjectID *ModelPublicProjects `bun:"rel:has-one,join:project_id=id" json:"relprojectid,omitempty"` // Has one ModelPublicProjects
|
RelProjectID *ModelPublicProjects `bun:"rel:has-one,join:project_id=id" json:"relprojectid,omitempty"` // Has one ModelPublicProjects
|
||||||
RelSkillID *ModelPublicAgentSkills `bun:"rel:has-one,join:skill_id=id" json:"relskillid,omitempty"` // Has one ModelPublicAgentSkills
|
RelSkillID *ModelPublicAgentSkills `bun:"rel:has-one,join:skill_id=id" json:"relskillid,omitempty"` // Has one ModelPublicAgentSkills
|
||||||
}
|
}
|
||||||
|
|
||||||
// TableName returns the table name for ModelPublicProjectSkills
|
// TableName returns the table name for ModelPublicProjectSkills
|
||||||
|
|||||||
@@ -3,19 +3,20 @@ package generatedmodels
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
resolvespec_common "github.com/bitechdev/ResolveSpec/pkg/spectypes"
|
sql_types "git.warky.dev/wdevs/relspecgo/pkg/sqltypes"
|
||||||
"github.com/uptrace/bun"
|
"github.com/uptrace/bun"
|
||||||
)
|
)
|
||||||
|
|
||||||
type ModelPublicProjects struct {
|
type ModelPublicProjects struct {
|
||||||
bun.BaseModel `bun:"table:public.projects,alias:projects"`
|
bun.BaseModel `bun:"table:public.projects,alias:projects"`
|
||||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
ID sql_types.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||||
CreatedAt resolvespec_common.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),nullzero," json:"created_at"`
|
CreatedAt sql_types.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),nullzero," json:"created_at"`
|
||||||
Description resolvespec_common.SqlString `bun:"description,type:text,nullzero," json:"description"`
|
Description sql_types.SqlString `bun:"description,type:text,nullzero," json:"description"`
|
||||||
GUID resolvespec_common.SqlUUID `bun:"guid,type:uuid,default:gen_random_uuid(),notnull," json:"guid"`
|
GUID sql_types.SqlUUID `bun:"guid,type:uuid,default:gen_random_uuid(),notnull," json:"guid"`
|
||||||
LastActiveAt resolvespec_common.SqlTimeStamp `bun:"last_active_at,type:timestamptz,default:now(),nullzero," json:"last_active_at"`
|
LastActiveAt sql_types.SqlTimeStamp `bun:"last_active_at,type:timestamptz,default:now(),nullzero," json:"last_active_at"`
|
||||||
Name resolvespec_common.SqlString `bun:"name,type:text,notnull," json:"name"`
|
Name sql_types.SqlString `bun:"name,type:text,notnull,unique:uidx_projects_tenant_id_name," json:"name"`
|
||||||
ThoughtCount resolvespec_common.SqlInt64 `bun:"thought_count,scanonly" json:"thought_count"`
|
TenantID sql_types.SqlString `bun:"tenant_id,type:text,nullzero,unique:uidx_projects_tenant_id_name," json:"tenant_id"`
|
||||||
|
RelTenantID *ModelPublicTenants `bun:"rel:has-one,join:tenant_id=id" json:"reltenantid,omitempty"` // Has one ModelPublicTenants
|
||||||
RelProjectIDPublicProjectPersonas []*ModelPublicProjectPersonas `bun:"rel:has-many,join:id=project_id" json:"relprojectidpublicprojectpersonas,omitempty"` // Has many ModelPublicProjectPersonas
|
RelProjectIDPublicProjectPersonas []*ModelPublicProjectPersonas `bun:"rel:has-many,join:id=project_id" json:"relprojectidpublicprojectpersonas,omitempty"` // Has many ModelPublicProjectPersonas
|
||||||
RelProjectIDPublicThoughts []*ModelPublicThoughts `bun:"rel:has-many,join:id=project_id" json:"relprojectidpublicthoughts,omitempty"` // Has many ModelPublicThoughts
|
RelProjectIDPublicThoughts []*ModelPublicThoughts `bun:"rel:has-many,join:id=project_id" json:"relprojectidpublicthoughts,omitempty"` // Has many ModelPublicThoughts
|
||||||
RelProjectIDPublicStoredFiles []*ModelPublicStoredFiles `bun:"rel:has-many,join:id=project_id" json:"relprojectidpublicstoredfiles,omitempty"` // Has many ModelPublicStoredFiles
|
RelProjectIDPublicStoredFiles []*ModelPublicStoredFiles `bun:"rel:has-many,join:id=project_id" json:"relprojectidpublicstoredfiles,omitempty"` // Has many ModelPublicStoredFiles
|
||||||
|
|||||||
@@ -3,27 +3,29 @@ package generatedmodels
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
resolvespec_common "github.com/bitechdev/ResolveSpec/pkg/spectypes"
|
sql_types "git.warky.dev/wdevs/relspecgo/pkg/sqltypes"
|
||||||
"github.com/uptrace/bun"
|
"github.com/uptrace/bun"
|
||||||
)
|
)
|
||||||
|
|
||||||
type ModelPublicStoredFiles struct {
|
type ModelPublicStoredFiles struct {
|
||||||
bun.BaseModel `bun:"table:public.stored_files,alias:stored_files"`
|
bun.BaseModel `bun:"table:public.stored_files,alias:stored_files"`
|
||||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
ID sql_types.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||||
Content []byte `bun:"content,type:bytea,notnull," json:"content"`
|
Content []byte `bun:"content,type:bytea,notnull," json:"content"`
|
||||||
CreatedAt resolvespec_common.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
CreatedAt sql_types.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
||||||
Encoding resolvespec_common.SqlString `bun:"encoding,type:text,default:'base64',notnull," json:"encoding"`
|
Encoding sql_types.SqlString `bun:"encoding,type:text,default:'base64',notnull," json:"encoding"`
|
||||||
GUID resolvespec_common.SqlUUID `bun:"guid,type:uuid,default:gen_random_uuid(),notnull," json:"guid"`
|
GUID sql_types.SqlUUID `bun:"guid,type:uuid,default:gen_random_uuid(),notnull," json:"guid"`
|
||||||
Kind resolvespec_common.SqlString `bun:"kind,type:text,default:'file',notnull," json:"kind"`
|
Kind sql_types.SqlString `bun:"kind,type:text,default:'file',notnull," json:"kind"`
|
||||||
MediaType resolvespec_common.SqlString `bun:"media_type,type:text,notnull," json:"media_type"`
|
MediaType sql_types.SqlString `bun:"media_type,type:text,notnull," json:"media_type"`
|
||||||
Name resolvespec_common.SqlString `bun:"name,type:text,notnull," json:"name"`
|
Name sql_types.SqlString `bun:"name,type:text,notnull," json:"name"`
|
||||||
ProjectID resolvespec_common.SqlInt64 `bun:"project_id,type:bigint,nullzero," json:"project_id"`
|
ProjectID sql_types.SqlInt64 `bun:"project_id,type:bigint,nullzero," json:"project_id"`
|
||||||
Sha256 resolvespec_common.SqlString `bun:"sha256,type:text,notnull," json:"sha256"`
|
Sha256 sql_types.SqlString `bun:"sha256,type:text,notnull," json:"sha256"`
|
||||||
SizeBytes int64 `bun:"size_bytes,type:bigint,notnull," json:"size_bytes"`
|
SizeBytes int64 `bun:"size_bytes,type:bigint,notnull," json:"size_bytes"`
|
||||||
ThoughtID resolvespec_common.SqlInt64 `bun:"thought_id,type:bigint,nullzero," json:"thought_id"`
|
TenantID sql_types.SqlString `bun:"tenant_id,type:text,nullzero," json:"tenant_id"`
|
||||||
UpdatedAt resolvespec_common.SqlTimeStamp `bun:"updated_at,type:timestamptz,default:now(),notnull," json:"updated_at"`
|
ThoughtID sql_types.SqlInt64 `bun:"thought_id,type:bigint,nullzero," json:"thought_id"`
|
||||||
RelProjectID *ModelPublicProjects `bun:"rel:has-one,join:project_id=id" json:"relprojectid,omitempty"` // Has one ModelPublicProjects
|
UpdatedAt sql_types.SqlTimeStamp `bun:"updated_at,type:timestamptz,default:now(),notnull," json:"updated_at"`
|
||||||
RelThoughtID *ModelPublicThoughts `bun:"rel:has-one,join:thought_id=id" json:"relthoughtid,omitempty"` // Has one ModelPublicThoughts
|
RelProjectID *ModelPublicProjects `bun:"rel:has-one,join:project_id=id" json:"relprojectid,omitempty"` // Has one ModelPublicProjects
|
||||||
|
RelTenantID *ModelPublicTenants `bun:"rel:has-one,join:tenant_id=id" json:"reltenantid,omitempty"` // Has one ModelPublicTenants
|
||||||
|
RelThoughtID *ModelPublicThoughts `bun:"rel:has-one,join:thought_id=id" json:"relthoughtid,omitempty"` // Has one ModelPublicThoughts
|
||||||
}
|
}
|
||||||
|
|
||||||
// TableName returns the table name for ModelPublicStoredFiles
|
// TableName returns the table name for ModelPublicStoredFiles
|
||||||
|
|||||||
@@ -0,0 +1,63 @@
|
|||||||
|
// Code generated by relspecgo. DO NOT EDIT.
|
||||||
|
package generatedmodels
|
||||||
|
|
||||||
|
import (
|
||||||
|
sql_types "git.warky.dev/wdevs/relspecgo/pkg/sqltypes"
|
||||||
|
"github.com/uptrace/bun"
|
||||||
|
)
|
||||||
|
|
||||||
|
type ModelPublicTenantUsers struct {
|
||||||
|
bun.BaseModel `bun:"table:public.tenant_users,alias:tenant_users"`
|
||||||
|
ID sql_types.SqlString `bun:"id,type:text,pk," json:"id"`
|
||||||
|
CreatedAt sql_types.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
||||||
|
Email sql_types.SqlString `bun:"email,type:text,nullzero,unique:uidx_tenant_users_tenant_id_email," json:"email"`
|
||||||
|
Name sql_types.SqlString `bun:"name,type:text,notnull," json:"name"`
|
||||||
|
TenantID sql_types.SqlString `bun:"tenant_id,type:text,notnull,unique:uidx_tenant_users_tenant_id_email," json:"tenant_id"`
|
||||||
|
RelTenantID *ModelPublicTenants `bun:"rel:has-one,join:tenant_id=id" json:"reltenantid,omitempty"` // Has one ModelPublicTenants
|
||||||
|
RelUserIDPublicAPIKeyAssignments []*ModelPublicAPIKeyAssignments `bun:"rel:has-many,join:id=user_id" json:"reluseridpublicapikeyassignments,omitempty"` // Has many ModelPublicAPIKeyAssignments
|
||||||
|
}
|
||||||
|
|
||||||
|
// TableName returns the table name for ModelPublicTenantUsers
|
||||||
|
func (m ModelPublicTenantUsers) TableName() string {
|
||||||
|
return "public.tenant_users"
|
||||||
|
}
|
||||||
|
|
||||||
|
// TableNameOnly returns the table name without schema for ModelPublicTenantUsers
|
||||||
|
func (m ModelPublicTenantUsers) TableNameOnly() string {
|
||||||
|
return "tenant_users"
|
||||||
|
}
|
||||||
|
|
||||||
|
// SchemaName returns the schema name for ModelPublicTenantUsers
|
||||||
|
func (m ModelPublicTenantUsers) SchemaName() string {
|
||||||
|
return "public"
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetID returns the primary key value
|
||||||
|
func (m ModelPublicTenantUsers) GetID() string {
|
||||||
|
return m.ID.String()
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetIDStr returns the primary key as a string
|
||||||
|
func (m ModelPublicTenantUsers) GetIDStr() string {
|
||||||
|
return m.ID.String()
|
||||||
|
}
|
||||||
|
|
||||||
|
// SetID sets the primary key value
|
||||||
|
func (m ModelPublicTenantUsers) SetID(newid string) {
|
||||||
|
m.UpdateID(newid)
|
||||||
|
}
|
||||||
|
|
||||||
|
// UpdateID updates the primary key value
|
||||||
|
func (m *ModelPublicTenantUsers) UpdateID(newid string) {
|
||||||
|
m.ID.FromString(newid)
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetIDName returns the name of the primary key column
|
||||||
|
func (m ModelPublicTenantUsers) GetIDName() string {
|
||||||
|
return "id"
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetPrefix returns the table prefix
|
||||||
|
func (m ModelPublicTenantUsers) GetPrefix() string {
|
||||||
|
return "TUE"
|
||||||
|
}
|
||||||
@@ -0,0 +1,73 @@
|
|||||||
|
// Code generated by relspecgo. DO NOT EDIT.
|
||||||
|
package generatedmodels
|
||||||
|
|
||||||
|
import (
|
||||||
|
sql_types "git.warky.dev/wdevs/relspecgo/pkg/sqltypes"
|
||||||
|
"github.com/uptrace/bun"
|
||||||
|
)
|
||||||
|
|
||||||
|
type ModelPublicTenants struct {
|
||||||
|
bun.BaseModel `bun:"table:public.tenants,alias:tenants"`
|
||||||
|
ID sql_types.SqlString `bun:"id,type:text,pk," json:"id"`
|
||||||
|
CreatedAt sql_types.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
||||||
|
Name sql_types.SqlString `bun:"name,type:text,notnull," json:"name"`
|
||||||
|
RelTenantIDPublicAgentPersonas []*ModelPublicAgentPersonas `bun:"rel:has-many,join:id=tenant_id" json:"reltenantidpublicagentpersonas,omitempty"` // Has many ModelPublicAgentPersonas
|
||||||
|
RelTenantIDPublicAgentParts []*ModelPublicAgentParts `bun:"rel:has-many,join:id=tenant_id" json:"reltenantidpublicagentparts,omitempty"` // Has many ModelPublicAgentParts
|
||||||
|
RelTenantIDPublicAgentTraits []*ModelPublicAgentTraits `bun:"rel:has-many,join:id=tenant_id" json:"reltenantidpublicagenttraits,omitempty"` // Has many ModelPublicAgentTraits
|
||||||
|
RelTenantIDPublicCharacterArcs []*ModelPublicCharacterArcs `bun:"rel:has-many,join:id=tenant_id" json:"reltenantidpubliccharacterarcs,omitempty"` // Has many ModelPublicCharacterArcs
|
||||||
|
RelTenantIDPublicThoughts []*ModelPublicThoughts `bun:"rel:has-many,join:id=tenant_id" json:"reltenantidpublicthoughts,omitempty"` // Has many ModelPublicThoughts
|
||||||
|
RelTenantIDPublicProjects []*ModelPublicProjects `bun:"rel:has-many,join:id=tenant_id" json:"reltenantidpublicprojects,omitempty"` // Has many ModelPublicProjects
|
||||||
|
RelTenantIDPublicStoredFiles []*ModelPublicStoredFiles `bun:"rel:has-many,join:id=tenant_id" json:"reltenantidpublicstoredfiles,omitempty"` // Has many ModelPublicStoredFiles
|
||||||
|
RelTenantIDPublicTenantUsers []*ModelPublicTenantUsers `bun:"rel:has-many,join:id=tenant_id" json:"reltenantidpublictenantusers,omitempty"` // Has many ModelPublicTenantUsers
|
||||||
|
RelTenantIDPublicAPIKeyAssignments []*ModelPublicAPIKeyAssignments `bun:"rel:has-many,join:id=tenant_id" json:"reltenantidpublicapikeyassignments,omitempty"` // Has many ModelPublicAPIKeyAssignments
|
||||||
|
RelTenantIDPublicChatHistories []*ModelPublicChatHistories `bun:"rel:has-many,join:id=tenant_id" json:"reltenantidpublicchathistories,omitempty"` // Has many ModelPublicChatHistories
|
||||||
|
RelTenantIDPublicLearnings []*ModelPublicLearnings `bun:"rel:has-many,join:id=tenant_id" json:"reltenantidpubliclearnings,omitempty"` // Has many ModelPublicLearnings
|
||||||
|
RelTenantIDPublicPlans []*ModelPublicPlans `bun:"rel:has-many,join:id=tenant_id" json:"reltenantidpublicplans,omitempty"` // Has many ModelPublicPlans
|
||||||
|
RelTenantIDPublicAgentSkills []*ModelPublicAgentSkills `bun:"rel:has-many,join:id=tenant_id" json:"reltenantidpublicagentskills,omitempty"` // Has many ModelPublicAgentSkills
|
||||||
|
RelTenantIDPublicAgentGuardrails []*ModelPublicAgentGuardrails `bun:"rel:has-many,join:id=tenant_id" json:"reltenantidpublicagentguardrails,omitempty"` // Has many ModelPublicAgentGuardrails
|
||||||
|
}
|
||||||
|
|
||||||
|
// TableName returns the table name for ModelPublicTenants
|
||||||
|
func (m ModelPublicTenants) TableName() string {
|
||||||
|
return "public.tenants"
|
||||||
|
}
|
||||||
|
|
||||||
|
// TableNameOnly returns the table name without schema for ModelPublicTenants
|
||||||
|
func (m ModelPublicTenants) TableNameOnly() string {
|
||||||
|
return "tenants"
|
||||||
|
}
|
||||||
|
|
||||||
|
// SchemaName returns the schema name for ModelPublicTenants
|
||||||
|
func (m ModelPublicTenants) SchemaName() string {
|
||||||
|
return "public"
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetID returns the primary key value
|
||||||
|
func (m ModelPublicTenants) GetID() string {
|
||||||
|
return m.ID.String()
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetIDStr returns the primary key as a string
|
||||||
|
func (m ModelPublicTenants) GetIDStr() string {
|
||||||
|
return m.ID.String()
|
||||||
|
}
|
||||||
|
|
||||||
|
// SetID sets the primary key value
|
||||||
|
func (m ModelPublicTenants) SetID(newid string) {
|
||||||
|
m.UpdateID(newid)
|
||||||
|
}
|
||||||
|
|
||||||
|
// UpdateID updates the primary key value
|
||||||
|
func (m *ModelPublicTenants) UpdateID(newid string) {
|
||||||
|
m.ID.FromString(newid)
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetIDName returns the name of the primary key column
|
||||||
|
func (m ModelPublicTenants) GetIDName() string {
|
||||||
|
return "id"
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetPrefix returns the table prefix
|
||||||
|
func (m ModelPublicTenants) GetPrefix() string {
|
||||||
|
return "TEN"
|
||||||
|
}
|
||||||
@@ -0,0 +1,64 @@
|
|||||||
|
// Code generated by relspecgo. DO NOT EDIT.
|
||||||
|
package generatedmodels
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
sql_types "git.warky.dev/wdevs/relspecgo/pkg/sqltypes"
|
||||||
|
"github.com/uptrace/bun"
|
||||||
|
)
|
||||||
|
|
||||||
|
type ModelPublicThoughtLearningLinks struct {
|
||||||
|
bun.BaseModel `bun:"table:public.thought_learning_links,alias:thought_learning_links"`
|
||||||
|
ID sql_types.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||||
|
CreatedAt sql_types.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
||||||
|
LearningID int64 `bun:"learning_id,type:bigint,notnull,unique:uidx_thought_learning_links_thought_id_learning_id," json:"learning_id"`
|
||||||
|
Relation sql_types.SqlString `bun:"relation,type:text,default:'source',notnull," json:"relation"`
|
||||||
|
ThoughtID int64 `bun:"thought_id,type:bigint,notnull,unique:uidx_thought_learning_links_thought_id_learning_id," json:"thought_id"`
|
||||||
|
RelLearningID *ModelPublicLearnings `bun:"rel:has-one,join:learning_id=id" json:"rellearningid,omitempty"` // Has one ModelPublicLearnings
|
||||||
|
RelThoughtID *ModelPublicThoughts `bun:"rel:has-one,join:thought_id=id" json:"relthoughtid,omitempty"` // Has one ModelPublicThoughts
|
||||||
|
}
|
||||||
|
|
||||||
|
// TableName returns the table name for ModelPublicThoughtLearningLinks
|
||||||
|
func (m ModelPublicThoughtLearningLinks) TableName() string {
|
||||||
|
return "public.thought_learning_links"
|
||||||
|
}
|
||||||
|
|
||||||
|
// TableNameOnly returns the table name without schema for ModelPublicThoughtLearningLinks
|
||||||
|
func (m ModelPublicThoughtLearningLinks) TableNameOnly() string {
|
||||||
|
return "thought_learning_links"
|
||||||
|
}
|
||||||
|
|
||||||
|
// SchemaName returns the schema name for ModelPublicThoughtLearningLinks
|
||||||
|
func (m ModelPublicThoughtLearningLinks) SchemaName() string {
|
||||||
|
return "public"
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetID returns the primary key value
|
||||||
|
func (m ModelPublicThoughtLearningLinks) GetID() int64 {
|
||||||
|
return m.ID.Int64()
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetIDStr returns the primary key as a string
|
||||||
|
func (m ModelPublicThoughtLearningLinks) GetIDStr() string {
|
||||||
|
return m.ID.String()
|
||||||
|
}
|
||||||
|
|
||||||
|
// SetID sets the primary key value
|
||||||
|
func (m ModelPublicThoughtLearningLinks) SetID(newid int64) {
|
||||||
|
m.UpdateID(newid)
|
||||||
|
}
|
||||||
|
|
||||||
|
// UpdateID updates the primary key value
|
||||||
|
func (m *ModelPublicThoughtLearningLinks) UpdateID(newid int64) {
|
||||||
|
m.ID.FromString(fmt.Sprintf("%d", newid))
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetIDName returns the name of the primary key column
|
||||||
|
func (m ModelPublicThoughtLearningLinks) GetIDName() string {
|
||||||
|
return "id"
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetPrefix returns the table prefix
|
||||||
|
func (m ModelPublicThoughtLearningLinks) GetPrefix() string {
|
||||||
|
return "TLL"
|
||||||
|
}
|
||||||
@@ -3,19 +3,19 @@ package generatedmodels
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
resolvespec_common "github.com/bitechdev/ResolveSpec/pkg/spectypes"
|
sql_types "git.warky.dev/wdevs/relspecgo/pkg/sqltypes"
|
||||||
"github.com/uptrace/bun"
|
"github.com/uptrace/bun"
|
||||||
)
|
)
|
||||||
|
|
||||||
type ModelPublicThoughtLinks struct {
|
type ModelPublicThoughtLinks struct {
|
||||||
bun.BaseModel `bun:"table:public.thought_links,alias:thought_links"`
|
bun.BaseModel `bun:"table:public.thought_links,alias:thought_links"`
|
||||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
ID sql_types.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||||
CreatedAt resolvespec_common.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),nullzero," json:"created_at"`
|
CreatedAt sql_types.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),nullzero," json:"created_at"`
|
||||||
FromID int64 `bun:"from_id,type:bigint,notnull," json:"from_id"`
|
FromID int64 `bun:"from_id,type:bigint,notnull," json:"from_id"`
|
||||||
Relation resolvespec_common.SqlString `bun:"relation,type:text,notnull," json:"relation"`
|
Relation sql_types.SqlString `bun:"relation,type:text,notnull," json:"relation"`
|
||||||
ToID int64 `bun:"to_id,type:bigint,notnull," json:"to_id"`
|
ToID int64 `bun:"to_id,type:bigint,notnull," json:"to_id"`
|
||||||
RelFromID *ModelPublicThoughts `bun:"rel:has-one,join:from_id=id" json:"relfromid,omitempty"` // Has one ModelPublicThoughts
|
RelFromID *ModelPublicThoughts `bun:"rel:has-one,join:from_id=id" json:"relfromid,omitempty"` // Has one ModelPublicThoughts
|
||||||
RelToID *ModelPublicThoughts `bun:"rel:has-one,join:to_id=id" json:"reltoid,omitempty"` // Has one ModelPublicThoughts
|
RelToID *ModelPublicThoughts `bun:"rel:has-one,join:to_id=id" json:"reltoid,omitempty"` // Has one ModelPublicThoughts
|
||||||
}
|
}
|
||||||
|
|
||||||
// TableName returns the table name for ModelPublicThoughtLinks
|
// TableName returns the table name for ModelPublicThoughtLinks
|
||||||
|
|||||||
@@ -3,26 +3,29 @@ package generatedmodels
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
resolvespec_common "github.com/bitechdev/ResolveSpec/pkg/spectypes"
|
sql_types "git.warky.dev/wdevs/relspecgo/pkg/sqltypes"
|
||||||
"github.com/uptrace/bun"
|
"github.com/uptrace/bun"
|
||||||
)
|
)
|
||||||
|
|
||||||
type ModelPublicThoughts struct {
|
type ModelPublicThoughts struct {
|
||||||
bun.BaseModel `bun:"table:public.thoughts,alias:thoughts"`
|
bun.BaseModel `bun:"table:public.thoughts,alias:thoughts"`
|
||||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
ID sql_types.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||||
ArchivedAt resolvespec_common.SqlTimeStamp `bun:"archived_at,type:timestamptz,nullzero," json:"archived_at"`
|
ArchivedAt sql_types.SqlTimeStamp `bun:"archived_at,type:timestamptz,nullzero," json:"archived_at"`
|
||||||
Content resolvespec_common.SqlString `bun:"content,type:text,notnull," json:"content"`
|
Content sql_types.SqlString `bun:"content,type:text,notnull," json:"content"`
|
||||||
CreatedAt resolvespec_common.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),nullzero," json:"created_at"`
|
CreatedAt sql_types.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),nullzero," json:"created_at"`
|
||||||
GUID resolvespec_common.SqlUUID `bun:"guid,type:uuid,default:gen_random_uuid(),notnull," json:"guid"`
|
GUID sql_types.SqlUUID `bun:"guid,type:uuid,default:gen_random_uuid(),notnull," json:"guid"`
|
||||||
Metadata resolvespec_common.SqlJSONB `bun:"metadata,type:jsonb,default:{}::jsonb,nullzero," json:"metadata"`
|
Metadata sql_types.SqlJSONB `bun:"metadata,type:jsonb,default:{}::jsonb,nullzero," json:"metadata"`
|
||||||
ProjectID resolvespec_common.SqlInt64 `bun:"project_id,type:bigint,nullzero," json:"project_id"`
|
ProjectID sql_types.SqlInt64 `bun:"project_id,type:bigint,nullzero," json:"project_id"`
|
||||||
UpdatedAt resolvespec_common.SqlTimeStamp `bun:"updated_at,type:timestamptz,default:now(),nullzero," json:"updated_at"`
|
TenantID sql_types.SqlString `bun:"tenant_id,type:text,nullzero," json:"tenant_id"`
|
||||||
RelProjectID *ModelPublicProjects `bun:"rel:has-one,join:project_id=id" json:"relprojectid,omitempty"` // Has one ModelPublicProjects
|
UpdatedAt sql_types.SqlTimeStamp `bun:"updated_at,type:timestamptz,default:now(),nullzero," json:"updated_at"`
|
||||||
RelFromIDPublicThoughtLinks []*ModelPublicThoughtLinks `bun:"rel:has-many,join:id=from_id" json:"relfromidpublicthoughtlinks,omitempty"` // Has many ModelPublicThoughtLinks
|
RelProjectID *ModelPublicProjects `bun:"rel:has-one,join:project_id=id" json:"relprojectid,omitempty"` // Has one ModelPublicProjects
|
||||||
RelToIDPublicThoughtLinks []*ModelPublicThoughtLinks `bun:"rel:has-many,join:id=to_id" json:"reltoidpublicthoughtlinks,omitempty"` // Has many ModelPublicThoughtLinks
|
RelTenantID *ModelPublicTenants `bun:"rel:has-one,join:tenant_id=id" json:"reltenantid,omitempty"` // Has one ModelPublicTenants
|
||||||
RelThoughtIDPublicEmbeddings []*ModelPublicEmbeddings `bun:"rel:has-many,join:id=thought_id" json:"relthoughtidpublicembeddings,omitempty"` // Has many ModelPublicEmbeddings
|
RelFromIDPublicThoughtLinks []*ModelPublicThoughtLinks `bun:"rel:has-many,join:id=from_id" json:"relfromidpublicthoughtlinks,omitempty"` // Has many ModelPublicThoughtLinks
|
||||||
RelThoughtIDPublicStoredFiles []*ModelPublicStoredFiles `bun:"rel:has-many,join:id=thought_id" json:"relthoughtidpublicstoredfiles,omitempty"` // Has many ModelPublicStoredFiles
|
RelToIDPublicThoughtLinks []*ModelPublicThoughtLinks `bun:"rel:has-many,join:id=to_id" json:"reltoidpublicthoughtlinks,omitempty"` // Has many ModelPublicThoughtLinks
|
||||||
RelRelatedThoughtIDPublicLearnings []*ModelPublicLearnings `bun:"rel:has-many,join:id=related_thought_id" json:"relrelatedthoughtidpubliclearnings,omitempty"` // Has many ModelPublicLearnings
|
RelThoughtIDPublicThoughtLearningLinks []*ModelPublicThoughtLearningLinks `bun:"rel:has-many,join:id=thought_id" json:"relthoughtidpublicthoughtlearninglinks,omitempty"` // Has many ModelPublicThoughtLearningLinks
|
||||||
|
RelThoughtIDPublicEmbeddings []*ModelPublicEmbeddings `bun:"rel:has-many,join:id=thought_id" json:"relthoughtidpublicembeddings,omitempty"` // Has many ModelPublicEmbeddings
|
||||||
|
RelThoughtIDPublicStoredFiles []*ModelPublicStoredFiles `bun:"rel:has-many,join:id=thought_id" json:"relthoughtidpublicstoredfiles,omitempty"` // Has many ModelPublicStoredFiles
|
||||||
|
RelRelatedThoughtIDPublicLearnings []*ModelPublicLearnings `bun:"rel:has-many,join:id=related_thought_id" json:"relrelatedthoughtidpubliclearnings,omitempty"` // Has many ModelPublicLearnings
|
||||||
}
|
}
|
||||||
|
|
||||||
// TableName returns the table name for ModelPublicThoughts
|
// TableName returns the table name for ModelPublicThoughts
|
||||||
|
|||||||
@@ -3,17 +3,17 @@ package generatedmodels
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
resolvespec_common "github.com/bitechdev/ResolveSpec/pkg/spectypes"
|
sql_types "git.warky.dev/wdevs/relspecgo/pkg/sqltypes"
|
||||||
"github.com/uptrace/bun"
|
"github.com/uptrace/bun"
|
||||||
)
|
)
|
||||||
|
|
||||||
type ModelPublicToolAnnotations struct {
|
type ModelPublicToolAnnotations struct {
|
||||||
bun.BaseModel `bun:"table:public.tool_annotations,alias:tool_annotations"`
|
bun.BaseModel `bun:"table:public.tool_annotations,alias:tool_annotations"`
|
||||||
ID resolvespec_common.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
ID sql_types.SqlInt64 `bun:"id,type:bigserial,pk,autoincrement," json:"id"`
|
||||||
CreatedAt resolvespec_common.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
CreatedAt sql_types.SqlTimeStamp `bun:"created_at,type:timestamptz,default:now(),notnull," json:"created_at"`
|
||||||
Notes resolvespec_common.SqlString `bun:"notes,type:text,default:'',notnull," json:"notes"`
|
Notes sql_types.SqlString `bun:"notes,type:text,default:'',notnull," json:"notes"`
|
||||||
ToolName resolvespec_common.SqlString `bun:"tool_name,type:text,notnull," json:"tool_name"`
|
ToolName sql_types.SqlString `bun:"tool_name,type:text,notnull," json:"tool_name"`
|
||||||
UpdatedAt resolvespec_common.SqlTimeStamp `bun:"updated_at,type:timestamptz,default:now(),notnull," json:"updated_at"`
|
UpdatedAt sql_types.SqlTimeStamp `bun:"updated_at,type:timestamptz,default:now(),notnull," json:"updated_at"`
|
||||||
}
|
}
|
||||||
|
|
||||||
// TableName returns the table name for ModelPublicToolAnnotations
|
// TableName returns the table name for ModelPublicToolAnnotations
|
||||||
|
|||||||
@@ -15,7 +15,7 @@ import (
|
|||||||
|
|
||||||
func (db *DB) InsertStoredFile(ctx context.Context, file thoughttypes.StoredFile) (thoughttypes.StoredFile, error) {
|
func (db *DB) InsertStoredFile(ctx context.Context, file thoughttypes.StoredFile) (thoughttypes.StoredFile, error) {
|
||||||
row := db.pool.QueryRow(ctx, `
|
row := db.pool.QueryRow(ctx, `
|
||||||
insert into stored_files (thought_id, project_id, tenant_key, name, media_type, kind, encoding, size_bytes, sha256, content)
|
insert into stored_files (thought_id, project_id, tenant_id, name, media_type, kind, encoding, size_bytes, sha256, content)
|
||||||
values ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10)
|
values ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10)
|
||||||
returning id, guid, thought_id, project_id, name, media_type, kind, encoding, size_bytes, sha256, created_at, updated_at
|
returning id, guid, thought_id, project_id, name, media_type, kind, encoding, size_bytes, sha256, created_at, updated_at
|
||||||
`, file.ThoughtID, file.ProjectID, tenantKeyPtr(ctx), file.Name, file.MediaType, file.Kind, file.Encoding, file.SizeBytes, file.SHA256, file.Content)
|
`, file.ThoughtID, file.ProjectID, tenantKeyPtr(ctx), file.Name, file.MediaType, file.Kind, file.Encoding, file.SizeBytes, file.SHA256, file.Content)
|
||||||
@@ -46,7 +46,7 @@ func (db *DB) GetStoredFile(ctx context.Context, id uuid.UUID) (thoughttypes.Sto
|
|||||||
row := db.pool.QueryRow(ctx, `
|
row := db.pool.QueryRow(ctx, `
|
||||||
select id, guid, thought_id, project_id, name, media_type, kind, encoding, size_bytes, sha256, content, created_at, updated_at
|
select id, guid, thought_id, project_id, name, media_type, kind, encoding, size_bytes, sha256, content, created_at, updated_at
|
||||||
from stored_files
|
from stored_files
|
||||||
where guid = $1`+tenantSQL(ctx, &args, "tenant_key"), args...)
|
where guid = $1`+tenantSQL(ctx, &args, "tenant_id"), args...)
|
||||||
|
|
||||||
var model generatedmodels.ModelPublicStoredFiles
|
var model generatedmodels.ModelPublicStoredFiles
|
||||||
if err := row.Scan(
|
if err := row.Scan(
|
||||||
@@ -77,7 +77,7 @@ func (db *DB) ListStoredFiles(ctx context.Context, filter thoughttypes.StoredFil
|
|||||||
args := make([]any, 0, 4)
|
args := make([]any, 0, 4)
|
||||||
conditions := make([]string, 0, 3)
|
conditions := make([]string, 0, 3)
|
||||||
|
|
||||||
addTenantCondition(ctx, &args, &conditions, "tenant_key")
|
addTenantCondition(ctx, &args, &conditions, "tenant_id")
|
||||||
if filter.ThoughtID != nil {
|
if filter.ThoughtID != nil {
|
||||||
args = append(args, *filter.ThoughtID)
|
args = append(args, *filter.ThoughtID)
|
||||||
conditions = append(conditions, fmt.Sprintf("thought_id = $%d", len(args)))
|
conditions = append(conditions, fmt.Sprintf("thought_id = $%d", len(args)))
|
||||||
|
|||||||
@@ -475,4 +475,3 @@ func canonicalPlanPair(a, b int64) (int64, int64) {
|
|||||||
}
|
}
|
||||||
return b, a
|
return b, a
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -14,7 +14,7 @@ import (
|
|||||||
|
|
||||||
func (db *DB) CreateProject(ctx context.Context, name, description string) (thoughttypes.Project, error) {
|
func (db *DB) CreateProject(ctx context.Context, name, description string) (thoughttypes.Project, error) {
|
||||||
row := db.pool.QueryRow(ctx, `
|
row := db.pool.QueryRow(ctx, `
|
||||||
insert into projects (name, description, tenant_key)
|
insert into projects (name, description, tenant_id)
|
||||||
values ($1, $2, $3)
|
values ($1, $2, $3)
|
||||||
returning id, guid, name, description, created_at, last_active_at
|
returning id, guid, name, description, created_at, last_active_at
|
||||||
`, name, description, tenantKeyPtr(ctx))
|
`, name, description, tenantKeyPtr(ctx))
|
||||||
@@ -49,7 +49,7 @@ func (db *DB) getProjectByGUID(ctx context.Context, id uuid.UUID) (thoughttypes.
|
|||||||
row := db.pool.QueryRow(ctx, `
|
row := db.pool.QueryRow(ctx, `
|
||||||
select id, guid, name, description, created_at, last_active_at
|
select id, guid, name, description, created_at, last_active_at
|
||||||
from projects
|
from projects
|
||||||
where guid = $1`+tenantSQL(ctx, &args, "tenant_key"), args...)
|
where guid = $1`+tenantSQL(ctx, &args, "tenant_id"), args...)
|
||||||
return scanProject(row)
|
return scanProject(row)
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -58,7 +58,7 @@ func (db *DB) getProjectByName(ctx context.Context, name string) (thoughttypes.P
|
|||||||
row := db.pool.QueryRow(ctx, `
|
row := db.pool.QueryRow(ctx, `
|
||||||
select id, guid, name, description, created_at, last_active_at
|
select id, guid, name, description, created_at, last_active_at
|
||||||
from projects
|
from projects
|
||||||
where name = $1`+tenantSQL(ctx, &args, "tenant_key"), args...)
|
where name = $1`+tenantSQL(ctx, &args, "tenant_id"), args...)
|
||||||
return scanProject(row)
|
return scanProject(row)
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -78,7 +78,7 @@ func (db *DB) ListProjects(ctx context.Context) ([]thoughttypes.ProjectSummary,
|
|||||||
where := ""
|
where := ""
|
||||||
if key, ok := tenantKey(ctx); ok {
|
if key, ok := tenantKey(ctx); ok {
|
||||||
args = append(args, key)
|
args = append(args, key)
|
||||||
where = "where p.tenant_key = $1"
|
where = "where p.tenant_id = $1"
|
||||||
}
|
}
|
||||||
rows, err := db.pool.Query(ctx, `
|
rows, err := db.pool.Query(ctx, `
|
||||||
select p.id, p.guid, p.name, p.description, p.created_at, p.last_active_at, count(t.id) as thought_count
|
select p.id, p.guid, p.name, p.description, p.created_at, p.last_active_at, count(t.id) as thought_count
|
||||||
@@ -113,7 +113,7 @@ func (db *DB) ListProjects(ctx context.Context) ([]thoughttypes.ProjectSummary,
|
|||||||
|
|
||||||
func (db *DB) TouchProject(ctx context.Context, id int64) error {
|
func (db *DB) TouchProject(ctx context.Context, id int64) error {
|
||||||
args := []any{id}
|
args := []any{id}
|
||||||
tag, err := db.pool.Exec(ctx, `update projects set last_active_at = now() where id = $1`+tenantSQL(ctx, &args, "tenant_key"), args...)
|
tag, err := db.pool.Exec(ctx, `update projects set last_active_at = now() where id = $1`+tenantSQL(ctx, &args, "tenant_id"), args...)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("touch project: %w", err)
|
return fmt.Errorf("touch project: %w", err)
|
||||||
}
|
}
|
||||||
|
|||||||
+25
-15
@@ -28,10 +28,10 @@ func (db *DB) AddSkill(ctx context.Context, skill ext.AgentSkill) (ext.AgentSkil
|
|||||||
skill.DomainTags = []string{}
|
skill.DomainTags = []string{}
|
||||||
}
|
}
|
||||||
row := db.pool.QueryRow(ctx, `
|
row := db.pool.QueryRow(ctx, `
|
||||||
insert into agent_skills (name, description, content, tags, language_tags, library_tags, framework_tags, domain_tags)
|
insert into agent_skills (name, description, content, tenant_id, tags, language_tags, library_tags, framework_tags, domain_tags)
|
||||||
values ($1, $2, $3, $4, $5, $6, $7, $8)
|
values ($1, $2, $3, $4, $5, $6, $7, $8, $9)
|
||||||
returning id, guid, created_at, updated_at
|
returning id, guid, created_at, updated_at
|
||||||
`, skill.Name, skill.Description, skill.Content, skill.Tags,
|
`, skill.Name, skill.Description, skill.Content, tenantKeyPtr(ctx), skill.Tags,
|
||||||
skill.LanguageTags, skill.LibraryTags, skill.FrameworkTags, skill.DomainTags)
|
skill.LanguageTags, skill.LibraryTags, skill.FrameworkTags, skill.DomainTags)
|
||||||
|
|
||||||
created := skill
|
created := skill
|
||||||
@@ -47,7 +47,8 @@ func (db *DB) AddSkill(ctx context.Context, skill ext.AgentSkill) (ext.AgentSkil
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (db *DB) RemoveSkill(ctx context.Context, id int64) error {
|
func (db *DB) RemoveSkill(ctx context.Context, id int64) error {
|
||||||
tag, err := db.pool.Exec(ctx, `delete from agent_skills where id = $1`, id)
|
args := []any{id}
|
||||||
|
tag, err := db.pool.Exec(ctx, `delete from agent_skills where id = $1`+tenantSQL(ctx, &args, "tenant_id"), args...)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("delete agent skill: %w", err)
|
return fmt.Errorf("delete agent skill: %w", err)
|
||||||
}
|
}
|
||||||
@@ -60,9 +61,14 @@ func (db *DB) RemoveSkill(ctx context.Context, id int64) error {
|
|||||||
func (db *DB) ListSkills(ctx context.Context, tag string) ([]ext.AgentSkill, error) {
|
func (db *DB) ListSkills(ctx context.Context, tag string) ([]ext.AgentSkill, error) {
|
||||||
q := `select id, name, description, content, tags::text[], language_tags::text[], library_tags::text[], framework_tags::text[], domain_tags::text[], created_at, updated_at from agent_skills`
|
q := `select id, name, description, content, tags::text[], language_tags::text[], library_tags::text[], framework_tags::text[], domain_tags::text[], created_at, updated_at from agent_skills`
|
||||||
args := []any{}
|
args := []any{}
|
||||||
|
conditions := []string{}
|
||||||
if t := strings.TrimSpace(tag); t != "" {
|
if t := strings.TrimSpace(tag); t != "" {
|
||||||
args = append(args, t)
|
args = append(args, t)
|
||||||
q += fmt.Sprintf(" where $%d = any(tags) or $%d = any(language_tags) or $%d = any(library_tags) or $%d = any(framework_tags) or $%d = any(domain_tags)", len(args), len(args), len(args), len(args), len(args))
|
conditions = append(conditions, fmt.Sprintf("($%d = any(tags) or $%d = any(language_tags) or $%d = any(library_tags) or $%d = any(framework_tags) or $%d = any(domain_tags))", len(args), len(args), len(args), len(args), len(args)))
|
||||||
|
}
|
||||||
|
addTenantCondition(ctx, &args, &conditions, "tenant_id")
|
||||||
|
if len(conditions) > 0 {
|
||||||
|
q += " where " + strings.Join(conditions, " and ")
|
||||||
}
|
}
|
||||||
q += " order by name"
|
q += " order by name"
|
||||||
|
|
||||||
@@ -135,7 +141,8 @@ func normalizeSkillSlices(skill *ext.AgentSkill) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (db *DB) GetSkill(ctx context.Context, id int64) (ext.AgentSkill, error) {
|
func (db *DB) GetSkill(ctx context.Context, id int64) (ext.AgentSkill, error) {
|
||||||
row := db.pool.QueryRow(ctx, `select `+skillSelectCols+` from agent_skills where id = $1`, id)
|
args := []any{id}
|
||||||
|
row := db.pool.QueryRow(ctx, `select `+skillSelectCols+` from agent_skills where id = $1`+tenantSQL(ctx, &args, "tenant_id"), args...)
|
||||||
s, err := scanSkill(row)
|
s, err := scanSkill(row)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return ext.AgentSkill{}, fmt.Errorf("get agent skill: %w", err)
|
return ext.AgentSkill{}, fmt.Errorf("get agent skill: %w", err)
|
||||||
@@ -144,7 +151,8 @@ func (db *DB) GetSkill(ctx context.Context, id int64) (ext.AgentSkill, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (db *DB) GetSkillByName(ctx context.Context, name string) (ext.AgentSkill, error) {
|
func (db *DB) GetSkillByName(ctx context.Context, name string) (ext.AgentSkill, error) {
|
||||||
row := db.pool.QueryRow(ctx, `select `+skillSelectCols+` from agent_skills where name = $1`, name)
|
args := []any{name}
|
||||||
|
row := db.pool.QueryRow(ctx, `select `+skillSelectCols+` from agent_skills where name = $1`+tenantSQL(ctx, &args, "tenant_id"), args...)
|
||||||
s, err := scanSkill(row)
|
s, err := scanSkill(row)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return ext.AgentSkill{}, fmt.Errorf("get agent skill by name: %w", err)
|
return ext.AgentSkill{}, fmt.Errorf("get agent skill by name: %w", err)
|
||||||
@@ -153,10 +161,10 @@ func (db *DB) GetSkillByName(ctx context.Context, name string) (ext.AgentSkill,
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (db *DB) GetGuardrailByName(ctx context.Context, name string) (ext.AgentGuardrail, error) {
|
func (db *DB) GetGuardrailByName(ctx context.Context, name string) (ext.AgentGuardrail, error) {
|
||||||
|
args := []any{name}
|
||||||
row := db.pool.QueryRow(ctx, `
|
row := db.pool.QueryRow(ctx, `
|
||||||
select id, name, description, content, severity, tags::text[], created_at, updated_at
|
select id, name, description, content, severity, tags::text[], created_at, updated_at
|
||||||
from agent_guardrails where name = $1
|
from agent_guardrails where name = $1`+tenantSQL(ctx, &args, "tenant_id"), args...)
|
||||||
`, name)
|
|
||||||
|
|
||||||
var model generatedmodels.ModelPublicAgentGuardrails
|
var model generatedmodels.ModelPublicAgentGuardrails
|
||||||
var tags []string
|
var tags []string
|
||||||
@@ -189,10 +197,10 @@ func (db *DB) AddGuardrail(ctx context.Context, g ext.AgentGuardrail) (ext.Agent
|
|||||||
g.Severity = "medium"
|
g.Severity = "medium"
|
||||||
}
|
}
|
||||||
row := db.pool.QueryRow(ctx, `
|
row := db.pool.QueryRow(ctx, `
|
||||||
insert into agent_guardrails (name, description, content, severity, tags)
|
insert into agent_guardrails (name, description, content, severity, tenant_id, tags)
|
||||||
values ($1, $2, $3, $4, $5)
|
values ($1, $2, $3, $4, $5, $6)
|
||||||
returning id, guid, created_at, updated_at
|
returning id, guid, created_at, updated_at
|
||||||
`, g.Name, g.Description, g.Content, g.Severity, g.Tags)
|
`, g.Name, g.Description, g.Content, g.Severity, tenantKeyPtr(ctx), g.Tags)
|
||||||
|
|
||||||
created := g
|
created := g
|
||||||
var model generatedmodels.ModelPublicAgentGuardrails
|
var model generatedmodels.ModelPublicAgentGuardrails
|
||||||
@@ -207,7 +215,8 @@ func (db *DB) AddGuardrail(ctx context.Context, g ext.AgentGuardrail) (ext.Agent
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (db *DB) RemoveGuardrail(ctx context.Context, id int64) error {
|
func (db *DB) RemoveGuardrail(ctx context.Context, id int64) error {
|
||||||
tag, err := db.pool.Exec(ctx, `delete from agent_guardrails where id = $1`, id)
|
args := []any{id}
|
||||||
|
tag, err := db.pool.Exec(ctx, `delete from agent_guardrails where id = $1`+tenantSQL(ctx, &args, "tenant_id"), args...)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("delete agent guardrail: %w", err)
|
return fmt.Errorf("delete agent guardrail: %w", err)
|
||||||
}
|
}
|
||||||
@@ -229,6 +238,7 @@ func (db *DB) ListGuardrails(ctx context.Context, tag, severity string) ([]ext.A
|
|||||||
args = append(args, s)
|
args = append(args, s)
|
||||||
conditions = append(conditions, fmt.Sprintf("severity = $%d", len(args)))
|
conditions = append(conditions, fmt.Sprintf("severity = $%d", len(args)))
|
||||||
}
|
}
|
||||||
|
addTenantCondition(ctx, &args, &conditions, "tenant_id")
|
||||||
|
|
||||||
q := `select id, name, description, content, severity, tags::text[], created_at, updated_at from agent_guardrails`
|
q := `select id, name, description, content, severity, tags::text[], created_at, updated_at from agent_guardrails`
|
||||||
if len(conditions) > 0 {
|
if len(conditions) > 0 {
|
||||||
@@ -268,10 +278,10 @@ func (db *DB) ListGuardrails(ctx context.Context, tag, severity string) ([]ext.A
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (db *DB) GetGuardrail(ctx context.Context, id int64) (ext.AgentGuardrail, error) {
|
func (db *DB) GetGuardrail(ctx context.Context, id int64) (ext.AgentGuardrail, error) {
|
||||||
|
args := []any{id}
|
||||||
row := db.pool.QueryRow(ctx, `
|
row := db.pool.QueryRow(ctx, `
|
||||||
select id, name, description, content, severity, tags::text[], created_at, updated_at
|
select id, name, description, content, severity, tags::text[], created_at, updated_at
|
||||||
from agent_guardrails where id = $1
|
from agent_guardrails where id = $1`+tenantSQL(ctx, &args, "tenant_id"), args...)
|
||||||
`, id)
|
|
||||||
|
|
||||||
var model generatedmodels.ModelPublicAgentGuardrails
|
var model generatedmodels.ModelPublicAgentGuardrails
|
||||||
var tags []string
|
var tags []string
|
||||||
|
|||||||
+14
-14
@@ -31,7 +31,7 @@ func (db *DB) InsertThought(ctx context.Context, thought thoughttypes.Thought, e
|
|||||||
}()
|
}()
|
||||||
|
|
||||||
row := tx.QueryRow(ctx, `
|
row := tx.QueryRow(ctx, `
|
||||||
insert into thoughts (content, metadata, project_id, tenant_key)
|
insert into thoughts (content, metadata, project_id, tenant_id)
|
||||||
values ($1, $2::jsonb, $3, $4)
|
values ($1, $2::jsonb, $3, $4)
|
||||||
returning id, guid, created_at, updated_at
|
returning id, guid, created_at, updated_at
|
||||||
`, thought.Content, metadata, thought.ProjectID, tenantKeyPtr(ctx))
|
`, thought.Content, metadata, thought.ProjectID, tenantKeyPtr(ctx))
|
||||||
@@ -123,7 +123,7 @@ func (db *DB) ListThoughts(ctx context.Context, filter thoughttypes.ListFilter)
|
|||||||
args := make([]any, 0, 6)
|
args := make([]any, 0, 6)
|
||||||
conditions := []string{}
|
conditions := []string{}
|
||||||
|
|
||||||
addTenantCondition(ctx, &args, &conditions, "tenant_key")
|
addTenantCondition(ctx, &args, &conditions, "tenant_id")
|
||||||
if !filter.IncludeArchived {
|
if !filter.IncludeArchived {
|
||||||
conditions = append(conditions, "archived_at is null")
|
conditions = append(conditions, "archived_at is null")
|
||||||
}
|
}
|
||||||
@@ -189,7 +189,7 @@ func (db *DB) Stats(ctx context.Context) (thoughttypes.ThoughtStats, error) {
|
|||||||
var total int
|
var total int
|
||||||
statsArgs := []any{}
|
statsArgs := []any{}
|
||||||
statsConditions := []string{"archived_at is null"}
|
statsConditions := []string{"archived_at is null"}
|
||||||
addTenantCondition(ctx, &statsArgs, &statsConditions, "tenant_key")
|
addTenantCondition(ctx, &statsArgs, &statsConditions, "tenant_id")
|
||||||
if err := db.pool.QueryRow(ctx, `select count(*) from thoughts where `+strings.Join(statsConditions, " and "), statsArgs...).Scan(&total); err != nil {
|
if err := db.pool.QueryRow(ctx, `select count(*) from thoughts where `+strings.Join(statsConditions, " and "), statsArgs...).Scan(&total); err != nil {
|
||||||
return thoughttypes.ThoughtStats{}, fmt.Errorf("count thoughts: %w", err)
|
return thoughttypes.ThoughtStats{}, fmt.Errorf("count thoughts: %w", err)
|
||||||
}
|
}
|
||||||
@@ -241,7 +241,7 @@ func (db *DB) GetThought(ctx context.Context, id uuid.UUID) (thoughttypes.Though
|
|||||||
row := db.pool.QueryRow(ctx, `
|
row := db.pool.QueryRow(ctx, `
|
||||||
select id, guid, content, metadata, project_id, archived_at, created_at, updated_at
|
select id, guid, content, metadata, project_id, archived_at, created_at, updated_at
|
||||||
from thoughts
|
from thoughts
|
||||||
where guid = $1`+tenantSQL(ctx, &args, "tenant_key"), args...)
|
where guid = $1`+tenantSQL(ctx, &args, "tenant_id"), args...)
|
||||||
|
|
||||||
var model generatedmodels.ModelPublicThoughts
|
var model generatedmodels.ModelPublicThoughts
|
||||||
if err := row.Scan(&model.ID, &model.GUID, &model.Content, &model.Metadata, &model.ProjectID, &model.ArchivedAt, &model.CreatedAt, &model.UpdatedAt); err != nil {
|
if err := row.Scan(&model.ID, &model.GUID, &model.Content, &model.Metadata, &model.ProjectID, &model.ArchivedAt, &model.CreatedAt, &model.UpdatedAt); err != nil {
|
||||||
@@ -264,7 +264,7 @@ func (db *DB) GetThoughtByID(ctx context.Context, id int64) (thoughttypes.Though
|
|||||||
row := db.pool.QueryRow(ctx, `
|
row := db.pool.QueryRow(ctx, `
|
||||||
select id, guid, content, metadata, project_id, archived_at, created_at, updated_at
|
select id, guid, content, metadata, project_id, archived_at, created_at, updated_at
|
||||||
from thoughts
|
from thoughts
|
||||||
where id = $1`+tenantSQL(ctx, &args, "tenant_key"), args...)
|
where id = $1`+tenantSQL(ctx, &args, "tenant_id"), args...)
|
||||||
|
|
||||||
var model generatedmodels.ModelPublicThoughts
|
var model generatedmodels.ModelPublicThoughts
|
||||||
if err := row.Scan(&model.ID, &model.GUID, &model.Content, &model.Metadata, &model.ProjectID, &model.ArchivedAt, &model.CreatedAt, &model.UpdatedAt); err != nil {
|
if err := row.Scan(&model.ID, &model.GUID, &model.Content, &model.Metadata, &model.ProjectID, &model.ArchivedAt, &model.CreatedAt, &model.UpdatedAt); err != nil {
|
||||||
@@ -303,7 +303,7 @@ func (db *DB) UpdateThought(ctx context.Context, id uuid.UUID, content string, e
|
|||||||
metadata = $3::jsonb,
|
metadata = $3::jsonb,
|
||||||
project_id = $4,
|
project_id = $4,
|
||||||
updated_at = now()
|
updated_at = now()
|
||||||
where guid = $1`+tenantSQL(ctx, &args, "tenant_key"), args...)
|
where guid = $1`+tenantSQL(ctx, &args, "tenant_id"), args...)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return thoughttypes.Thought{}, fmt.Errorf("update thought: %w", err)
|
return thoughttypes.Thought{}, fmt.Errorf("update thought: %w", err)
|
||||||
}
|
}
|
||||||
@@ -342,7 +342,7 @@ func (db *DB) UpdateThoughtMetadata(ctx context.Context, id int64, metadata thou
|
|||||||
update thoughts
|
update thoughts
|
||||||
set metadata = $2::jsonb,
|
set metadata = $2::jsonb,
|
||||||
updated_at = now()
|
updated_at = now()
|
||||||
where id = $1`+tenantSQL(ctx, &args, "tenant_key"), args...)
|
where id = $1`+tenantSQL(ctx, &args, "tenant_id"), args...)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return thoughttypes.Thought{}, fmt.Errorf("update thought metadata: %w", err)
|
return thoughttypes.Thought{}, fmt.Errorf("update thought metadata: %w", err)
|
||||||
}
|
}
|
||||||
@@ -355,7 +355,7 @@ func (db *DB) UpdateThoughtMetadata(ctx context.Context, id int64, metadata thou
|
|||||||
|
|
||||||
func (db *DB) DeleteThought(ctx context.Context, id uuid.UUID) error {
|
func (db *DB) DeleteThought(ctx context.Context, id uuid.UUID) error {
|
||||||
args := []any{id}
|
args := []any{id}
|
||||||
tag, err := db.pool.Exec(ctx, `delete from thoughts where guid = $1`+tenantSQL(ctx, &args, "tenant_key"), args...)
|
tag, err := db.pool.Exec(ctx, `delete from thoughts where guid = $1`+tenantSQL(ctx, &args, "tenant_id"), args...)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("delete thought: %w", err)
|
return fmt.Errorf("delete thought: %w", err)
|
||||||
}
|
}
|
||||||
@@ -367,7 +367,7 @@ func (db *DB) DeleteThought(ctx context.Context, id uuid.UUID) error {
|
|||||||
|
|
||||||
func (db *DB) ArchiveThought(ctx context.Context, id uuid.UUID) error {
|
func (db *DB) ArchiveThought(ctx context.Context, id uuid.UUID) error {
|
||||||
args := []any{id}
|
args := []any{id}
|
||||||
tag, err := db.pool.Exec(ctx, `update thoughts set archived_at = now(), updated_at = now() where guid = $1`+tenantSQL(ctx, &args, "tenant_key"), args...)
|
tag, err := db.pool.Exec(ctx, `update thoughts set archived_at = now(), updated_at = now() where guid = $1`+tenantSQL(ctx, &args, "tenant_id"), args...)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("archive thought: %w", err)
|
return fmt.Errorf("archive thought: %w", err)
|
||||||
}
|
}
|
||||||
@@ -446,7 +446,7 @@ func (db *DB) SearchSimilarThoughts(ctx context.Context, embedding []float32, em
|
|||||||
"1 - (e.embedding <=> $1) > $2",
|
"1 - (e.embedding <=> $1) > $2",
|
||||||
"e.model = $3",
|
"e.model = $3",
|
||||||
}
|
}
|
||||||
addTenantCondition(ctx, &args, &conditions, "t.tenant_key")
|
addTenantCondition(ctx, &args, &conditions, "t.tenant_id")
|
||||||
if projectID != nil {
|
if projectID != nil {
|
||||||
args = append(args, *projectID)
|
args = append(args, *projectID)
|
||||||
conditions = append(conditions, fmt.Sprintf("t.project_id = $%d", len(args)))
|
conditions = append(conditions, fmt.Sprintf("t.project_id = $%d", len(args)))
|
||||||
@@ -495,7 +495,7 @@ func (db *DB) HasEmbeddingsForModel(ctx context.Context, model string, projectID
|
|||||||
"e.model = $1",
|
"e.model = $1",
|
||||||
"t.archived_at is null",
|
"t.archived_at is null",
|
||||||
}
|
}
|
||||||
addTenantCondition(ctx, &args, &conditions, "t.tenant_key")
|
addTenantCondition(ctx, &args, &conditions, "t.tenant_id")
|
||||||
if projectID != nil {
|
if projectID != nil {
|
||||||
args = append(args, *projectID)
|
args = append(args, *projectID)
|
||||||
conditions = append(conditions, fmt.Sprintf("t.project_id = $%d", len(args)))
|
conditions = append(conditions, fmt.Sprintf("t.project_id = $%d", len(args)))
|
||||||
@@ -514,7 +514,7 @@ func (db *DB) HasEmbeddingsForModel(ctx context.Context, model string, projectID
|
|||||||
func (db *DB) ListThoughtsMissingEmbedding(ctx context.Context, model string, limit int, projectID *int64, includeArchived bool, olderThanDays int) ([]thoughttypes.Thought, error) {
|
func (db *DB) ListThoughtsMissingEmbedding(ctx context.Context, model string, limit int, projectID *int64, includeArchived bool, olderThanDays int) ([]thoughttypes.Thought, error) {
|
||||||
args := []any{model}
|
args := []any{model}
|
||||||
conditions := []string{"e.id is null"}
|
conditions := []string{"e.id is null"}
|
||||||
addTenantCondition(ctx, &args, &conditions, "t.tenant_key")
|
addTenantCondition(ctx, &args, &conditions, "t.tenant_id")
|
||||||
|
|
||||||
if !includeArchived {
|
if !includeArchived {
|
||||||
conditions = append(conditions, "t.archived_at is null")
|
conditions = append(conditions, "t.archived_at is null")
|
||||||
@@ -564,7 +564,7 @@ func (db *DB) ListThoughtsMissingEmbedding(ctx context.Context, model string, li
|
|||||||
func (db *DB) ListThoughtsForMetadataReparse(ctx context.Context, limit int, projectID *int64, includeArchived bool, olderThanDays int) ([]thoughttypes.Thought, error) {
|
func (db *DB) ListThoughtsForMetadataReparse(ctx context.Context, limit int, projectID *int64, includeArchived bool, olderThanDays int) ([]thoughttypes.Thought, error) {
|
||||||
args := make([]any, 0, 3)
|
args := make([]any, 0, 3)
|
||||||
conditions := make([]string, 0, 4)
|
conditions := make([]string, 0, 4)
|
||||||
addTenantCondition(ctx, &args, &conditions, "tenant_key")
|
addTenantCondition(ctx, &args, &conditions, "tenant_id")
|
||||||
|
|
||||||
if !includeArchived {
|
if !includeArchived {
|
||||||
conditions = append(conditions, "archived_at is null")
|
conditions = append(conditions, "archived_at is null")
|
||||||
@@ -634,7 +634,7 @@ func (db *DB) SearchThoughtsText(ctx context.Context, query string, limit int, p
|
|||||||
"t.archived_at is null",
|
"t.archived_at is null",
|
||||||
"(to_tsvector('simple', t.content) || to_tsvector('simple', coalesce(p.name, ''))) @@ websearch_to_tsquery('simple', $1)",
|
"(to_tsvector('simple', t.content) || to_tsvector('simple', coalesce(p.name, ''))) @@ websearch_to_tsquery('simple', $1)",
|
||||||
}
|
}
|
||||||
addTenantCondition(ctx, &args, &conditions, "t.tenant_key")
|
addTenantCondition(ctx, &args, &conditions, "t.tenant_id")
|
||||||
if projectID != nil {
|
if projectID != nil {
|
||||||
args = append(args, *projectID)
|
args = append(args, *projectID)
|
||||||
conditions = append(conditions, fmt.Sprintf("t.project_id = $%d", len(args)))
|
conditions = append(conditions, fmt.Sprintf("t.project_id = $%d", len(args)))
|
||||||
|
|||||||
+1350
-113
File diff suppressed because it is too large
Load Diff
@@ -0,0 +1,31 @@
|
|||||||
|
create table if not exists tenants (
|
||||||
|
id text primary key,
|
||||||
|
name text not null unique,
|
||||||
|
created_at timestamptz not null default now()
|
||||||
|
);
|
||||||
|
|
||||||
|
create table if not exists tenant_users (
|
||||||
|
id text primary key,
|
||||||
|
tenant_id text not null references tenants(id),
|
||||||
|
name text not null,
|
||||||
|
email text,
|
||||||
|
created_at timestamptz not null default now(),
|
||||||
|
unique (tenant_id, email)
|
||||||
|
);
|
||||||
|
|
||||||
|
create index if not exists tenant_users_tenant_id_idx on tenant_users (tenant_id);
|
||||||
|
|
||||||
|
create table if not exists api_key_assignments (
|
||||||
|
key_id text primary key,
|
||||||
|
tenant_id text not null references tenants(id),
|
||||||
|
user_id text references tenant_users(id),
|
||||||
|
description text not null default '',
|
||||||
|
source text not null check (source in ('configured', 'managed')),
|
||||||
|
enabled boolean not null default true,
|
||||||
|
created_at timestamptz not null default now()
|
||||||
|
);
|
||||||
|
|
||||||
|
create table if not exists managed_api_keys (
|
||||||
|
key_id text primary key references api_key_assignments(key_id) on delete cascade,
|
||||||
|
secret_hash text not null
|
||||||
|
);
|
||||||
@@ -0,0 +1,53 @@
|
|||||||
|
-- Convert tenant_key from an opaque authentication value into a real tenant
|
||||||
|
-- relationship. Preserve legacy rows and create placeholders for historical
|
||||||
|
-- key IDs before adding foreign keys.
|
||||||
|
insert into tenants (id, name)
|
||||||
|
select tenant_key, tenant_key
|
||||||
|
from (
|
||||||
|
select tenant_key from projects
|
||||||
|
union select tenant_key from thoughts
|
||||||
|
union select tenant_key from stored_files
|
||||||
|
union select tenant_key from learnings
|
||||||
|
union select tenant_key from plans
|
||||||
|
union select tenant_key from chat_histories
|
||||||
|
) tenant_keys
|
||||||
|
where tenant_key is not null and tenant_key <> ''
|
||||||
|
on conflict (id) do nothing;
|
||||||
|
|
||||||
|
alter table agent_skills add column if not exists tenant_key text references tenants(id);
|
||||||
|
alter table agent_guardrails add column if not exists tenant_key text references tenants(id);
|
||||||
|
alter table agent_personas add column if not exists tenant_key text references tenants(id);
|
||||||
|
alter table agent_parts add column if not exists tenant_key text references tenants(id);
|
||||||
|
alter table agent_traits add column if not exists tenant_key text references tenants(id);
|
||||||
|
alter table character_arcs add column if not exists tenant_key text references tenants(id);
|
||||||
|
|
||||||
|
alter table agent_skills drop constraint if exists ukey_agent_skills_name;
|
||||||
|
alter table agent_guardrails drop constraint if exists ukey_agent_guardrails_name;
|
||||||
|
alter table agent_personas drop constraint if exists ukey_agent_personas_name;
|
||||||
|
alter table agent_parts drop constraint if exists ukey_agent_parts_name;
|
||||||
|
alter table agent_traits drop constraint if exists ukey_agent_traits_name;
|
||||||
|
alter table character_arcs drop constraint if exists ukey_character_arcs_name;
|
||||||
|
|
||||||
|
create unique index if not exists agent_skills_tenant_key_name_idx on agent_skills (coalesce(tenant_key, ''), name);
|
||||||
|
create unique index if not exists agent_guardrails_tenant_key_name_idx on agent_guardrails (coalesce(tenant_key, ''), name);
|
||||||
|
create unique index if not exists agent_personas_tenant_key_name_idx on agent_personas (coalesce(tenant_key, ''), name);
|
||||||
|
create unique index if not exists agent_parts_tenant_key_name_idx on agent_parts (coalesce(tenant_key, ''), name);
|
||||||
|
create unique index if not exists agent_traits_tenant_key_name_idx on agent_traits (coalesce(tenant_key, ''), name);
|
||||||
|
create unique index if not exists character_arcs_tenant_key_name_idx on character_arcs (coalesce(tenant_key, ''), name);
|
||||||
|
|
||||||
|
create index if not exists agent_skills_tenant_key_idx on agent_skills (tenant_key);
|
||||||
|
create index if not exists agent_guardrails_tenant_key_idx on agent_guardrails (tenant_key);
|
||||||
|
create index if not exists agent_personas_tenant_key_idx on agent_personas (tenant_key);
|
||||||
|
create index if not exists agent_parts_tenant_key_idx on agent_parts (tenant_key);
|
||||||
|
create index if not exists agent_traits_tenant_key_idx on agent_traits (tenant_key);
|
||||||
|
create index if not exists character_arcs_tenant_key_idx on character_arcs (tenant_key);
|
||||||
|
|
||||||
|
do $$
|
||||||
|
declare table_name text;
|
||||||
|
begin
|
||||||
|
foreach table_name in array array['projects', 'thoughts', 'stored_files', 'learnings', 'plans', 'chat_histories']
|
||||||
|
loop
|
||||||
|
execute format('alter table %I drop constraint if exists %I', table_name, table_name || '_tenant_key_fkey');
|
||||||
|
execute format('alter table %I add constraint %I foreign key (tenant_key) references tenants(id)', table_name, table_name || '_tenant_key_fkey');
|
||||||
|
end loop;
|
||||||
|
end $$;
|
||||||
@@ -0,0 +1,44 @@
|
|||||||
|
-- Tenant data is not in use yet, so replace the earlier opaque tenant_key
|
||||||
|
-- column name with the relational tenant_id convention everywhere.
|
||||||
|
do $$
|
||||||
|
declare
|
||||||
|
tbl text;
|
||||||
|
con text;
|
||||||
|
begin
|
||||||
|
foreach tbl in array array[
|
||||||
|
'projects', 'thoughts', 'stored_files', 'learnings', 'plans', 'chat_histories',
|
||||||
|
'agent_skills', 'agent_guardrails', 'agent_personas', 'agent_parts',
|
||||||
|
'agent_traits', 'character_arcs'
|
||||||
|
] loop
|
||||||
|
-- Fresh installs already have tenant_id from the regenerated schema;
|
||||||
|
-- discard that empty column so the historical migration chain converges.
|
||||||
|
if exists (
|
||||||
|
select 1 from information_schema.columns
|
||||||
|
where table_schema = 'public' and table_name = tbl and column_name = 'tenant_id'
|
||||||
|
) and exists (
|
||||||
|
select 1 from information_schema.columns
|
||||||
|
where table_schema = 'public' and table_name = tbl and column_name = 'tenant_key'
|
||||||
|
) then
|
||||||
|
execute format('alter table %I drop column tenant_id', tbl);
|
||||||
|
end if;
|
||||||
|
if exists (
|
||||||
|
select 1 from information_schema.columns
|
||||||
|
where table_schema = 'public' and table_name = tbl and column_name = 'tenant_key'
|
||||||
|
) then
|
||||||
|
execute format('alter table %I rename column tenant_key to tenant_id', tbl);
|
||||||
|
end if;
|
||||||
|
|
||||||
|
for con in
|
||||||
|
select c.conname
|
||||||
|
from pg_constraint c
|
||||||
|
join pg_class r on r.oid = c.conrelid
|
||||||
|
join pg_namespace n on n.oid = r.relnamespace
|
||||||
|
join unnest(c.conkey) as k(attnum) on true
|
||||||
|
join pg_attribute a on a.attrelid = r.oid and a.attnum = k.attnum
|
||||||
|
where n.nspname = 'public' and r.relname = tbl and c.contype = 'f' and a.attname = 'tenant_id'
|
||||||
|
loop
|
||||||
|
execute format('alter table %I drop constraint %I', tbl, con);
|
||||||
|
end loop;
|
||||||
|
execute format('alter table %I add constraint %I foreign key (tenant_id) references tenants(id)', tbl, 'fk_' || tbl || '_tenant_id');
|
||||||
|
end loop;
|
||||||
|
end $$;
|
||||||
@@ -1,7 +1,8 @@
|
|||||||
Table agent_personas {
|
Table agent_personas {
|
||||||
id bigserial [pk]
|
id bigserial [pk]
|
||||||
guid uuid [unique, not null, default: `gen_random_uuid()`]
|
guid uuid [unique, not null, default: `gen_random_uuid()`]
|
||||||
name text [unique, not null]
|
name text [not null]
|
||||||
|
tenant_id text [ref: > tenants.id]
|
||||||
description text [not null, default: '']
|
description text [not null, default: '']
|
||||||
summary text [not null]
|
summary text [not null]
|
||||||
detail text [not null, default: '']
|
detail text [not null, default: '']
|
||||||
@@ -11,12 +12,15 @@ Table agent_personas {
|
|||||||
tags "text[]" [not null, default: `'{}'`]
|
tags "text[]" [not null, default: `'{}'`]
|
||||||
created_at timestamptz [not null, default: `now()`]
|
created_at timestamptz [not null, default: `now()`]
|
||||||
updated_at timestamptz [not null, default: `now()`]
|
updated_at timestamptz [not null, default: `now()`]
|
||||||
|
|
||||||
|
indexes { (tenant_id, name) [unique] tenant_id }
|
||||||
}
|
}
|
||||||
|
|
||||||
Table agent_parts {
|
Table agent_parts {
|
||||||
id bigserial [pk]
|
id bigserial [pk]
|
||||||
guid uuid [unique, not null, default: `gen_random_uuid()`]
|
guid uuid [unique, not null, default: `gen_random_uuid()`]
|
||||||
name text [unique, not null]
|
name text [not null]
|
||||||
|
tenant_id text [ref: > tenants.id]
|
||||||
part_type text [not null]
|
part_type text [not null]
|
||||||
description text [not null, default: '']
|
description text [not null, default: '']
|
||||||
summary text [not null]
|
summary text [not null]
|
||||||
@@ -24,6 +28,8 @@ Table agent_parts {
|
|||||||
tags "text[]" [not null, default: `'{}'`]
|
tags "text[]" [not null, default: `'{}'`]
|
||||||
created_at timestamptz [not null, default: `now()`]
|
created_at timestamptz [not null, default: `now()`]
|
||||||
updated_at timestamptz [not null, default: `now()`]
|
updated_at timestamptz [not null, default: `now()`]
|
||||||
|
|
||||||
|
indexes { (tenant_id, name) [unique] tenant_id }
|
||||||
}
|
}
|
||||||
|
|
||||||
Table agent_persona_parts {
|
Table agent_persona_parts {
|
||||||
@@ -77,13 +83,16 @@ Table agent_persona_guardrails {
|
|||||||
Table agent_traits {
|
Table agent_traits {
|
||||||
id bigserial [pk]
|
id bigserial [pk]
|
||||||
guid uuid [unique, not null, default: `gen_random_uuid()`]
|
guid uuid [unique, not null, default: `gen_random_uuid()`]
|
||||||
name text [unique, not null]
|
name text [not null]
|
||||||
|
tenant_id text [ref: > tenants.id]
|
||||||
trait_type text [not null]
|
trait_type text [not null]
|
||||||
description text [not null, default: '']
|
description text [not null, default: '']
|
||||||
instruction text [not null, default: '']
|
instruction text [not null, default: '']
|
||||||
tags "text[]" [not null, default: `'{}'`]
|
tags "text[]" [not null, default: `'{}'`]
|
||||||
created_at timestamptz [not null, default: `now()`]
|
created_at timestamptz [not null, default: `now()`]
|
||||||
updated_at timestamptz [not null, default: `now()`]
|
updated_at timestamptz [not null, default: `now()`]
|
||||||
|
|
||||||
|
indexes { (tenant_id, name) [unique] tenant_id }
|
||||||
}
|
}
|
||||||
|
|
||||||
Table agent_persona_traits {
|
Table agent_persona_traits {
|
||||||
@@ -98,11 +107,14 @@ Table agent_persona_traits {
|
|||||||
|
|
||||||
Table character_arcs {
|
Table character_arcs {
|
||||||
id bigserial [pk]
|
id bigserial [pk]
|
||||||
name text [unique, not null]
|
name text [not null]
|
||||||
|
tenant_id text [ref: > tenants.id]
|
||||||
description text [not null, default: '']
|
description text [not null, default: '']
|
||||||
summary text [not null, default: '']
|
summary text [not null, default: '']
|
||||||
created_at timestamptz [not null, default: `now()`]
|
created_at timestamptz [not null, default: `now()`]
|
||||||
updated_at timestamptz [not null, default: `now()`]
|
updated_at timestamptz [not null, default: `now()`]
|
||||||
|
|
||||||
|
indexes { (tenant_id, name) [unique] tenant_id }
|
||||||
}
|
}
|
||||||
|
|
||||||
Table arc_stages {
|
Table arc_stages {
|
||||||
@@ -127,7 +139,7 @@ Table arc_stage_parts {
|
|||||||
|
|
||||||
Table persona_arc {
|
Table persona_arc {
|
||||||
id bigserial [pk]
|
id bigserial [pk]
|
||||||
persona_id bigint [pk, ref: > agent_personas.id]
|
persona_id bigint [unique, not null, ref: > agent_personas.id]
|
||||||
arc_id bigint [not null, ref: > character_arcs.id]
|
arc_id bigint [not null, ref: > character_arcs.id]
|
||||||
current_stage_id bigint [not null, ref: > arc_stages.id]
|
current_stage_id bigint [not null, ref: > arc_stages.id]
|
||||||
updated_at timestamptz [not null, default: `now()`]
|
updated_at timestamptz [not null, default: `now()`]
|
||||||
|
|||||||
+6
-6
@@ -6,12 +6,12 @@ Table thoughts {
|
|||||||
created_at timestamptz [default: `now()`]
|
created_at timestamptz [default: `now()`]
|
||||||
updated_at timestamptz [default: `now()`]
|
updated_at timestamptz [default: `now()`]
|
||||||
project_id bigint [ref: > projects.id]
|
project_id bigint [ref: > projects.id]
|
||||||
tenant_key text
|
tenant_id text [ref: > tenants.id]
|
||||||
archived_at timestamptz
|
archived_at timestamptz
|
||||||
|
|
||||||
indexes {
|
indexes {
|
||||||
tenant_key
|
tenant_id
|
||||||
(tenant_key, project_id)
|
(tenant_id, project_id)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -20,13 +20,13 @@ Table projects {
|
|||||||
guid uuid [unique, not null, default: `gen_random_uuid()`]
|
guid uuid [unique, not null, default: `gen_random_uuid()`]
|
||||||
name text [not null]
|
name text [not null]
|
||||||
description text
|
description text
|
||||||
tenant_key text
|
tenant_id text [ref: > tenants.id]
|
||||||
created_at timestamptz [default: `now()`]
|
created_at timestamptz [default: `now()`]
|
||||||
last_active_at timestamptz [default: `now()`]
|
last_active_at timestamptz [default: `now()`]
|
||||||
|
|
||||||
indexes {
|
indexes {
|
||||||
(tenant_key, name) [unique]
|
(tenant_id, name) [unique]
|
||||||
tenant_key
|
tenant_id
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
+2
-2
@@ -3,7 +3,7 @@ Table stored_files {
|
|||||||
guid uuid [unique, not null, default: `gen_random_uuid()`]
|
guid uuid [unique, not null, default: `gen_random_uuid()`]
|
||||||
thought_id bigint [ref: > thoughts.id]
|
thought_id bigint [ref: > thoughts.id]
|
||||||
project_id bigint [ref: > projects.id]
|
project_id bigint [ref: > projects.id]
|
||||||
tenant_key text
|
tenant_id text [ref: > tenants.id]
|
||||||
name text [not null]
|
name text [not null]
|
||||||
media_type text [not null]
|
media_type text [not null]
|
||||||
kind text [not null, default: 'file']
|
kind text [not null, default: 'file']
|
||||||
@@ -17,7 +17,7 @@ Table stored_files {
|
|||||||
indexes {
|
indexes {
|
||||||
thought_id
|
thought_id
|
||||||
project_id
|
project_id
|
||||||
tenant_key
|
tenant_id
|
||||||
sha256
|
sha256
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -0,0 +1,33 @@
|
|||||||
|
Table tenants {
|
||||||
|
id text [pk]
|
||||||
|
name text [not null, unique]
|
||||||
|
created_at timestamptz [not null, default: `now()`]
|
||||||
|
}
|
||||||
|
|
||||||
|
Table tenant_users {
|
||||||
|
id text [pk]
|
||||||
|
tenant_id text [not null, ref: > tenants.id]
|
||||||
|
name text [not null]
|
||||||
|
email text
|
||||||
|
created_at timestamptz [not null, default: `now()`]
|
||||||
|
|
||||||
|
Indexes {
|
||||||
|
tenant_id
|
||||||
|
(tenant_id, email) [unique]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Table api_key_assignments {
|
||||||
|
key_id text [pk]
|
||||||
|
tenant_id text [not null, ref: > tenants.id]
|
||||||
|
user_id text [ref: > tenant_users.id]
|
||||||
|
description text [not null, default: '']
|
||||||
|
source text [not null]
|
||||||
|
enabled boolean [not null, default: true]
|
||||||
|
created_at timestamptz [not null, default: `now()`]
|
||||||
|
}
|
||||||
|
|
||||||
|
Table managed_api_keys {
|
||||||
|
key_id text [pk, ref: > api_key_assignments.key_id]
|
||||||
|
secret_hash text [not null]
|
||||||
|
}
|
||||||
+4
-4
@@ -6,7 +6,7 @@ Table chat_histories {
|
|||||||
channel text
|
channel text
|
||||||
agent_id text
|
agent_id text
|
||||||
project_id bigint [ref: > projects.id]
|
project_id bigint [ref: > projects.id]
|
||||||
tenant_key text
|
tenant_id text [ref: > tenants.id]
|
||||||
messages jsonb [not null, default: `'[]'`]
|
messages jsonb [not null, default: `'[]'`]
|
||||||
summary text
|
summary text
|
||||||
metadata jsonb [not null, default: `'{}'`]
|
metadata jsonb [not null, default: `'{}'`]
|
||||||
@@ -16,7 +16,7 @@ Table chat_histories {
|
|||||||
indexes {
|
indexes {
|
||||||
session_id
|
session_id
|
||||||
project_id
|
project_id
|
||||||
tenant_key
|
tenant_id
|
||||||
channel
|
channel
|
||||||
agent_id
|
agent_id
|
||||||
created_at
|
created_at
|
||||||
@@ -48,7 +48,7 @@ Table learnings {
|
|||||||
source_type text
|
source_type text
|
||||||
source_ref text
|
source_ref text
|
||||||
project_id bigint [ref: > projects.id]
|
project_id bigint [ref: > projects.id]
|
||||||
tenant_key text
|
tenant_id text [ref: > tenants.id]
|
||||||
related_thought_id bigint [ref: > thoughts.id]
|
related_thought_id bigint [ref: > thoughts.id]
|
||||||
related_skill_id bigint [ref: > agent_skills.id]
|
related_skill_id bigint [ref: > agent_skills.id]
|
||||||
reviewed_by text
|
reviewed_by text
|
||||||
@@ -61,7 +61,7 @@ Table learnings {
|
|||||||
|
|
||||||
indexes {
|
indexes {
|
||||||
project_id
|
project_id
|
||||||
tenant_key
|
tenant_id
|
||||||
category
|
category
|
||||||
area
|
area
|
||||||
status
|
status
|
||||||
|
|||||||
+2
-2
@@ -6,7 +6,7 @@ Table plans {
|
|||||||
status text [not null, default: 'draft'] // draft, active, blocked, completed, cancelled, superseded
|
status text [not null, default: 'draft'] // draft, active, blocked, completed, cancelled, superseded
|
||||||
priority text [not null, default: 'medium'] // low, medium, high, critical
|
priority text [not null, default: 'medium'] // low, medium, high, critical
|
||||||
project_id bigint [ref: > projects.id]
|
project_id bigint [ref: > projects.id]
|
||||||
tenant_key text
|
tenant_id text [ref: > tenants.id]
|
||||||
owner text
|
owner text
|
||||||
due_date timestamptz
|
due_date timestamptz
|
||||||
completed_at timestamptz
|
completed_at timestamptz
|
||||||
@@ -19,7 +19,7 @@ Table plans {
|
|||||||
|
|
||||||
indexes {
|
indexes {
|
||||||
project_id
|
project_id
|
||||||
tenant_key
|
tenant_id
|
||||||
status
|
status
|
||||||
priority
|
priority
|
||||||
owner
|
owner
|
||||||
|
|||||||
+8
-2
@@ -1,7 +1,8 @@
|
|||||||
Table agent_skills {
|
Table agent_skills {
|
||||||
id bigserial [pk]
|
id bigserial [pk]
|
||||||
guid uuid [unique, not null, default: `gen_random_uuid()`]
|
guid uuid [unique, not null, default: `gen_random_uuid()`]
|
||||||
name text [unique, not null]
|
name text [not null]
|
||||||
|
tenant_id text [ref: > tenants.id]
|
||||||
description text [not null, default: '']
|
description text [not null, default: '']
|
||||||
content text [not null]
|
content text [not null]
|
||||||
tags "text[]" [not null, default: `'{}'`]
|
tags "text[]" [not null, default: `'{}'`]
|
||||||
@@ -11,18 +12,23 @@ Table agent_skills {
|
|||||||
domain_tags "text[]" [not null, default: `'{}'`]
|
domain_tags "text[]" [not null, default: `'{}'`]
|
||||||
created_at timestamptz [not null, default: `now()`]
|
created_at timestamptz [not null, default: `now()`]
|
||||||
updated_at timestamptz [not null, default: `now()`]
|
updated_at timestamptz [not null, default: `now()`]
|
||||||
|
|
||||||
|
indexes { (tenant_id, name) [unique] tenant_id }
|
||||||
}
|
}
|
||||||
|
|
||||||
Table agent_guardrails {
|
Table agent_guardrails {
|
||||||
id bigserial [pk]
|
id bigserial [pk]
|
||||||
guid uuid [unique, not null, default: `gen_random_uuid()`]
|
guid uuid [unique, not null, default: `gen_random_uuid()`]
|
||||||
name text [unique, not null]
|
name text [not null]
|
||||||
|
tenant_id text [ref: > tenants.id]
|
||||||
description text [not null, default: '']
|
description text [not null, default: '']
|
||||||
content text [not null]
|
content text [not null]
|
||||||
severity text [not null, default: 'medium']
|
severity text [not null, default: 'medium']
|
||||||
tags "text[]" [not null, default: `'{}'`]
|
tags "text[]" [not null, default: `'{}'`]
|
||||||
created_at timestamptz [not null, default: `now()`]
|
created_at timestamptz [not null, default: `now()`]
|
||||||
updated_at timestamptz [not null, default: `now()`]
|
updated_at timestamptz [not null, default: `now()`]
|
||||||
|
|
||||||
|
indexes { (tenant_id, name) [unique] tenant_id }
|
||||||
}
|
}
|
||||||
|
|
||||||
Table project_skills {
|
Table project_skills {
|
||||||
|
|||||||
+10
-10
@@ -11,22 +11,22 @@
|
|||||||
"preview": "vite preview"
|
"preview": "vite preview"
|
||||||
},
|
},
|
||||||
"devDependencies": {
|
"devDependencies": {
|
||||||
"@sveltejs/vite-plugin-svelte": "^7.0.0",
|
"@sveltejs/vite-plugin-svelte": "^7.2.0",
|
||||||
"@tailwindcss/vite": "^4.2.4",
|
"@tailwindcss/vite": "^4.3.3",
|
||||||
"@types/node": "^25.6.0",
|
"@types/node": "^26.1.1",
|
||||||
"svelte": "^5.55.5",
|
"svelte": "^5.56.6",
|
||||||
"svelte-check": "^4.4.6",
|
"svelte-check": "^4.7.3",
|
||||||
"tailwindcss": "^4.2.4",
|
"tailwindcss": "^4.3.3",
|
||||||
"typescript": "^6.0.3",
|
"typescript": "^6.0.3",
|
||||||
"vite": "^8.0.10"
|
"vite": "^8.1.5"
|
||||||
},
|
},
|
||||||
"dependencies": {
|
"dependencies": {
|
||||||
"@sentry/svelte": "^10.51.0",
|
"@sentry/svelte": "^10.67.0",
|
||||||
"@skeletonlabs/skeleton": "^4.15.2",
|
"@skeletonlabs/skeleton": "^4.15.2",
|
||||||
"@skeletonlabs/skeleton-svelte": "^4.15.2",
|
"@skeletonlabs/skeleton-svelte": "^4.15.2",
|
||||||
"@tanstack/svelte-virtual": "^3.13.24",
|
"@tanstack/svelte-virtual": "^3.13.33",
|
||||||
"@warkypublic/artemis-kit": "^1.0.10",
|
"@warkypublic/artemis-kit": "^1.0.10",
|
||||||
"@warkypublic/resolvespec-js": "^1.0.1",
|
"@warkypublic/resolvespec-js": "^1.0.1",
|
||||||
"@warkypublic/svelix": "^0.1.40"
|
"@warkypublic/svelix": "^0.2.5"
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
Generated
+435
-390
File diff suppressed because it is too large
Load Diff
+84
-3
@@ -1,8 +1,9 @@
|
|||||||
import { GlobalStateStore } from './shellState';
|
import { GlobalStateStore } from './shellState';
|
||||||
|
import { currentTenantID, tenantScopeHeaders } from './tenantScope';
|
||||||
|
|
||||||
function authHeaders(): HeadersInit {
|
function authHeaders(): HeadersInit {
|
||||||
const token = GlobalStateStore.getState().session.authToken;
|
const token = GlobalStateStore.getState().session.authToken;
|
||||||
return token ? { Authorization: `Bearer ${token}` } : {};
|
return { ...(token ? { Authorization: `Bearer ${token}` } : {}), ...tenantScopeHeaders() };
|
||||||
}
|
}
|
||||||
|
|
||||||
type ResolveSpecResponse<T> = {
|
type ResolveSpecResponse<T> = {
|
||||||
@@ -18,6 +19,62 @@ type ResolveSpecFilter = {
|
|||||||
value?: unknown;
|
value?: unknown;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
const TENANT_OWNED_RESOLVE_SPEC_ENTITIES = new Set([
|
||||||
|
'projects',
|
||||||
|
'thoughts',
|
||||||
|
'learnings',
|
||||||
|
'plans',
|
||||||
|
'stored_files',
|
||||||
|
'agent_skills',
|
||||||
|
'agent_guardrails',
|
||||||
|
'agent_personas',
|
||||||
|
'agent_parts',
|
||||||
|
'agent_traits',
|
||||||
|
'character_arcs',
|
||||||
|
'chat_histories'
|
||||||
|
]);
|
||||||
|
|
||||||
|
function resolveSpecEntity(path: string): string | null {
|
||||||
|
const match = path.match(/^\/api\/rs\/public\/([^/?]+)/);
|
||||||
|
return match?.[1] ?? null;
|
||||||
|
}
|
||||||
|
|
||||||
|
function tenantScopedResolveSpecPayload(
|
||||||
|
path: string,
|
||||||
|
operation: 'read' | 'create' | 'update' | 'delete',
|
||||||
|
payload?: { data?: unknown; options?: unknown }
|
||||||
|
): { data?: unknown; options?: unknown } | undefined {
|
||||||
|
const entity = resolveSpecEntity(path);
|
||||||
|
const tenantID = currentTenantID();
|
||||||
|
if (!entity || !tenantID || !TENANT_OWNED_RESOLVE_SPEC_ENTITIES.has(entity)) return payload;
|
||||||
|
|
||||||
|
if (operation === 'create') {
|
||||||
|
const data = payload?.data;
|
||||||
|
return {
|
||||||
|
...payload,
|
||||||
|
// A selected tenant is authoritative; do not permit a form value to
|
||||||
|
// accidentally create data in a different tenant.
|
||||||
|
data: data && typeof data === 'object' && !Array.isArray(data) ? { ...data, tenant_id: tenantID } : data
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
const existingOptions = payload?.options && typeof payload.options === 'object' && !Array.isArray(payload.options)
|
||||||
|
? payload.options as Record<string, unknown>
|
||||||
|
: {};
|
||||||
|
const filters = Array.isArray(existingOptions.filters)
|
||||||
|
? existingOptions.filters.filter((filter): filter is ResolveSpecFilter =>
|
||||||
|
Boolean(filter) && typeof filter === 'object' && (filter as ResolveSpecFilter).column !== 'tenant_id')
|
||||||
|
: [];
|
||||||
|
|
||||||
|
return {
|
||||||
|
...payload,
|
||||||
|
options: {
|
||||||
|
...existingOptions,
|
||||||
|
filters: [...filters, { column: 'tenant_id', operator: 'eq', value: tenantID }]
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
function normalizeTags(value: unknown): string[] {
|
function normalizeTags(value: unknown): string[] {
|
||||||
if (Array.isArray(value)) {
|
if (Array.isArray(value)) {
|
||||||
return value.map((tag) => String(tag).trim()).filter(Boolean);
|
return value.map((tag) => String(tag).trim()).filter(Boolean);
|
||||||
@@ -70,18 +127,29 @@ async function del(path: string): Promise<void> {
|
|||||||
if (!res.ok) throw new Error(`${res.status} ${res.statusText}`);
|
if (!res.ok) throw new Error(`${res.status} ${res.statusText}`);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
async function patch<T>(path: string, body: unknown): Promise<T> {
|
||||||
|
const res = await fetch(path, {
|
||||||
|
method: 'PATCH',
|
||||||
|
headers: { 'Content-Type': 'application/json', ...authHeaders() },
|
||||||
|
body: JSON.stringify(body)
|
||||||
|
});
|
||||||
|
if (!res.ok) throw new Error(`${res.status} ${res.statusText}`);
|
||||||
|
return res.json() as Promise<T>;
|
||||||
|
}
|
||||||
|
|
||||||
async function rsCall<T>(
|
async function rsCall<T>(
|
||||||
path: string,
|
path: string,
|
||||||
operation: 'read' | 'create' | 'update' | 'delete',
|
operation: 'read' | 'create' | 'update' | 'delete',
|
||||||
payload?: { data?: unknown; options?: unknown }
|
payload?: { data?: unknown; options?: unknown }
|
||||||
): Promise<T> {
|
): Promise<T> {
|
||||||
|
const scopedPayload = tenantScopedResolveSpecPayload(path, operation, payload);
|
||||||
const res = await fetch(path, {
|
const res = await fetch(path, {
|
||||||
method: 'POST',
|
method: 'POST',
|
||||||
headers: { 'Content-Type': 'application/json', ...authHeaders() },
|
headers: { 'Content-Type': 'application/json', ...authHeaders() },
|
||||||
body: JSON.stringify({
|
body: JSON.stringify({
|
||||||
operation,
|
operation,
|
||||||
...(payload?.data !== undefined ? { data: payload.data } : {}),
|
...(scopedPayload?.data !== undefined ? { data: scopedPayload.data } : {}),
|
||||||
...(payload?.options !== undefined ? { options: payload.options } : {})
|
...(scopedPayload?.options !== undefined ? { options: scopedPayload.options } : {})
|
||||||
})
|
})
|
||||||
});
|
});
|
||||||
if (!res.ok) throw new Error(`${res.status} ${res.statusText}`);
|
if (!res.ok) throw new Error(`${res.status} ${res.statusText}`);
|
||||||
@@ -333,6 +401,19 @@ export const api = {
|
|||||||
dry_run: input?.dry_run ?? false
|
dry_run: input?.dry_run ?? false
|
||||||
})
|
})
|
||||||
},
|
},
|
||||||
|
identity: {
|
||||||
|
get: () => get<import('./types').IdentityData>('/api/admin/identity'),
|
||||||
|
createTenant: (name: string) =>
|
||||||
|
post<import('./types').Tenant>('/api/admin/identity/tenants', { name }),
|
||||||
|
adoptLegacy: (id: string) =>
|
||||||
|
post<unknown>(`/api/admin/identity/tenants/${id}/adopt-legacy`, {}),
|
||||||
|
createUser: (data: { tenant_id: string; name: string; email: string }) =>
|
||||||
|
post<import('./types').TenantUser>('/api/admin/identity/users', data),
|
||||||
|
createKey: (data: { tenant_id: string; user_id?: string; description: string }) =>
|
||||||
|
post<{ key: import('./types').IdentityKey; secret: string }>('/api/admin/identity/keys', data),
|
||||||
|
updateKey: (id: string, data: { tenant_id: string; user_id?: string | null; description?: string; enabled?: boolean }) =>
|
||||||
|
patch<import('./types').IdentityKey>(`/api/admin/identity/keys/${id}`, data)
|
||||||
|
},
|
||||||
plans: {
|
plans: {
|
||||||
list: async (params?: { status?: string; priority?: string; project_id?: string; limit?: number }) => {
|
list: async (params?: { status?: string; priority?: string; project_id?: string; limit?: number }) => {
|
||||||
const filters: ResolveSpecFilter[] = [];
|
const filters: ResolveSpecFilter[] = [];
|
||||||
|
|||||||
@@ -0,0 +1,196 @@
|
|||||||
|
<script lang="ts">
|
||||||
|
import { onMount } from 'svelte';
|
||||||
|
import { api } from '../../api';
|
||||||
|
import type { IdentityData, IdentityKey, Tenant, TenantUser } from '../../types';
|
||||||
|
import BooleanStatusBadge from '../shared/BooleanStatusBadge.svelte';
|
||||||
|
|
||||||
|
let identity = $state<IdentityData>({ tenants: [], users: [], keys: [] });
|
||||||
|
let loading = $state(true);
|
||||||
|
let error = $state('');
|
||||||
|
let message = $state('');
|
||||||
|
let busy = $state(false);
|
||||||
|
let tenantName = $state('');
|
||||||
|
let userTenantID = $state('');
|
||||||
|
let userName = $state('');
|
||||||
|
let userEmail = $state('');
|
||||||
|
let keyTenantID = $state('');
|
||||||
|
let keyUserID = $state('');
|
||||||
|
let keyDescription = $state('');
|
||||||
|
let revealedSecret = $state<string | null>(null);
|
||||||
|
|
||||||
|
const usersForTenant = (tenantID: string): TenantUser[] =>
|
||||||
|
identity.users.filter((user) => user.tenant_id === tenantID);
|
||||||
|
|
||||||
|
function tenantNameFor(id: string): string {
|
||||||
|
return identity.tenants.find((tenant) => tenant.id === id)?.name ?? id;
|
||||||
|
}
|
||||||
|
|
||||||
|
function userNameFor(id?: string): string {
|
||||||
|
if (!id) return 'Unassigned';
|
||||||
|
const user = identity.users.find((candidate) => candidate.id === id);
|
||||||
|
return user ? `${user.name} (${user.email})` : id;
|
||||||
|
}
|
||||||
|
|
||||||
|
function formatDate(value: string): string {
|
||||||
|
const date = new Date(value);
|
||||||
|
return Number.isNaN(date.getTime()) ? '—' : date.toLocaleString();
|
||||||
|
}
|
||||||
|
|
||||||
|
async function load() {
|
||||||
|
loading = true;
|
||||||
|
error = '';
|
||||||
|
try {
|
||||||
|
identity = await api.identity.get();
|
||||||
|
if (!userTenantID && identity.tenants[0]) userTenantID = identity.tenants[0].id;
|
||||||
|
if (!keyTenantID && identity.tenants[0]) keyTenantID = identity.tenants[0].id;
|
||||||
|
} catch (cause) {
|
||||||
|
error = cause instanceof Error ? cause.message : 'Failed to load identity data.';
|
||||||
|
} finally {
|
||||||
|
loading = false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async function createTenant() {
|
||||||
|
if (!tenantName.trim()) return;
|
||||||
|
busy = true; error = ''; message = '';
|
||||||
|
try {
|
||||||
|
const tenant = await api.identity.createTenant(tenantName.trim());
|
||||||
|
identity.tenants = [...identity.tenants, tenant];
|
||||||
|
userTenantID ||= tenant.id;
|
||||||
|
keyTenantID ||= tenant.id;
|
||||||
|
tenantName = '';
|
||||||
|
message = `Created tenant ${tenant.name}.`;
|
||||||
|
} catch (cause) {
|
||||||
|
error = cause instanceof Error ? cause.message : 'Failed to create tenant.';
|
||||||
|
} finally { busy = false; }
|
||||||
|
}
|
||||||
|
|
||||||
|
async function createUser() {
|
||||||
|
if (!userTenantID || !userName.trim() || !userEmail.trim()) return;
|
||||||
|
busy = true; error = ''; message = '';
|
||||||
|
try {
|
||||||
|
const user = await api.identity.createUser({ tenant_id: userTenantID, name: userName.trim(), email: userEmail.trim() });
|
||||||
|
identity.users = [...identity.users, user];
|
||||||
|
userName = ''; userEmail = '';
|
||||||
|
message = `Created user ${user.name}.`;
|
||||||
|
} catch (cause) {
|
||||||
|
error = cause instanceof Error ? cause.message : 'Failed to create user.';
|
||||||
|
} finally { busy = false; }
|
||||||
|
}
|
||||||
|
|
||||||
|
async function createKey() {
|
||||||
|
if (!keyTenantID || !keyDescription.trim()) return;
|
||||||
|
busy = true; error = ''; message = ''; revealedSecret = null;
|
||||||
|
try {
|
||||||
|
const result = await api.identity.createKey({
|
||||||
|
tenant_id: keyTenantID,
|
||||||
|
...(keyUserID ? { user_id: keyUserID } : {}),
|
||||||
|
description: keyDescription.trim()
|
||||||
|
});
|
||||||
|
identity.keys = [...identity.keys, result.key];
|
||||||
|
keyDescription = '';
|
||||||
|
revealedSecret = result.secret;
|
||||||
|
message = `Created key ${result.key.id}. Copy the secret now; it will not be shown again.`;
|
||||||
|
} catch (cause) {
|
||||||
|
error = cause instanceof Error ? cause.message : 'Failed to create key.';
|
||||||
|
} finally { busy = false; }
|
||||||
|
}
|
||||||
|
|
||||||
|
async function adoptLegacy(tenant: Tenant) {
|
||||||
|
if (!window.confirm(`Move all unassigned legacy data to ${tenant.name}? This cannot be undone from the UI.`)) return;
|
||||||
|
busy = true; error = ''; message = '';
|
||||||
|
try {
|
||||||
|
await api.identity.adoptLegacy(tenant.id);
|
||||||
|
message = `Moved legacy data to ${tenant.name}.`;
|
||||||
|
} catch (cause) {
|
||||||
|
error = cause instanceof Error ? cause.message : 'Failed to adopt legacy data.';
|
||||||
|
} finally { busy = false; }
|
||||||
|
}
|
||||||
|
|
||||||
|
async function updateKey(key: IdentityKey, field: 'tenant' | 'user' | 'description' | 'enabled', value: string | boolean) {
|
||||||
|
busy = true; error = ''; message = '';
|
||||||
|
try {
|
||||||
|
const data = {
|
||||||
|
tenant_id: field === 'tenant' ? String(value) : key.tenant_id,
|
||||||
|
...(field === 'user' ? { user_id: value ? String(value) : null } : { user_id: key.user_id }),
|
||||||
|
...(field === 'description' ? { description: String(value) } : {}),
|
||||||
|
...(field === 'enabled' ? { enabled: Boolean(value) } : {})
|
||||||
|
};
|
||||||
|
const updated = await api.identity.updateKey(key.id, data);
|
||||||
|
identity.keys = identity.keys.map((candidate) => candidate.id === updated.id ? updated : candidate);
|
||||||
|
message = `Updated key ${updated.id}.`;
|
||||||
|
} catch (cause) {
|
||||||
|
error = cause instanceof Error ? cause.message : 'Failed to update key.';
|
||||||
|
} finally { busy = false; }
|
||||||
|
}
|
||||||
|
|
||||||
|
onMount(load);
|
||||||
|
</script>
|
||||||
|
|
||||||
|
<div class="space-y-6">
|
||||||
|
<div class="flex flex-col gap-3 sm:flex-row sm:items-end sm:justify-between">
|
||||||
|
<div>
|
||||||
|
<h2 class="text-2xl font-semibold text-white">Identity</h2>
|
||||||
|
<p class="mt-1 text-sm text-slate-400">Group multiple API keys under a tenant, optionally assigning each key to a user.</p>
|
||||||
|
</div>
|
||||||
|
<button class="rounded-xl border border-white/10 bg-white/5 px-4 py-2 text-sm text-slate-200 hover:bg-white/10" onclick={load} disabled={loading || busy}>Refresh</button>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{#if error}<div class="rounded-xl border border-rose-400/30 bg-rose-400/10 px-4 py-3 text-sm text-rose-100">{error}</div>{/if}
|
||||||
|
{#if message}<div class="rounded-xl border border-emerald-400/30 bg-emerald-400/10 px-4 py-3 text-sm text-emerald-100">{message}</div>{/if}
|
||||||
|
{#if revealedSecret}
|
||||||
|
<section class="rounded-2xl border border-amber-300/30 bg-amber-400/10 p-4">
|
||||||
|
<h3 class="font-semibold text-amber-100">New API key secret</h3>
|
||||||
|
<p class="mt-1 text-sm text-amber-50/80">Copy this value now. It is shown only once.</p>
|
||||||
|
<code class="mt-3 block overflow-x-auto rounded-lg bg-slate-950/80 p-3 text-sm text-amber-100">{revealedSecret}</code>
|
||||||
|
</section>
|
||||||
|
{/if}
|
||||||
|
|
||||||
|
<div class="grid gap-4 xl:grid-cols-3">
|
||||||
|
<form class="rounded-2xl border border-white/10 bg-slate-900/70 p-4" onsubmit={(event) => { event.preventDefault(); void createTenant(); }}>
|
||||||
|
<h3 class="font-semibold text-white">New tenant</h3>
|
||||||
|
<label class="mt-3 block text-sm text-slate-300" for="tenant-name">Tenant name</label>
|
||||||
|
<input id="tenant-name" class="mt-1 w-full rounded-lg border border-white/10 bg-slate-950 px-3 py-2 text-white" bind:value={tenantName} required />
|
||||||
|
<button class="mt-3 rounded-xl border border-cyan-300/30 bg-cyan-400/10 px-4 py-2 text-sm text-cyan-100 disabled:opacity-50" disabled={busy}>Create tenant</button>
|
||||||
|
</form>
|
||||||
|
|
||||||
|
<form class="rounded-2xl border border-white/10 bg-slate-900/70 p-4" onsubmit={(event) => { event.preventDefault(); void createUser(); }}>
|
||||||
|
<h3 class="font-semibold text-white">New user</h3>
|
||||||
|
<label class="mt-3 block text-sm text-slate-300" for="user-tenant">Tenant</label>
|
||||||
|
<select id="user-tenant" class="mt-1 w-full rounded-lg border border-white/10 bg-slate-950 px-3 py-2 text-white" bind:value={userTenantID} required>
|
||||||
|
<option value="" disabled>Select a tenant</option>{#each identity.tenants as tenant}<option value={tenant.id}>{tenant.name}</option>{/each}
|
||||||
|
</select>
|
||||||
|
<label class="mt-3 block text-sm text-slate-300" for="user-name">Name</label><input id="user-name" class="mt-1 w-full rounded-lg border border-white/10 bg-slate-950 px-3 py-2 text-white" bind:value={userName} required />
|
||||||
|
<label class="mt-3 block text-sm text-slate-300" for="user-email">Email</label><input id="user-email" type="email" class="mt-1 w-full rounded-lg border border-white/10 bg-slate-950 px-3 py-2 text-white" bind:value={userEmail} required />
|
||||||
|
<button class="mt-3 rounded-xl border border-cyan-300/30 bg-cyan-400/10 px-4 py-2 text-sm text-cyan-100 disabled:opacity-50" disabled={busy || identity.tenants.length === 0}>Create user</button>
|
||||||
|
</form>
|
||||||
|
|
||||||
|
<form class="rounded-2xl border border-white/10 bg-slate-900/70 p-4" onsubmit={(event) => { event.preventDefault(); void createKey(); }}>
|
||||||
|
<h3 class="font-semibold text-white">New managed key</h3>
|
||||||
|
<label class="mt-3 block text-sm text-slate-300" for="key-tenant">Tenant</label>
|
||||||
|
<select id="key-tenant" class="mt-1 w-full rounded-lg border border-white/10 bg-slate-950 px-3 py-2 text-white" bind:value={keyTenantID} onchange={() => { keyUserID = ''; }} required>
|
||||||
|
<option value="" disabled>Select a tenant</option>{#each identity.tenants as tenant}<option value={tenant.id}>{tenant.name}</option>{/each}
|
||||||
|
</select>
|
||||||
|
<label class="mt-3 block text-sm text-slate-300" for="key-user">User (optional)</label>
|
||||||
|
<select id="key-user" class="mt-1 w-full rounded-lg border border-white/10 bg-slate-950 px-3 py-2 text-white" bind:value={keyUserID}><option value="">Unassigned</option>{#each usersForTenant(keyTenantID) as user}<option value={user.id}>{user.name} ({user.email})</option>{/each}</select>
|
||||||
|
<label class="mt-3 block text-sm text-slate-300" for="key-description">Description</label><input id="key-description" class="mt-1 w-full rounded-lg border border-white/10 bg-slate-950 px-3 py-2 text-white" bind:value={keyDescription} placeholder="e.g. production deployer" required />
|
||||||
|
<button class="mt-3 rounded-xl border border-cyan-300/30 bg-cyan-400/10 px-4 py-2 text-sm text-cyan-100 disabled:opacity-50" disabled={busy || identity.tenants.length === 0}>Create key</button>
|
||||||
|
</form>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{#if loading}
|
||||||
|
<div class="rounded-2xl border border-dashed border-white/10 py-12 text-center text-slate-400">Loading identity data…</div>
|
||||||
|
{:else}
|
||||||
|
<section class="rounded-2xl border border-white/10 bg-slate-900/70 p-4">
|
||||||
|
<h3 class="font-semibold text-white">Tenants ({identity.tenants.length})</h3>
|
||||||
|
<p class="mt-1 text-sm text-slate-400">Adopting legacy data is explicit: it assigns every unassigned pre-tenancy record to one tenant.</p>
|
||||||
|
<div class="mt-3 overflow-x-auto"><table class="min-w-full text-sm"><thead class="text-left text-xs uppercase tracking-wider text-slate-500"><tr><th class="pb-2 pr-4">Name</th><th class="pb-2 pr-4">Users</th><th class="pb-2 pr-4">Created</th><th class="pb-2"></th></tr></thead><tbody class="divide-y divide-white/5">{#each identity.tenants as tenant}<tr><td class="py-3 pr-4 font-medium text-white">{tenant.name}<div class="mt-1 font-mono text-xs text-slate-500">{tenant.id}</div></td><td class="py-3 pr-4 text-slate-300">{usersForTenant(tenant.id).length}</td><td class="py-3 pr-4 text-slate-400">{formatDate(tenant.created_at)}</td><td class="py-3 text-right"><button class="rounded-lg border border-amber-300/30 bg-amber-400/10 px-3 py-1.5 text-xs text-amber-100 hover:bg-amber-400/20 disabled:opacity-50" onclick={() => void adoptLegacy(tenant)} disabled={busy}>Adopt legacy data</button></td></tr>{:else}<tr><td class="py-5 text-slate-500" colspan="4">No tenants yet.</td></tr>{/each}</tbody></table></div>
|
||||||
|
</section>
|
||||||
|
|
||||||
|
<section class="rounded-2xl border border-white/10 bg-slate-900/70 p-4">
|
||||||
|
<h3 class="font-semibold text-white">API keys ({identity.keys.length})</h3>
|
||||||
|
<p class="mt-1 text-sm text-slate-400">Configured keys can be reassigned here; their secret values are never displayed.</p>
|
||||||
|
<div class="mt-3 overflow-x-auto"><table class="min-w-[860px] w-full text-sm"><thead class="text-left text-xs uppercase tracking-wider text-slate-500"><tr><th class="pb-2 pr-4">Key</th><th class="pb-2 pr-4">Tenant</th><th class="pb-2 pr-4">User</th><th class="pb-2 pr-4">Description</th><th class="pb-2 pr-4">Status</th><th class="pb-2">Created</th></tr></thead><tbody class="divide-y divide-white/5">{#each identity.keys as key}<tr><td class="py-3 pr-4"><code class="text-xs text-cyan-100">{key.id}</code><div class="mt-1 text-xs text-slate-500">{key.source}</div></td><td class="py-3 pr-4"><select aria-label={`Tenant for ${key.id}`} class="w-full rounded-lg border border-white/10 bg-slate-950 px-2 py-1.5 text-slate-200" value={key.tenant_id} onchange={(event) => void updateKey(key, 'tenant', event.currentTarget.value)} disabled={busy}>{#each identity.tenants as tenant}<option value={tenant.id}>{tenant.name}</option>{/each}</select></td><td class="py-3 pr-4"><select aria-label={`User for ${key.id}`} class="w-full rounded-lg border border-white/10 bg-slate-950 px-2 py-1.5 text-slate-200" value={key.user_id ?? ''} onchange={(event) => void updateKey(key, 'user', event.currentTarget.value)} disabled={busy}><option value="">Unassigned</option>{#each usersForTenant(key.tenant_id) as user}<option value={user.id}>{user.name}</option>{/each}</select></td><td class="py-3 pr-4"><input aria-label={`Description for ${key.id}`} class="w-full rounded-lg border border-white/10 bg-slate-950 px-2 py-1.5 text-slate-200" value={key.description} onchange={(event) => void updateKey(key, 'description', event.currentTarget.value)} disabled={busy} /></td><td class="py-3 pr-4"><label class="flex items-center gap-2 text-slate-300"><input type="checkbox" class="accent-cyan-400" checked={key.enabled} onchange={(event) => void updateKey(key, 'enabled', event.currentTarget.checked)} disabled={busy} />{#if key.enabled}Enabled{:else}<BooleanStatusBadge value={key.enabled} falseLabel="Disabled" />{/if}</label></td><td class="py-3 text-slate-400">{formatDate(key.created_at)}</td></tr>{:else}<tr><td class="py-5 text-slate-500" colspan="6">No keys available.</td></tr>{/each}</tbody></table></div>
|
||||||
|
</section>
|
||||||
|
{/if}
|
||||||
|
</div>
|
||||||
@@ -0,0 +1,14 @@
|
|||||||
|
<script lang="ts">
|
||||||
|
type Props = {
|
||||||
|
value: boolean;
|
||||||
|
falseLabel: 'Disabled' | 'Inactive';
|
||||||
|
};
|
||||||
|
|
||||||
|
let { value, falseLabel }: Props = $props();
|
||||||
|
</script>
|
||||||
|
|
||||||
|
{#if !value}
|
||||||
|
<span class="inline-flex rounded-full bg-slate-700 px-2 py-0.5 text-xs font-medium text-slate-200">
|
||||||
|
{falseLabel}
|
||||||
|
</span>
|
||||||
|
{/if}
|
||||||
@@ -10,7 +10,10 @@
|
|||||||
import ProjectsPage from '../projects/ProjectsPage.svelte';
|
import ProjectsPage from '../projects/ProjectsPage.svelte';
|
||||||
import SkillsPage from '../skills/SkillsPage.svelte';
|
import SkillsPage from '../skills/SkillsPage.svelte';
|
||||||
import ThoughtsPage from '../thoughts/ThoughtsPage.svelte';
|
import ThoughtsPage from '../thoughts/ThoughtsPage.svelte';
|
||||||
|
import IdentityPage from '../identity/IdentityPage.svelte';
|
||||||
import AppSidebar from './AppSidebar.svelte';
|
import AppSidebar from './AppSidebar.svelte';
|
||||||
|
import { fromStore } from 'svelte/store';
|
||||||
|
import { selectedTenantID } from '../../tenantScope';
|
||||||
|
|
||||||
const {
|
const {
|
||||||
currentPage,
|
currentPage,
|
||||||
@@ -29,14 +32,19 @@
|
|||||||
onnavigate: (page: ShellPage) => void;
|
onnavigate: (page: ShellPage) => void;
|
||||||
onrefresh: () => void;
|
onrefresh: () => void;
|
||||||
} = $props();
|
} = $props();
|
||||||
|
|
||||||
|
const tenantScope = fromStore(selectedTenantID);
|
||||||
</script>
|
</script>
|
||||||
|
|
||||||
<div class="grid min-h-screen lg:grid-cols-[17rem_1fr]">
|
<div class="grid min-h-screen lg:grid-cols-[17rem_1fr]">
|
||||||
<AppSidebar {currentPage} {onnavigate} {onlogout} />
|
<AppSidebar {currentPage} {onnavigate} {onlogout} />
|
||||||
|
|
||||||
<main class="px-4 py-6 sm:px-6 lg:px-8">
|
<main class="px-4 py-6 sm:px-6 lg:px-8">
|
||||||
|
{#key tenantScope.current}
|
||||||
{#if currentPage === 'dashboard'}
|
{#if currentPage === 'dashboard'}
|
||||||
<DashboardPage {data} {loading} {error} {onrefresh} />
|
<DashboardPage {data} {loading} {error} {onrefresh} />
|
||||||
|
{:else if currentPage === 'identity'}
|
||||||
|
<IdentityPage />
|
||||||
{:else if currentPage === 'projects'}
|
{:else if currentPage === 'projects'}
|
||||||
<ProjectsPage />
|
<ProjectsPage />
|
||||||
{:else if currentPage === 'thoughts'}
|
{:else if currentPage === 'thoughts'}
|
||||||
@@ -56,5 +64,6 @@
|
|||||||
{:else if currentPage === 'maintenance'}
|
{:else if currentPage === 'maintenance'}
|
||||||
<MaintenancePage />
|
<MaintenancePage />
|
||||||
{/if}
|
{/if}
|
||||||
|
{/key}
|
||||||
</main>
|
</main>
|
||||||
</div>
|
</div>
|
||||||
|
|||||||
@@ -1,5 +1,12 @@
|
|||||||
<script lang="ts">
|
<script lang="ts">
|
||||||
|
import { onMount } from 'svelte';
|
||||||
import type { NavItem, ShellPage } from '../../types';
|
import type { NavItem, ShellPage } from '../../types';
|
||||||
|
import { api } from '../../api';
|
||||||
|
import { selectedTenantID, setSelectedTenantID } from '../../tenantScope';
|
||||||
|
|
||||||
|
let tenants = $state<{ id: string; name: string }[]>([]);
|
||||||
|
let tenantError = $state('');
|
||||||
|
let tenantLoading = $state(false);
|
||||||
|
|
||||||
const {
|
const {
|
||||||
currentPage,
|
currentPage,
|
||||||
@@ -13,6 +20,7 @@
|
|||||||
|
|
||||||
const navItems: NavItem[] = [
|
const navItems: NavItem[] = [
|
||||||
{ id: 'dashboard', label: 'Dashboard', description: 'System overview and status.' },
|
{ id: 'dashboard', label: 'Dashboard', description: 'System overview and status.' },
|
||||||
|
{ id: 'identity', label: 'Identity', description: 'Tenants, users, and API keys.' },
|
||||||
{ id: 'projects', label: 'Projects', description: 'Browse and manage projects.' },
|
{ id: 'projects', label: 'Projects', description: 'Browse and manage projects.' },
|
||||||
{ id: 'thoughts', label: 'Thoughts', description: 'Search and inspect thoughts.' },
|
{ id: 'thoughts', label: 'Thoughts', description: 'Search and inspect thoughts.' },
|
||||||
{ id: 'learnings', label: 'Learnings', description: 'Curated insights and outcomes.' },
|
{ id: 'learnings', label: 'Learnings', description: 'Curated insights and outcomes.' },
|
||||||
@@ -23,6 +31,25 @@
|
|||||||
{ id: 'files', label: 'Files', description: 'Stored file inventory.' },
|
{ id: 'files', label: 'Files', description: 'Stored file inventory.' },
|
||||||
{ id: 'maintenance', label: 'Maintenance', description: 'Task state and upkeep actions.' }
|
{ id: 'maintenance', label: 'Maintenance', description: 'Task state and upkeep actions.' }
|
||||||
];
|
];
|
||||||
|
|
||||||
|
async function loadTenants(): Promise<void> {
|
||||||
|
tenantLoading = true;
|
||||||
|
tenantError = '';
|
||||||
|
try {
|
||||||
|
tenants = (await api.identity.get()).tenants;
|
||||||
|
if ($selectedTenantID && !tenants.some((tenant) => tenant.id === $selectedTenantID)) {
|
||||||
|
setSelectedTenantID('');
|
||||||
|
}
|
||||||
|
} catch (cause) {
|
||||||
|
tenantError = cause instanceof Error ? cause.message : 'Failed to load tenants.';
|
||||||
|
} finally {
|
||||||
|
tenantLoading = false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
onMount(() => {
|
||||||
|
void loadTenants();
|
||||||
|
});
|
||||||
</script>
|
</script>
|
||||||
|
|
||||||
<aside class="border-r border-white/10 bg-slate-900/90 p-6">
|
<aside class="border-r border-white/10 bg-slate-900/90 p-6">
|
||||||
@@ -32,6 +59,26 @@
|
|||||||
<p class="mt-2 text-sm text-slate-400">Memory server control panel.</p>
|
<p class="mt-2 text-sm text-slate-400">Memory server control panel.</p>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
|
<div class="mt-6 rounded-2xl border border-white/10 bg-slate-950/40 p-3">
|
||||||
|
<label class="block text-xs font-semibold uppercase tracking-wider text-slate-400" for="tenant-selector">Tenant scope</label>
|
||||||
|
<select
|
||||||
|
id="tenant-selector"
|
||||||
|
class="mt-2 w-full rounded-lg border border-white/10 bg-slate-950 px-3 py-2 text-sm text-slate-100 disabled:opacity-50"
|
||||||
|
value={$selectedTenantID}
|
||||||
|
onchange={(event) => setSelectedTenantID(event.currentTarget.value)}
|
||||||
|
disabled={tenantLoading}
|
||||||
|
>
|
||||||
|
<option value="">Default key tenant</option>
|
||||||
|
{#each tenants as tenant}
|
||||||
|
<option value={tenant.id}>{tenant.name}</option>
|
||||||
|
{/each}
|
||||||
|
</select>
|
||||||
|
<p class="mt-2 text-xs text-slate-500">Applies to tenant data in this admin session.</p>
|
||||||
|
{#if tenantError}
|
||||||
|
<p class="mt-2 text-xs text-rose-300">{tenantError}</p>
|
||||||
|
{/if}
|
||||||
|
</div>
|
||||||
|
|
||||||
<nav class="mt-8 space-y-1">
|
<nav class="mt-8 space-y-1">
|
||||||
{#each navItems as item}
|
{#each navItems as item}
|
||||||
<button
|
<button
|
||||||
|
|||||||
@@ -576,7 +576,6 @@
|
|||||||
adapter={projectBoxerAdapter}
|
adapter={projectBoxerAdapter}
|
||||||
value={state.values?.project_id ?? null}
|
value={state.values?.project_id ?? null}
|
||||||
clearable
|
clearable
|
||||||
searchable
|
|
||||||
onChange={(v) => state.setState('values', { ...state.values, project_id: v || undefined })}
|
onChange={(v) => state.setState('values', { ...state.values, project_id: v || undefined })}
|
||||||
/>
|
/>
|
||||||
</div>
|
</div>
|
||||||
|
|||||||
@@ -1,6 +1,9 @@
|
|||||||
import './app.css';
|
import './app.css';
|
||||||
import App from './App.svelte';
|
import App from './App.svelte';
|
||||||
import { mount } from 'svelte';
|
import { mount } from 'svelte';
|
||||||
|
import { installTenantScopedFetch } from './tenantScope';
|
||||||
|
|
||||||
|
installTenantScopedFetch();
|
||||||
|
|
||||||
const app = mount(App, {
|
const app = mount(App, {
|
||||||
target: document.getElementById('app')!
|
target: document.getElementById('app')!
|
||||||
|
|||||||
@@ -0,0 +1,66 @@
|
|||||||
|
import { get, writable } from 'svelte/store';
|
||||||
|
|
||||||
|
const STORAGE_KEY = 'amcs.admin.selected-tenant-id';
|
||||||
|
const TENANT_HEADER = 'X-AMCS-Tenant-ID';
|
||||||
|
let tenantScopedFetchInstalled = false;
|
||||||
|
|
||||||
|
function initialTenantID(): string {
|
||||||
|
if (typeof window === 'undefined') return '';
|
||||||
|
return window.localStorage.getItem(STORAGE_KEY)?.trim() ?? '';
|
||||||
|
}
|
||||||
|
|
||||||
|
export const selectedTenantID = writable<string>(initialTenantID());
|
||||||
|
|
||||||
|
export function setSelectedTenantID(tenantID: string): void {
|
||||||
|
const normalized = tenantID.trim();
|
||||||
|
selectedTenantID.set(normalized);
|
||||||
|
|
||||||
|
if (typeof window === 'undefined') return;
|
||||||
|
if (normalized) {
|
||||||
|
window.localStorage.setItem(STORAGE_KEY, normalized);
|
||||||
|
} else {
|
||||||
|
window.localStorage.removeItem(STORAGE_KEY);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export function tenantScopeHeaders(): Record<string, string> {
|
||||||
|
const tenantID = currentTenantID();
|
||||||
|
return tenantID ? { [TENANT_HEADER]: tenantID } : {};
|
||||||
|
}
|
||||||
|
|
||||||
|
export function currentTenantID(): string {
|
||||||
|
return get(selectedTenantID).trim();
|
||||||
|
}
|
||||||
|
|
||||||
|
function shouldScopeRequest(input: RequestInfo | URL): boolean {
|
||||||
|
if (typeof window === 'undefined') return false;
|
||||||
|
|
||||||
|
const rawURL = input instanceof Request ? input.url : input.toString();
|
||||||
|
const url = new URL(rawURL, window.location.origin);
|
||||||
|
return url.origin === window.location.origin && (url.pathname.startsWith('/api/rs') || url.pathname.startsWith('/api/admin'));
|
||||||
|
}
|
||||||
|
|
||||||
|
// Gridler and Former perform their own fetches. Scope those requests here so
|
||||||
|
// their ResolveSpec calls use the same tenant as the rest of the admin UI.
|
||||||
|
export function installTenantScopedFetch(): void {
|
||||||
|
if (typeof window === 'undefined' || tenantScopedFetchInstalled) return;
|
||||||
|
|
||||||
|
const baseFetch = window.fetch.bind(window);
|
||||||
|
|
||||||
|
async function scopedFetch(input: RequestInfo | URL, init?: RequestInit): Promise<Response> {
|
||||||
|
const headersToAdd = tenantScopeHeaders();
|
||||||
|
if (!Object.keys(headersToAdd).length || !shouldScopeRequest(input)) {
|
||||||
|
return baseFetch(input, init);
|
||||||
|
}
|
||||||
|
|
||||||
|
const headers = new Headers(input instanceof Request ? input.headers : undefined);
|
||||||
|
if (init?.headers) {
|
||||||
|
new Headers(init.headers).forEach((value, name) => headers.set(name, value));
|
||||||
|
}
|
||||||
|
Object.entries(headersToAdd).forEach(([name, value]) => headers.set(name, value));
|
||||||
|
return baseFetch(input, { ...init, headers });
|
||||||
|
}
|
||||||
|
|
||||||
|
window.fetch = scopedFetch;
|
||||||
|
tenantScopedFetchInstalled = true;
|
||||||
|
}
|
||||||
+31
-1
@@ -66,7 +66,7 @@ export type NavItem = {
|
|||||||
disabled?: boolean;
|
disabled?: boolean;
|
||||||
};
|
};
|
||||||
|
|
||||||
export type ShellPage = 'dashboard' | 'projects' | 'thoughts' | 'learnings' | 'plans' | 'skills' | 'guardrails' | 'personas' | 'files' | 'maintenance';
|
export type ShellPage = 'dashboard' | 'identity' | 'projects' | 'thoughts' | 'learnings' | 'plans' | 'skills' | 'guardrails' | 'personas' | 'files' | 'maintenance';
|
||||||
|
|
||||||
export type Project = {
|
export type Project = {
|
||||||
id: string;
|
id: string;
|
||||||
@@ -295,3 +295,33 @@ export type MetadataRetryResult = {
|
|||||||
failed: number;
|
failed: number;
|
||||||
dry_run: boolean;
|
dry_run: boolean;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
export type Tenant = {
|
||||||
|
id: string;
|
||||||
|
name: string;
|
||||||
|
created_at: string;
|
||||||
|
};
|
||||||
|
|
||||||
|
export type TenantUser = {
|
||||||
|
id: string;
|
||||||
|
tenant_id: string;
|
||||||
|
name: string;
|
||||||
|
email: string;
|
||||||
|
created_at: string;
|
||||||
|
};
|
||||||
|
|
||||||
|
export type IdentityKey = {
|
||||||
|
id: string;
|
||||||
|
tenant_id: string;
|
||||||
|
user_id?: string;
|
||||||
|
description: string;
|
||||||
|
source: 'configured' | 'managed';
|
||||||
|
enabled: boolean;
|
||||||
|
created_at: string;
|
||||||
|
};
|
||||||
|
|
||||||
|
export type IdentityData = {
|
||||||
|
tenants: Tenant[];
|
||||||
|
users: TenantUser[];
|
||||||
|
keys: IdentityKey[];
|
||||||
|
};
|
||||||
|
|||||||
+73
@@ -0,0 +1,73 @@
|
|||||||
|
Apache License
|
||||||
|
Version 2.0, January 2004
|
||||||
|
http://www.apache.org/licenses/
|
||||||
|
|
||||||
|
TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
|
||||||
|
|
||||||
|
1. Definitions.
|
||||||
|
|
||||||
|
"License" shall mean the terms and conditions for use, reproduction, and distribution as defined by Sections 1 through 9 of this document.
|
||||||
|
|
||||||
|
"Licensor" shall mean the copyright owner or entity authorized by the copyright owner that is granting the License.
|
||||||
|
|
||||||
|
"Legal Entity" shall mean the union of the acting entity and all other entities that control, are controlled by, or are under common control with that entity. For the purposes of this definition, "control" means (i) the power, direct or indirect, to cause the direction or management of such entity, whether by contract or otherwise, or (ii) ownership of fifty percent (50%) or more of the outstanding shares, or (iii) beneficial ownership of such entity.
|
||||||
|
|
||||||
|
"You" (or "Your") shall mean an individual or Legal Entity exercising permissions granted by this License.
|
||||||
|
|
||||||
|
"Source" form shall mean the preferred form for making modifications, including but not limited to software source code, documentation source, and configuration files.
|
||||||
|
|
||||||
|
"Object" form shall mean any form resulting from mechanical transformation or translation of a Source form, including but not limited to compiled object code, generated documentation, and conversions to other media types.
|
||||||
|
|
||||||
|
"Work" shall mean the work of authorship, whether in Source or Object form, made available under the License, as indicated by a copyright notice that is included in or attached to the work (an example is provided in the Appendix below).
|
||||||
|
|
||||||
|
"Derivative Works" shall mean any work, whether in Source or Object form, that is based on (or derived from) the Work and for which the editorial revisions, annotations, elaborations, or other modifications represent, as a whole, an original work of authorship. For the purposes of this License, Derivative Works shall not include works that remain separable from, or merely link (or bind by name) to the interfaces of, the Work and Derivative Works thereof.
|
||||||
|
|
||||||
|
"Contribution" shall mean any work of authorship, including the original version of the Work and any modifications or additions to that Work or Derivative Works thereof, that is intentionally submitted to Licensor for inclusion in the Work by the copyright owner or by an individual or Legal Entity authorized to submit on behalf of the copyright owner. For the purposes of this definition, "submitted" means any form of electronic, verbal, or written communication sent to the Licensor or its representatives, including but not limited to communication on electronic mailing lists, source code control systems, and issue tracking systems that are managed by, or on behalf of, the Licensor for the purpose of discussing and improving the Work, but excluding communication that is conspicuously marked or otherwise designated in writing by the copyright owner as "Not a Contribution."
|
||||||
|
|
||||||
|
"Contributor" shall mean Licensor and any individual or Legal Entity on behalf of whom a Contribution has been received by Licensor and subsequently incorporated within the Work.
|
||||||
|
|
||||||
|
2. Grant of Copyright License. Subject to the terms and conditions of this License, each Contributor hereby grants to You a perpetual, worldwide, non-exclusive, no-charge, royalty-free, irrevocable copyright license to reproduce, prepare Derivative Works of, publicly display, publicly perform, sublicense, and distribute the Work and such Derivative Works in Source or Object form.
|
||||||
|
|
||||||
|
3. Grant of Patent License. Subject to the terms and conditions of this License, each Contributor hereby grants to You a perpetual, worldwide, non-exclusive, no-charge, royalty-free, irrevocable (except as stated in this section) patent license to make, have made, use, offer to sell, sell, import, and otherwise transfer the Work, where such license applies only to those patent claims licensable by such Contributor that are necessarily infringed by their Contribution(s) alone or by combination of their Contribution(s) with the Work to which such Contribution(s) was submitted. If You institute patent litigation against any entity (including a cross-claim or counterclaim in a lawsuit) alleging that the Work or a Contribution incorporated within the Work constitutes direct or contributory patent infringement, then any patent licenses granted to You under this License for that Work shall terminate as of the date such litigation is filed.
|
||||||
|
|
||||||
|
4. Redistribution. You may reproduce and distribute copies of the Work or Derivative Works thereof in any medium, with or without modifications, and in Source or Object form, provided that You meet the following conditions:
|
||||||
|
|
||||||
|
(a) You must give any other recipients of the Work or Derivative Works a copy of this License; and
|
||||||
|
|
||||||
|
(b) You must cause any modified files to carry prominent notices stating that You changed the files; and
|
||||||
|
|
||||||
|
(c) You must retain, in the Source form of any Derivative Works that You distribute, all copyright, patent, trademark, and attribution notices from the Source form of the Work, excluding those notices that do not pertain to any part of the Derivative Works; and
|
||||||
|
|
||||||
|
(d) If the Work includes a "NOTICE" text file as part of its distribution, then any Derivative Works that You distribute must include a readable copy of the attribution notices contained within such NOTICE file, excluding those notices that do not pertain to any part of the Derivative Works, in at least one of the following places: within a NOTICE text file distributed as part of the Derivative Works; within the Source form or documentation, if provided along with the Derivative Works; or, within a display generated by the Derivative Works, if and wherever such third-party notices normally appear. The contents of the NOTICE file are for informational purposes only and do not modify the License. You may add Your own attribution notices within Derivative Works that You distribute, alongside or as an addendum to the NOTICE text from the Work, provided that such additional attribution notices cannot be construed as modifying the License.
|
||||||
|
|
||||||
|
You may add Your own copyright statement to Your modifications and may provide additional or different license terms and conditions for use, reproduction, or distribution of Your modifications, or for any such Derivative Works as a whole, provided Your use, reproduction, and distribution of the Work otherwise complies with the conditions stated in this License.
|
||||||
|
|
||||||
|
5. Submission of Contributions. Unless You explicitly state otherwise, any Contribution intentionally submitted for inclusion in the Work by You to the Licensor shall be under the terms and conditions of this License, without any additional terms or conditions. Notwithstanding the above, nothing herein shall supersede or modify the terms of any separate license agreement you may have executed with Licensor regarding such Contributions.
|
||||||
|
|
||||||
|
6. Trademarks. This License does not grant permission to use the trade names, trademarks, service marks, or product names of the Licensor, except as required for reasonable and customary use in describing the origin of the Work and reproducing the content of the NOTICE file.
|
||||||
|
|
||||||
|
7. Disclaimer of Warranty. Unless required by applicable law or agreed to in writing, Licensor provides the Work (and each Contributor provides its Contributions) on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied, including, without limitation, any warranties or conditions of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A PARTICULAR PURPOSE. You are solely responsible for determining the appropriateness of using or redistributing the Work and assume any risks associated with Your exercise of permissions under this License.
|
||||||
|
|
||||||
|
8. Limitation of Liability. In no event and under no legal theory, whether in tort (including negligence), contract, or otherwise, unless required by applicable law (such as deliberate and grossly negligent acts) or agreed to in writing, shall any Contributor be liable to You for damages, including any direct, indirect, special, incidental, or consequential damages of any character arising as a result of this License or out of the use or inability to use the Work (including but not limited to damages for loss of goodwill, work stoppage, computer failure or malfunction, or any and all other commercial damages or losses), even if such Contributor has been advised of the possibility of such damages.
|
||||||
|
|
||||||
|
9. Accepting Warranty or Additional Liability. While redistributing the Work or Derivative Works thereof, You may choose to offer, and charge a fee for, acceptance of support, warranty, indemnity, or other liability obligations and/or rights consistent with this License. However, in accepting such obligations, You may act only on Your own behalf and on Your sole responsibility, not on behalf of any other Contributor, and only if You agree to indemnify, defend, and hold each Contributor harmless for any liability incurred by, or claims asserted against, such Contributor by reason of your accepting any such warranty or additional liability.
|
||||||
|
|
||||||
|
END OF TERMS AND CONDITIONS
|
||||||
|
|
||||||
|
APPENDIX: How to apply the Apache License to your work.
|
||||||
|
|
||||||
|
To apply the Apache License to your work, attach the following boilerplate notice, with the fields enclosed by brackets "[]" replaced with your own identifying information. (Don't include the brackets!) The text should be enclosed in the appropriate comment syntax for the file format. We also recommend that a file or class name and description of purpose be included on the same "printed page" as the copyright notice for easier identification within third-party archives.
|
||||||
|
|
||||||
|
Copyright 2025 wdevs
|
||||||
|
|
||||||
|
Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
you may not use this file except in compliance with the License.
|
||||||
|
You may obtain a copy of the License at
|
||||||
|
|
||||||
|
http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
|
||||||
|
Unless required by applicable law or agreed to in writing, software
|
||||||
|
distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
See the License for the specific language governing permissions and
|
||||||
|
limitations under the License.
|
||||||
@@ -0,0 +1,136 @@
|
|||||||
|
# sqltypes
|
||||||
|
|
||||||
|
Nullable SQL types for hand-written or generated Go models. Each type wraps a
|
||||||
|
value with a `Valid` flag and implements `database/sql.Scanner`,
|
||||||
|
`driver.Valuer`, `encoding/json`, `gopkg.in/yaml.v3`, and `encoding/xml`
|
||||||
|
marshalling — so a single struct field can be scanned from a database row,
|
||||||
|
round-tripped through JSON/YAML/XML, and written back to the database without
|
||||||
|
any per-format glue code.
|
||||||
|
|
||||||
|
This package is what the `bun` and `gorm` writers emit when generating models
|
||||||
|
with `--types sqltypes` (see [`pkg/writers/bun`](../writers/bun/README.md) and
|
||||||
|
[`pkg/writers/gorm`](../writers/gorm/README.md)). It can also be imported
|
||||||
|
directly in hand-written models.
|
||||||
|
|
||||||
|
## Import
|
||||||
|
|
||||||
|
```go
|
||||||
|
import sql_types "git.warky.dev/wdevs/relspecgo/pkg/sqltypes"
|
||||||
|
```
|
||||||
|
|
||||||
|
## Scalar types
|
||||||
|
|
||||||
|
All scalar types are instantiations of the generic `SqlNull[T]`:
|
||||||
|
|
||||||
|
| Type | Underlying | Typical SQL type |
|
||||||
|
|---|---|---|
|
||||||
|
| `SqlInt16` | `int16` | `smallint` |
|
||||||
|
| `SqlInt32` | `int32` | `integer` |
|
||||||
|
| `SqlInt64` | `int64` | `bigint` |
|
||||||
|
| `SqlFloat32` | `float32` | `real`, `float4` |
|
||||||
|
| `SqlFloat64` | `float64` | `double precision`, `numeric`, `decimal`, `money` |
|
||||||
|
| `SqlBool` | `bool` | `boolean` |
|
||||||
|
| `SqlString` | `string` | `text`, `varchar`, `char`, `citext`, `inet`, `cidr`, `macaddr` |
|
||||||
|
| `SqlByteArray` | `[]byte` | `bytea` (base64-encoded in JSON/YAML/XML) |
|
||||||
|
| `SqlUUID` | `uuid.UUID` (`github.com/google/uuid`) | `uuid` |
|
||||||
|
|
||||||
|
You can also instantiate `SqlNull[T]` directly for any type not covered
|
||||||
|
above, e.g. `SqlNull[MyEnum]`.
|
||||||
|
|
||||||
|
### Date/time types
|
||||||
|
|
||||||
|
Plain `time.Time` doesn't distinguish date-only, time-only, and timestamp
|
||||||
|
semantics, and its zero value marshals to a confusing `0001-01-01T00:00:00Z`.
|
||||||
|
These wrapper types fix both problems:
|
||||||
|
|
||||||
|
| Type | Format | Notes |
|
||||||
|
|---|---|---|
|
||||||
|
| `SqlTimeStamp` | `2006-01-02T15:04:05` | Full timestamp |
|
||||||
|
| `SqlDate` | `2006-01-02` | Date only |
|
||||||
|
| `SqlTime` | `15:04:05` | Time only |
|
||||||
|
|
||||||
|
Zero/pre-epoch values (`time.Time{}` or anything before `0002-01-01`) marshal
|
||||||
|
to `null` and `Value()` returns `nil`, instead of leaking Go's zero-time
|
||||||
|
sentinel into the database or API responses.
|
||||||
|
|
||||||
|
### JSON types
|
||||||
|
|
||||||
|
| Type | Underlying | Notes |
|
||||||
|
|---|---|---|
|
||||||
|
| `SqlJSONB` | `[]byte` | Raw JSON bytes; `MarshalYAML` decodes to native YAML mappings/sequences instead of an embedded JSON string |
|
||||||
|
| `SqlJSON` | `= SqlJSONB` | Alias — PostgreSQL's `json` and `jsonb` share the same Go representation |
|
||||||
|
|
||||||
|
`SqlJSONB` has `AsMap()` / `AsSlice()` helpers for pulling out
|
||||||
|
`map[string]any` / `[]any` without a separate `json.Unmarshal` call.
|
||||||
|
|
||||||
|
### Vector type (pgvector)
|
||||||
|
|
||||||
|
`SqlVector` wraps `[]float32` for the `vector` column type ([pgvector](https://github.com/pgvector/pgvector)),
|
||||||
|
scanning/writing the `[1,2,3]` literal format pgvector uses over the wire.
|
||||||
|
|
||||||
|
## Array types
|
||||||
|
|
||||||
|
PostgreSQL array columns (`text[]`, `integer[]`, …) map to `SqlXxxArray`
|
||||||
|
types, each wrapping `Val []T` + `Valid bool` and handling PostgreSQL's
|
||||||
|
`{a,b,c}` array literal format on `Scan`/`Value`:
|
||||||
|
|
||||||
|
`SqlStringArray`, `SqlInt16Array`, `SqlInt32Array`, `SqlInt64Array`,
|
||||||
|
`SqlFloat32Array`, `SqlFloat64Array`, `SqlBoolArray`, `SqlUUIDArray`.
|
||||||
|
|
||||||
|
## Constructing values
|
||||||
|
|
||||||
|
Every type has a `NewSqlXxx(v)` constructor that sets `Valid: true`:
|
||||||
|
|
||||||
|
```go
|
||||||
|
name := sql_types.NewSqlString("Ada Lovelace")
|
||||||
|
age := sql_types.NewSqlInt32(36)
|
||||||
|
tags := sql_types.NewSqlStringArray([]string{"engineer", "mathematician"})
|
||||||
|
```
|
||||||
|
|
||||||
|
The zero value of any type (`sql_types.SqlString{}`) is null/invalid — use it
|
||||||
|
directly for a `NULL` field instead of a separate constructor.
|
||||||
|
|
||||||
|
Generic helpers:
|
||||||
|
|
||||||
|
```go
|
||||||
|
sql_types.Null(v, valid) // SqlNull[T]{Val: v, Valid: valid}
|
||||||
|
sql_types.NewSql[T](anyValue) // best-effort conversion from any Go value
|
||||||
|
```
|
||||||
|
|
||||||
|
## Reading values back
|
||||||
|
|
||||||
|
Each scalar type has typed accessors that return the zero value instead of
|
||||||
|
panicking when `Valid` is false:
|
||||||
|
|
||||||
|
```go
|
||||||
|
n.Int64() // SqlInt16/32/64, SqlFloat32/64, SqlBool, SqlString → int64
|
||||||
|
n.Float64() // → float64
|
||||||
|
n.Bool() // → bool
|
||||||
|
n.Time() // SqlNull[time.Time]-based types → time.Time
|
||||||
|
n.UUID() // SqlUUID → uuid.UUID
|
||||||
|
n.String() // fmt.Stringer — empty string when invalid
|
||||||
|
```
|
||||||
|
|
||||||
|
## Example
|
||||||
|
|
||||||
|
```go
|
||||||
|
type User struct {
|
||||||
|
ID sql_types.SqlUUID `json:"id"`
|
||||||
|
Name sql_types.SqlString `json:"name"`
|
||||||
|
Tags sql_types.SqlStringArray `json:"tags"`
|
||||||
|
Metadata sql_types.SqlJSONB `json:"metadata"`
|
||||||
|
CreatedAt sql_types.SqlTimeStamp `json:"created_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
u := User{
|
||||||
|
ID: sql_types.NewSqlUUID(uuid.New()),
|
||||||
|
Name: sql_types.NewSqlString("Ada Lovelace"),
|
||||||
|
Tags: sql_types.NewSqlStringArray([]string{"engineer"}),
|
||||||
|
CreatedAt: sql_types.SqlTimeStampNow(),
|
||||||
|
}
|
||||||
|
// Metadata left as the zero value → serializes as null, scans as NULL.
|
||||||
|
```
|
||||||
|
|
||||||
|
Every type implements `sql.Scanner` and `driver.Valuer`, so these fields can
|
||||||
|
be used directly as struct fields with `database/sql`, `bun`, or `gorm`
|
||||||
|
without additional tags or hooks.
|
||||||
+305
-1
@@ -1,15 +1,85 @@
|
|||||||
package spectypes
|
package sqltypes
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"database/sql/driver"
|
"database/sql/driver"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
|
"encoding/xml"
|
||||||
"fmt"
|
"fmt"
|
||||||
"strconv"
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
"github.com/google/uuid"
|
"github.com/google/uuid"
|
||||||
|
"gopkg.in/yaml.v3"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// marshalYAMLSlice returns the value used by yaml.Marshaler implementations
|
||||||
|
// for the nullable array types below: nil when invalid, the slice otherwise.
|
||||||
|
func marshalYAMLSlice[T any](valid bool, vals []T) (any, error) {
|
||||||
|
if !valid {
|
||||||
|
return nil, nil
|
||||||
|
}
|
||||||
|
return vals, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// unmarshalYAMLSlice returns the value used by yaml.Unmarshaler
|
||||||
|
// implementations for the nullable array types below.
|
||||||
|
func unmarshalYAMLSlice[T any](value *yaml.Node) (vals []T, valid bool, err error) {
|
||||||
|
if value == nil || value.Tag == "!!null" {
|
||||||
|
return nil, false, nil
|
||||||
|
}
|
||||||
|
if err := value.Decode(&vals); err != nil {
|
||||||
|
return nil, false, err
|
||||||
|
}
|
||||||
|
return vals, true, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// xmlArrayItemName is the element name used for each item when a nullable
|
||||||
|
// array type is encoded as XML, since XML has no native list type.
|
||||||
|
var xmlArrayItemName = xml.Name{Local: "item"}
|
||||||
|
|
||||||
|
// marshalXMLSlice writes a nullable array type as XML: an empty element when
|
||||||
|
// invalid, otherwise the start tag followed by one <item> child per element.
|
||||||
|
func marshalXMLSlice[T any](e *xml.Encoder, start xml.StartElement, valid bool, vals []T) error {
|
||||||
|
if !valid {
|
||||||
|
return e.EncodeElement("", start)
|
||||||
|
}
|
||||||
|
if err := e.EncodeToken(start); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
for _, v := range vals {
|
||||||
|
if err := e.EncodeElement(v, xml.StartElement{Name: xmlArrayItemName}); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return e.EncodeToken(start.End())
|
||||||
|
}
|
||||||
|
|
||||||
|
// unmarshalXMLSlice reads back the XML written by marshalXMLSlice.
|
||||||
|
//
|
||||||
|
// XML has no native null representation: an element with no <item> children
|
||||||
|
// unmarshals to a valid, empty slice rather than an invalid one.
|
||||||
|
func unmarshalXMLSlice[T any](d *xml.Decoder, start xml.StartElement) ([]T, error) {
|
||||||
|
var vals []T
|
||||||
|
for {
|
||||||
|
tok, err := d.Token()
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
switch t := tok.(type) {
|
||||||
|
case xml.StartElement:
|
||||||
|
var v T
|
||||||
|
if err := d.DecodeElement(&v, &t); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
vals = append(vals, v)
|
||||||
|
case xml.EndElement:
|
||||||
|
if t.Name == start.Name {
|
||||||
|
return vals, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// parsePostgresArrayElements parses a PostgreSQL array literal (e.g. `{a,"b,c",d}`)
|
// parsePostgresArrayElements parses a PostgreSQL array literal (e.g. `{a,"b,c",d}`)
|
||||||
// into a slice of raw string elements. Each element retains its unquoted/unescaped value.
|
// into a slice of raw string elements. Each element retains its unquoted/unescaped value.
|
||||||
func parsePostgresArrayElements(s string) ([]string, error) {
|
func parsePostgresArrayElements(s string) ([]string, error) {
|
||||||
@@ -140,6 +210,32 @@ func (a *SqlStringArray) UnmarshalJSON(b []byte) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (a SqlStringArray) MarshalYAML() (any, error) {
|
||||||
|
return marshalYAMLSlice(a.Valid, a.Val)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *SqlStringArray) UnmarshalYAML(value *yaml.Node) error {
|
||||||
|
vals, valid, err := unmarshalYAMLSlice[string](value)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
a.Val, a.Valid = vals, valid
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a SqlStringArray) MarshalXML(e *xml.Encoder, start xml.StartElement) error {
|
||||||
|
return marshalXMLSlice(e, start, a.Valid, a.Val)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *SqlStringArray) UnmarshalXML(d *xml.Decoder, start xml.StartElement) error {
|
||||||
|
vals, err := unmarshalXMLSlice[string](d, start)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
a.Val, a.Valid = vals, true
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
func NewSqlStringArray(v []string) SqlStringArray {
|
func NewSqlStringArray(v []string) SqlStringArray {
|
||||||
return SqlStringArray{Val: v, Valid: true}
|
return SqlStringArray{Val: v, Valid: true}
|
||||||
}
|
}
|
||||||
@@ -215,6 +311,32 @@ func (a *SqlInt16Array) UnmarshalJSON(b []byte) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (a SqlInt16Array) MarshalYAML() (any, error) {
|
||||||
|
return marshalYAMLSlice(a.Valid, a.Val)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *SqlInt16Array) UnmarshalYAML(value *yaml.Node) error {
|
||||||
|
vals, valid, err := unmarshalYAMLSlice[int16](value)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
a.Val, a.Valid = vals, valid
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a SqlInt16Array) MarshalXML(e *xml.Encoder, start xml.StartElement) error {
|
||||||
|
return marshalXMLSlice(e, start, a.Valid, a.Val)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *SqlInt16Array) UnmarshalXML(d *xml.Decoder, start xml.StartElement) error {
|
||||||
|
vals, err := unmarshalXMLSlice[int16](d, start)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
a.Val, a.Valid = vals, true
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
func NewSqlInt16Array(v []int16) SqlInt16Array {
|
func NewSqlInt16Array(v []int16) SqlInt16Array {
|
||||||
return SqlInt16Array{Val: v, Valid: true}
|
return SqlInt16Array{Val: v, Valid: true}
|
||||||
}
|
}
|
||||||
@@ -290,6 +412,32 @@ func (a *SqlInt32Array) UnmarshalJSON(b []byte) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (a SqlInt32Array) MarshalYAML() (any, error) {
|
||||||
|
return marshalYAMLSlice(a.Valid, a.Val)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *SqlInt32Array) UnmarshalYAML(value *yaml.Node) error {
|
||||||
|
vals, valid, err := unmarshalYAMLSlice[int32](value)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
a.Val, a.Valid = vals, valid
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a SqlInt32Array) MarshalXML(e *xml.Encoder, start xml.StartElement) error {
|
||||||
|
return marshalXMLSlice(e, start, a.Valid, a.Val)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *SqlInt32Array) UnmarshalXML(d *xml.Decoder, start xml.StartElement) error {
|
||||||
|
vals, err := unmarshalXMLSlice[int32](d, start)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
a.Val, a.Valid = vals, true
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
func NewSqlInt32Array(v []int32) SqlInt32Array {
|
func NewSqlInt32Array(v []int32) SqlInt32Array {
|
||||||
return SqlInt32Array{Val: v, Valid: true}
|
return SqlInt32Array{Val: v, Valid: true}
|
||||||
}
|
}
|
||||||
@@ -365,6 +513,32 @@ func (a *SqlInt64Array) UnmarshalJSON(b []byte) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (a SqlInt64Array) MarshalYAML() (any, error) {
|
||||||
|
return marshalYAMLSlice(a.Valid, a.Val)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *SqlInt64Array) UnmarshalYAML(value *yaml.Node) error {
|
||||||
|
vals, valid, err := unmarshalYAMLSlice[int64](value)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
a.Val, a.Valid = vals, valid
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a SqlInt64Array) MarshalXML(e *xml.Encoder, start xml.StartElement) error {
|
||||||
|
return marshalXMLSlice(e, start, a.Valid, a.Val)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *SqlInt64Array) UnmarshalXML(d *xml.Decoder, start xml.StartElement) error {
|
||||||
|
vals, err := unmarshalXMLSlice[int64](d, start)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
a.Val, a.Valid = vals, true
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
func NewSqlInt64Array(v []int64) SqlInt64Array {
|
func NewSqlInt64Array(v []int64) SqlInt64Array {
|
||||||
return SqlInt64Array{Val: v, Valid: true}
|
return SqlInt64Array{Val: v, Valid: true}
|
||||||
}
|
}
|
||||||
@@ -440,6 +614,32 @@ func (a *SqlFloat32Array) UnmarshalJSON(b []byte) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (a SqlFloat32Array) MarshalYAML() (any, error) {
|
||||||
|
return marshalYAMLSlice(a.Valid, a.Val)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *SqlFloat32Array) UnmarshalYAML(value *yaml.Node) error {
|
||||||
|
vals, valid, err := unmarshalYAMLSlice[float32](value)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
a.Val, a.Valid = vals, valid
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a SqlFloat32Array) MarshalXML(e *xml.Encoder, start xml.StartElement) error {
|
||||||
|
return marshalXMLSlice(e, start, a.Valid, a.Val)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *SqlFloat32Array) UnmarshalXML(d *xml.Decoder, start xml.StartElement) error {
|
||||||
|
vals, err := unmarshalXMLSlice[float32](d, start)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
a.Val, a.Valid = vals, true
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
func NewSqlFloat32Array(v []float32) SqlFloat32Array {
|
func NewSqlFloat32Array(v []float32) SqlFloat32Array {
|
||||||
return SqlFloat32Array{Val: v, Valid: true}
|
return SqlFloat32Array{Val: v, Valid: true}
|
||||||
}
|
}
|
||||||
@@ -515,6 +715,32 @@ func (a *SqlFloat64Array) UnmarshalJSON(b []byte) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (a SqlFloat64Array) MarshalYAML() (any, error) {
|
||||||
|
return marshalYAMLSlice(a.Valid, a.Val)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *SqlFloat64Array) UnmarshalYAML(value *yaml.Node) error {
|
||||||
|
vals, valid, err := unmarshalYAMLSlice[float64](value)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
a.Val, a.Valid = vals, valid
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a SqlFloat64Array) MarshalXML(e *xml.Encoder, start xml.StartElement) error {
|
||||||
|
return marshalXMLSlice(e, start, a.Valid, a.Val)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *SqlFloat64Array) UnmarshalXML(d *xml.Decoder, start xml.StartElement) error {
|
||||||
|
vals, err := unmarshalXMLSlice[float64](d, start)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
a.Val, a.Valid = vals, true
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
func NewSqlFloat64Array(v []float64) SqlFloat64Array {
|
func NewSqlFloat64Array(v []float64) SqlFloat64Array {
|
||||||
return SqlFloat64Array{Val: v, Valid: true}
|
return SqlFloat64Array{Val: v, Valid: true}
|
||||||
}
|
}
|
||||||
@@ -591,6 +817,32 @@ func (a *SqlBoolArray) UnmarshalJSON(b []byte) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (a SqlBoolArray) MarshalYAML() (any, error) {
|
||||||
|
return marshalYAMLSlice(a.Valid, a.Val)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *SqlBoolArray) UnmarshalYAML(value *yaml.Node) error {
|
||||||
|
vals, valid, err := unmarshalYAMLSlice[bool](value)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
a.Val, a.Valid = vals, valid
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a SqlBoolArray) MarshalXML(e *xml.Encoder, start xml.StartElement) error {
|
||||||
|
return marshalXMLSlice(e, start, a.Valid, a.Val)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *SqlBoolArray) UnmarshalXML(d *xml.Decoder, start xml.StartElement) error {
|
||||||
|
vals, err := unmarshalXMLSlice[bool](d, start)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
a.Val, a.Valid = vals, true
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
func NewSqlBoolArray(v []bool) SqlBoolArray {
|
func NewSqlBoolArray(v []bool) SqlBoolArray {
|
||||||
return SqlBoolArray{Val: v, Valid: true}
|
return SqlBoolArray{Val: v, Valid: true}
|
||||||
}
|
}
|
||||||
@@ -666,6 +918,32 @@ func (a *SqlUUIDArray) UnmarshalJSON(b []byte) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (a SqlUUIDArray) MarshalYAML() (any, error) {
|
||||||
|
return marshalYAMLSlice(a.Valid, a.Val)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *SqlUUIDArray) UnmarshalYAML(value *yaml.Node) error {
|
||||||
|
vals, valid, err := unmarshalYAMLSlice[uuid.UUID](value)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
a.Val, a.Valid = vals, valid
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a SqlUUIDArray) MarshalXML(e *xml.Encoder, start xml.StartElement) error {
|
||||||
|
return marshalXMLSlice(e, start, a.Valid, a.Val)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *SqlUUIDArray) UnmarshalXML(d *xml.Decoder, start xml.StartElement) error {
|
||||||
|
vals, err := unmarshalXMLSlice[uuid.UUID](d, start)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
a.Val, a.Valid = vals, true
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
func NewSqlUUIDArray(v []uuid.UUID) SqlUUIDArray {
|
func NewSqlUUIDArray(v []uuid.UUID) SqlUUIDArray {
|
||||||
return SqlUUIDArray{Val: v, Valid: true}
|
return SqlUUIDArray{Val: v, Valid: true}
|
||||||
}
|
}
|
||||||
@@ -750,6 +1028,32 @@ func (v *SqlVector) UnmarshalJSON(b []byte) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (v SqlVector) MarshalYAML() (any, error) {
|
||||||
|
return marshalYAMLSlice(v.Valid, v.Val)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (v *SqlVector) UnmarshalYAML(value *yaml.Node) error {
|
||||||
|
vals, valid, err := unmarshalYAMLSlice[float32](value)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
v.Val, v.Valid = vals, valid
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (v SqlVector) MarshalXML(e *xml.Encoder, start xml.StartElement) error {
|
||||||
|
return marshalXMLSlice(e, start, v.Valid, v.Val)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (v *SqlVector) UnmarshalXML(d *xml.Decoder, start xml.StartElement) error {
|
||||||
|
vals, err := unmarshalXMLSlice[float32](d, start)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
v.Val, v.Valid = vals, true
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
func NewSqlVector(val []float32) SqlVector {
|
func NewSqlVector(val []float32) SqlVector {
|
||||||
return SqlVector{Val: val, Valid: true}
|
return SqlVector{Val: val, Valid: true}
|
||||||
}
|
}
|
||||||
+331
-8
@@ -1,11 +1,12 @@
|
|||||||
// Package spectypes provides nullable SQL types with automatic casting and conversion methods.
|
// Package sqltypes provides nullable SQL types with automatic casting and conversion methods.
|
||||||
package spectypes
|
package sqltypes
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"database/sql"
|
"database/sql"
|
||||||
"database/sql/driver"
|
"database/sql/driver"
|
||||||
"encoding/base64"
|
"encoding/base64"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
|
"encoding/xml"
|
||||||
"fmt"
|
"fmt"
|
||||||
"reflect"
|
"reflect"
|
||||||
"strconv"
|
"strconv"
|
||||||
@@ -13,6 +14,7 @@ import (
|
|||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/google/uuid"
|
"github.com/google/uuid"
|
||||||
|
"gopkg.in/yaml.v3"
|
||||||
)
|
)
|
||||||
|
|
||||||
// tryParseDT attempts to parse a string into a time.Time using various formats.
|
// tryParseDT attempts to parse a string into a time.Time using various formats.
|
||||||
@@ -120,15 +122,22 @@ func (n *SqlNull[T]) FromString(s string) error {
|
|||||||
|
|
||||||
var zero T
|
var zero T
|
||||||
switch any(zero).(type) {
|
switch any(zero).(type) {
|
||||||
case int, int8, int16, int32, int64, uint, uint8, uint16, uint32, uint64:
|
case int, int8, int16, int32, int64:
|
||||||
if i, err := strconv.ParseInt(s, 10, 64); err == nil {
|
if i, err := strconv.ParseInt(s, 10, 64); err == nil {
|
||||||
reflect.ValueOf(&n.Val).Elem().SetInt(i)
|
reflect.ValueOf(&n.Val).Elem().SetInt(i)
|
||||||
n.Valid = true
|
n.Valid = true
|
||||||
}
|
} else if f, err := strconv.ParseFloat(s, 64); err == nil {
|
||||||
if f, err := strconv.ParseFloat(s, 64); err == nil {
|
|
||||||
reflect.ValueOf(&n.Val).Elem().SetInt(int64(f))
|
reflect.ValueOf(&n.Val).Elem().SetInt(int64(f))
|
||||||
n.Valid = true
|
n.Valid = true
|
||||||
}
|
}
|
||||||
|
case uint, uint8, uint16, uint32, uint64:
|
||||||
|
if u, err := strconv.ParseUint(s, 10, 64); err == nil {
|
||||||
|
reflect.ValueOf(&n.Val).Elem().SetUint(u)
|
||||||
|
n.Valid = true
|
||||||
|
} else if f, err := strconv.ParseFloat(s, 64); err == nil && f >= 0 {
|
||||||
|
reflect.ValueOf(&n.Val).Elem().SetUint(uint64(f))
|
||||||
|
n.Valid = true
|
||||||
|
}
|
||||||
case float32, float64:
|
case float32, float64:
|
||||||
if f, err := strconv.ParseFloat(s, 64); err == nil {
|
if f, err := strconv.ParseFloat(s, 64); err == nil {
|
||||||
reflect.ValueOf(&n.Val).Elem().SetFloat(f)
|
reflect.ValueOf(&n.Val).Elem().SetFloat(f)
|
||||||
@@ -232,6 +241,104 @@ func (n *SqlNull[T]) UnmarshalJSON(b []byte) error {
|
|||||||
return fmt.Errorf("cannot unmarshal %s into SqlNull[%T]", b, n.Val)
|
return fmt.Errorf("cannot unmarshal %s into SqlNull[%T]", b, n.Val)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// MarshalYAML implements yaml.Marshaler.
|
||||||
|
func (n SqlNull[T]) MarshalYAML() (any, error) {
|
||||||
|
if !n.Valid {
|
||||||
|
return nil, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check if T is []byte, and encode to base64 (mirrors MarshalJSON).
|
||||||
|
if b, ok := any(n.Val).([]byte); ok {
|
||||||
|
return base64.StdEncoding.EncodeToString(b), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return n.Val, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// UnmarshalYAML implements yaml.Unmarshaler.
|
||||||
|
func (n *SqlNull[T]) UnmarshalYAML(value *yaml.Node) error {
|
||||||
|
if value == nil || value.Tag == "!!null" {
|
||||||
|
n.Valid = false
|
||||||
|
n.Val = *new(T)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check if T is []byte, and decode from base64.
|
||||||
|
var zero T
|
||||||
|
if _, ok := any(zero).([]byte); ok {
|
||||||
|
var s string
|
||||||
|
if err := value.Decode(&s); err == nil {
|
||||||
|
if decoded, err := base64.StdEncoding.DecodeString(s); err == nil {
|
||||||
|
n.Val = any(decoded).(T)
|
||||||
|
n.Valid = true
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
n.Val = any([]byte(s)).(T)
|
||||||
|
n.Valid = true
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
var val T
|
||||||
|
if err := value.Decode(&val); err == nil {
|
||||||
|
n.Val = val
|
||||||
|
n.Valid = true
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Fallback: decode as string and parse.
|
||||||
|
var s string
|
||||||
|
if err := value.Decode(&s); err == nil {
|
||||||
|
return n.FromString(s)
|
||||||
|
}
|
||||||
|
|
||||||
|
return fmt.Errorf("cannot unmarshal %q into SqlNull[%T]", value.Value, n.Val)
|
||||||
|
}
|
||||||
|
|
||||||
|
// MarshalXML implements xml.Marshaler.
|
||||||
|
func (n SqlNull[T]) MarshalXML(e *xml.Encoder, start xml.StartElement) error {
|
||||||
|
if !n.Valid {
|
||||||
|
return e.EncodeElement("", start)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check if T is []byte, and encode to base64 (mirrors MarshalJSON).
|
||||||
|
if b, ok := any(n.Val).([]byte); ok {
|
||||||
|
return e.EncodeElement(base64.StdEncoding.EncodeToString(b), start)
|
||||||
|
}
|
||||||
|
|
||||||
|
return e.EncodeElement(n.Val, start)
|
||||||
|
}
|
||||||
|
|
||||||
|
// UnmarshalXML implements xml.Unmarshaler.
|
||||||
|
//
|
||||||
|
// XML has no native null representation, so an empty element unmarshals to
|
||||||
|
// an invalid (null) value rather than a zero-value-but-valid one.
|
||||||
|
func (n *SqlNull[T]) UnmarshalXML(d *xml.Decoder, start xml.StartElement) error {
|
||||||
|
var s string
|
||||||
|
if err := d.DecodeElement(&s, &start); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if s == "" {
|
||||||
|
n.Valid = false
|
||||||
|
n.Val = *new(T)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
var zero T
|
||||||
|
if _, ok := any(zero).([]byte); ok {
|
||||||
|
if decoded, err := base64.StdEncoding.DecodeString(s); err == nil {
|
||||||
|
n.Val = any(decoded).(T)
|
||||||
|
n.Valid = true
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
n.Val = any([]byte(s)).(T)
|
||||||
|
n.Valid = true
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return n.FromString(s)
|
||||||
|
}
|
||||||
|
|
||||||
// String implements fmt.Stringer.
|
// String implements fmt.Stringer.
|
||||||
func (n SqlNull[T]) String() string {
|
func (n SqlNull[T]) String() string {
|
||||||
if !n.Valid {
|
if !n.Valid {
|
||||||
@@ -329,6 +436,7 @@ type (
|
|||||||
SqlInt16 = SqlNull[int16]
|
SqlInt16 = SqlNull[int16]
|
||||||
SqlInt32 = SqlNull[int32]
|
SqlInt32 = SqlNull[int32]
|
||||||
SqlInt64 = SqlNull[int64]
|
SqlInt64 = SqlNull[int64]
|
||||||
|
SqlFloat32 = SqlNull[float32]
|
||||||
SqlFloat64 = SqlNull[float64]
|
SqlFloat64 = SqlNull[float64]
|
||||||
SqlBool = SqlNull[bool]
|
SqlBool = SqlNull[bool]
|
||||||
SqlString = SqlNull[string]
|
SqlString = SqlNull[string]
|
||||||
@@ -343,7 +451,7 @@ func (t SqlTimeStamp) MarshalJSON() ([]byte, error) {
|
|||||||
if !t.Valid || t.Val.IsZero() || t.Val.Before(time.Date(0002, 1, 1, 0, 0, 0, 0, time.UTC)) {
|
if !t.Valid || t.Val.IsZero() || t.Val.Before(time.Date(0002, 1, 1, 0, 0, 0, 0, time.UTC)) {
|
||||||
return []byte("null"), nil
|
return []byte("null"), nil
|
||||||
}
|
}
|
||||||
return []byte(fmt.Sprintf(`"%s"`, t.Val.Format("2006-01-02T15:04:05"))), nil
|
return fmt.Appendf(nil, `"%s"`, t.Val.Format("2006-01-02T15:04:05")), nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (t *SqlTimeStamp) UnmarshalJSON(b []byte) error {
|
func (t *SqlTimeStamp) UnmarshalJSON(b []byte) error {
|
||||||
@@ -363,6 +471,49 @@ func (t SqlTimeStamp) Value() (driver.Value, error) {
|
|||||||
return t.Val.Format("2006-01-02T15:04:05"), nil
|
return t.Val.Format("2006-01-02T15:04:05"), nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (t SqlTimeStamp) MarshalYAML() (any, error) {
|
||||||
|
if !t.Valid || t.Val.IsZero() || t.Val.Before(time.Date(0002, 1, 1, 0, 0, 0, 0, time.UTC)) {
|
||||||
|
return nil, nil
|
||||||
|
}
|
||||||
|
return t.Val.Format("2006-01-02T15:04:05"), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *SqlTimeStamp) UnmarshalYAML(value *yaml.Node) error {
|
||||||
|
if err := t.SqlNull.UnmarshalYAML(value); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if t.Valid && (t.Val.IsZero() || t.Val.Format("2006-01-02T15:04:05") == "0001-01-01T00:00:00") {
|
||||||
|
t.Valid = false
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t SqlTimeStamp) MarshalXML(e *xml.Encoder, start xml.StartElement) error {
|
||||||
|
if !t.Valid || t.Val.IsZero() || t.Val.Before(time.Date(0002, 1, 1, 0, 0, 0, 0, time.UTC)) {
|
||||||
|
return e.EncodeElement("", start)
|
||||||
|
}
|
||||||
|
return e.EncodeElement(t.Val.Format("2006-01-02T15:04:05"), start)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *SqlTimeStamp) UnmarshalXML(d *xml.Decoder, start xml.StartElement) error {
|
||||||
|
var s string
|
||||||
|
if err := d.DecodeElement(&s, &start); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if s == "" {
|
||||||
|
t.Valid = false
|
||||||
|
t.Val = time.Time{}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
tm, err := tryParseDT(s)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
t.Val = tm
|
||||||
|
t.Valid = !tm.IsZero() && tm.Format("2006-01-02T15:04:05") != "0001-01-01T00:00:00"
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
func SqlTimeStampNow() SqlTimeStamp {
|
func SqlTimeStampNow() SqlTimeStamp {
|
||||||
return SqlTimeStamp{SqlNull: SqlNull[time.Time]{Val: time.Now(), Valid: true}}
|
return SqlTimeStamp{SqlNull: SqlNull[time.Time]{Val: time.Now(), Valid: true}}
|
||||||
}
|
}
|
||||||
@@ -378,7 +529,7 @@ func (d SqlDate) MarshalJSON() ([]byte, error) {
|
|||||||
if strings.HasPrefix(s, "0001-01-01") {
|
if strings.HasPrefix(s, "0001-01-01") {
|
||||||
return []byte("null"), nil
|
return []byte("null"), nil
|
||||||
}
|
}
|
||||||
return []byte(fmt.Sprintf(`"%s"`, s)), nil
|
return fmt.Appendf(nil, `"%s"`, s), nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *SqlDate) UnmarshalJSON(b []byte) error {
|
func (d *SqlDate) UnmarshalJSON(b []byte) error {
|
||||||
@@ -413,6 +564,57 @@ func (d SqlDate) String() string {
|
|||||||
return s
|
return s
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (d SqlDate) MarshalYAML() (any, error) {
|
||||||
|
if !d.Valid || d.Val.IsZero() {
|
||||||
|
return nil, nil
|
||||||
|
}
|
||||||
|
s := d.Val.Format("2006-01-02")
|
||||||
|
if strings.HasPrefix(s, "0001-01-01") {
|
||||||
|
return nil, nil
|
||||||
|
}
|
||||||
|
return s, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *SqlDate) UnmarshalYAML(value *yaml.Node) error {
|
||||||
|
if err := d.SqlNull.UnmarshalYAML(value); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if d.Valid && d.Val.Format("2006-01-02") <= "0001-01-01" {
|
||||||
|
d.Valid = false
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d SqlDate) MarshalXML(e *xml.Encoder, start xml.StartElement) error {
|
||||||
|
if !d.Valid || d.Val.IsZero() {
|
||||||
|
return e.EncodeElement("", start)
|
||||||
|
}
|
||||||
|
s := d.Val.Format("2006-01-02")
|
||||||
|
if strings.HasPrefix(s, "0001-01-01") {
|
||||||
|
return e.EncodeElement("", start)
|
||||||
|
}
|
||||||
|
return e.EncodeElement(s, start)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *SqlDate) UnmarshalXML(dec *xml.Decoder, start xml.StartElement) error {
|
||||||
|
var s string
|
||||||
|
if err := dec.DecodeElement(&s, &start); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if s == "" {
|
||||||
|
d.Valid = false
|
||||||
|
d.Val = time.Time{}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
tm, err := tryParseDT(s)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
d.Val = tm
|
||||||
|
d.Valid = !tm.IsZero() && tm.Format("2006-01-02") > "0001-01-01"
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
func SqlDateNow() SqlDate {
|
func SqlDateNow() SqlDate {
|
||||||
return SqlDate{SqlNull: SqlNull[time.Time]{Val: time.Now(), Valid: true}}
|
return SqlDate{SqlNull: SqlNull[time.Time]{Val: time.Now(), Valid: true}}
|
||||||
}
|
}
|
||||||
@@ -428,7 +630,7 @@ func (t SqlTime) MarshalJSON() ([]byte, error) {
|
|||||||
if s == "00:00:00" {
|
if s == "00:00:00" {
|
||||||
return []byte("null"), nil
|
return []byte("null"), nil
|
||||||
}
|
}
|
||||||
return []byte(fmt.Sprintf(`"%s"`, s)), nil
|
return fmt.Appendf(nil, `"%s"`, s), nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (t *SqlTime) UnmarshalJSON(b []byte) error {
|
func (t *SqlTime) UnmarshalJSON(b []byte) error {
|
||||||
@@ -455,6 +657,57 @@ func (t SqlTime) String() string {
|
|||||||
return t.Val.Format("15:04:05")
|
return t.Val.Format("15:04:05")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (t SqlTime) MarshalYAML() (any, error) {
|
||||||
|
if !t.Valid || t.Val.IsZero() {
|
||||||
|
return nil, nil
|
||||||
|
}
|
||||||
|
s := t.Val.Format("15:04:05")
|
||||||
|
if s == "00:00:00" {
|
||||||
|
return nil, nil
|
||||||
|
}
|
||||||
|
return s, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *SqlTime) UnmarshalYAML(value *yaml.Node) error {
|
||||||
|
if err := t.SqlNull.UnmarshalYAML(value); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if t.Valid && t.Val.Format("15:04:05") == "00:00:00" {
|
||||||
|
t.Valid = false
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t SqlTime) MarshalXML(e *xml.Encoder, start xml.StartElement) error {
|
||||||
|
if !t.Valid || t.Val.IsZero() {
|
||||||
|
return e.EncodeElement("", start)
|
||||||
|
}
|
||||||
|
s := t.Val.Format("15:04:05")
|
||||||
|
if s == "00:00:00" {
|
||||||
|
return e.EncodeElement("", start)
|
||||||
|
}
|
||||||
|
return e.EncodeElement(s, start)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *SqlTime) UnmarshalXML(d *xml.Decoder, start xml.StartElement) error {
|
||||||
|
var s string
|
||||||
|
if err := d.DecodeElement(&s, &start); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if s == "" {
|
||||||
|
t.Valid = false
|
||||||
|
t.Val = time.Time{}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
tm, err := tryParseDT(s)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
t.Val = tm
|
||||||
|
t.Valid = !tm.IsZero() && tm.Format("15:04:05") != "00:00:00"
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
func SqlTimeNow() SqlTime {
|
func SqlTimeNow() SqlTime {
|
||||||
return SqlTime{SqlNull: SqlNull[time.Time]{Val: time.Now(), Valid: true}}
|
return SqlTime{SqlNull: SqlNull[time.Time]{Val: time.Now(), Valid: true}}
|
||||||
}
|
}
|
||||||
@@ -462,6 +715,11 @@ func SqlTimeNow() SqlTime {
|
|||||||
// SqlJSONB - Nullable JSONB as []byte.
|
// SqlJSONB - Nullable JSONB as []byte.
|
||||||
type SqlJSONB []byte
|
type SqlJSONB []byte
|
||||||
|
|
||||||
|
// SqlJSON - Nullable JSON as []byte. PostgreSQL's json and jsonb types share
|
||||||
|
// the same textual representation and Go marshalling behavior, differing only
|
||||||
|
// in server-side storage, so SqlJSON is an alias of SqlJSONB.
|
||||||
|
type SqlJSON = SqlJSONB
|
||||||
|
|
||||||
// Scan implements sql.Scanner.
|
// Scan implements sql.Scanner.
|
||||||
func (n *SqlJSONB) Scan(value any) error {
|
func (n *SqlJSONB) Scan(value any) error {
|
||||||
if value == nil {
|
if value == nil {
|
||||||
@@ -518,6 +776,67 @@ func (n *SqlJSONB) UnmarshalJSON(b []byte) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// MarshalYAML implements yaml.Marshaler. The underlying JSON is decoded into
|
||||||
|
// a generic value first so it renders as native YAML mappings/sequences
|
||||||
|
// rather than an embedded JSON string.
|
||||||
|
func (n SqlJSONB) MarshalYAML() (any, error) {
|
||||||
|
if len(n) == 0 {
|
||||||
|
return nil, nil
|
||||||
|
}
|
||||||
|
var v any
|
||||||
|
if err := json.Unmarshal(n, &v); err != nil {
|
||||||
|
return nil, nil
|
||||||
|
}
|
||||||
|
return v, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// UnmarshalYAML implements yaml.Unmarshaler.
|
||||||
|
func (n *SqlJSONB) UnmarshalYAML(value *yaml.Node) error {
|
||||||
|
if value == nil || value.Tag == "!!null" {
|
||||||
|
*n = nil
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
var v any
|
||||||
|
if err := value.Decode(&v); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
b, err := json.Marshal(v)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
*n = b
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// MarshalXML implements xml.Marshaler. JSON has no clean structural mapping
|
||||||
|
// to XML, so the raw JSON text is emitted as the element's text content.
|
||||||
|
func (n SqlJSONB) MarshalXML(e *xml.Encoder, start xml.StartElement) error {
|
||||||
|
if len(n) == 0 {
|
||||||
|
return e.EncodeElement("", start)
|
||||||
|
}
|
||||||
|
var obj any
|
||||||
|
if err := json.Unmarshal(n, &obj); err != nil {
|
||||||
|
return e.EncodeElement("", start)
|
||||||
|
}
|
||||||
|
return e.EncodeElement(string(n), start)
|
||||||
|
}
|
||||||
|
|
||||||
|
// UnmarshalXML implements xml.Unmarshaler, reading back the raw JSON text
|
||||||
|
// written by MarshalXML.
|
||||||
|
func (n *SqlJSONB) UnmarshalXML(d *xml.Decoder, start xml.StartElement) error {
|
||||||
|
var s string
|
||||||
|
if err := d.DecodeElement(&s, &start); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
s = strings.TrimSpace(s)
|
||||||
|
if s == "" {
|
||||||
|
*n = nil
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
*n = []byte(s)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
func (n SqlJSONB) AsMap() (map[string]any, error) {
|
func (n SqlJSONB) AsMap() (map[string]any, error) {
|
||||||
if len(n) == 0 {
|
if len(n) == 0 {
|
||||||
return nil, nil
|
return nil, nil
|
||||||
@@ -625,6 +944,10 @@ func NewSqlInt64(v int64) SqlInt64 {
|
|||||||
return SqlInt64{Val: v, Valid: true}
|
return SqlInt64{Val: v, Valid: true}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func NewSqlFloat32(v float32) SqlFloat32 {
|
||||||
|
return SqlFloat32{Val: v, Valid: true}
|
||||||
|
}
|
||||||
|
|
||||||
func NewSqlFloat64(v float64) SqlFloat64 {
|
func NewSqlFloat64(v float64) SqlFloat64 {
|
||||||
return SqlFloat64{Val: v, Valid: true}
|
return SqlFloat64{Val: v, Valid: true}
|
||||||
}
|
}
|
||||||
+3
-3
@@ -508,8 +508,8 @@ func (h *Handler) handleRead(ctx context.Context, w common.ResponseWriter, id st
|
|||||||
singleResult := reflect.New(modelType).Interface()
|
singleResult := reflect.New(modelType).Interface()
|
||||||
pkName := reflection.GetPrimaryKeyName(singleResult)
|
pkName := reflection.GetPrimaryKeyName(singleResult)
|
||||||
|
|
||||||
query = query.Where(fmt.Sprintf("%s = ?", common.QuoteIdent(pkName)), targetID)
|
query = query.Model(singleResult).Where(fmt.Sprintf("%s = ?", common.QuoteIdent(pkName)), targetID)
|
||||||
if err := query.Scan(ctx, singleResult); err != nil {
|
if err := query.ScanModel(ctx); err != nil {
|
||||||
logger.Error("Error querying record: %v", err)
|
logger.Error("Error querying record: %v", err)
|
||||||
h.sendError(w, http.StatusInternalServerError, "query_error", "Error executing query", err)
|
h.sendError(w, http.StatusInternalServerError, "query_error", "Error executing query", err)
|
||||||
return
|
return
|
||||||
@@ -518,7 +518,7 @@ func (h *Handler) handleRead(ctx context.Context, w common.ResponseWriter, id st
|
|||||||
} else {
|
} else {
|
||||||
logger.Debug("Querying multiple records")
|
logger.Debug("Querying multiple records")
|
||||||
// Use the modelPtr already created and set on the query
|
// Use the modelPtr already created and set on the query
|
||||||
if err := query.Scan(ctx, modelPtr); err != nil {
|
if err := query.ScanModel(ctx); err != nil {
|
||||||
logger.Error("Error querying records: %v", err)
|
logger.Error("Error querying records: %v", err)
|
||||||
h.sendError(w, http.StatusInternalServerError, "query_error", "Error executing query", err)
|
h.sendError(w, http.StatusInternalServerError, "query_error", "Error executing query", err)
|
||||||
return
|
return
|
||||||
|
|||||||
+11
@@ -78,6 +78,17 @@ func (s *securityContext) GetUserID() (int, bool) {
|
|||||||
return security.GetUserID(s.ctx.Context)
|
return security.GetUserID(s.ctx.Context)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// GetUserRef returns an opaque user identifier for row security lookups.
|
||||||
|
// It prefers the full *security.UserContext (so providers can read JWT claims,
|
||||||
|
// e.g. a UUID subject) and falls back to the int user ID.
|
||||||
|
func (s *securityContext) GetUserRef() (any, bool) {
|
||||||
|
if userCtx, ok := security.GetUserContext(s.ctx.Context); ok {
|
||||||
|
return userCtx, true
|
||||||
|
}
|
||||||
|
userID, ok := security.GetUserID(s.ctx.Context)
|
||||||
|
return userID, ok
|
||||||
|
}
|
||||||
|
|
||||||
func (s *securityContext) GetSchema() string {
|
func (s *securityContext) GetSchema() string {
|
||||||
return s.ctx.Schema
|
return s.ctx.Schema
|
||||||
}
|
}
|
||||||
|
|||||||
+90
-1
@@ -11,7 +11,8 @@ Type-safe, composable security system for ResolveSpec with support for authentic
|
|||||||
- ✅ **No Global State** - Each handler has its own security configuration
|
- ✅ **No Global State** - Each handler has its own security configuration
|
||||||
- ✅ **Testable** - Easy to mock and test
|
- ✅ **Testable** - Easy to mock and test
|
||||||
- ✅ **Extensible** - Implement custom providers for your needs
|
- ✅ **Extensible** - Implement custom providers for your needs
|
||||||
- ✅ **Stored Procedures** - All database operations use PostgreSQL stored procedures for security and maintainability
|
- ✅ **Stored Procedures** - Database operations use PostgreSQL stored procedures where available, for security and maintainability
|
||||||
|
- ✅ **Direct Mode** - Portable Go/SQL fallback for SQLite, MySQL, or Postgres without the stored procedures installed — no code changes required
|
||||||
- ✅ **OAuth2 Authorization Server** - Built-in OAuth 2.1 + PKCE server (RFC 8414, 7591, 7009, 7662) with login form and external provider federation
|
- ✅ **OAuth2 Authorization Server** - Built-in OAuth 2.1 + PKCE server (RFC 8414, 7591, 7009, 7662) with login form and external provider federation
|
||||||
- ✅ **Password Reset** - Self-service password reset with secure token generation and session invalidation
|
- ✅ **Password Reset** - Self-service password reset with secure token generation and session invalidation
|
||||||
|
|
||||||
@@ -51,6 +52,94 @@ Type-safe, composable security system for ResolveSpec with support for authentic
|
|||||||
|
|
||||||
See `database_schema.sql` for complete stored procedure definitions and examples.
|
See `database_schema.sql` for complete stored procedure definitions and examples.
|
||||||
|
|
||||||
|
**Not on Postgres, or don't have the procedures installed?** See [Direct Mode](#direct-mode-portable-sql-without-stored-procedures) below — every provider that calls a `resolvespec_*` procedure also has a portable Go/SQL implementation that works on SQLite, MySQL, or plain Postgres.
|
||||||
|
|
||||||
|
## Direct Mode (portable SQL without stored procedures)
|
||||||
|
|
||||||
|
Every database-backed provider (`DatabaseAuthenticator`, `JWTAuthenticator`, `DatabaseTwoFactorProvider`, `DatabasePasskeyProvider`, the OAuth2 methods/server, `DatabaseKeyStore`) has two code paths:
|
||||||
|
|
||||||
|
- **Procedure mode** — calls the configured `resolvespec_*` stored procedure (original behavior, Postgres-only).
|
||||||
|
- **Direct mode** — reimplements the same logic in Go using plain parameterized SQL against configurable table names. Works on SQLite, MySQL, or a Postgres database where the procedures were never deployed.
|
||||||
|
|
||||||
|
### QueryMode
|
||||||
|
|
||||||
|
Selection is controlled per-provider by a `QueryMode`:
|
||||||
|
|
||||||
|
```go
|
||||||
|
type QueryMode int
|
||||||
|
|
||||||
|
const (
|
||||||
|
ModeAuto QueryMode = iota // default
|
||||||
|
ModeProcedure
|
||||||
|
ModeDirect
|
||||||
|
)
|
||||||
|
```
|
||||||
|
|
||||||
|
- **`ModeAuto`** (default, zero value) — auto-detects per connection:
|
||||||
|
- SQLite/MySQL drivers → Direct mode, no probing.
|
||||||
|
- Postgres drivers (`lib/pq`, `pgx`) → probes `pg_proc` for the configured procedure name and uses it **only if it actually exists**; otherwise falls back to Direct mode. The result is cached per procedure name and reset on reconnect.
|
||||||
|
- Any other/unrecognized driver (including `sqlmock` test doubles) → defaults to Procedure mode, preserving existing behavior for callers that don't expose an identifiable driver type.
|
||||||
|
- **`ModeProcedure`** — always calls the stored procedure, regardless of dialect.
|
||||||
|
- **`ModeDirect`** — always uses the portable Go/SQL path, never the stored procedure.
|
||||||
|
|
||||||
|
Set it via the provider's `Options` struct or `With...` chain method:
|
||||||
|
|
||||||
|
```go
|
||||||
|
auth := security.NewDatabaseAuthenticatorWithOptions(db, security.DatabaseAuthenticatorOptions{
|
||||||
|
QueryMode: security.ModeDirect, // force Direct mode, e.g. for SQLite
|
||||||
|
})
|
||||||
|
|
||||||
|
tfaProvider := security.NewDatabaseTwoFactorProvider(sqliteDB, nil).
|
||||||
|
WithQueryMode(security.ModeDirect)
|
||||||
|
```
|
||||||
|
|
||||||
|
On a real SQLite/MySQL connection you can usually leave `QueryMode` unset — `ModeAuto` detects the dialect and uses Direct mode automatically.
|
||||||
|
|
||||||
|
### TableNames / KeyStoreTableNames
|
||||||
|
|
||||||
|
Direct mode reads/writes plain tables instead of calling procedures, so table names are configurable the same way procedure names are (`SQLNames`):
|
||||||
|
|
||||||
|
```go
|
||||||
|
type TableNames struct {
|
||||||
|
Users string // default: "users"
|
||||||
|
UserSessions string // default: "user_sessions"
|
||||||
|
TokenBlacklist string // default: "token_blacklist"
|
||||||
|
UserTOTPBackupCodes string // default: "user_totp_backup_codes"
|
||||||
|
UserPasskeyCredentials string // default: "user_passkey_credentials"
|
||||||
|
UserPasswordResets string // default: "user_password_resets"
|
||||||
|
OAuthClients string // default: "oauth_clients"
|
||||||
|
OAuthCodes string // default: "oauth_codes"
|
||||||
|
}
|
||||||
|
|
||||||
|
type KeyStoreTableNames struct {
|
||||||
|
UserKeys string // default: "user_keys" — used by DatabaseKeyStore
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
`DefaultTableNames()` / `MergeTableNames()` / `ValidateTableNames()` mirror `DefaultSQLNames()` / `MergeSQLNames()` / `ValidateSQLNames()`. Set custom names via the same `Options`/`With...` surface as `QueryMode`:
|
||||||
|
|
||||||
|
```go
|
||||||
|
auth := security.NewDatabaseAuthenticatorWithOptions(db, security.DatabaseAuthenticatorOptions{
|
||||||
|
TableNames: &security.TableNames{Users: "app_users"}, // only override what differs
|
||||||
|
})
|
||||||
|
```
|
||||||
|
|
||||||
|
`oauth2_methods.go` and `oauth_server_db.go` are methods on `*DatabaseAuthenticator` and reuse its `TableNames`/`QueryMode`; there's no separate config for them.
|
||||||
|
|
||||||
|
### Schema
|
||||||
|
|
||||||
|
`database_schema_sqlite.sql` is the portable companion to `database_schema.sql` — plain `CREATE TABLE` statements only (no functions, no triggers, no `jsonb`/`bytea`/array types), covering every table Direct mode reads or writes. Use it to stand up a SQLite (or adapt for MySQL) database for Direct mode.
|
||||||
|
|
||||||
|
### What's NOT covered
|
||||||
|
|
||||||
|
`ColumnSecurityProvider`/`RowSecurityProvider` (`resolvespec_column_security` / `resolvespec_row_security`) query an external `core.secaccess`/`core.hub_link` schema this package doesn't own. Direct mode has no portable equivalent to fabricate for these and returns `security.ErrDirectModeUnsupported` — use `ConfigColumnSecurityProvider`/`ConfigRowSecurityProvider` instead when not running against Postgres with those procedures installed.
|
||||||
|
|
||||||
|
### Behavioral notes
|
||||||
|
|
||||||
|
- Direct mode matches Procedure mode's current behavior exactly, including its TODOs — e.g. passwords are compared as-is (the stored procedures don't verify bcrypt hashes yet either; see the TODO in `resolvespec_login`/`resolvespec_password_reset`).
|
||||||
|
- Session tokens generated by Direct mode use the same `sess_<hex>_<unix-timestamp>` shape as the plpgsql procedures.
|
||||||
|
- `bytea`/array/`jsonb` Postgres-only columns (passkey credentials, OAuth2 client scopes, keystore `meta`) are stored as base64/JSON-encoded `TEXT` in Direct mode — transparent to callers, since the Go-level API already deals in those same encodings.
|
||||||
|
|
||||||
## Quick Start
|
## Quick Start
|
||||||
|
|
||||||
```go
|
```go
|
||||||
|
|||||||
+2
-2
@@ -74,8 +74,8 @@ func (c *CompositeSecurityProvider) GetColumnSecurity(ctx context.Context, userI
|
|||||||
}
|
}
|
||||||
|
|
||||||
// GetRowSecurity delegates to the row security provider
|
// GetRowSecurity delegates to the row security provider
|
||||||
func (c *CompositeSecurityProvider) GetRowSecurity(ctx context.Context, userID int, schema, table string) (RowSecurity, error) {
|
func (c *CompositeSecurityProvider) GetRowSecurity(ctx context.Context, userRef any, schema, table string) (RowSecurity, error) {
|
||||||
return c.rowSec.GetRowSecurity(ctx, userID, schema, table)
|
return c.rowSec.GetRowSecurity(ctx, userRef, schema, table)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Optional interface implementations (if wrapped providers support them)
|
// Optional interface implementations (if wrapped providers support them)
|
||||||
|
|||||||
+141
@@ -0,0 +1,141 @@
|
|||||||
|
-- Portable schema for Direct-mode (non-stored-procedure) operation.
|
||||||
|
-- Plain CREATE TABLE statements only, no functions/triggers, using types
|
||||||
|
-- understood by SQLite (and portable to MySQL). Used by Direct-mode tests
|
||||||
|
-- and as a reference for deployments without Postgres.
|
||||||
|
|
||||||
|
CREATE TABLE IF NOT EXISTS users (
|
||||||
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
username VARCHAR(255) NOT NULL UNIQUE,
|
||||||
|
email VARCHAR(255) NOT NULL UNIQUE,
|
||||||
|
password VARCHAR(255),
|
||||||
|
user_level INTEGER DEFAULT 0,
|
||||||
|
roles VARCHAR(500),
|
||||||
|
is_active BOOLEAN DEFAULT 1,
|
||||||
|
created_at TIMESTAMP,
|
||||||
|
updated_at TIMESTAMP,
|
||||||
|
last_login_at TIMESTAMP,
|
||||||
|
program_user_id INTEGER DEFAULT 0,
|
||||||
|
program_user_table VARCHAR(255) DEFAULT '',
|
||||||
|
remote_id VARCHAR(255),
|
||||||
|
auth_provider VARCHAR(50),
|
||||||
|
totp_secret VARCHAR(255),
|
||||||
|
totp_enabled BOOLEAN DEFAULT 0,
|
||||||
|
totp_enabled_at TIMESTAMP
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE TABLE IF NOT EXISTS user_sessions (
|
||||||
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
session_token VARCHAR(500) NOT NULL UNIQUE,
|
||||||
|
user_id INTEGER NOT NULL,
|
||||||
|
expires_at TIMESTAMP NOT NULL,
|
||||||
|
created_at TIMESTAMP,
|
||||||
|
last_activity_at TIMESTAMP,
|
||||||
|
ip_address VARCHAR(45),
|
||||||
|
user_agent TEXT,
|
||||||
|
access_token TEXT,
|
||||||
|
refresh_token TEXT,
|
||||||
|
token_type VARCHAR(50) DEFAULT 'Bearer',
|
||||||
|
auth_provider VARCHAR(50)
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_session_token ON user_sessions(session_token);
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_user_id ON user_sessions(user_id);
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_expires_at ON user_sessions(expires_at);
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_refresh_token ON user_sessions(refresh_token);
|
||||||
|
|
||||||
|
CREATE TABLE IF NOT EXISTS token_blacklist (
|
||||||
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
token VARCHAR(500) NOT NULL,
|
||||||
|
user_id INTEGER,
|
||||||
|
expires_at TIMESTAMP NOT NULL,
|
||||||
|
created_at TIMESTAMP
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE TABLE IF NOT EXISTS user_totp_backup_codes (
|
||||||
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
user_id INTEGER NOT NULL,
|
||||||
|
code_hash VARCHAR(64) NOT NULL,
|
||||||
|
used BOOLEAN DEFAULT 0,
|
||||||
|
used_at TIMESTAMP,
|
||||||
|
created_at TIMESTAMP
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_totp_user_id ON user_totp_backup_codes(user_id);
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_totp_code_hash ON user_totp_backup_codes(code_hash);
|
||||||
|
|
||||||
|
CREATE TABLE IF NOT EXISTS user_passkey_credentials (
|
||||||
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
user_id INTEGER NOT NULL,
|
||||||
|
credential_id TEXT NOT NULL UNIQUE, -- base64 text (Direct mode), not native bytea
|
||||||
|
public_key TEXT NOT NULL, -- base64 text
|
||||||
|
attestation_type VARCHAR(50) DEFAULT 'none',
|
||||||
|
aaguid TEXT, -- base64 text
|
||||||
|
sign_count INTEGER DEFAULT 0,
|
||||||
|
clone_warning BOOLEAN DEFAULT 0,
|
||||||
|
transports TEXT, -- JSON-encoded []string
|
||||||
|
backup_eligible BOOLEAN DEFAULT 0,
|
||||||
|
backup_state BOOLEAN DEFAULT 0,
|
||||||
|
name VARCHAR(255),
|
||||||
|
created_at TIMESTAMP,
|
||||||
|
last_used_at TIMESTAMP
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_passkey_user_id ON user_passkey_credentials(user_id);
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_passkey_credential_id ON user_passkey_credentials(credential_id);
|
||||||
|
|
||||||
|
CREATE TABLE IF NOT EXISTS user_password_resets (
|
||||||
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
user_id INTEGER NOT NULL,
|
||||||
|
token_hash VARCHAR(64) NOT NULL UNIQUE,
|
||||||
|
expires_at TIMESTAMP NOT NULL,
|
||||||
|
created_at TIMESTAMP,
|
||||||
|
used BOOLEAN DEFAULT 0,
|
||||||
|
used_at TIMESTAMP
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE TABLE IF NOT EXISTS oauth_clients (
|
||||||
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
client_id VARCHAR(255) NOT NULL UNIQUE,
|
||||||
|
redirect_uris TEXT NOT NULL, -- JSON-encoded []string
|
||||||
|
client_name VARCHAR(255),
|
||||||
|
grant_types TEXT, -- JSON-encoded []string
|
||||||
|
allowed_scopes TEXT, -- JSON-encoded []string
|
||||||
|
is_active BOOLEAN DEFAULT 1,
|
||||||
|
created_at TIMESTAMP
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE TABLE IF NOT EXISTS oauth_codes (
|
||||||
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
code VARCHAR(255) NOT NULL UNIQUE,
|
||||||
|
client_id VARCHAR(255) NOT NULL,
|
||||||
|
redirect_uri TEXT NOT NULL,
|
||||||
|
client_state TEXT,
|
||||||
|
code_challenge VARCHAR(255) NOT NULL,
|
||||||
|
code_challenge_method VARCHAR(10) DEFAULT 'S256',
|
||||||
|
session_token TEXT NOT NULL,
|
||||||
|
refresh_token TEXT,
|
||||||
|
scopes TEXT, -- JSON-encoded []string
|
||||||
|
expires_at TIMESTAMP NOT NULL,
|
||||||
|
created_at TIMESTAMP
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_oauth_codes_code ON oauth_codes(code);
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_oauth_codes_expires ON oauth_codes(expires_at);
|
||||||
|
|
||||||
|
CREATE TABLE IF NOT EXISTS user_keys (
|
||||||
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
user_id INTEGER NOT NULL,
|
||||||
|
key_type VARCHAR(50) NOT NULL,
|
||||||
|
key_hash VARCHAR(64) NOT NULL UNIQUE,
|
||||||
|
name VARCHAR(255) NOT NULL DEFAULT '',
|
||||||
|
scopes TEXT, -- JSON-encoded []string
|
||||||
|
meta TEXT, -- JSON-encoded map
|
||||||
|
expires_at TIMESTAMP,
|
||||||
|
created_at TIMESTAMP,
|
||||||
|
last_used_at TIMESTAMP,
|
||||||
|
is_active BOOLEAN DEFAULT 1
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_user_keys_user_id ON user_keys(user_id);
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_user_keys_key_hash ON user_keys(key_hash);
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_user_keys_key_type ON user_keys(key_type);
|
||||||
+23
-9
@@ -14,6 +14,11 @@ import (
|
|||||||
type SecurityContext interface {
|
type SecurityContext interface {
|
||||||
GetContext() context.Context
|
GetContext() context.Context
|
||||||
GetUserID() (int, bool)
|
GetUserID() (int, bool)
|
||||||
|
// GetUserRef returns an opaque user identifier for row security lookups.
|
||||||
|
// Unlike GetUserID, it is not required to be an integer: implementations backed by
|
||||||
|
// non-integer identifiers (e.g. UUIDs) can return a string, or the full
|
||||||
|
// *security.UserContext so a RowSecurityProvider can read JWT claims directly.
|
||||||
|
GetUserRef() (any, bool)
|
||||||
GetSchema() string
|
GetSchema() string
|
||||||
GetEntity() string
|
GetEntity() string
|
||||||
GetModel() interface{}
|
GetModel() interface{}
|
||||||
@@ -45,8 +50,13 @@ func loadSecurityRules(secCtx SecurityContext, securityList *SecurityList) error
|
|||||||
// return err
|
// return err
|
||||||
}
|
}
|
||||||
|
|
||||||
// Load row security rules using the provider
|
// Load row security rules using the provider. Row security uses the opaque
|
||||||
_, err = securityList.LoadRowSecurity(secCtx.GetContext(), userID, schema, tablename, false)
|
// user ref (not the int-only user ID) so non-integer user identifiers work.
|
||||||
|
userRef, refOK := secCtx.GetUserRef()
|
||||||
|
if !refOK {
|
||||||
|
userRef = userID
|
||||||
|
}
|
||||||
|
_, err = securityList.LoadRowSecurity(secCtx.GetContext(), userRef, schema, tablename, false)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Warn("Failed to load row security: %v", err)
|
logger.Warn("Failed to load row security: %v", err)
|
||||||
// Don't fail the request if no security rules exist
|
// Don't fail the request if no security rules exist
|
||||||
@@ -58,25 +68,29 @@ func loadSecurityRules(secCtx SecurityContext, securityList *SecurityList) error
|
|||||||
|
|
||||||
// applyRowSecurity applies row-level security filters to the query (generic version)
|
// applyRowSecurity applies row-level security filters to the query (generic version)
|
||||||
func applyRowSecurity(secCtx SecurityContext, securityList *SecurityList) error {
|
func applyRowSecurity(secCtx SecurityContext, securityList *SecurityList) error {
|
||||||
userID, ok := secCtx.GetUserID()
|
userRef, ok := secCtx.GetUserRef()
|
||||||
if !ok {
|
if !ok {
|
||||||
return nil // No user context, skip
|
userID, idOK := secCtx.GetUserID()
|
||||||
|
if !idOK {
|
||||||
|
return nil // No user context, skip
|
||||||
|
}
|
||||||
|
userRef = userID
|
||||||
}
|
}
|
||||||
|
|
||||||
schema := secCtx.GetSchema()
|
schema := secCtx.GetSchema()
|
||||||
tablename := secCtx.GetEntity()
|
tablename := secCtx.GetEntity()
|
||||||
|
|
||||||
// Get row security template
|
// Get row security template
|
||||||
rowSec, err := securityList.GetRowSecurityTemplate(userID, schema, tablename)
|
rowSec, err := securityList.GetRowSecurityTemplate(userRef, schema, tablename)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
// No row security defined, allow query to proceed
|
// No row security defined, allow query to proceed
|
||||||
logger.Debug("No row security for %s.%s@%d: %v", schema, tablename, userID, err)
|
logger.Debug("No row security for %s.%s@%v: %v", schema, tablename, userRef, err)
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// Check if user has a blocking rule
|
// Check if user has a blocking rule
|
||||||
if rowSec.HasBlock {
|
if rowSec.HasBlock {
|
||||||
logger.Warn("User %d blocked from accessing %s.%s", userID, schema, tablename)
|
logger.Warn("User %v blocked from accessing %s.%s", userRef, schema, tablename)
|
||||||
return fmt.Errorf("access denied to %s", tablename)
|
return fmt.Errorf("access denied to %s", tablename)
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -112,8 +126,8 @@ func applyRowSecurity(secCtx SecurityContext, securityList *SecurityList) error
|
|||||||
// Generate the WHERE clause from template
|
// Generate the WHERE clause from template
|
||||||
whereClause := rowSec.GetTemplate(pkName, modelType)
|
whereClause := rowSec.GetTemplate(pkName, modelType)
|
||||||
|
|
||||||
logger.Info("Applying row security filter for user %d on %s.%s: %s",
|
logger.Info("Applying row security filter for user %v on %s.%s: %s",
|
||||||
userID, schema, tablename, whereClause)
|
userRef, schema, tablename, whereClause)
|
||||||
|
|
||||||
// Apply the WHERE clause to the query
|
// Apply the WHERE clause to the query
|
||||||
query := secCtx.GetQuery()
|
query := secCtx.GetQuery()
|
||||||
|
|||||||
+6
-2
@@ -121,8 +121,12 @@ type ColumnSecurityProvider interface {
|
|||||||
|
|
||||||
// RowSecurityProvider handles row-level security (filtering)
|
// RowSecurityProvider handles row-level security (filtering)
|
||||||
type RowSecurityProvider interface {
|
type RowSecurityProvider interface {
|
||||||
// GetRowSecurity loads row security rules for a user and entity
|
// GetRowSecurity loads row security rules for a user and entity.
|
||||||
GetRowSecurity(ctx context.Context, userID int, schema, table string) (RowSecurity, error)
|
// userRef identifies the user and is opaque to the caller: it may be an int ID,
|
||||||
|
// a string/UUID, or the full *security.UserContext (see SecurityContext.GetUserRef),
|
||||||
|
// so providers backed by non-integer user identifiers (e.g. UUIDs) or that need
|
||||||
|
// access to JWT claims can implement row security without relying on a numeric ID.
|
||||||
|
GetRowSecurity(ctx context.Context, userRef any, schema, table string) (RowSecurity, error)
|
||||||
}
|
}
|
||||||
|
|
||||||
// SecurityProvider is the main interface combining all security concerns
|
// SecurityProvider is the main interface combining all security concerns
|
||||||
|
|||||||
+52
-11
@@ -23,6 +23,10 @@ type DatabaseKeyStoreOptions struct {
|
|||||||
CacheTTL time.Duration
|
CacheTTL time.Duration
|
||||||
// SQLNames provides custom procedure names. If nil, uses DefaultKeyStoreSQLNames().
|
// SQLNames provides custom procedure names. If nil, uses DefaultKeyStoreSQLNames().
|
||||||
SQLNames *KeyStoreSQLNames
|
SQLNames *KeyStoreSQLNames
|
||||||
|
// TableNames provides custom table names for Direct mode. If nil, uses DefaultKeyStoreTableNames().
|
||||||
|
TableNames *KeyStoreTableNames
|
||||||
|
// QueryMode selects stored-procedure vs Direct-mode SQL. Default (zero value) is ModeAuto.
|
||||||
|
QueryMode QueryMode
|
||||||
// DBFactory is called to obtain a fresh *sql.DB when the existing connection is closed.
|
// DBFactory is called to obtain a fresh *sql.DB when the existing connection is closed.
|
||||||
// If nil, reconnection is disabled.
|
// If nil, reconnection is disabled.
|
||||||
DBFactory func() (*sql.DB, error)
|
DBFactory func() (*sql.DB, error)
|
||||||
@@ -38,12 +42,15 @@ type DatabaseKeyStoreOptions struct {
|
|||||||
// cache TTL, a deleted key may continue to authenticate for up to CacheTTL
|
// cache TTL, a deleted key may continue to authenticate for up to CacheTTL
|
||||||
// (default 2 minutes) if the cache entry cannot be invalidated.
|
// (default 2 minutes) if the cache entry cannot be invalidated.
|
||||||
type DatabaseKeyStore struct {
|
type DatabaseKeyStore struct {
|
||||||
db *sql.DB
|
db *sql.DB
|
||||||
dbMu sync.RWMutex
|
dbMu sync.RWMutex
|
||||||
dbFactory func() (*sql.DB, error)
|
dbFactory func() (*sql.DB, error)
|
||||||
sqlNames *KeyStoreSQLNames
|
sqlNames *KeyStoreSQLNames
|
||||||
cache *cache.Cache
|
tableNames *KeyStoreTableNames
|
||||||
cacheTTL time.Duration
|
queryMode QueryMode
|
||||||
|
capability *dbCapability
|
||||||
|
cache *cache.Cache
|
||||||
|
cacheTTL time.Duration
|
||||||
}
|
}
|
||||||
|
|
||||||
// NewDatabaseKeyStore creates a DatabaseKeyStore with optional configuration.
|
// NewDatabaseKeyStore creates a DatabaseKeyStore with optional configuration.
|
||||||
@@ -60,12 +67,16 @@ func NewDatabaseKeyStore(db *sql.DB, opts ...DatabaseKeyStoreOptions) *DatabaseK
|
|||||||
c = cache.GetDefaultCache()
|
c = cache.GetDefaultCache()
|
||||||
}
|
}
|
||||||
names := MergeKeyStoreSQLNames(DefaultKeyStoreSQLNames(), o.SQLNames)
|
names := MergeKeyStoreSQLNames(DefaultKeyStoreSQLNames(), o.SQLNames)
|
||||||
|
tableNames := resolveKeyStoreTableNames(o.TableNames)
|
||||||
return &DatabaseKeyStore{
|
return &DatabaseKeyStore{
|
||||||
db: db,
|
db: db,
|
||||||
dbFactory: o.DBFactory,
|
dbFactory: o.DBFactory,
|
||||||
sqlNames: names,
|
sqlNames: names,
|
||||||
cache: c,
|
tableNames: tableNames,
|
||||||
cacheTTL: o.CacheTTL,
|
queryMode: o.QueryMode,
|
||||||
|
capability: newDBCapability(),
|
||||||
|
cache: c,
|
||||||
|
cacheTTL: o.CacheTTL,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -86,6 +97,9 @@ func (ks *DatabaseKeyStore) reconnectDB() error {
|
|||||||
ks.dbMu.Lock()
|
ks.dbMu.Lock()
|
||||||
ks.db = newDB
|
ks.db = newDB
|
||||||
ks.dbMu.Unlock()
|
ks.dbMu.Unlock()
|
||||||
|
if ks.capability != nil {
|
||||||
|
ks.capability.reset()
|
||||||
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -99,6 +113,14 @@ func (ks *DatabaseKeyStore) CreateKey(ctx context.Context, req CreateKeyRequest)
|
|||||||
rawKey := base64.RawURLEncoding.EncodeToString(rawBytes)
|
rawKey := base64.RawURLEncoding.EncodeToString(rawBytes)
|
||||||
hash := hashSHA256Hex(rawKey)
|
hash := hashSHA256Hex(rawKey)
|
||||||
|
|
||||||
|
if !ks.capability.ShouldUseProcedure(ctx, ks.queryMode, ks.getDB(), ks.sqlNames.CreateKey) {
|
||||||
|
key, err := ks.createKeyDirect(ctx, req, hash)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return &CreateKeyResponse{Key: *key, RawKey: rawKey}, nil
|
||||||
|
}
|
||||||
|
|
||||||
type createRequest struct {
|
type createRequest struct {
|
||||||
UserID int `json:"user_id"`
|
UserID int `json:"user_id"`
|
||||||
KeyType KeyType `json:"key_type"`
|
KeyType KeyType `json:"key_type"`
|
||||||
@@ -145,6 +167,10 @@ func (ks *DatabaseKeyStore) CreateKey(ctx context.Context, req CreateKeyRequest)
|
|||||||
// GetUserKeys returns all active, non-expired keys for the given user.
|
// GetUserKeys returns all active, non-expired keys for the given user.
|
||||||
// Pass an empty KeyType to return all types.
|
// Pass an empty KeyType to return all types.
|
||||||
func (ks *DatabaseKeyStore) GetUserKeys(ctx context.Context, userID int, keyType KeyType) ([]UserKey, error) {
|
func (ks *DatabaseKeyStore) GetUserKeys(ctx context.Context, userID int, keyType KeyType) ([]UserKey, error) {
|
||||||
|
if !ks.capability.ShouldUseProcedure(ctx, ks.queryMode, ks.getDB(), ks.sqlNames.GetUserKeys) {
|
||||||
|
return ks.getUserKeysDirect(ctx, userID, keyType)
|
||||||
|
}
|
||||||
|
|
||||||
var success bool
|
var success bool
|
||||||
var errorMsg sql.NullString
|
var errorMsg sql.NullString
|
||||||
var keysJSON sql.NullString
|
var keysJSON sql.NullString
|
||||||
@@ -173,6 +199,10 @@ func (ks *DatabaseKeyStore) GetUserKeys(ctx context.Context, userID int, keyType
|
|||||||
// The delete procedure returns the key_hash so no separate lookup is needed.
|
// The delete procedure returns the key_hash so no separate lookup is needed.
|
||||||
// Note: cache invalidation is best-effort; a cached entry may persist for up to CacheTTL.
|
// Note: cache invalidation is best-effort; a cached entry may persist for up to CacheTTL.
|
||||||
func (ks *DatabaseKeyStore) DeleteKey(ctx context.Context, userID int, keyID int64) error {
|
func (ks *DatabaseKeyStore) DeleteKey(ctx context.Context, userID int, keyID int64) error {
|
||||||
|
if !ks.capability.ShouldUseProcedure(ctx, ks.queryMode, ks.getDB(), ks.sqlNames.DeleteKey) {
|
||||||
|
return ks.deleteKeyDirect(ctx, userID, keyID)
|
||||||
|
}
|
||||||
|
|
||||||
var success bool
|
var success bool
|
||||||
var errorMsg sql.NullString
|
var errorMsg sql.NullString
|
||||||
var keyHash sql.NullString
|
var keyHash sql.NullString
|
||||||
@@ -207,6 +237,17 @@ func (ks *DatabaseKeyStore) ValidateKey(ctx context.Context, rawKey string, keyT
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if !ks.capability.ShouldUseProcedure(ctx, ks.queryMode, ks.getDB(), ks.sqlNames.ValidateKey) {
|
||||||
|
key, err := ks.validateKeyDirect(ctx, hash, keyType)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if ks.cache != nil {
|
||||||
|
_ = ks.cache.Set(ctx, cacheKey, *key, ks.cacheTTL)
|
||||||
|
}
|
||||||
|
return key, nil
|
||||||
|
}
|
||||||
|
|
||||||
var success bool
|
var success bool
|
||||||
var errorMsg sql.NullString
|
var errorMsg sql.NullString
|
||||||
var keyJSON sql.NullString
|
var keyJSON sql.NullString
|
||||||
|
|||||||
+216
@@ -0,0 +1,216 @@
|
|||||||
|
package security
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"database/sql"
|
||||||
|
"encoding/json"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Direct-mode implementations mirroring the resolvespec_keystore_* stored
|
||||||
|
// procedures in keystore_schema.sql using plain SQL against
|
||||||
|
// TableNames.UserKeys. meta/scopes are stored as JSON-encoded TEXT instead
|
||||||
|
// of Postgres JSONB.
|
||||||
|
|
||||||
|
func (ks *DatabaseKeyStore) createKeyDirect(ctx context.Context, req CreateKeyRequest, keyHash string) (*UserKey, error) {
|
||||||
|
scopesJSON, err := json.Marshal(req.Scopes)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("failed to marshal scopes: %w", err)
|
||||||
|
}
|
||||||
|
var metaJSON []byte
|
||||||
|
if req.Meta != nil {
|
||||||
|
metaJSON, err = json.Marshal(req.Meta)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("failed to marshal meta: %w", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
now := time.Now()
|
||||||
|
var id int64
|
||||||
|
err = ks.runDBOpWithReconnect(func(db *sql.DB) error {
|
||||||
|
query := rewritePlaceholders(db, fmt.Sprintf(
|
||||||
|
`INSERT INTO %s (user_id, key_type, key_hash, name, scopes, meta, expires_at, created_at, is_active) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)`,
|
||||||
|
ks.tableNames.UserKeys))
|
||||||
|
res, err := db.ExecContext(ctx, query, req.UserID, string(req.KeyType), keyHash, req.Name, string(scopesJSON), nullableString(metaJSON), req.ExpiresAt, now, true)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
id, err = res.LastInsertId()
|
||||||
|
return err
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("create key query failed: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return &UserKey{
|
||||||
|
ID: id,
|
||||||
|
UserID: req.UserID,
|
||||||
|
KeyType: req.KeyType,
|
||||||
|
KeyHash: keyHash,
|
||||||
|
Name: req.Name,
|
||||||
|
Scopes: req.Scopes,
|
||||||
|
Meta: req.Meta,
|
||||||
|
ExpiresAt: req.ExpiresAt,
|
||||||
|
CreatedAt: now,
|
||||||
|
IsActive: true,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ks *DatabaseKeyStore) runDBOpWithReconnect(run func(*sql.DB) error) error {
|
||||||
|
db := ks.getDB()
|
||||||
|
if db == nil {
|
||||||
|
return fmt.Errorf("database connection is nil")
|
||||||
|
}
|
||||||
|
err := run(db)
|
||||||
|
if isDBClosed(err) {
|
||||||
|
if reconnErr := ks.reconnectDB(); reconnErr == nil {
|
||||||
|
err = run(ks.getDB())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ks *DatabaseKeyStore) getUserKeysDirect(ctx context.Context, userID int, keyType KeyType) ([]UserKey, error) {
|
||||||
|
keys := []UserKey{}
|
||||||
|
err := ks.runDBOpWithReconnect(func(db *sql.DB) error {
|
||||||
|
var query string
|
||||||
|
var args []any
|
||||||
|
if keyType == "" {
|
||||||
|
query = rewritePlaceholders(db, fmt.Sprintf(
|
||||||
|
`SELECT id, user_id, key_type, name, scopes, meta, expires_at, created_at, last_used_at, is_active
|
||||||
|
FROM %s WHERE user_id = ? AND is_active = ? AND (expires_at IS NULL OR expires_at > ?)`, ks.tableNames.UserKeys))
|
||||||
|
args = []any{userID, true, time.Now()}
|
||||||
|
} else {
|
||||||
|
query = rewritePlaceholders(db, fmt.Sprintf(
|
||||||
|
`SELECT id, user_id, key_type, name, scopes, meta, expires_at, created_at, last_used_at, is_active
|
||||||
|
FROM %s WHERE user_id = ? AND is_active = ? AND (expires_at IS NULL OR expires_at > ?) AND key_type = ?`, ks.tableNames.UserKeys))
|
||||||
|
args = []any{userID, true, time.Now(), string(keyType)}
|
||||||
|
}
|
||||||
|
|
||||||
|
rows, err := db.QueryContext(ctx, query, args...)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
|
||||||
|
for rows.Next() {
|
||||||
|
var k UserKey
|
||||||
|
var kt string
|
||||||
|
var scopesJSON, metaJSON sql.NullString
|
||||||
|
var expiresAt, lastUsedAt sql.NullTime
|
||||||
|
if err := rows.Scan(&k.ID, &k.UserID, &kt, &k.Name, &scopesJSON, &metaJSON, &expiresAt, &k.CreatedAt, &lastUsedAt, &k.IsActive); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
k.KeyType = KeyType(kt)
|
||||||
|
if scopesJSON.Valid && scopesJSON.String != "" {
|
||||||
|
_ = json.Unmarshal([]byte(scopesJSON.String), &k.Scopes)
|
||||||
|
}
|
||||||
|
if metaJSON.Valid && metaJSON.String != "" {
|
||||||
|
_ = json.Unmarshal([]byte(metaJSON.String), &k.Meta)
|
||||||
|
}
|
||||||
|
if expiresAt.Valid {
|
||||||
|
t := expiresAt.Time
|
||||||
|
k.ExpiresAt = &t
|
||||||
|
}
|
||||||
|
if lastUsedAt.Valid {
|
||||||
|
t := lastUsedAt.Time
|
||||||
|
k.LastUsedAt = &t
|
||||||
|
}
|
||||||
|
keys = append(keys, k)
|
||||||
|
}
|
||||||
|
return rows.Err()
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("get user keys query failed: %w", err)
|
||||||
|
}
|
||||||
|
return keys, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ks *DatabaseKeyStore) deleteKeyDirect(ctx context.Context, userID int, keyID int64) error {
|
||||||
|
var keyHash string
|
||||||
|
err := ks.runDBOpWithReconnect(func(db *sql.DB) error {
|
||||||
|
selQuery := rewritePlaceholders(db, fmt.Sprintf(`SELECT key_hash FROM %s WHERE id = ? AND user_id = ? AND is_active = ?`, ks.tableNames.UserKeys))
|
||||||
|
if err := db.QueryRowContext(ctx, selQuery, keyID, userID, true).Scan(&keyHash); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
updQuery := rewritePlaceholders(db, fmt.Sprintf(`UPDATE %s SET is_active = ? WHERE id = ? AND user_id = ? AND is_active = ?`, ks.tableNames.UserKeys))
|
||||||
|
_, err := db.ExecContext(ctx, updQuery, false, keyID, userID, true)
|
||||||
|
return err
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
if errors.Is(err, sql.ErrNoRows) {
|
||||||
|
return errors.New("key not found or already deleted")
|
||||||
|
}
|
||||||
|
return fmt.Errorf("delete key query failed: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if keyHash != "" && ks.cache != nil {
|
||||||
|
_ = ks.cache.Delete(ctx, keystoreCacheKey(keyHash))
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ks *DatabaseKeyStore) validateKeyDirect(ctx context.Context, keyHash string, keyType KeyType) (*UserKey, error) {
|
||||||
|
var k UserKey
|
||||||
|
var kt string
|
||||||
|
var scopesJSON, metaJSON sql.NullString
|
||||||
|
var expiresAt, lastUsedAt sql.NullTime
|
||||||
|
|
||||||
|
err := ks.runDBOpWithReconnect(func(db *sql.DB) error {
|
||||||
|
var query string
|
||||||
|
var args []any
|
||||||
|
if keyType == "" {
|
||||||
|
query = rewritePlaceholders(db, fmt.Sprintf(
|
||||||
|
`SELECT id, user_id, key_type, name, scopes, meta, expires_at, created_at, is_active
|
||||||
|
FROM %s WHERE key_hash = ? AND is_active = ? AND (expires_at IS NULL OR expires_at > ?)`, ks.tableNames.UserKeys))
|
||||||
|
args = []any{keyHash, true, time.Now()}
|
||||||
|
} else {
|
||||||
|
query = rewritePlaceholders(db, fmt.Sprintf(
|
||||||
|
`SELECT id, user_id, key_type, name, scopes, meta, expires_at, created_at, is_active
|
||||||
|
FROM %s WHERE key_hash = ? AND is_active = ? AND (expires_at IS NULL OR expires_at > ?) AND key_type = ?`, ks.tableNames.UserKeys))
|
||||||
|
args = []any{keyHash, true, time.Now(), string(keyType)}
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := db.QueryRowContext(ctx, query, args...).Scan(&k.ID, &k.UserID, &kt, &k.Name, &scopesJSON, &metaJSON, &expiresAt, &k.CreatedAt, &k.IsActive); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
now := time.Now()
|
||||||
|
updQuery := rewritePlaceholders(db, fmt.Sprintf(`UPDATE %s SET last_used_at = ? WHERE id = ?`, ks.tableNames.UserKeys))
|
||||||
|
_, err := db.ExecContext(ctx, updQuery, now, k.ID)
|
||||||
|
return err
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
if errors.Is(err, sql.ErrNoRows) {
|
||||||
|
return nil, errors.New("invalid or expired key")
|
||||||
|
}
|
||||||
|
return nil, fmt.Errorf("validate key query failed: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
k.KeyType = KeyType(kt)
|
||||||
|
k.KeyHash = keyHash
|
||||||
|
if scopesJSON.Valid && scopesJSON.String != "" {
|
||||||
|
_ = json.Unmarshal([]byte(scopesJSON.String), &k.Scopes)
|
||||||
|
}
|
||||||
|
if metaJSON.Valid && metaJSON.String != "" {
|
||||||
|
_ = json.Unmarshal([]byte(metaJSON.String), &k.Meta)
|
||||||
|
}
|
||||||
|
if expiresAt.Valid {
|
||||||
|
t := expiresAt.Time
|
||||||
|
k.ExpiresAt = &t
|
||||||
|
}
|
||||||
|
_ = lastUsedAt
|
||||||
|
now := time.Now()
|
||||||
|
k.LastUsedAt = &now
|
||||||
|
|
||||||
|
return &k, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func nullableString(b []byte) any {
|
||||||
|
if b == nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
return string(b)
|
||||||
|
}
|
||||||
+44
@@ -0,0 +1,44 @@
|
|||||||
|
package security
|
||||||
|
|
||||||
|
import "fmt"
|
||||||
|
|
||||||
|
// KeyStoreTableNames holds the configurable table name used by DatabaseKeyStore
|
||||||
|
// in Direct mode. Use DefaultKeyStoreTableNames() for defaults and
|
||||||
|
// MergeKeyStoreTableNames() for partial overrides.
|
||||||
|
type KeyStoreTableNames struct {
|
||||||
|
UserKeys string // default: "user_keys"
|
||||||
|
}
|
||||||
|
|
||||||
|
// DefaultKeyStoreTableNames returns a KeyStoreTableNames with default table names.
|
||||||
|
func DefaultKeyStoreTableNames() *KeyStoreTableNames {
|
||||||
|
return &KeyStoreTableNames{
|
||||||
|
UserKeys: "user_keys",
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MergeKeyStoreTableNames returns a copy of base with any non-empty fields from override applied.
|
||||||
|
// If override is nil, a copy of base is returned.
|
||||||
|
func MergeKeyStoreTableNames(base, override *KeyStoreTableNames) *KeyStoreTableNames {
|
||||||
|
if override == nil {
|
||||||
|
copied := *base
|
||||||
|
return &copied
|
||||||
|
}
|
||||||
|
merged := *base
|
||||||
|
if override.UserKeys != "" {
|
||||||
|
merged.UserKeys = override.UserKeys
|
||||||
|
}
|
||||||
|
return &merged
|
||||||
|
}
|
||||||
|
|
||||||
|
// ValidateKeyStoreTableNames checks that all non-empty table names are valid SQL identifiers.
|
||||||
|
func ValidateKeyStoreTableNames(names *KeyStoreTableNames) error {
|
||||||
|
if names.UserKeys != "" && !validSQLIdentifier.MatchString(names.UserKeys) {
|
||||||
|
return fmt.Errorf("KeyStoreTableNames.UserKeys contains invalid characters: %q", names.UserKeys)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// resolveKeyStoreTableNames merges an optional override with defaults.
|
||||||
|
func resolveKeyStoreTableNames(override *KeyStoreTableNames) *KeyStoreTableNames {
|
||||||
|
return MergeKeyStoreTableNames(DefaultKeyStoreTableNames(), override)
|
||||||
|
}
|
||||||
+15
-83
@@ -226,6 +226,10 @@ func (a *DatabaseAuthenticator) getOAuth2Provider(providerName string) (*OAuth2P
|
|||||||
|
|
||||||
// oauth2GetOrCreateUser finds or creates a user based on OAuth2 info using stored procedure
|
// oauth2GetOrCreateUser finds or creates a user based on OAuth2 info using stored procedure
|
||||||
func (a *DatabaseAuthenticator) oauth2GetOrCreateUser(ctx context.Context, userCtx *UserContext, providerName string) (int, error) {
|
func (a *DatabaseAuthenticator) oauth2GetOrCreateUser(ctx context.Context, userCtx *UserContext, providerName string) (int, error) {
|
||||||
|
if !a.capability.ShouldUseProcedure(ctx, a.queryMode, a.getDB(), a.sqlNames.OAuthGetOrCreateUser) {
|
||||||
|
return a.oauth2GetOrCreateUserDirect(ctx, userCtx, providerName)
|
||||||
|
}
|
||||||
|
|
||||||
userData := map[string]interface{}{
|
userData := map[string]interface{}{
|
||||||
"username": userCtx.UserName,
|
"username": userCtx.UserName,
|
||||||
"email": userCtx.Email,
|
"email": userCtx.Email,
|
||||||
@@ -269,6 +273,10 @@ func (a *DatabaseAuthenticator) oauth2GetOrCreateUser(ctx context.Context, userC
|
|||||||
|
|
||||||
// oauth2CreateSession creates a new OAuth2 session using stored procedure
|
// oauth2CreateSession creates a new OAuth2 session using stored procedure
|
||||||
func (a *DatabaseAuthenticator) oauth2CreateSession(ctx context.Context, sessionToken string, userID int, token *oauth2.Token, expiresAt time.Time, providerName string) error {
|
func (a *DatabaseAuthenticator) oauth2CreateSession(ctx context.Context, sessionToken string, userID int, token *oauth2.Token, expiresAt time.Time, providerName string) error {
|
||||||
|
if !a.capability.ShouldUseProcedure(ctx, a.queryMode, a.getDB(), a.sqlNames.OAuthCreateSession) {
|
||||||
|
return a.oauth2CreateSessionDirect(ctx, sessionToken, userID, token, expiresAt, providerName)
|
||||||
|
}
|
||||||
|
|
||||||
sessionData := map[string]interface{}{
|
sessionData := map[string]interface{}{
|
||||||
"session_token": sessionToken,
|
"session_token": sessionToken,
|
||||||
"user_id": userID,
|
"user_id": userID,
|
||||||
@@ -381,35 +389,9 @@ func (a *DatabaseAuthenticator) OAuth2RefreshToken(ctx context.Context, refreshT
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Get session by refresh token from database
|
// Get session by refresh token from database
|
||||||
var success bool
|
session, err := a.oauthGetByRefreshToken(ctx, refreshToken)
|
||||||
var errMsg *string
|
|
||||||
var sessionData []byte
|
|
||||||
|
|
||||||
err = a.getDB().QueryRowContext(ctx, fmt.Sprintf(`
|
|
||||||
SELECT p_success, p_error, p_data::text
|
|
||||||
FROM %s($1)
|
|
||||||
`, a.sqlNames.OAuthGetRefreshToken), refreshToken).Scan(&success, &errMsg, &sessionData)
|
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("failed to get session by refresh token: %w", err)
|
return nil, err
|
||||||
}
|
|
||||||
|
|
||||||
if !success {
|
|
||||||
if errMsg != nil {
|
|
||||||
return nil, fmt.Errorf("%s", *errMsg)
|
|
||||||
}
|
|
||||||
return nil, fmt.Errorf("invalid or expired refresh token")
|
|
||||||
}
|
|
||||||
|
|
||||||
// Parse session data
|
|
||||||
var session struct {
|
|
||||||
UserID int `json:"user_id"`
|
|
||||||
AccessToken string `json:"access_token"`
|
|
||||||
TokenType string `json:"token_type"`
|
|
||||||
Expiry time.Time `json:"expiry"`
|
|
||||||
}
|
|
||||||
if err := json.Unmarshal(sessionData, &session); err != nil {
|
|
||||||
return nil, fmt.Errorf("failed to parse session data: %w", err)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Create oauth2.Token from stored data
|
// Create oauth2.Token from stored data
|
||||||
@@ -434,64 +416,14 @@ func (a *DatabaseAuthenticator) OAuth2RefreshToken(ctx context.Context, refreshT
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Update session in database with new tokens
|
// Update session in database with new tokens
|
||||||
updateData := map[string]interface{}{
|
if err := a.oauthUpdateRefreshTokenRecord(ctx, session.UserID, refreshToken, newSessionToken, newToken.AccessToken, newToken.RefreshToken, newToken.Expiry); err != nil {
|
||||||
"user_id": session.UserID,
|
return nil, err
|
||||||
"old_refresh_token": refreshToken,
|
|
||||||
"new_session_token": newSessionToken,
|
|
||||||
"new_access_token": newToken.AccessToken,
|
|
||||||
"new_refresh_token": newToken.RefreshToken,
|
|
||||||
"expires_at": newToken.Expiry,
|
|
||||||
}
|
|
||||||
|
|
||||||
updateJSON, err := json.Marshal(updateData)
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("failed to marshal update data: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
var updateSuccess bool
|
|
||||||
var updateErrMsg *string
|
|
||||||
|
|
||||||
err = a.getDB().QueryRowContext(ctx, fmt.Sprintf(`
|
|
||||||
SELECT p_success, p_error
|
|
||||||
FROM %s($1::jsonb)
|
|
||||||
`, a.sqlNames.OAuthUpdateRefreshToken), updateJSON).Scan(&updateSuccess, &updateErrMsg)
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("failed to update session: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if !updateSuccess {
|
|
||||||
if updateErrMsg != nil {
|
|
||||||
return nil, fmt.Errorf("%s", *updateErrMsg)
|
|
||||||
}
|
|
||||||
return nil, fmt.Errorf("failed to update session")
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Get user data
|
// Get user data
|
||||||
var userSuccess bool
|
userCtx, err := a.oauthGetUserByID(ctx, session.UserID)
|
||||||
var userErrMsg *string
|
|
||||||
var userData []byte
|
|
||||||
|
|
||||||
err = a.getDB().QueryRowContext(ctx, fmt.Sprintf(`
|
|
||||||
SELECT p_success, p_error, p_data::text
|
|
||||||
FROM %s($1)
|
|
||||||
`, a.sqlNames.OAuthGetUser), session.UserID).Scan(&userSuccess, &userErrMsg, &userData)
|
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("failed to get user data: %w", err)
|
return nil, err
|
||||||
}
|
|
||||||
|
|
||||||
if !userSuccess {
|
|
||||||
if userErrMsg != nil {
|
|
||||||
return nil, fmt.Errorf("%s", *userErrMsg)
|
|
||||||
}
|
|
||||||
return nil, fmt.Errorf("failed to get user data")
|
|
||||||
}
|
|
||||||
|
|
||||||
// Parse user context
|
|
||||||
var userCtx UserContext
|
|
||||||
if err := json.Unmarshal(userData, &userCtx); err != nil {
|
|
||||||
return nil, fmt.Errorf("failed to parse user context: %w", err)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
userCtx.SessionID = newSessionToken
|
userCtx.SessionID = newSessionToken
|
||||||
@@ -499,7 +431,7 @@ func (a *DatabaseAuthenticator) OAuth2RefreshToken(ctx context.Context, refreshT
|
|||||||
return &LoginResponse{
|
return &LoginResponse{
|
||||||
Token: newSessionToken,
|
Token: newSessionToken,
|
||||||
RefreshToken: newToken.RefreshToken,
|
RefreshToken: newToken.RefreshToken,
|
||||||
User: &userCtx,
|
User: userCtx,
|
||||||
ExpiresIn: int64(time.Until(newToken.Expiry).Seconds()),
|
ExpiresIn: int64(time.Until(newToken.Expiry).Seconds()),
|
||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|||||||
+242
@@ -0,0 +1,242 @@
|
|||||||
|
package security
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"database/sql"
|
||||||
|
"encoding/json"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"golang.org/x/oauth2"
|
||||||
|
)
|
||||||
|
|
||||||
|
// oauthRefreshSession is the session data needed to refresh an OAuth2 token,
|
||||||
|
// shared by both the stored-procedure and Direct-mode code paths.
|
||||||
|
type oauthRefreshSession struct {
|
||||||
|
UserID int `json:"user_id"`
|
||||||
|
AccessToken string `json:"access_token"`
|
||||||
|
TokenType string `json:"token_type"`
|
||||||
|
Expiry time.Time `json:"expiry"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// oauth2GetOrCreateUserDirect mirrors resolvespec_oauth_getorcreateuser.
|
||||||
|
func (a *DatabaseAuthenticator) oauth2GetOrCreateUserDirect(ctx context.Context, userCtx *UserContext, providerName string) (int, error) {
|
||||||
|
rolesStr := strings.Join(userCtx.Roles, ",")
|
||||||
|
var userID int
|
||||||
|
|
||||||
|
err := a.runDBOpWithReconnect(func(db *sql.DB) error {
|
||||||
|
query := rewritePlaceholders(db, fmt.Sprintf(`SELECT id FROM %s WHERE email = ?`, a.tableNames.Users))
|
||||||
|
err := db.QueryRowContext(ctx, query, userCtx.Email).Scan(&userID)
|
||||||
|
if err == nil {
|
||||||
|
now := time.Now()
|
||||||
|
updQuery := rewritePlaceholders(db, fmt.Sprintf(
|
||||||
|
`UPDATE %s SET last_login_at = ?, updated_at = ?, remote_id = COALESCE(remote_id, ?), auth_provider = COALESCE(auth_provider, ?) WHERE id = ?`,
|
||||||
|
a.tableNames.Users))
|
||||||
|
_, err := db.ExecContext(ctx, updQuery, now, now, userCtx.RemoteID, providerName, userID)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if !errors.Is(err, sql.ErrNoRows) {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
now := time.Now()
|
||||||
|
insQuery := rewritePlaceholders(db, fmt.Sprintf(
|
||||||
|
`INSERT INTO %s (username, email, password, user_level, roles, is_active, created_at, updated_at, last_login_at, remote_id, auth_provider) VALUES (?, ?, NULL, ?, ?, ?, ?, ?, ?, ?, ?)`,
|
||||||
|
a.tableNames.Users))
|
||||||
|
res, err := db.ExecContext(ctx, insQuery, userCtx.UserName, userCtx.Email, userCtx.UserLevel, rolesStr, true, now, now, now, userCtx.RemoteID, providerName)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
id, err := res.LastInsertId()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
userID = int(id)
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return 0, fmt.Errorf("failed to get or create user: %w", err)
|
||||||
|
}
|
||||||
|
return userID, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// oauth2CreateSessionDirect mirrors resolvespec_oauth_createsession (insert-or-update by session_token).
|
||||||
|
func (a *DatabaseAuthenticator) oauth2CreateSessionDirect(ctx context.Context, sessionToken string, userID int, token *oauth2.Token, expiresAt time.Time, providerName string) error {
|
||||||
|
return a.runDBOpWithReconnect(func(db *sql.DB) error {
|
||||||
|
var exists int
|
||||||
|
checkQuery := rewritePlaceholders(db, fmt.Sprintf(`SELECT 1 FROM %s WHERE session_token = ?`, a.tableNames.UserSessions))
|
||||||
|
err := db.QueryRowContext(ctx, checkQuery, sessionToken).Scan(&exists)
|
||||||
|
now := time.Now()
|
||||||
|
if err == nil {
|
||||||
|
updQuery := rewritePlaceholders(db, fmt.Sprintf(
|
||||||
|
`UPDATE %s SET access_token = ?, refresh_token = ?, token_type = ?, expires_at = ?, last_activity_at = ? WHERE session_token = ?`,
|
||||||
|
a.tableNames.UserSessions))
|
||||||
|
_, err := db.ExecContext(ctx, updQuery, token.AccessToken, token.RefreshToken, token.TokenType, expiresAt, now, sessionToken)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if !errors.Is(err, sql.ErrNoRows) {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
insQuery := rewritePlaceholders(db, fmt.Sprintf(
|
||||||
|
`INSERT INTO %s (session_token, user_id, expires_at, created_at, last_activity_at, access_token, refresh_token, token_type, auth_provider) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)`,
|
||||||
|
a.tableNames.UserSessions))
|
||||||
|
_, err = db.ExecContext(ctx, insQuery, sessionToken, userID, expiresAt, now, now, token.AccessToken, token.RefreshToken, token.TokenType, providerName)
|
||||||
|
return err
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// oauthGetByRefreshToken retrieves the session for a refresh token, dispatching between
|
||||||
|
// the resolvespec_oauth_getrefreshtoken stored procedure and Direct-mode SQL.
|
||||||
|
func (a *DatabaseAuthenticator) oauthGetByRefreshToken(ctx context.Context, refreshToken string) (*oauthRefreshSession, error) {
|
||||||
|
if !a.capability.ShouldUseProcedure(ctx, a.queryMode, a.getDB(), a.sqlNames.OAuthGetRefreshToken) {
|
||||||
|
var session oauthRefreshSession
|
||||||
|
err := a.runDBOpWithReconnect(func(db *sql.DB) error {
|
||||||
|
query := rewritePlaceholders(db, fmt.Sprintf(
|
||||||
|
`SELECT user_id, access_token, token_type, expires_at FROM %s WHERE refresh_token = ? AND expires_at > ?`,
|
||||||
|
a.tableNames.UserSessions))
|
||||||
|
return db.QueryRowContext(ctx, query, refreshToken, time.Now()).Scan(&session.UserID, &session.AccessToken, &session.TokenType, &session.Expiry)
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
if errors.Is(err, sql.ErrNoRows) {
|
||||||
|
return nil, fmt.Errorf("refresh token not found or expired")
|
||||||
|
}
|
||||||
|
return nil, fmt.Errorf("failed to get session by refresh token: %w", err)
|
||||||
|
}
|
||||||
|
return &session, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
var success bool
|
||||||
|
var errMsg *string
|
||||||
|
var sessionData []byte
|
||||||
|
|
||||||
|
err := a.getDB().QueryRowContext(ctx, fmt.Sprintf(`
|
||||||
|
SELECT p_success, p_error, p_data::text
|
||||||
|
FROM %s($1)
|
||||||
|
`, a.sqlNames.OAuthGetRefreshToken), refreshToken).Scan(&success, &errMsg, &sessionData)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("failed to get session by refresh token: %w", err)
|
||||||
|
}
|
||||||
|
if !success {
|
||||||
|
if errMsg != nil {
|
||||||
|
return nil, fmt.Errorf("%s", *errMsg)
|
||||||
|
}
|
||||||
|
return nil, fmt.Errorf("invalid or expired refresh token")
|
||||||
|
}
|
||||||
|
|
||||||
|
var session oauthRefreshSession
|
||||||
|
if err := json.Unmarshal(sessionData, &session); err != nil {
|
||||||
|
return nil, fmt.Errorf("failed to parse session data: %w", err)
|
||||||
|
}
|
||||||
|
return &session, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// oauthUpdateRefreshTokenRecord updates a session with new tokens, dispatching between
|
||||||
|
// the resolvespec_oauth_updaterefreshtoken stored procedure and Direct-mode SQL.
|
||||||
|
func (a *DatabaseAuthenticator) oauthUpdateRefreshTokenRecord(ctx context.Context, userID int, oldRefreshToken, newSessionToken, newAccessToken, newRefreshToken string, expiresAt time.Time) error {
|
||||||
|
if !a.capability.ShouldUseProcedure(ctx, a.queryMode, a.getDB(), a.sqlNames.OAuthUpdateRefreshToken) {
|
||||||
|
var rows int64
|
||||||
|
err := a.runDBOpWithReconnect(func(db *sql.DB) error {
|
||||||
|
query := rewritePlaceholders(db, fmt.Sprintf(
|
||||||
|
`UPDATE %s SET session_token = ?, access_token = ?, refresh_token = ?, expires_at = ?, last_activity_at = ? WHERE user_id = ? AND refresh_token = ?`,
|
||||||
|
a.tableNames.UserSessions))
|
||||||
|
res, err := db.ExecContext(ctx, query, newSessionToken, newAccessToken, newRefreshToken, expiresAt, time.Now(), userID, oldRefreshToken)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
rows, err = res.RowsAffected()
|
||||||
|
return err
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to update session: %w", err)
|
||||||
|
}
|
||||||
|
if rows == 0 {
|
||||||
|
return fmt.Errorf("session not found")
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
updateData := map[string]interface{}{
|
||||||
|
"user_id": userID,
|
||||||
|
"old_refresh_token": oldRefreshToken,
|
||||||
|
"new_session_token": newSessionToken,
|
||||||
|
"new_access_token": newAccessToken,
|
||||||
|
"new_refresh_token": newRefreshToken,
|
||||||
|
"expires_at": expiresAt,
|
||||||
|
}
|
||||||
|
updateJSON, err := json.Marshal(updateData)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to marshal update data: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var updateSuccess bool
|
||||||
|
var updateErrMsg *string
|
||||||
|
err = a.getDB().QueryRowContext(ctx, fmt.Sprintf(`
|
||||||
|
SELECT p_success, p_error
|
||||||
|
FROM %s($1::jsonb)
|
||||||
|
`, a.sqlNames.OAuthUpdateRefreshToken), updateJSON).Scan(&updateSuccess, &updateErrMsg)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to update session: %w", err)
|
||||||
|
}
|
||||||
|
if !updateSuccess {
|
||||||
|
if updateErrMsg != nil {
|
||||||
|
return fmt.Errorf("%s", *updateErrMsg)
|
||||||
|
}
|
||||||
|
return fmt.Errorf("failed to update session")
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// oauthGetUserByID retrieves user data by ID, dispatching between the
|
||||||
|
// resolvespec_oauth_getuser stored procedure and Direct-mode SQL.
|
||||||
|
func (a *DatabaseAuthenticator) oauthGetUserByID(ctx context.Context, userID int) (*UserContext, error) {
|
||||||
|
if !a.capability.ShouldUseProcedure(ctx, a.queryMode, a.getDB(), a.sqlNames.OAuthGetUser) {
|
||||||
|
var username, email, roles, programUserTable sql.NullString
|
||||||
|
var userLevel, programUserID sql.NullInt64
|
||||||
|
err := a.runDBOpWithReconnect(func(db *sql.DB) error {
|
||||||
|
query := rewritePlaceholders(db, fmt.Sprintf(
|
||||||
|
`SELECT username, email, user_level, roles, program_user_id, program_user_table FROM %s WHERE id = ? AND is_active = ?`,
|
||||||
|
a.tableNames.Users))
|
||||||
|
return db.QueryRowContext(ctx, query, userID, true).Scan(&username, &email, &userLevel, &roles, &programUserID, &programUserTable)
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
if errors.Is(err, sql.ErrNoRows) {
|
||||||
|
return nil, fmt.Errorf("user not found")
|
||||||
|
}
|
||||||
|
return nil, fmt.Errorf("failed to get user data: %w", err)
|
||||||
|
}
|
||||||
|
return &UserContext{
|
||||||
|
UserID: userID,
|
||||||
|
UserName: username.String,
|
||||||
|
Email: email.String,
|
||||||
|
UserLevel: int(userLevel.Int64),
|
||||||
|
Roles: parseRoles(roles.String),
|
||||||
|
ProgramUserID: int(programUserID.Int64),
|
||||||
|
ProgramUserTable: programUserTable.String,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
var userSuccess bool
|
||||||
|
var userErrMsg *string
|
||||||
|
var userData []byte
|
||||||
|
err := a.getDB().QueryRowContext(ctx, fmt.Sprintf(`
|
||||||
|
SELECT p_success, p_error, p_data::text
|
||||||
|
FROM %s($1)
|
||||||
|
`, a.sqlNames.OAuthGetUser), userID).Scan(&userSuccess, &userErrMsg, &userData)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("failed to get user data: %w", err)
|
||||||
|
}
|
||||||
|
if !userSuccess {
|
||||||
|
if userErrMsg != nil {
|
||||||
|
return nil, fmt.Errorf("%s", *userErrMsg)
|
||||||
|
}
|
||||||
|
return nil, fmt.Errorf("failed to get user data")
|
||||||
|
}
|
||||||
|
var userCtx UserContext
|
||||||
|
if err := json.Unmarshal(userData, &userCtx); err != nil {
|
||||||
|
return nil, fmt.Errorf("failed to parse user context: %w", err)
|
||||||
|
}
|
||||||
|
return &userCtx, nil
|
||||||
|
}
|
||||||
+24
@@ -44,6 +44,10 @@ type OAuthTokenInfo struct {
|
|||||||
|
|
||||||
// OAuthRegisterClient persists an OAuth2 client registration.
|
// OAuthRegisterClient persists an OAuth2 client registration.
|
||||||
func (a *DatabaseAuthenticator) OAuthRegisterClient(ctx context.Context, client *OAuthServerClient) (*OAuthServerClient, error) {
|
func (a *DatabaseAuthenticator) OAuthRegisterClient(ctx context.Context, client *OAuthServerClient) (*OAuthServerClient, error) {
|
||||||
|
if !a.capability.ShouldUseProcedure(ctx, a.queryMode, a.getDB(), a.sqlNames.OAuthRegisterClient) {
|
||||||
|
return a.oauthRegisterClientDirect(ctx, client)
|
||||||
|
}
|
||||||
|
|
||||||
input, err := json.Marshal(client)
|
input, err := json.Marshal(client)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("failed to marshal client: %w", err)
|
return nil, fmt.Errorf("failed to marshal client: %w", err)
|
||||||
@@ -76,6 +80,10 @@ func (a *DatabaseAuthenticator) OAuthRegisterClient(ctx context.Context, client
|
|||||||
|
|
||||||
// OAuthGetClient retrieves a registered client by ID.
|
// OAuthGetClient retrieves a registered client by ID.
|
||||||
func (a *DatabaseAuthenticator) OAuthGetClient(ctx context.Context, clientID string) (*OAuthServerClient, error) {
|
func (a *DatabaseAuthenticator) OAuthGetClient(ctx context.Context, clientID string) (*OAuthServerClient, error) {
|
||||||
|
if !a.capability.ShouldUseProcedure(ctx, a.queryMode, a.getDB(), a.sqlNames.OAuthGetClient) {
|
||||||
|
return a.oauthGetClientDirect(ctx, clientID)
|
||||||
|
}
|
||||||
|
|
||||||
var success bool
|
var success bool
|
||||||
var errMsg *string
|
var errMsg *string
|
||||||
var data []byte
|
var data []byte
|
||||||
@@ -103,6 +111,10 @@ func (a *DatabaseAuthenticator) OAuthGetClient(ctx context.Context, clientID str
|
|||||||
|
|
||||||
// OAuthSaveCode persists an authorization code.
|
// OAuthSaveCode persists an authorization code.
|
||||||
func (a *DatabaseAuthenticator) OAuthSaveCode(ctx context.Context, code *OAuthCode) error {
|
func (a *DatabaseAuthenticator) OAuthSaveCode(ctx context.Context, code *OAuthCode) error {
|
||||||
|
if !a.capability.ShouldUseProcedure(ctx, a.queryMode, a.getDB(), a.sqlNames.OAuthSaveCode) {
|
||||||
|
return a.oauthSaveCodeDirect(ctx, code)
|
||||||
|
}
|
||||||
|
|
||||||
input, err := json.Marshal(code)
|
input, err := json.Marshal(code)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("failed to marshal code: %w", err)
|
return fmt.Errorf("failed to marshal code: %w", err)
|
||||||
@@ -129,6 +141,10 @@ func (a *DatabaseAuthenticator) OAuthSaveCode(ctx context.Context, code *OAuthCo
|
|||||||
|
|
||||||
// OAuthExchangeCode retrieves and deletes an authorization code (single use).
|
// OAuthExchangeCode retrieves and deletes an authorization code (single use).
|
||||||
func (a *DatabaseAuthenticator) OAuthExchangeCode(ctx context.Context, code string) (*OAuthCode, error) {
|
func (a *DatabaseAuthenticator) OAuthExchangeCode(ctx context.Context, code string) (*OAuthCode, error) {
|
||||||
|
if !a.capability.ShouldUseProcedure(ctx, a.queryMode, a.getDB(), a.sqlNames.OAuthExchangeCode) {
|
||||||
|
return a.oauthExchangeCodeDirect(ctx, code)
|
||||||
|
}
|
||||||
|
|
||||||
var success bool
|
var success bool
|
||||||
var errMsg *string
|
var errMsg *string
|
||||||
var data []byte
|
var data []byte
|
||||||
@@ -157,6 +173,10 @@ func (a *DatabaseAuthenticator) OAuthExchangeCode(ctx context.Context, code stri
|
|||||||
|
|
||||||
// OAuthIntrospectToken validates a token and returns its metadata (RFC 7662).
|
// OAuthIntrospectToken validates a token and returns its metadata (RFC 7662).
|
||||||
func (a *DatabaseAuthenticator) OAuthIntrospectToken(ctx context.Context, token string) (*OAuthTokenInfo, error) {
|
func (a *DatabaseAuthenticator) OAuthIntrospectToken(ctx context.Context, token string) (*OAuthTokenInfo, error) {
|
||||||
|
if !a.capability.ShouldUseProcedure(ctx, a.queryMode, a.getDB(), a.sqlNames.OAuthIntrospect) {
|
||||||
|
return a.oauthIntrospectTokenDirect(ctx, token)
|
||||||
|
}
|
||||||
|
|
||||||
var success bool
|
var success bool
|
||||||
var errMsg *string
|
var errMsg *string
|
||||||
var data []byte
|
var data []byte
|
||||||
@@ -184,6 +204,10 @@ func (a *DatabaseAuthenticator) OAuthIntrospectToken(ctx context.Context, token
|
|||||||
|
|
||||||
// OAuthRevokeToken revokes a token by deleting the session (RFC 7009).
|
// OAuthRevokeToken revokes a token by deleting the session (RFC 7009).
|
||||||
func (a *DatabaseAuthenticator) OAuthRevokeToken(ctx context.Context, token string) error {
|
func (a *DatabaseAuthenticator) OAuthRevokeToken(ctx context.Context, token string) error {
|
||||||
|
if !a.capability.ShouldUseProcedure(ctx, a.queryMode, a.getDB(), a.sqlNames.OAuthRevoke) {
|
||||||
|
return a.oauthRevokeTokenDirect(ctx, token)
|
||||||
|
}
|
||||||
|
|
||||||
var success bool
|
var success bool
|
||||||
var errMsg *string
|
var errMsg *string
|
||||||
|
|
||||||
|
|||||||
+188
@@ -0,0 +1,188 @@
|
|||||||
|
package security
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"database/sql"
|
||||||
|
"encoding/json"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Direct-mode implementations mirroring the OAuth2 server stored procedures
|
||||||
|
// (resolvespec_oauth_register_client, etc.) in database_schema.sql, using
|
||||||
|
// plain SQL against TableNames.OAuthClients / TableNames.OAuthCodes.
|
||||||
|
// Array columns (redirect_uris, grant_types, allowed_scopes, scopes) are
|
||||||
|
// JSON-encoded TEXT instead of native Postgres arrays.
|
||||||
|
|
||||||
|
func (a *DatabaseAuthenticator) oauthRegisterClientDirect(ctx context.Context, client *OAuthServerClient) (*OAuthServerClient, error) {
|
||||||
|
grantTypes := client.GrantTypes
|
||||||
|
if len(grantTypes) == 0 {
|
||||||
|
grantTypes = []string{"authorization_code"}
|
||||||
|
}
|
||||||
|
allowedScopes := client.AllowedScopes
|
||||||
|
if len(allowedScopes) == 0 {
|
||||||
|
allowedScopes = []string{"openid", "profile", "email"}
|
||||||
|
}
|
||||||
|
|
||||||
|
redirectURIsJSON, err := json.Marshal(client.RedirectURIs)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("failed to marshal redirect_uris: %w", err)
|
||||||
|
}
|
||||||
|
grantTypesJSON, err := json.Marshal(grantTypes)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("failed to marshal grant_types: %w", err)
|
||||||
|
}
|
||||||
|
allowedScopesJSON, err := json.Marshal(allowedScopes)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("failed to marshal allowed_scopes: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
err = a.runDBOpWithReconnect(func(db *sql.DB) error {
|
||||||
|
query := rewritePlaceholders(db, fmt.Sprintf(
|
||||||
|
`INSERT INTO %s (client_id, redirect_uris, client_name, grant_types, allowed_scopes, is_active, created_at) VALUES (?, ?, ?, ?, ?, ?, ?)`,
|
||||||
|
a.tableNames.OAuthClients))
|
||||||
|
_, err := db.ExecContext(ctx, query, client.ClientID, string(redirectURIsJSON), client.ClientName, string(grantTypesJSON), string(allowedScopesJSON), true, time.Now())
|
||||||
|
return err
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("failed to register client: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return &OAuthServerClient{
|
||||||
|
ClientID: client.ClientID,
|
||||||
|
RedirectURIs: client.RedirectURIs,
|
||||||
|
ClientName: client.ClientName,
|
||||||
|
GrantTypes: grantTypes,
|
||||||
|
AllowedScopes: allowedScopes,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *DatabaseAuthenticator) oauthGetClientDirect(ctx context.Context, clientID string) (*OAuthServerClient, error) {
|
||||||
|
var redirectURIsJSON, grantTypesJSON, allowedScopesJSON sql.NullString
|
||||||
|
var clientName sql.NullString
|
||||||
|
|
||||||
|
err := a.runDBOpWithReconnect(func(db *sql.DB) error {
|
||||||
|
query := rewritePlaceholders(db, fmt.Sprintf(
|
||||||
|
`SELECT redirect_uris, client_name, grant_types, allowed_scopes FROM %s WHERE client_id = ? AND is_active = ?`,
|
||||||
|
a.tableNames.OAuthClients))
|
||||||
|
return db.QueryRowContext(ctx, query, clientID, true).Scan(&redirectURIsJSON, &clientName, &grantTypesJSON, &allowedScopesJSON)
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
if errors.Is(err, sql.ErrNoRows) {
|
||||||
|
return nil, fmt.Errorf("client not found")
|
||||||
|
}
|
||||||
|
return nil, fmt.Errorf("failed to get client: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
result := &OAuthServerClient{ClientID: clientID, ClientName: clientName.String}
|
||||||
|
if redirectURIsJSON.Valid {
|
||||||
|
_ = json.Unmarshal([]byte(redirectURIsJSON.String), &result.RedirectURIs)
|
||||||
|
}
|
||||||
|
if grantTypesJSON.Valid {
|
||||||
|
_ = json.Unmarshal([]byte(grantTypesJSON.String), &result.GrantTypes)
|
||||||
|
}
|
||||||
|
if allowedScopesJSON.Valid {
|
||||||
|
_ = json.Unmarshal([]byte(allowedScopesJSON.String), &result.AllowedScopes)
|
||||||
|
}
|
||||||
|
return result, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *DatabaseAuthenticator) oauthSaveCodeDirect(ctx context.Context, code *OAuthCode) error {
|
||||||
|
scopesJSON, err := json.Marshal(code.Scopes)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to marshal scopes: %w", err)
|
||||||
|
}
|
||||||
|
method := code.CodeChallengeMethod
|
||||||
|
if method == "" {
|
||||||
|
method = "S256"
|
||||||
|
}
|
||||||
|
|
||||||
|
return a.runDBOpWithReconnect(func(db *sql.DB) error {
|
||||||
|
query := rewritePlaceholders(db, fmt.Sprintf(
|
||||||
|
`INSERT INTO %s (code, client_id, redirect_uri, client_state, code_challenge, code_challenge_method, session_token, refresh_token, scopes, expires_at, created_at)
|
||||||
|
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`, a.tableNames.OAuthCodes))
|
||||||
|
_, err := db.ExecContext(ctx, query, code.Code, code.ClientID, code.RedirectURI, code.ClientState, code.CodeChallenge,
|
||||||
|
method, code.SessionToken, code.RefreshToken, string(scopesJSON), code.ExpiresAt, time.Now())
|
||||||
|
return err
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *DatabaseAuthenticator) oauthExchangeCodeDirect(ctx context.Context, code string) (*OAuthCode, error) {
|
||||||
|
var result OAuthCode
|
||||||
|
var clientState, refreshToken sql.NullString
|
||||||
|
var scopesJSON sql.NullString
|
||||||
|
|
||||||
|
err := a.runDBOpWithReconnect(func(db *sql.DB) error {
|
||||||
|
query := rewritePlaceholders(db, fmt.Sprintf(
|
||||||
|
`SELECT client_id, redirect_uri, client_state, code_challenge, code_challenge_method, session_token, refresh_token, scopes
|
||||||
|
FROM %s WHERE code = ? AND expires_at > ?`, a.tableNames.OAuthCodes))
|
||||||
|
err := db.QueryRowContext(ctx, query, code, time.Now()).Scan(
|
||||||
|
&result.ClientID, &result.RedirectURI, &clientState, &result.CodeChallenge, &result.CodeChallengeMethod, &result.SessionToken, &refreshToken, &scopesJSON)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
delQuery := rewritePlaceholders(db, fmt.Sprintf(`DELETE FROM %s WHERE code = ?`, a.tableNames.OAuthCodes))
|
||||||
|
_, err = db.ExecContext(ctx, delQuery, code)
|
||||||
|
return err
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
if errors.Is(err, sql.ErrNoRows) {
|
||||||
|
return nil, fmt.Errorf("invalid or expired code")
|
||||||
|
}
|
||||||
|
return nil, fmt.Errorf("failed to exchange code: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
result.Code = code
|
||||||
|
result.ClientState = clientState.String
|
||||||
|
result.RefreshToken = refreshToken.String
|
||||||
|
if scopesJSON.Valid {
|
||||||
|
_ = json.Unmarshal([]byte(scopesJSON.String), &result.Scopes)
|
||||||
|
}
|
||||||
|
return &result, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *DatabaseAuthenticator) oauthIntrospectTokenDirect(ctx context.Context, token string) (*OAuthTokenInfo, error) {
|
||||||
|
var info OAuthTokenInfo
|
||||||
|
var roles sql.NullString
|
||||||
|
var exp, iat sql.NullTime
|
||||||
|
|
||||||
|
err := a.runDBOpWithReconnect(func(db *sql.DB) error {
|
||||||
|
query := rewritePlaceholders(db, fmt.Sprintf(
|
||||||
|
`SELECT u.id, u.username, u.email, u.user_level, u.roles, s.expires_at, s.created_at
|
||||||
|
FROM %s s JOIN %s u ON u.id = s.user_id
|
||||||
|
WHERE s.session_token = ? AND s.expires_at > ? AND u.is_active = ?`,
|
||||||
|
a.tableNames.UserSessions, a.tableNames.Users))
|
||||||
|
var userID int
|
||||||
|
err := db.QueryRowContext(ctx, query, token, time.Now(), true).Scan(&userID, &info.Username, &info.Email, &info.UserLevel, &roles, &exp, &iat)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
info.Sub = fmt.Sprintf("%d", userID)
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
if errors.Is(err, sql.ErrNoRows) {
|
||||||
|
return &OAuthTokenInfo{Active: false}, nil
|
||||||
|
}
|
||||||
|
return nil, fmt.Errorf("failed to introspect token: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
info.Active = true
|
||||||
|
info.Roles = parseRoles(roles.String)
|
||||||
|
if exp.Valid {
|
||||||
|
info.Exp = exp.Time.Unix()
|
||||||
|
}
|
||||||
|
if iat.Valid {
|
||||||
|
info.Iat = iat.Time.Unix()
|
||||||
|
}
|
||||||
|
return &info, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *DatabaseAuthenticator) oauthRevokeTokenDirect(ctx context.Context, token string) error {
|
||||||
|
return a.runDBOpWithReconnect(func(db *sql.DB) error {
|
||||||
|
query := rewritePlaceholders(db, fmt.Sprintf(`DELETE FROM %s WHERE session_token = ?`, a.tableNames.UserSessions))
|
||||||
|
_, err := db.ExecContext(ctx, query, token)
|
||||||
|
return err
|
||||||
|
})
|
||||||
|
}
|
||||||
Some files were not shown because too many files have changed in this diff Show More
Reference in New Issue
Block a user