package search

import (
	"encoding/base64"
	"encoding/json"
	"fmt"
	"strings"
	"sync"
	"time"

	"github.com/anotherhadi/eleakxir/backend/search/dataleak"
	"github.com/anotherhadi/eleakxir/backend/search/osint"
	"github.com/anotherhadi/eleakxir/backend/server"
)

type Query struct {
	Text           string
	Column         string // The column to search in (e.g., "email", "password", etc.
	ExactMatch     bool   // Whether to search for an exact match
	Folders        []string
	IncludeFolders []bool

	// Services
	Datawells     bool // Whether to include datawells in the search
	GithubRecon   bool // Whether to include github-recon in the search
	GravatarRecon bool // Whether to include gravatar-recon in the search
}

type Result struct {
	Id           string
	Date         time.Time
	Status       string // "queued", "pending", "completed", "error", "cancelled"
	Query        Query
	ResultsCount int // Total number of results found across all services

	LeakResult     dataleak.LeakResult
	GithubResult   osint.GithubResult
	GravatarResult osint.GravatarResult
}

func Search(s *server.Server, q Query, r *Result, mu *sync.RWMutex) {
	var wg sync.WaitGroup

	mu.Lock()
	r.Status = "pending"
	r.ResultsCount = 0
	mu.Unlock()

	wg.Add(3)
	go func() {
		if !q.Datawells {
			mu.Lock()
			r.LeakResult = dataleak.LeakResult{Inactive: true}
			mu.Unlock()
			wg.Done()
			return
		}
		leakResult := dataleak.Search(s, q.Text, q.Column, q.ExactMatch, q.IncludeFolders)
		mu.Lock()
		r.LeakResult = leakResult
		r.ResultsCount += len(leakResult.Rows)
		mu.Unlock()
		wg.Done()
	}()

	cleanQueryText := strings.TrimPrefix(q.Text, "^")
	cleanQueryText = strings.TrimSuffix(q.Text, "$")
	isEmail := false
	isUsername := false

	if q.Column == "email" || strings.HasSuffix(q.Column, "_email") ||
		q.Column == "username" || strings.HasSuffix(q.Column, "_username") ||
		q.Column == "" || q.Column == "all" {
		if isValidEmail(cleanQueryText) {
			isEmail = true
		} else if isValidUsername(cleanQueryText) {
			isUsername = true
		}
	}

	go func() {
		if !q.GithubRecon || !s.Settings.GithubRecon || (!isEmail && !isUsername) {
			mu.Lock()
			r.GithubResult = osint.GithubResult{Inactive: true}
			mu.Unlock()
			wg.Done()
			return
		}
		var githubResult osint.GithubResult
		if isEmail {
			githubResult = osint.GithubSearch(s, cleanQueryText, "email")
		} else if isUsername {
			githubResult = osint.GithubSearch(s, cleanQueryText, "username")
		}
		mu.Lock()
		r.GithubResult = githubResult
		if githubResult.EmailResult != nil && githubResult.EmailResult.Commits != nil {
			r.ResultsCount += len(githubResult.EmailResult.Commits)
		}
		if githubResult.EmailResult != nil && githubResult.EmailResult.Spoofing != nil && githubResult.EmailResult.Spoofing.Username != "" {
			r.ResultsCount += 1
		}
		if githubResult.UsernameResult != nil && githubResult.UsernameResult.Commits != nil {
			r.ResultsCount += len(githubResult.UsernameResult.Commits)
		}
		mu.Unlock()
		wg.Done()
	}()

	go func() {
		if !q.GravatarRecon || !s.Settings.GravatarRecon || !isEmail {
			mu.Lock()
			r.GravatarResult = osint.GravatarResult{Inactive: true}
			mu.Unlock()
			wg.Done()
			return
		}
		gravatarResult := osint.GravatarSearch(s, cleanQueryText)
		mu.Lock()
		r.GravatarResult = gravatarResult
		if gravatarResult.Results != nil {
			r.ResultsCount += len(gravatarResult.Results)
		}
		mu.Unlock()
		wg.Done()
	}()

	wg.Wait()

	mu.Lock()
	if r.LeakResult.Error != "" {
		r.Status = "error"
	} else {
		r.Status = "completed"
	}
	mu.Unlock()
}

func EncodeQueryID(q Query, dataleaksCount uint64) string {
	raw, _ := json.Marshal(q)
	return fmt.Sprintf("%d:%s", dataleaksCount, base64.URLEncoding.EncodeToString(raw))
}

func isValidEmail(email string) bool {
	if !strings.Contains(email, "@") || !strings.Contains(email, ".") {
		return false
	}
	if strings.HasPrefix(email, "@") || strings.HasSuffix(email, "@") {
		return false
	}
	if strings.Contains(email, " ") {
		return false
	}
	return true
}

func isValidUsername(username string) bool {
	if len(username) < 1 || len(username) > 39 {
		return false
	}
	if strings.Contains(username, " ") {
		return false
	}
	return true
}