Added hunter.io functions and file writes.

2025-05-16 23:46:55 -04:00
parent 59ca1d4e92
commit 2caccbee9d
14 changed files with 2514 additions and 23 deletions
@@ -0,0 +1,161 @@
+package export
+
+import (
+	"dehasher/internal/files"
+	"dehasher/internal/sqlite"
+	"encoding/json"
+	"encoding/xml"
+	"fmt"
+	"gopkg.in/yaml.v3"
+	"os"
+)
+
+func WriteHunterDomainToFile(result sqlite.HunterDomainData, outputFile string, fileType files.FileType) error {
+	var data []byte
+	var err error
+
+	switch fileType {
+	case files.JSON:
+		data, err = json.MarshalIndent(result, "", "  ")
+	case files.XML:
+		data, err = xml.MarshalIndent(result, "", "  ")
+	case files.YAML:
+		data, err = yaml.Marshal(result)
+	case files.TEXT:
+		data = []byte(result.String())
+	default:
+		return err
+	}
+
+	if err != nil {
+		return err
+	}
+
+	filePath := fmt.Sprintf("%s.%s", outputFile, fileType.String())
+	return os.WriteFile(filePath, data, 0644)
+}
+
+func WriteHunterEmailToFile(result sqlite.HunterEmailFinderData, outputFile string, fileType files.FileType) error {
+	var data []byte
+	var err error
+
+	switch fileType {
+	case files.JSON:
+		data, err = json.MarshalIndent(result, "", "  ")
+	case files.XML:
+		data, err = xml.MarshalIndent(result, "", "  ")
+	case files.YAML:
+		data, err = yaml.Marshal(result)
+	case files.TEXT:
+		data = []byte(result.String())
+	default:
+		return err
+	}
+
+	if err != nil {
+		return err
+	}
+
+	filePath := fmt.Sprintf("%s.%s", outputFile, fileType.String())
+	return os.WriteFile(filePath, data, 0644)
+}
+
+func WriteHunterEmailVerifyToFile(result sqlite.HunterEmailVerifyData, outputFile string, fileType files.FileType) error {
+	var data []byte
+	var err error
+
+	switch fileType {
+	case files.JSON:
+		data, err = json.MarshalIndent(result, "", "  ")
+	case files.XML:
+		data, err = xml.MarshalIndent(result, "", "  ")
+	case files.YAML:
+		data, err = yaml.Marshal(result)
+	case files.TEXT:
+		data = []byte(result.String())
+	default:
+		return err
+	}
+
+	if err != nil {
+		return err
+	}
+
+	filePath := fmt.Sprintf("%s.%s", outputFile, fileType.String())
+	return os.WriteFile(filePath, data, 0644)
+}
+
+func WriteHunterCompanyEnrichmentToFile(result sqlite.CompanyData, outputFile string, fileType files.FileType) error {
+	var data []byte
+	var err error
+
+	switch fileType {
+	case files.JSON:
+		data, err = json.MarshalIndent(result, "", "  ")
+	case files.XML:
+		data, err = xml.MarshalIndent(result, "", "  ")
+	case files.YAML:
+		data, err = yaml.Marshal(result)
+	case files.TEXT:
+		data = []byte(result.String())
+	default:
+		return err
+	}
+
+	if err != nil {
+		return err
+	}
+
+	filePath := fmt.Sprintf("%s.%s", outputFile, fileType.String())
+	return os.WriteFile(filePath, data, 0644)
+}
+
+func WriteHunterPersonEnrichmentToFile(result sqlite.PersonData, outputFile string, fileType files.FileType) error {
+	var data []byte
+	var err error
+
+	switch fileType {
+	case files.JSON:
+		data, err = json.MarshalIndent(result, "", "  ")
+	case files.XML:
+		data, err = xml.MarshalIndent(result, "", "  ")
+	case files.YAML:
+		data, err = yaml.Marshal(result)
+	case files.TEXT:
+		data = []byte(result.String())
+	default:
+		return err
+	}
+
+	if err != nil {
+		return err
+	}
+
+	filePath := fmt.Sprintf("%s.%s", outputFile, fileType.String())
+	return os.WriteFile(filePath, data, 0644)
+}
+
+func WriteHunterCombinedEnrichmentToFile(result sqlite.CombinedData, outputFile string, fileType files.FileType) error {
+	var data []byte
+	var err error
+
+	switch fileType {
+	case files.JSON:
+		data, err = json.MarshalIndent(result, "", "  ")
+	case files.XML:
+		data, err = xml.MarshalIndent(result, "", "  ")
+	case files.YAML:
+		data, err = yaml.Marshal(result)
+	case files.TEXT:
+		data = []byte(result.String())
+	default:
+		return err
+	}
+
+	if err != nil {
+		return err
+	}
+
+	filePath := fmt.Sprintf("%s.%s", outputFile, fileType.String())
+	return os.WriteFile(filePath, data, 0644)
+}