package settings

import (
	"github.com/anytypeio/any-sync/commonspace/object/treegetter"
	"github.com/anytypeio/any-sync/commonspace/settings/settingsstate"
	"github.com/anytypeio/any-sync/util/slice"
	"go.uber.org/zap"
	"time"
)

type SpaceIdsProvider interface {
	AllIds() []string
}

type SpaceDeleter interface {
	DeleteSpace(spaceId string)
}

type DeletionManager interface {
	UpdateState(state *settingsstate.State) (err error)
}

func newDeletionManager(
	spaceId string,
	settingsId string,
	isResponsible bool,
	deletionInterval time.Duration,
	deletionState settingsstate.ObjectDeletionState,
	provider SpaceIdsProvider,
	onSpaceDelete func()) DeletionManager {
	return &deletionManager{
		isResponsible:    isResponsible,
		spaceId:          spaceId,
		settingsId:       settingsId,
		deletionState:    deletionState,
		provider:         provider,
		deletionInterval: deletionInterval,
		onSpaceDelete:    onSpaceDelete,
	}
}

type deletionManager struct {
	deletionState    settingsstate.ObjectDeletionState
	provider         SpaceIdsProvider
	treeGetter       treegetter.TreeGetter
	deletionInterval time.Duration
	spaceId          string
	settingsId       string
	isResponsible    bool
	onSpaceDelete    func()
}

func (d *deletionManager) UpdateState(state *settingsstate.State) (err error) {
	err = d.deletionState.Add(state.DeletedIds)
	if err != nil {
		log.Warn("failed to add deleted ids to deletion state")
	}

	if state.DeleterId != "" {
		log.Debug("deleting space", zap.String("spaceId", d.spaceId))
		spaceDeleter, ok := d.treeGetter.(SpaceDeleter)
		if ok {
			spaceDeleter.DeleteSpace(d.spaceId)
		}
		if d.isResponsible {
			allIds := slice.DiscardFromSlice(d.provider.AllIds(), func(id string) bool {
				return id == d.settingsId
			})
			err = d.deletionState.Add(allIds)
			if err != nil {
				return
			}
		}
		d.onSpaceDelete()
	}
	return
}