package main

import (
	"errors"
	"fmt"
	"os"
	"path/filepath"

	"github.com/projectdiscovery/uncover/testutils"
	folderutil "github.com/projectdiscovery/utils/folder"
)

var (
	ConfigFile = filepath.Join(folderutil.AppConfigDirOrDefault(".uncover-config", "uncover"), "provider-config.yaml")
)

type censysTestcases struct{}

func (h censysTestcases) Execute() error {
	token := os.Getenv("CENSYS_API_KEY")
	if token == "" {
		return errors.New("missing censys api key")
	}
	censysToken := fmt.Sprintf(`censys: [%s]`, token)
	_ = os.WriteFile(ConfigFile, []byte(censysToken), 0644)
	defer func() {
		_ = os.RemoveAll(ConfigFile)
	}()
	results, err := testutils.RunUncoverAndGetResults(debug, "-censys", "'services.software.vendor=Grafana'")
	if err != nil {
		return err
	}
	return expectResultsGreaterThanCount(results, 0)
}

type shodanTestcases struct{}

func (h shodanTestcases) Execute() error {
	token := os.Getenv("SHODAN_API_KEY")
	if token == "" {
		return errors.New("missing shodan api key")
	}
	shodanToken := fmt.Sprintf(`shodan: [%s]`, token)
	_ = os.WriteFile(ConfigFile, []byte(shodanToken), 0644)
	defer func() {
		_ = os.RemoveAll(ConfigFile)
	}()
	results, err := testutils.RunUncoverAndGetResults(debug, "-shodan", "'title:\"Grafana\"'")
	if err != nil {
		return err
	}
	err = expectResultsGreaterThanCount(results, 0)
	if err != nil {
		return err
	}
	results, err = testutils.RunUncoverAndGetResults(debug, "-shodan", "'org:\"Something, Inc.\"'")
	if err != nil {
		return err
	}
	return expectResultsGreaterThanCount(results, 1)
}

type zoomeyeTestcases struct{}

func (h zoomeyeTestcases) Execute() error {
	token := os.Getenv("ZOOMEYE_API_KEY")
	if token == "" {
		return errors.New("missing zoomeye api key")
	}
	zoomeyeToken := fmt.Sprintf(`zoomeye: [%s]`, token)
	_ = os.WriteFile(ConfigFile, []byte(zoomeyeToken), 0644)
	defer func() {
		_ = os.RemoveAll(ConfigFile)
	}()
	results, err := testutils.RunUncoverAndGetResults(debug, "-zoomeye", "'app:\"Atlassian JIRA\"'")
	if err != nil {
		return err
	}
	return expectResultsGreaterThanCount(results, 0)
}

type fofaTestcases struct{}

func (h fofaTestcases) Execute() error {
	token := os.Getenv("FOFA_API_KEY")
	if token == "" {
		return errors.New("missing fofa api key")
	}
	fofaToken := fmt.Sprintf(`fofa: [%s]`, token)
	_ = os.WriteFile(ConfigFile, []byte(fofaToken), 0644)
	defer func() {
		_ = os.RemoveAll(ConfigFile)
	}()
	results, err := testutils.RunUncoverAndGetResults(debug, "-fofa", "'app=Grafana'")
	if err != nil {
		return err
	}
	return expectResultsGreaterThanCount(results, 0)
}

// type hunterTestcases struct{}

// func (h hunterTestcases) Execute() error {
// 	results, err := testutils.RunUncoverAndGetResults(debug, "-hunter", "'Grafana'")
// 	if err != nil {
// 		return err
// 	}
// 	return expectResultsGreaterOrEqualToCount(results, 0)
// }

type quakeTestcases struct{}

func (h quakeTestcases) Execute() error {
	token := os.Getenv("QUAKE_API_KEY")
	if token == "" {
		return errors.New("missing quake api key")
	}
	quakeToken := fmt.Sprintf(`quake: [%s]`, token)
	_ = os.WriteFile(ConfigFile, []byte(quakeToken), 0644)
	defer func() {
		_ = os.RemoveAll(ConfigFile)
	}()
	results, err := testutils.RunUncoverAndGetResults(debug, "-quake", "'Grafana'")
	if err != nil {
		return err
	}
	return expectResultsGreaterThanCount(results, 0)
}

type netlasTestcases struct{}

func (h netlasTestcases) Execute() error {
	token := os.Getenv("NETLAS_API_KEY")
	if token == "" {
		return errors.New("missing netlas api key")
	}
	netlasToken := fmt.Sprintf(`netlas: [%s]`, token)
	_ = os.WriteFile(ConfigFile, []byte(netlasToken), 0644)
	defer func() {
		_ = os.RemoveAll(ConfigFile)
	}()
	results, err := testutils.RunUncoverAndGetResults(debug, "-netlas", "'Grafana'")
	if err != nil {
		return err
	}
	return expectResultsGreaterThanCount(results, 0)
}

type criminalipTestcases struct{}

func (h criminalipTestcases) Execute() error {
	token := os.Getenv("CRIMINALIP_API_KEY")
	if token == "" {
		return errors.New("missing criminalip api key")
	}
	criminalipToken := fmt.Sprintf(`criminalip: [%s]`, token)
	_ = os.WriteFile(ConfigFile, []byte(criminalipToken), 0644)
	defer func() {
		_ = os.RemoveAll(ConfigFile)
	}()
	results, err := testutils.RunUncoverAndGetResults(debug, "-criminalip", "'Grafana'")
	if err != nil {
		return err
	}
	return expectResultsGreaterThanCount(results, 0)
}

type hunterhowTestcases struct{}

func (h hunterhowTestcases) Execute() error {
	token := os.Getenv("HUNTERHOW_API_KEY")
	if token == "" {
		return errors.New("missing hunterhow api key")
	}
	hunterhowApiKey := fmt.Sprintf(`hunterhow: [%s]`, token)
	_ = os.WriteFile(ConfigFile, []byte(hunterhowApiKey), 0644)
	defer func() {
		_ = os.RemoveAll(ConfigFile)
	}()
	results, err := testutils.RunUncoverAndGetResults(debug, "-hunterhow", "'web.body=\"ElasticJob\"'")
	if err != nil {
		return err
	}
	return expectResultsGreaterThanCount(results, 0)
}

type outputTestcases struct{}

func (h outputTestcases) Execute() error {
	results, err := testutils.RunUncoverAndGetResults(debug, "-q", "'element'", "-j", "-silent")
	if err != nil {
		return err
	}
	err = expectResultsGreaterThanCount(results, 0)
	if err != nil {
		return err
	}
	results, err = testutils.RunUncoverAndGetResults(debug, "-q", "'element'", "-r", "-silent")
	if err != nil {
		return err
	}
	return expectResultsGreaterThanCount(results, 0)
}

type googleTestcases struct{}

func (h googleTestcases) Execute() error {
	token := os.Getenv("GOOGLE_API_KEY")
	if token == "" {
		return errors.New("missing google api key")
	}
	engineId := os.Getenv("GOOGLE_API_CX")
	if engineId == "" {
		return errors.New("missing google engine id")
	}
	googleToken := fmt.Sprintf(`google: [%s,%s]`, token, engineId)
	_ = os.WriteFile(ConfigFile, []byte(googleToken), 0644)
	defer func() {
		_ = os.RemoveAll(ConfigFile)
	}()
	results, err := testutils.RunUncoverAndGetResults(debug, "-google", "site:*.hackerone.com")
	if err != nil {
		return err
	}
	return expectResultsGreaterThanCount(results, 0)
}

type odinTestcases struct{}

func (h odinTestcases) Execute() error {
	token := os.Getenv("ODIN_API_KEY")
	if token == "" {
		return errors.New("missing odin api key")
	}
	odinToken := fmt.Sprintf(`odin: [%s]`, token)
	if err := os.WriteFile(ConfigFile, []byte(odinToken), 0644); err != nil {
		return err
	}
	defer func() {
		_ = os.RemoveAll(ConfigFile)
	}()
	results, err := testutils.RunUncoverAndGetResults(debug, "-odin", "nginx")

	if err != nil {
		return err
	}
	return expectResultsGreaterThanCount(results, 0)
}

type binaryedgeTestcases struct{}

func (h binaryedgeTestcases) Execute() error {
	token := os.Getenv("BINARYEDGE_API_KEY")
	if token == "" {
		return errors.New("missing binaryedge api key")
	}

	binaryedgeToken := fmt.Sprintf(`binaryedge: [%s]`, token)
	_ = os.WriteFile(ConfigFile, []byte(binaryedgeToken), 0644)
	defer func() {
		_ = os.RemoveAll(ConfigFile)
	}()
	results, err := testutils.RunUncoverAndGetResults(debug, "-binaryedge", "1.1.1.1")

	if err != nil {
		return err
	}
	return expectResultsGreaterThanCount(results, 0)
}

type onypheTestcases struct{}

func (h onypheTestcases) Execute() error {
	token := os.Getenv("ONYPHE_API_KEY")
	if token == "" {
		return errors.New("missing onyphe api key")
	}

	onypheToken := fmt.Sprintf(`onyphe: [%s]`, token)
	_ = os.WriteFile(ConfigFile, []byte(onypheToken), 0644)
	defer func() {
		_ = os.RemoveAll(ConfigFile)
	}()
	results, err := testutils.RunUncoverAndGetResults(debug, "-onyphe", "google.com")

	if err != nil {
		return err
	}
	return expectResultsGreaterThanCount(results, 0)
}

type nerdydataTestcases struct{}

func (h nerdydataTestcases) Execute() error {
	token := os.Getenv("NERDYDATA_API_KEY")
	if token == "" {
		return errors.New("missing NerdyData api key")
	}
	nerdydataToken := fmt.Sprintf(`nerdydata: [%s]`, token)
	_ = os.WriteFile(ConfigFile, []byte(nerdydataToken), 0644)
	defer func() { _ = os.RemoveAll(ConfigFile) }()
	results, err := testutils.RunUncoverAndGetResults(debug, "-nerdydata", "jquery")
	if err != nil {
		return err
	}
	return expectResultsGreaterThanCount(results, 0)
}

type greynoiseTestcases struct{}

func (h greynoiseTestcases) Execute() error {
	token := os.Getenv("GREYNOISE_API_KEY")
	if token == "" {
		return errors.New("missing greynoise api key")
	}

	greynoiseToken := fmt.Sprintf(`greynoise: [%s]`, token)
	_ = os.WriteFile(ConfigFile, []byte(greynoiseToken), 0644)
	defer os.RemoveAll(ConfigFile)

	results, err := testutils.RunUncoverAndGetResults(debug, "-e", "greynoise", "-q", "tag:scanner")
	if err != nil {
		fmt.Fprintf(os.Stderr, "WARNING: greynoise query failed: %v\n", err)
		fmt.Fprintln(os.Stderr, "INFO: This may happen if you are using a Community API key. GNQL queries require an Enterprise API key.")
		return nil
	}

	if len(results) == 0 {
		fmt.Fprintln(os.Stderr, "INFO: greynoise returned 0 results.")
		fmt.Fprintln(os.Stderr, "NOTE: Community API keys cannot access GNQL queries. An Enterprise API key is required for this test to return data.")
		return nil
	}

	return nil
}