Elydora-Go-SDK/cmd/elydora/plugins/kiroide_status.go

package plugins

import (
	"bytes"
	"fmt"
	"path/filepath"
	"runtime"
	"sort"
	"strings"
)

func readKiroIdeRuntimeConfig(path string) (map[string]any, bool, error) {
	snapshot, err := readManagedFile(path, "Elydora runtime config", maxRuntimeConfigBytes)
	if err != nil || snapshot == nil {
		return nil, snapshot != nil, err
	}
	label := fmt.Sprintf("Elydora runtime config at %s", path)
	config, err := decodeStrictJSONObject(snapshot.contents, label)
	return config, true, err
}

func requireKiroIdeRuntimeString(
	config map[string]any,
	field string,
	configPath string,
) (string, error) {
	value, ok := config[field].(string)
	if !ok || strings.TrimSpace(value) == "" {
		return "", fmt.Errorf(
			"Elydora runtime config %s is invalid: %s",
			field,
			configPath,
		)
	}
	return value, nil
}

func validateKiroIdeRuntimeConfig(
	config map[string]any,
	expectedAgentID string,
	configPath string,
) error {
	supported := stringSet(
		"org_id", "agent_id", "kid", "base_url", "token", "agent_name",
	)
	fields := make([]string, 0, len(config))
	for field := range config {
		fields = append(fields, field)
	}
	sort.Strings(fields)
	for _, field := range fields {
		if _, ok := supported[field]; !ok {
			return fmt.Errorf(
				`Elydora runtime config has unsupported field %q: %s`,
				field,
				configPath,
			)
		}
	}
	if _, err := requireKiroIdeRuntimeString(config, "org_id", configPath); err != nil {
		return err
	}
	if _, err := requireKiroIdeRuntimeString(config, "kid", configPath); err != nil {
		return err
	}
	agentID, err := requireKiroIdeRuntimeString(config, "agent_id", configPath)
	if err != nil {
		return err
	}
	if !sameKiroIdeAgentID(agentID, expectedAgentID) ||
		config["agent_name"] != kiroIdeAgentKey {
		return fmt.Errorf(
			"Elydora runtime identity does not match Kiro IDE hooks: %s",
			configPath,
		)
	}
	if _, exists := config["token"]; exists {
		if _, err := requireKiroIdeRuntimeString(config, "token", configPath); err != nil {
			return err
		}
	}
	baseURL, err := requireKiroIdeRuntimeString(config, "base_url", configPath)
	if err != nil {
		return err
	}
	if err := validateManagedBaseURL(baseURL); err != nil {
		return fmt.Errorf(
			"Elydora runtime config base URL is invalid at %s: %w",
			configPath,
			err,
		)
	}
	return nil
}

func validateKiroIdeRuntimeIdentity(agentDirectory, agentID string) error {
	runtimeRoot := filepath.Dir(agentDirectory)
	rootExists, err := managedPhysicalDirectoryExists(
		runtimeRoot,
		"Elydora runtime directory",
	)
	if err != nil || !rootExists {
		return err
	}
	directoryExists, err := managedPhysicalDirectoryExists(
		agentDirectory,
		"Elydora agent runtime directory",
	)
	if err != nil || !directoryExists {
		return err
	}
	configPath := filepath.Join(agentDirectory, "config.json")
	config, configExists, err := readKiroIdeRuntimeConfig(configPath)
	if err != nil {
		return err
	}
	artifactExists := false
	for _, item := range []struct {
		path  string
		label string
		limit int64
	}{
		{filepath.Join(agentDirectory, "private.key"), "Elydora private key", maxProtectedSecretBytes},
		{filepath.Join(agentDirectory, kiroIdeGuardScript), "Elydora guard runtime", maxManagedSourceBytes},
		{filepath.Join(agentDirectory, kiroIdeAuditScript), "Elydora audit runtime", maxManagedSourceBytes},
		{filepath.Join(agentDirectory, "chain-state.json"), "Elydora chain state", maxRuntimeConfigBytes},
		{filepath.Join(agentDirectory, "status-cache.json"), "Elydora status cache", maxRuntimeConfigBytes},
		{filepath.Join(agentDirectory, "error.log"), "Elydora error log", maxManagedSourceBytes},
	} {
		exists, inspectErr := managedPhysicalFileExists(item.path, item.label, item.limit)
		if inspectErr != nil {
			return inspectErr
		}
		artifactExists = artifactExists || exists
	}
	if !configExists {
		if artifactExists {
			return fmt.Errorf(
				"Elydora runtime identity cannot be verified without config.json: %s",
				agentDirectory,
			)
		}
		return nil
	}
	configuredID, ok := config["agent_id"].(string)
	if !ok || config["agent_name"] != kiroIdeAgentKey ||
		!sameKiroIdeAgentID(configuredID, agentID) {
		return fmt.Errorf(
			"Elydora runtime config identity does not match Kiro IDE agent %s: %s",
			agentID,
			configPath,
		)
	}
	return nil
}

func kiroIdeRuntimeFileMode(snapshot *managedFileSnapshot, expected uint32) bool {
	if runtime.GOOS == "windows" {
		return true
	}
	return uint32(snapshot.mode.Perm()) == expected
}

func kiroIdeRuntimeContractExists(contract kiroIdeRuntimeContract) (bool, error) {
	runtimeRoot, err := AgentRuntimeRoot()
	if err != nil {
		return false, err
	}
	agentDirectory := filepath.Dir(contract.guardPath)
	if !sameKiroIdePath(filepath.Dir(agentDirectory), runtimeRoot) ||
		!sameKiroIdeAgentID(filepath.Base(agentDirectory), contract.agentID) ||
		!sameKiroIdePath(
			contract.guardPath,
			filepath.Join(agentDirectory, kiroIdeGuardScript),
		) ||
		!sameKiroIdePath(
			contract.auditPath,
			filepath.Join(agentDirectory, kiroIdeAuditScript),
		) {
		return false, nil
	}
	rootExists, err := managedPhysicalDirectoryExists(
		runtimeRoot,
		"Elydora runtime directory",
	)
	if err != nil || !rootExists {
		return false, err
	}
	directoryExists, err := managedPhysicalDirectoryExists(
		agentDirectory,
		"Elydora agent runtime directory",
	)
	if err != nil || !directoryExists {
		return false, err
	}
	configPath := filepath.Join(agentDirectory, "config.json")
	keyPath := filepath.Join(agentDirectory, "private.key")
	configSnapshot, err := readManagedFile(
		configPath,
		"Elydora runtime config",
		maxRuntimeConfigBytes,
	)
	if err != nil {
		return false, err
	}
	key, err := readManagedFile(keyPath, "Elydora private key", maxProtectedSecretBytes)
	if err != nil {
		return false, err
	}
	guard, err := readManagedFile(
		contract.guardPath,
		"Elydora guard runtime",
		maxManagedSourceBytes,
	)
	if err != nil {
		return false, err
	}
	audit, err := readManagedFile(
		contract.auditPath,
		"Elydora audit runtime",
		maxManagedSourceBytes,
	)
	if err != nil {
		return false, err
	}
	if configSnapshot == nil || key == nil || guard == nil || audit == nil {
		return false, nil
	}
	config, err := decodeStrictJSONObject(
		configSnapshot.contents,
		fmt.Sprintf("Elydora runtime config at %s", configPath),
	)
	if err != nil {
		return false, err
	}
	if err := validateKiroIdeRuntimeConfig(config, contract.agentID, configPath); err != nil {
		return false, err
	}
	if err := validateManagedPrivateKey(string(key.contents)); err != nil {
		return false, fmt.Errorf("Elydora private key at %s: %w", keyPath, err)
	}
	if !kiroIdeRuntimeFileMode(configSnapshot, 0600) ||
		!kiroIdeRuntimeFileMode(key, 0600) ||
		!kiroIdeRuntimeFileMode(guard, 0700) ||
		!kiroIdeRuntimeFileMode(audit, 0700) {
		return false, nil
	}
	expectedGuard := []byte(generateGuardScript(
		kiroIdeAgentKey,
		contract.agentID,
		"",
		false,
		"",
	))
	expectedAudit := []byte(buildHookScriptWithOutput(
		kiroIdeAgentKey,
		contract.agentID,
		"",
		false,
		true,
	))
	return bytes.Equal(guard.contents, expectedGuard) &&
		bytes.Equal(audit.contents, expectedAudit), nil
}

func kiroIdeRuntimeFilesExist(contracts []kiroIdeRuntimeContract) (bool, error) {
	for _, contract := range contracts {
		exists, err := kiroIdeRuntimeContractExists(contract)
		if err != nil {
			return false, err
		}
		if exists {
			return true, nil
		}
	}
	return false, nil
}