package consul

import (
	"github.com/hashicorp/consul/acl"
	"github.com/hashicorp/consul/agent/consul/fsm"
	"github.com/hashicorp/consul/agent/structs"
)

// FSMDataStore implements the DataStore interface using the Consul server and finite state manager.
type FSMDataStore struct {
	server *Server
	fsm    *fsm.FSM
}

func NewFSMDataStore(server *Server, fsm *fsm.FSM) *FSMDataStore {
	return &FSMDataStore{
		server: server,
		fsm:    fsm,
	}
}

// GetConfigEntry takes in a kind, name, and meta and returns a configentry and an error from the FSM state
func (f *FSMDataStore) GetConfigEntry(kind string, name string, meta *acl.EnterpriseMeta) (structs.ConfigEntry, error) {
	store := f.fsm.State()

	_, entry, err := store.ConfigEntry(nil, kind, name, meta)
	if err != nil {
		return nil, err
	}
	return entry, nil
}

// GetConfigEntriesByKind takes in a kind and returns all instances of that kind of config entry from the FSM state
func (f *FSMDataStore) GetConfigEntriesByKind(kind string) ([]structs.ConfigEntry, error) {
	store := f.fsm.State()

	_, entries, err := store.ConfigEntriesByKind(nil, kind, acl.WildcardEnterpriseMeta())
	if err != nil {
		return nil, err
	}
	return entries, nil
}

// Update takes a config entry and upserts it in the FSM state
func (f *FSMDataStore) Update(entry structs.ConfigEntry) error {
	_, err := f.server.leaderRaftApply("ConfigEntry.Apply", structs.ConfigEntryRequestType, &structs.ConfigEntryRequest{
		Op:    structs.ConfigEntryUpsertCAS,
		Entry: entry,
	})
	return err
}

// UpdateStatus takes a config entry, an error, and updates the status field as needed in the FSM state
func (f *FSMDataStore) UpdateStatus(entry structs.ControlledConfigEntry, err error) error {
	if err == nil {
		//TODO additional status messages for success?
		return nil
	}
	status := structs.Status{
		Conditions: []structs.Condition{{

			Status: err.Error() + ": Accepted == false",
		},
		},
	}
	entry.SetStatus(status)
	return f.Update(entry)
}

// Delete takes a config entry and deletes it from the FSM state
func (f *FSMDataStore) Delete(entry structs.ConfigEntry) error {
	_, err := f.server.leaderRaftApply("ConfigEntry.Delete", structs.ConfigEntryRequestType, &structs.ConfigEntryRequest{
		Op:    structs.ConfigEntryDelete,
		Entry: entry,
	})
	return err
}