session_codec.go

package codebuddy

import (
	"crypto/sha256"
	"fmt"
	"strings"

	"github.com/agent-dance/agent-adaptor/driver"
)

type sessionCodec struct{}

func (adapter) SessionCodec() driver.SessionCodec {
	return sessionCodec{}
}

func (sessionCodec) Name() string { return DriverType }

func (sessionCodec) ToParams(state *driver.SessionState) driver.SessionParams {
	if state == nil {
		return driver.SessionParams{}
	}
	return driver.SessionParams{
		ResumeID:  state.ResumeID,
		DisplayID: displayID(state),
		Values:    cloneData(state.Data),
	}
}

func (sessionCodec) FromParams(params driver.SessionParams) *driver.SessionState {
	if params.ResumeID == "" && params.DisplayID == "" && len(params.Values) == 0 {
		return nil
	}
	display := params.DisplayID
	if display == "" {
		display = params.ResumeID
	}
	return &driver.SessionState{
		ResumeID:  params.ResumeID,
		DisplayID: display,
		Data:      cloneData(params.Values),
	}
}

func (sessionCodec) GuardFingerprint(params driver.SessionParams) string {
	return guardHash(params.Values,
		driver.SessionParamCWD,
		driver.SessionParamWorkspaceID,
		driver.SessionParamProfileFingerprint,
	)
}

func displayID(state *driver.SessionState) string {
	if state.DisplayID != "" {
		return state.DisplayID
	}
	return state.ResumeID
}

func cloneData(values map[string]string) map[string]string {
	if len(values) == 0 {
		return nil
	}
	out := make(map[string]string, len(values))
	for key, value := range values {
		out[key] = value
	}
	return out
}

func guardHash(values map[string]string, keys ...string) string {
	builder := strings.Builder{}
	for index, key := range keys {
		if index > 0 {
			builder.WriteByte(0)
		}
		builder.WriteString(key)
		builder.WriteByte('=')
		builder.WriteString(values[key])
	}
	sum := sha256.Sum256([]byte(builder.String()))
	return fmt.Sprintf("%x", sum[:])
}