config.go

package main

import (
	"encoding/json"
	"os"
	"strings"
	"time"
)

type config struct {
	Website string `json:"website"`
	Proxy   string `json:"proxy"`
	Threads int    `json:"threads"`

	Headers struct {
		UserAgent string `json:"user_agent"`
		Accept    string `json:"accept"`
	} `json:"headers"`

	Print struct {
		Enabled bool `json:"enabled"`
		GeoInfo bool `json:"geo_info"`
	} `json:"print"`

	Timeout struct {
		HTTP   int `json:"http"`
		Socks4 int `json:"socks4"`
		Socks5 int `json:"socks5"`
	} `json:"timeout"`
}

func (c *config) getTimeout() time.Duration {
	var sec int
	switch c.Proxy {
	case "socks4":
		sec = c.Timeout.Socks4
	case "socks5":
		sec = c.Timeout.Socks5
	default:
		sec = c.Timeout.HTTP
	}
	if sec < 1 {
		sec = 10
	}
	return time.Duration(sec) * time.Second
}

func loadConfig(path string) (*config, error) {
	data, err := os.ReadFile(path)
	if err != nil {
		return nil, err
	}

	cfg := &config{}
	if err := json.Unmarshal(data, cfg); err != nil {
		return nil, err
	}

	cfg.Proxy = strings.ToLower(cfg.Proxy)
	if cfg.Threads < 1 {
		cfg.Threads = 1000
	}

	return cfg, nil
}