package sources

import (
	"strings"

	tea "github.com/charmbracelet/bubbletea"
	"github.com/trufflesecurity/trufflehog/v3/pkg/tui/sources/circleci"
	"github.com/trufflesecurity/trufflehog/v3/pkg/tui/sources/docker"
	"github.com/trufflesecurity/trufflehog/v3/pkg/tui/sources/elasticsearch"
	"github.com/trufflesecurity/trufflehog/v3/pkg/tui/sources/filesystem"
	"github.com/trufflesecurity/trufflehog/v3/pkg/tui/sources/gcs"
	"github.com/trufflesecurity/trufflehog/v3/pkg/tui/sources/git"
	"github.com/trufflesecurity/trufflehog/v3/pkg/tui/sources/github"
	"github.com/trufflesecurity/trufflehog/v3/pkg/tui/sources/gitlab"
	"github.com/trufflesecurity/trufflehog/v3/pkg/tui/sources/huggingface"
	"github.com/trufflesecurity/trufflehog/v3/pkg/tui/sources/jenkins"
	"github.com/trufflesecurity/trufflehog/v3/pkg/tui/sources/postman"
	"github.com/trufflesecurity/trufflehog/v3/pkg/tui/sources/s3"
	"github.com/trufflesecurity/trufflehog/v3/pkg/tui/sources/syslog"
)

func GetSourceNotes(sourceName string) string {
	source := strings.ToLower(sourceName)
	switch source {
	case "github":
		return github.GetNote()
	case "postman":
		return postman.GetNote()
	case "elasticsearch":
		return elasticsearch.GetNote()
	case "huggingface":
		return huggingface.GetNote()
	case "jenkins":
		return jenkins.GetNote()

	default:
		return ""
	}
}

type CmdModel interface {
	tea.Model
	Cmd() string
	Summary() string
}

func GetSourceFields(sourceName string) CmdModel {
	source := strings.ToLower(sourceName)

	switch source {
	case "aws s3":
		return s3.GetFields()
	case "circleci":
		return circleci.GetFields()
	case "docker":
		return docker.GetFields()
	case "elasticsearch":
		return elasticsearch.GetFields()
	case "filesystem":
		return filesystem.GetFields()
	case "gcs (google cloud storage)":
		return gcs.GetFields()
	case "git":
		return git.GetFields()
	case "github":
		return github.GetFields()
	case "gitlab":
		return gitlab.GetFields()
	case "hugging face":
		return huggingface.GetFields()
	case "jenkins":
		return jenkins.GetFields()
	case "postman":
		return postman.GetFields()
	case "syslog":
		return syslog.GetFields()
	}

	return nil
}