internal/enrollment/materialize.go

Rama main · réplica de lectura del repositorio original.

package enrollment

import (
	"encoding/json"
	"os"
	"path/filepath"
	"strings"

	"github.com/yeomyeonggeori/blueclaw/internal/config"
	"github.com/yeomyeonggeori/blueclaw/internal/llm"
)

func Materialize(home Home, enrollment Enrollment) error {
	if errorValue := enrollment.Validate(); errorValue != nil {
		return errorValue
	}
	if errorValue := os.MkdirAll(home.DirectoryPath, 0o700); errorValue != nil {
		return errorValue
	}
	if errorValue := os.MkdirAll(enrollment.WorkspaceRootPath, 0o755); errorValue != nil {
		return errorValue
	}
	if errorValue := writeMigrations(filepath.Join(home.DirectoryPath, "migrations")); errorValue != nil {
		return errorValue
	}
	if errorValue := writeModelAPIKey(home, enrollment.LanguageModel); errorValue != nil {
		return errorValue
	}
	if errorValue := writeJSONDocument(home.RuntimeConfigurationPath(), runtimeConfigurationFor(home, enrollment)); errorValue != nil {
		return errorValue
	}
	return writeJSONDocument(home.PolicyPath(), policyDocumentFor(enrollment))
}

func runtimeConfigurationFor(home Home, enrollment Enrollment) config.RuntimeConfiguration {
	runtimeConfiguration := config.RuntimeConfiguration{
		BaseURL: "http://" + availableListenAddress(),
		Database: config.DatabaseConfiguration{
			Driver:                 "postgres",
			ConnectionString:       enrollment.DatabaseConnectionString,
			MigrationDirectoryPath: filepath.Join(home.DirectoryPath, "migrations"),
		},
		Terminal: config.TerminalConfiguration{
			Mode:              "native",
			WorkspaceRootPath: enrollment.WorkspaceRootPath,
			TimeoutSecond:     600,
		},
		LanguageModel: languageModelConfigurationFor(home, enrollment.LanguageModel),
		Agent: config.AgentConfiguration{
			Harness: config.HarnessConfiguration{
				Name:             enrollment.Harness.Name,
				AgentCommandPath: enrollment.Harness.AgentCommandPath,
			},
		},
	}
	return runtimeConfiguration
}

func languageModelConfigurationFor(home Home, access LanguageModelAccess) config.LanguageModelConfiguration {
	tiers := make(map[string][]config.ModelEndpointConfiguration, len(llm.ModelTiers))
	for _, modelTierName := range llm.ModelTiers {
		tiers[modelTierName] = []config.ModelEndpointConfiguration{modelEndpointFor(home, access, access.ModelName)}
	}
	languageModel := config.LanguageModelConfiguration{Tiers: tiers}
	if strings.TrimSpace(access.EmbeddingModelName) != "" {
		languageModel.Embedding = modelEndpointFor(home, access, access.EmbeddingModelName)
	}
	return languageModel
}

func modelEndpointFor(home Home, access LanguageModelAccess, modelName string) config.ModelEndpointConfiguration {
	modelEndpoint := config.ModelEndpointConfiguration{
		Endpoint: strings.TrimSpace(access.EndpointURL),
		Model:    strings.TrimSpace(modelName),
	}
	if strings.TrimSpace(access.APIKey) != "" {
		modelEndpoint.APIKeyPath = home.ModelAPIKeyPath()
	}
	return modelEndpoint
}

func writeModelAPIKey(home Home, access LanguageModelAccess) error {
	apiKey := strings.TrimSpace(access.APIKey)
	if apiKey == "" {
		return nil
	}
	return os.WriteFile(home.ModelAPIKeyPath(), []byte(apiKey+"\n"), 0o600)
}

func policyDocumentFor(enrollment Enrollment) map[string]any {
	return map[string]any{
		"people": []map[string]any{{
			"personID":    enrollment.Operator.PersonID,
			"displayName": enrollment.Operator.DisplayName,
			"emails":      []string{enrollment.Operator.Email},
			"circles":     []string{"member"},
			"isAdmin":     true,
		}},
		"circles": []map[string]any{{
			"circleID":               "member",
			"displayName":            "Member",
			"workspaceDirectoryPath": filepath.Join(enrollment.WorkspaceRootPath, "circles", "member"),
		}},
	}
}

func writeJSONDocument(path string, document any) error {
	encodedDocument, errorValue := json.MarshalIndent(document, "", "  ")
	if errorValue != nil {
		return errorValue
	}
	return os.WriteFile(path, append(encodedDocument, '\n'), 0o600)
}