Files
Memoh/internal/bots/service.go
T
Ran 6acdd191c7 Squashed commit of the following:
commit bcdb026ae43e4f95d0b2c4f9bd440a2df9d6b514
Author: Ran <16112591+chen-ran@users.noreply.github.com>
Date:   Thu Feb 12 17:10:32 2026 +0800

    chore: update DEVELOPMENT.md

commit 30281742ef
Merge: ca5c6a1 5b05f13
Author: BBQ <bbq@BBQdeMacBook-Air.local>
Date:   Thu Feb 12 15:49:17 2026 +0800

    merge(github/main): integrate fx dependency injection framework

    Merge upstream fx refactor and adapt all services to use go.uber.org/fx
    for dependency injection. Resolve conflicts in main.go, server.go,
    and service constructors while preserving our domain model changes.

    - Fix telegram adapter panic on shutdown (double close channel)
    - Fix feishu adapter processing messages after stop
    - Increase directory lookup timeout from 2s to 5s

commit ca5c6a1866
Author: BBQ <bbq@BBQdeMacBook-Air.local>
Date:   Thu Feb 12 15:33:09 2026 +0800

    refactor(core): restructure conversation, channel and message domains

    - Rename chat module to conversation with flow-based architecture
    - Move channelidentities into channel/identities subpackage
    - Add channel/route for routing logic
    - Add message service with event hub
    - Add MCP providers: container, directory, schedule
    - Refactor Feishu/Telegram adapters with directory and stream support
    - Add platform management page and channel badges in web UI
    - Update database schema for conversations, messages and channel routes
    - Add @memoh/shared package for cross-package type definitions

commit 75e2ef0467
Merge: d99ba38 01cb6c8
Author: BBQ <bbq@BBQdeMacBook-Air.local>
Date:   Thu Feb 12 14:45:49 2026 +0800

    merge(github): merge github/main, resolve index.ts URL conflict

    Keep our defensive absolute-URL check in createAuthFetcher.

commit d99ba38b7d
Merge: 860e20f 35ce7d1
Author: BBQ <bbq@BBQdeMacBook-Air.local>
Date:   Thu Feb 12 05:20:18 2026 +0800

    merge(github): merge github/main, keep our code and docs/spec

commit 860e20fe70
Author: BBQ <bbq@BBQdeMacBook-Air.local>
Date:   Wed Feb 11 22:13:27 2026 +0800

    docs(docs): add concepts and style guides for VitePress site

    - Add concepts: identity-and-binding, index (en/zh)
    - Add style: terminology (en/zh)
    - Update index and zh/index
    - Update .vitepress/config.ts

commit a75fdb8040
Author: BBQ <bbq@BBQdeMacBook-Air.local>
Date:   Wed Feb 11 17:37:16 2026 +0800

    refactor(mcp): standardize unified tool gateway on go-sdk

    Split business executors from federation sources and migrate unified tool/federation transports to the official go-sdk for stricter MCP compliance and safer session lifecycle handling. Add targeted regression tests for accept compatibility, initialization retries, pending cleanup, and include updated swagger artifacts.

commit 02b33c8e85
Author: BBQ <bbq@BBQdeMacBook-Air.local>
Date:   Wed Feb 11 15:42:21 2026 +0800

    refactor(core): finalize user-centric identity and policy cleanup

    Unify auth and chat identity semantics around user_id, enforce personal-bot owner-only authorization, and remove legacy compatibility branches in integration tests.

commit 06e8619a37
Author: BBQ <bbq@BBQdeMacBook-Air.local>
Date:   Wed Feb 11 14:47:03 2026 +0800

    refactor(core): migrate channel identity and binding across app

    Align channel identity and bind flow across backend and app-facing layers, including generated swagger artifacts and package lock updates while excluding docs content changes.
2026-02-12 17:13:03 +08:00

860 lines
23 KiB
Go

package bots
import (
"context"
"encoding/json"
"errors"
"fmt"
"log/slog"
"os"
"strings"
"time"
"github.com/google/uuid"
"github.com/jackc/pgx/v5"
"github.com/jackc/pgx/v5/pgtype"
"github.com/memohai/memoh/internal/db"
"github.com/memohai/memoh/internal/db/sqlc"
)
// Service provides bot CRUD and membership management.
type Service struct {
queries *sqlc.Queries
logger *slog.Logger
containerLifecycle ContainerLifecycle
}
const (
botLifecycleOperationTimeout = 5 * time.Minute
)
var (
ErrBotNotFound = errors.New("bot not found")
ErrBotAccessDenied = errors.New("bot access denied")
ErrOwnerUserNotFound = errors.New("owner user not found")
)
// AccessPolicy controls bot access behavior.
type AccessPolicy struct {
AllowPublicMember bool
}
// NewService creates a new bot service.
func NewService(log *slog.Logger, queries *sqlc.Queries) *Service {
if log == nil {
log = slog.Default()
}
return &Service{
queries: queries,
logger: log.With(slog.String("service", "bots")),
}
}
// SetContainerLifecycle registers a container lifecycle handler for bot operations.
func (s *Service) SetContainerLifecycle(lc ContainerLifecycle) {
s.containerLifecycle = lc
}
// AuthorizeAccess checks whether userID may access the given bot.
func (s *Service) AuthorizeAccess(ctx context.Context, userID, botID string, isAdmin bool, policy AccessPolicy) (Bot, error) {
if s.queries == nil {
return Bot{}, fmt.Errorf("bot queries not configured")
}
bot, err := s.Get(ctx, botID)
if err != nil {
if errors.Is(err, pgx.ErrNoRows) {
return Bot{}, ErrBotNotFound
}
return Bot{}, err
}
if isAdmin || bot.OwnerUserID == userID {
return bot, nil
}
if policy.AllowPublicMember && bot.Type == BotTypePublic {
if _, err := s.GetMember(ctx, botID, userID); err == nil {
return bot, nil
}
}
return Bot{}, ErrBotAccessDenied
}
// Create creates a new bot owned by owner user.
func (s *Service) Create(ctx context.Context, ownerUserID string, req CreateBotRequest) (Bot, error) {
if s.queries == nil {
return Bot{}, fmt.Errorf("bot queries not configured")
}
ownerID := strings.TrimSpace(ownerUserID)
if ownerID == "" {
return Bot{}, fmt.Errorf("owner user id is required")
}
ownerUUID, err := db.ParseUUID(ownerID)
if err != nil {
return Bot{}, err
}
if err := s.ensureUserExists(ctx, ownerUUID); err != nil {
return Bot{}, err
}
normalizedType, err := normalizeBotType(req.Type)
if err != nil {
return Bot{}, err
}
displayName := strings.TrimSpace(req.DisplayName)
if displayName == "" {
displayName = "bot-" + uuid.NewString()
}
avatarURL := strings.TrimSpace(req.AvatarURL)
isActive := true
if req.IsActive != nil {
isActive = *req.IsActive
}
metadata := req.Metadata
if metadata == nil {
metadata = map[string]any{}
}
payload, err := json.Marshal(metadata)
if err != nil {
return Bot{}, err
}
row, err := s.queries.CreateBot(ctx, sqlc.CreateBotParams{
OwnerUserID: ownerUUID,
Type: normalizedType,
DisplayName: pgtype.Text{String: displayName, Valid: displayName != ""},
AvatarUrl: pgtype.Text{String: avatarURL, Valid: avatarURL != ""},
IsActive: isActive,
Metadata: payload,
Status: BotStatusCreating,
})
if err != nil {
return Bot{}, err
}
bot, err := toBot(row)
if err != nil {
return Bot{}, err
}
if err := s.attachCheckSummary(ctx, &bot, row); err != nil {
return Bot{}, err
}
s.enqueueCreateLifecycle(bot.ID)
return bot, nil
}
// Get returns a bot by its ID.
func (s *Service) Get(ctx context.Context, botID string) (Bot, error) {
if s.queries == nil {
return Bot{}, fmt.Errorf("bot queries not configured")
}
botUUID, err := db.ParseUUID(botID)
if err != nil {
return Bot{}, err
}
row, err := s.queries.GetBotByID(ctx, botUUID)
if err != nil {
return Bot{}, err
}
bot, err := toBot(row)
if err != nil {
return Bot{}, err
}
if err := s.attachCheckSummary(ctx, &bot, row); err != nil {
return Bot{}, err
}
return bot, nil
}
// ListByOwner returns bots owned by the given user.
func (s *Service) ListByOwner(ctx context.Context, ownerUserID string) ([]Bot, error) {
if s.queries == nil {
return nil, fmt.Errorf("bot queries not configured")
}
ownerUUID, err := db.ParseUUID(ownerUserID)
if err != nil {
return nil, err
}
rows, err := s.queries.ListBotsByOwner(ctx, ownerUUID)
if err != nil {
return nil, err
}
items := make([]Bot, 0, len(rows))
for _, row := range rows {
item, err := toBot(row)
if err != nil {
return nil, err
}
if err := s.attachCheckSummary(ctx, &item, row); err != nil {
return nil, err
}
items = append(items, item)
}
return items, nil
}
// ListByMember returns bots where the user is a member.
func (s *Service) ListByMember(ctx context.Context, channelIdentityID string) ([]Bot, error) {
if s.queries == nil {
return nil, fmt.Errorf("bot queries not configured")
}
memberUUID, err := db.ParseUUID(channelIdentityID)
if err != nil {
return nil, err
}
rows, err := s.queries.ListBotsByMember(ctx, memberUUID)
if err != nil {
return nil, err
}
items := make([]Bot, 0, len(rows))
for _, row := range rows {
item, err := toBot(row)
if err != nil {
return nil, err
}
if err := s.attachCheckSummary(ctx, &item, row); err != nil {
return nil, err
}
items = append(items, item)
}
return items, nil
}
// ListAccessible returns all bots the user can access (owned or member).
func (s *Service) ListAccessible(ctx context.Context, channelIdentityID string) ([]Bot, error) {
owned, err := s.ListByOwner(ctx, channelIdentityID)
if err != nil {
return nil, err
}
members, err := s.ListByMember(ctx, channelIdentityID)
if err != nil {
return nil, err
}
seen := map[string]Bot{}
for _, item := range owned {
seen[item.ID] = item
}
for _, item := range members {
if _, ok := seen[item.ID]; !ok {
seen[item.ID] = item
}
}
items := make([]Bot, 0, len(seen))
for _, item := range seen {
items = append(items, item)
}
return items, nil
}
// Update updates bot profile fields.
func (s *Service) Update(ctx context.Context, botID string, req UpdateBotRequest) (Bot, error) {
if s.queries == nil {
return Bot{}, fmt.Errorf("bot queries not configured")
}
botUUID, err := db.ParseUUID(botID)
if err != nil {
return Bot{}, err
}
existing, err := s.queries.GetBotByID(ctx, botUUID)
if err != nil {
return Bot{}, err
}
displayName := strings.TrimSpace(existing.DisplayName.String)
avatarURL := strings.TrimSpace(existing.AvatarUrl.String)
isActive := existing.IsActive
metadata, err := decodeMetadata(existing.Metadata)
if err != nil {
return Bot{}, err
}
if req.DisplayName != nil {
displayName = strings.TrimSpace(*req.DisplayName)
}
if req.AvatarURL != nil {
avatarURL = strings.TrimSpace(*req.AvatarURL)
}
if req.IsActive != nil {
isActive = *req.IsActive
}
if req.Metadata != nil {
metadata = req.Metadata
}
if displayName == "" {
displayName = "bot-" + uuid.NewString()
}
payload, err := json.Marshal(metadata)
if err != nil {
return Bot{}, err
}
row, err := s.queries.UpdateBotProfile(ctx, sqlc.UpdateBotProfileParams{
ID: botUUID,
DisplayName: pgtype.Text{String: displayName, Valid: displayName != ""},
AvatarUrl: pgtype.Text{String: avatarURL, Valid: avatarURL != ""},
IsActive: isActive,
Metadata: payload,
})
if err != nil {
return Bot{}, err
}
bot, err := toBot(row)
if err != nil {
return Bot{}, err
}
if err := s.attachCheckSummary(ctx, &bot, row); err != nil {
return Bot{}, err
}
return bot, nil
}
// TransferOwner transfers bot ownership to another user.
func (s *Service) TransferOwner(ctx context.Context, botID string, ownerUserID string) (Bot, error) {
if s.queries == nil {
return Bot{}, fmt.Errorf("bot queries not configured")
}
botUUID, err := db.ParseUUID(botID)
if err != nil {
return Bot{}, err
}
ownerUUID, err := db.ParseUUID(ownerUserID)
if err != nil {
return Bot{}, err
}
if err := s.ensureUserExists(ctx, ownerUUID); err != nil {
return Bot{}, err
}
row, err := s.queries.UpdateBotOwner(ctx, sqlc.UpdateBotOwnerParams{
ID: botUUID,
OwnerUserID: ownerUUID,
})
if err != nil {
return Bot{}, err
}
bot, err := toBot(row)
if err != nil {
return Bot{}, err
}
if err := s.attachCheckSummary(ctx, &bot, row); err != nil {
return Bot{}, err
}
return bot, nil
}
// Delete removes a bot and its associated resources.
func (s *Service) Delete(ctx context.Context, botID string) error {
if s.queries == nil {
return fmt.Errorf("bot queries not configured")
}
botUUID, err := db.ParseUUID(botID)
if err != nil {
return err
}
row, err := s.queries.GetBotByID(ctx, botUUID)
if err != nil {
return err
}
if strings.TrimSpace(row.Status) == BotStatusDeleting {
return nil
}
if err := s.queries.UpdateBotStatus(ctx, sqlc.UpdateBotStatusParams{
ID: botUUID,
Status: BotStatusDeleting,
}); err != nil {
return err
}
s.enqueueDeleteLifecycle(botID)
return nil
}
// ListChecks evaluates runtime resource checks for a bot.
func (s *Service) ListChecks(ctx context.Context, botID string) ([]BotCheck, error) {
if s.queries == nil {
return nil, fmt.Errorf("bot queries not configured")
}
botUUID, err := db.ParseUUID(botID)
if err != nil {
return nil, err
}
row, err := s.queries.GetBotByID(ctx, botUUID)
if err != nil {
return nil, err
}
return s.buildRuntimeChecks(ctx, row)
}
func (s *Service) enqueueCreateLifecycle(botID string) {
go func() {
ctx, cancel := context.WithTimeout(context.Background(), botLifecycleOperationTimeout)
defer cancel()
if s.containerLifecycle != nil {
if err := s.containerLifecycle.SetupBotContainer(ctx, botID); err != nil {
s.logger.Error("bot container setup failed",
slog.String("bot_id", botID),
slog.Any("error", err),
)
}
}
if err := s.updateStatus(ctx, botID, BotStatusReady); err != nil {
s.logger.Error("failed to update bot status to ready after create",
slog.String("bot_id", botID),
slog.Any("error", err),
)
}
}()
}
func (s *Service) enqueueDeleteLifecycle(botID string) {
go func() {
ctx, cancel := context.WithTimeout(context.Background(), botLifecycleOperationTimeout)
defer cancel()
if s.containerLifecycle != nil {
if err := s.containerLifecycle.CleanupBotContainer(ctx, botID); err != nil {
s.logger.Error("bot container cleanup failed",
slog.String("bot_id", botID),
slog.Any("error", err),
)
}
}
botUUID, err := db.ParseUUID(botID)
if err != nil {
s.logger.Error("invalid bot id while finalizing delete",
slog.String("bot_id", botID),
slog.Any("error", err),
)
_ = s.updateStatus(ctx, botID, BotStatusReady)
return
}
if err := s.queries.DeleteBotByID(ctx, botUUID); err != nil {
s.logger.Error("failed to delete bot after cleanup",
slog.String("bot_id", botID),
slog.Any("error", err),
)
_ = s.updateStatus(ctx, botID, BotStatusReady)
return
}
}()
}
func (s *Service) updateStatus(ctx context.Context, botID, status string) error {
if s.queries == nil {
return fmt.Errorf("bot queries not configured")
}
botUUID, err := db.ParseUUID(botID)
if err != nil {
return err
}
return s.queries.UpdateBotStatus(ctx, sqlc.UpdateBotStatusParams{
ID: botUUID,
Status: strings.TrimSpace(status),
})
}
func (s *Service) ensureUserExists(ctx context.Context, userID pgtype.UUID) error {
if s.queries == nil {
return fmt.Errorf("bot queries not configured")
}
_, err := s.queries.GetUserByID(ctx, userID)
if err != nil {
if errors.Is(err, pgx.ErrNoRows) {
return ErrOwnerUserNotFound
}
return err
}
return nil
}
// UpsertMember creates or updates a bot membership.
func (s *Service) UpsertMember(ctx context.Context, botID string, req UpsertMemberRequest) (BotMember, error) {
if s.queries == nil {
return BotMember{}, fmt.Errorf("bot queries not configured")
}
botUUID, err := db.ParseUUID(botID)
if err != nil {
return BotMember{}, err
}
memberUUID, err := db.ParseUUID(req.UserID)
if err != nil {
return BotMember{}, err
}
role, err := normalizeMemberRole(req.Role)
if err != nil {
return BotMember{}, err
}
row, err := s.queries.UpsertBotMember(ctx, sqlc.UpsertBotMemberParams{
BotID: botUUID,
UserID: memberUUID,
Role: role,
})
if err != nil {
return BotMember{}, err
}
return toBotMember(row), nil
}
// ListMembers returns all members of a bot.
func (s *Service) ListMembers(ctx context.Context, botID string) ([]BotMember, error) {
if s.queries == nil {
return nil, fmt.Errorf("bot queries not configured")
}
botUUID, err := db.ParseUUID(botID)
if err != nil {
return nil, err
}
rows, err := s.queries.ListBotMembers(ctx, botUUID)
if err != nil {
return nil, err
}
items := make([]BotMember, 0, len(rows))
for _, row := range rows {
items = append(items, toBotMember(row))
}
return items, nil
}
// GetMember returns a specific bot member.
func (s *Service) GetMember(ctx context.Context, botID, channelIdentityID string) (BotMember, error) {
if s.queries == nil {
return BotMember{}, fmt.Errorf("bot queries not configured")
}
botUUID, err := db.ParseUUID(botID)
if err != nil {
return BotMember{}, err
}
memberUUID, err := db.ParseUUID(channelIdentityID)
if err != nil {
return BotMember{}, err
}
row, err := s.queries.GetBotMember(ctx, sqlc.GetBotMemberParams{
BotID: botUUID,
UserID: memberUUID,
})
if err != nil {
return BotMember{}, err
}
return toBotMember(row), nil
}
// DeleteMember removes a member from a bot.
func (s *Service) DeleteMember(ctx context.Context, botID, channelIdentityID string) error {
if s.queries == nil {
return fmt.Errorf("bot queries not configured")
}
botUUID, err := db.ParseUUID(botID)
if err != nil {
return err
}
memberUUID, err := db.ParseUUID(channelIdentityID)
if err != nil {
return err
}
return s.queries.DeleteBotMember(ctx, sqlc.DeleteBotMemberParams{
BotID: botUUID,
UserID: memberUUID,
})
}
// UpsertMemberSimple creates or updates a bot membership with a direct channel identity ID and role.
// This satisfies the router.BotMemberService interface.
func (s *Service) UpsertMemberSimple(ctx context.Context, botID, channelIdentityID, role string) error {
_, err := s.UpsertMember(ctx, botID, UpsertMemberRequest{
UserID: channelIdentityID,
Role: role,
})
return err
}
// IsMember checks if a user is a member of a bot.
func (s *Service) IsMember(ctx context.Context, botID, channelIdentityID string) (bool, error) {
_, err := s.GetMember(ctx, botID, channelIdentityID)
if err != nil {
if errors.Is(err, pgx.ErrNoRows) {
return false, nil
}
return false, err
}
return true, nil
}
func normalizeBotType(raw string) (string, error) {
normalized := strings.ToLower(strings.TrimSpace(raw))
if normalized == "" {
return BotTypePersonal, nil
}
switch normalized {
case BotTypePersonal, BotTypePublic:
return normalized, nil
default:
return "", fmt.Errorf("invalid bot type: %s", raw)
}
}
func normalizeMemberRole(raw string) (string, error) {
role := strings.ToLower(strings.TrimSpace(raw))
if role == "" {
return MemberRoleMember, nil
}
switch role {
case MemberRoleOwner, MemberRoleAdmin, MemberRoleMember:
return role, nil
default:
return "", fmt.Errorf("invalid member role: %s", raw)
}
}
func toBot(row sqlc.Bot) (Bot, error) {
displayName := ""
if row.DisplayName.Valid {
displayName = row.DisplayName.String
}
avatarURL := ""
if row.AvatarUrl.Valid {
avatarURL = row.AvatarUrl.String
}
metadata, err := decodeMetadata(row.Metadata)
if err != nil {
return Bot{}, err
}
createdAt := time.Time{}
if row.CreatedAt.Valid {
createdAt = row.CreatedAt.Time
}
updatedAt := time.Time{}
if row.UpdatedAt.Valid {
updatedAt = row.UpdatedAt.Time
}
return Bot{
ID: row.ID.String(),
OwnerUserID: row.OwnerUserID.String(),
Type: row.Type,
DisplayName: displayName,
AvatarURL: avatarURL,
IsActive: row.IsActive,
Status: strings.TrimSpace(row.Status),
CheckState: BotCheckStateUnknown,
CheckIssueCount: 0,
Metadata: metadata,
CreatedAt: createdAt,
UpdatedAt: updatedAt,
}, nil
}
func toBotMember(row sqlc.BotMember) BotMember {
createdAt := time.Time{}
if row.CreatedAt.Valid {
createdAt = row.CreatedAt.Time
}
return BotMember{
BotID: row.BotID.String(),
UserID: row.UserID.String(),
Role: row.Role,
CreatedAt: createdAt,
}
}
func decodeMetadata(payload []byte) (map[string]any, error) {
if len(payload) == 0 {
return map[string]any{}, nil
}
var data map[string]any
if err := json.Unmarshal(payload, &data); err != nil {
return nil, err
}
if data == nil {
data = map[string]any{}
}
return data, nil
}
func (s *Service) attachCheckSummary(ctx context.Context, bot *Bot, row sqlc.Bot) error {
checks, err := s.buildRuntimeChecks(ctx, row)
if err != nil {
return err
}
checkState, issueCount := summarizeChecks(checks)
bot.CheckState = checkState
bot.CheckIssueCount = issueCount
return nil
}
func (s *Service) buildRuntimeChecks(ctx context.Context, row sqlc.Bot) ([]BotCheck, error) {
status := strings.TrimSpace(row.Status)
checks := make([]BotCheck, 0, 4)
if status == BotStatusCreating {
checks = append(checks, BotCheck{
CheckKey: BotCheckKeyContainerInit,
Status: BotCheckStatusUnknown,
Summary: "Initialization is in progress.",
Detail: "Bot resources are still being provisioned.",
})
checks = append(checks, BotCheck{
CheckKey: BotCheckKeyContainerRecord,
Status: BotCheckStatusUnknown,
Summary: "Container record is pending.",
Detail: "Container record will be checked after initialization.",
})
checks = append(checks, BotCheck{
CheckKey: BotCheckKeyContainerTask,
Status: BotCheckStatusUnknown,
Summary: "Container task state is pending.",
Detail: "Task state will be checked after initialization.",
})
checks = append(checks, BotCheck{
CheckKey: BotCheckKeyContainerData,
Status: BotCheckStatusUnknown,
Summary: "Container host path check is pending.",
Detail: "Data path will be checked after initialization.",
})
return checks, nil
}
if status == BotStatusDeleting {
checks = append(checks, BotCheck{
CheckKey: BotCheckKeyDelete,
Status: BotCheckStatusUnknown,
Summary: "Deletion is in progress.",
Detail: "Bot resources are being cleaned up.",
})
checks = append(checks, BotCheck{
CheckKey: BotCheckKeyContainerRecord,
Status: BotCheckStatusUnknown,
Summary: "Container record check is skipped.",
Detail: "Bot is deleting and container checks are paused.",
})
checks = append(checks, BotCheck{
CheckKey: BotCheckKeyContainerTask,
Status: BotCheckStatusUnknown,
Summary: "Container task check is skipped.",
Detail: "Bot is deleting and task checks are paused.",
})
checks = append(checks, BotCheck{
CheckKey: BotCheckKeyContainerData,
Status: BotCheckStatusUnknown,
Summary: "Container host path check is skipped.",
Detail: "Bot is deleting and data path checks are paused.",
})
return checks, nil
}
checks = append(checks, BotCheck{
CheckKey: BotCheckKeyContainerInit,
Status: BotCheckStatusOK,
Summary: "Initialization finished.",
})
containerRow, err := s.queries.GetContainerByBotID(ctx, row.ID)
if err != nil {
if errors.Is(err, pgx.ErrNoRows) {
checks = append(checks, BotCheck{
CheckKey: BotCheckKeyContainerRecord,
Status: BotCheckStatusError,
Summary: "Container record is missing.",
Detail: "No container is attached to this bot.",
})
checks = append(checks, BotCheck{
CheckKey: BotCheckKeyContainerTask,
Status: BotCheckStatusUnknown,
Summary: "Container task state is unknown.",
Detail: "Task state cannot be determined without a container record.",
})
checks = append(checks, BotCheck{
CheckKey: BotCheckKeyContainerData,
Status: BotCheckStatusUnknown,
Summary: "Container data path is unknown.",
Detail: "Data path cannot be determined without a container record.",
})
return checks, nil
}
return nil, err
}
checks = append(checks, BotCheck{
CheckKey: BotCheckKeyContainerRecord,
Status: BotCheckStatusOK,
Summary: "Container record exists.",
Detail: fmt.Sprintf("container_id=%s", strings.TrimSpace(containerRow.ContainerID)),
Metadata: map[string]any{
"container_id": strings.TrimSpace(containerRow.ContainerID),
"namespace": strings.TrimSpace(containerRow.Namespace),
"image": strings.TrimSpace(containerRow.Image),
},
})
taskStatus := strings.TrimSpace(strings.ToLower(containerRow.Status))
taskCheck := BotCheck{
CheckKey: BotCheckKeyContainerTask,
Status: BotCheckStatusWarn,
Summary: "Container task state needs attention.",
}
switch taskStatus {
case "running", "created", "stopped", "paused":
taskCheck.Status = BotCheckStatusOK
taskCheck.Summary = "Container task state is reported."
taskCheck.Detail = fmt.Sprintf("status=%s", taskStatus)
case "":
taskCheck.Detail = "status is empty"
default:
taskCheck.Detail = fmt.Sprintf("unexpected status=%s", taskStatus)
}
taskCheck.Metadata = map[string]any{"status": taskStatus}
checks = append(checks, taskCheck)
hostPath := ""
if containerRow.HostPath.Valid {
hostPath = strings.TrimSpace(containerRow.HostPath.String)
}
dataCheck := BotCheck{
CheckKey: BotCheckKeyContainerData,
Status: BotCheckStatusWarn,
Summary: "Container host path needs attention.",
Metadata: map[string]any{"host_path": hostPath},
}
if hostPath == "" {
dataCheck.Detail = "host path is empty"
checks = append(checks, dataCheck)
return checks, nil
}
info, statErr := os.Stat(hostPath)
switch {
case statErr == nil && info != nil && info.IsDir():
dataCheck.Status = BotCheckStatusOK
dataCheck.Summary = "Container host path is accessible."
dataCheck.Detail = hostPath
case statErr == nil:
dataCheck.Status = BotCheckStatusError
dataCheck.Summary = "Container host path is invalid."
dataCheck.Detail = "host path is not a directory"
case errors.Is(statErr, os.ErrNotExist):
dataCheck.Status = BotCheckStatusError
dataCheck.Summary = "Container host path does not exist."
dataCheck.Detail = hostPath
default:
dataCheck.Status = BotCheckStatusWarn
dataCheck.Summary = "Container host path cannot be checked."
dataCheck.Detail = statErr.Error()
}
checks = append(checks, dataCheck)
return checks, nil
}
func summarizeChecks(checks []BotCheck) (string, int32) {
if len(checks) == 0 {
return BotCheckStateUnknown, 0
}
var issueCount int32
unknownCount := 0
for _, check := range checks {
switch check.Status {
case BotCheckStatusWarn, BotCheckStatusError:
issueCount++
case BotCheckStatusUnknown:
unknownCount++
}
}
if issueCount > 0 {
return BotCheckStateIssue, issueCount
}
if unknownCount == len(checks) {
return BotCheckStateUnknown, 0
}
return BotCheckStateOK, 0
}