package main

import (
	"context"
	"encoding/json"

	"github.com/mudler/local-agent-framework/core/action"
	"github.com/mudler/local-agent-framework/pkg/xlog"

	. "github.com/mudler/local-agent-framework/core/agent"
	"github.com/mudler/local-agent-framework/services/actions"
)

const (
	// Actions
	ActionSearch              = "search"
	ActionCustom              = "custom"
	ActionGithubIssueLabeler  = "github-issue-labeler"
	ActionGithubIssueOpener   = "github-issue-opener"
	ActionGithubIssueCloser   = "github-issue-closer"
	ActionGithubIssueSearcher = "github-issue-searcher"
	ActionScraper             = "scraper"
	ActionWikipedia           = "wikipedia"
	ActionBrowse              = "browse"
	ActionSendMail            = "send_mail"
)

var AvailableActions = []string{
	ActionSearch,
	ActionCustom,
	ActionGithubIssueLabeler,
	ActionGithubIssueOpener,
	ActionGithubIssueCloser,
	ActionGithubIssueSearcher,
	ActionScraper,
	ActionBrowse,
	ActionWikipedia,
	ActionSendMail,
}

func (a *AgentConfig) availableActions(ctx context.Context) []Action {
	allActions := []Action{}

	for _, a := range a.Actions {
		var config map[string]string
		if err := json.Unmarshal([]byte(a.Config), &config); err != nil {
			xlog.Error("Error unmarshalling action config", "error", err)
			continue
		}

		switch a.Name {
		case ActionCustom:
			customAction, err := action.NewCustom(config, "")
			if err != nil {
				xlog.Error("Error creating custom action", "error", err)
				continue
			}
			allActions = append(allActions, customAction)
		case ActionSearch:
			allActions = append(allActions, actions.NewSearch(config))
		case ActionGithubIssueLabeler:
			allActions = append(allActions, actions.NewGithubIssueLabeler(ctx, config))
		case ActionGithubIssueOpener:
			allActions = append(allActions, actions.NewGithubIssueOpener(ctx, config))
		case ActionGithubIssueCloser:
			allActions = append(allActions, actions.NewGithubIssueCloser(ctx, config))
		case ActionGithubIssueSearcher:
			allActions = append(allActions, actions.NewGithubIssueSearch(ctx, config))
		case ActionScraper:
			allActions = append(allActions, actions.NewScraper(config))
		case ActionWikipedia:
			allActions = append(allActions, actions.NewWikipedia(config))
		case ActionBrowse:
			allActions = append(allActions, actions.NewBrowse(config))
		case ActionSendMail:
			allActions = append(allActions, actions.NewSendMail(config))
		}
	}

	return allActions
}