root
b16e9f0bd1
Plan 0 (workspaces): add workspaces + workspace_memberships schema, extend JWT/Actor/claims with primary_workspace_id, seed default workspace per tenant, thread workspace_id through tenant monitoring quota. Plan A (desktop skeleton): new Electron app (apps/desktop-client) with main/ preload/renderer, shared Vue component package (packages/ui-shared), and server surface — desktop client registration + token rotation + heartbeat, SSE task event stream, desktop accounts/tasks/content handlers, publish job endpoint, and supporting repositories, services, sqlc queries, and migrations. Hard cutover per plan: remove browser-extension monitoring callback endpoints, stub legacy media API in admin-web, and delete monitoring_callback_handler.go.
118 lines
3.1 KiB
Go
118 lines
3.1 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/geo-platform/tenant-api/internal/tenant/repository/generated"
|
|
)
|
|
|
|
type UserRecord struct {
|
|
ID int64
|
|
Email string
|
|
Phone *string
|
|
PasswordHash string
|
|
Name string
|
|
AvatarURL *string
|
|
Status string
|
|
CreatedAt time.Time
|
|
UpdatedAt time.Time
|
|
}
|
|
|
|
type MembershipRecord struct {
|
|
ID int64
|
|
TenantID int64
|
|
UserID int64
|
|
PrimaryWorkspaceID int64
|
|
TenantRole string
|
|
CreatedAt time.Time
|
|
}
|
|
|
|
type AuthRepository interface {
|
|
GetUserByEmail(ctx context.Context, email string) (*UserRecord, error)
|
|
GetUserByID(ctx context.Context, id int64) (*UserRecord, error)
|
|
GetTenantMembership(ctx context.Context, userID int64) (*MembershipRecord, error)
|
|
GetTenantMembershipByTenantAndUser(ctx context.Context, tenantID, userID int64) (*MembershipRecord, error)
|
|
}
|
|
|
|
type authRepository struct {
|
|
q generated.Querier
|
|
}
|
|
|
|
func NewAuthRepository(db generated.DBTX) AuthRepository {
|
|
return &authRepository{q: newQuerier(db)}
|
|
}
|
|
|
|
func (r *authRepository) GetUserByEmail(ctx context.Context, email string) (*UserRecord, error) {
|
|
row, err := r.q.GetUserByEmail(ctx, email)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &UserRecord{
|
|
ID: row.ID,
|
|
Email: row.Email,
|
|
Phone: nullableText(row.Phone),
|
|
PasswordHash: row.PasswordHash,
|
|
Name: row.Name,
|
|
AvatarURL: nullableText(row.AvatarUrl),
|
|
Status: row.Status,
|
|
CreatedAt: timeFromTimestamp(row.CreatedAt),
|
|
UpdatedAt: timeFromTimestamp(row.UpdatedAt),
|
|
}, nil
|
|
}
|
|
|
|
func (r *authRepository) GetUserByID(ctx context.Context, id int64) (*UserRecord, error) {
|
|
row, err := r.q.GetUserByID(ctx, id)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &UserRecord{
|
|
ID: row.ID,
|
|
Email: row.Email,
|
|
Phone: nullableText(row.Phone),
|
|
PasswordHash: row.PasswordHash,
|
|
Name: row.Name,
|
|
AvatarURL: nullableText(row.AvatarUrl),
|
|
Status: row.Status,
|
|
CreatedAt: timeFromTimestamp(row.CreatedAt),
|
|
UpdatedAt: timeFromTimestamp(row.UpdatedAt),
|
|
}, nil
|
|
}
|
|
|
|
func (r *authRepository) GetTenantMembership(ctx context.Context, userID int64) (*MembershipRecord, error) {
|
|
row, err := r.q.GetTenantMembership(ctx, userID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &MembershipRecord{
|
|
ID: row.ID,
|
|
TenantID: row.TenantID,
|
|
UserID: row.UserID,
|
|
PrimaryWorkspaceID: row.PrimaryWorkspaceID,
|
|
TenantRole: row.TenantRole,
|
|
CreatedAt: timeFromTimestamp(row.CreatedAt),
|
|
}, nil
|
|
}
|
|
|
|
func (r *authRepository) GetTenantMembershipByTenantAndUser(ctx context.Context, tenantID, userID int64) (*MembershipRecord, error) {
|
|
row, err := r.q.GetTenantMembershipByTenantAndUser(ctx, generated.GetTenantMembershipByTenantAndUserParams{
|
|
TenantID: tenantID,
|
|
UserID: userID,
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &MembershipRecord{
|
|
ID: row.ID,
|
|
TenantID: row.TenantID,
|
|
UserID: row.UserID,
|
|
PrimaryWorkspaceID: row.PrimaryWorkspaceID,
|
|
TenantRole: row.TenantRole,
|
|
CreatedAt: timeFromTimestamp(row.CreatedAt),
|
|
}, nil
|
|
}
|