package state

import (
"context"
)

type memoryStateValue struct {
	backend   *MemoryStateBackend
	stateName string
}

func (m *memoryStateValue) Value(ctx context.Context) (any, error) {
	kg := m.backend.currentGroupKey
	key := m.backend.currentKey
	ns := m.backend.currentNamespace
	if kgMap, exist := m.backend.states[m.stateName][kg][key]; exist {
		if v, hasKey := kgMap[ns]; hasKey {
			return v, nil
		}
	}

	return nil, nil
}

func (m *memoryStateValue) Update(ctx context.Context, newVal any) (any, error) {
	kg := m.backend.currentGroupKey
	key := m.backend.currentKey
	ns := m.backend.currentNamespace
	if _, exists := m.backend.states[m.stateName][kg][key]; !exists {
		m.backend.states[m.stateName][kg] = make(map[string]map[any]any)
	}

	m.backend.states[m.stateName][kg][key][ns] = newVal
	return newVal, nil
}

func (m *memoryStateValue) Clear(ctx context.Context) error {
	kg := m.backend.currentGroupKey
	key := m.backend.currentKey
	if kgMap, exists := m.backend.states[m.stateName][kg]; exists {
		delete(kgMap, key)
	}
	return nil
}