Function readURLsFromInput

cmd/root.go:193–227

readURLsFromInput checks if the input is a file path containing URLs. If it is, returns all URLs from the file. Otherwise returns the input as a single URL.

(input string)

// readURLsFromInput checks if the input is a file path containing URLs.
// If it is, returns all URLs from the file. Otherwise returns the input as a single URL.
func readURLsFromInput(input string) []string {
	// If input looks like a URL (has scheme), treat it as a single URL
	if strings.HasPrefix(input, "http://") || strings.HasPrefix(input, "https://") {
		return []string{input}
	}

	// Try to open as a file
	file, err := os.Open(input)
	if err != nil {
		// Not a file, treat as a URL
		return []string{input}
	}
	defer file.Close()

	var urls []string
	scanner := bufio.NewScanner(file)
	for scanner.Scan() {
		line := strings.TrimSpace(scanner.Text())
		if line == "" || strings.HasPrefix(line, "#") {
			continue
		}
		urls = append(urls, line)
	}
	if err := scanner.Err(); err != nil {
		log.Printf("[!] Error reading URL file: %v", err)
		return []string{input}
	}

	if len(urls) == 0 {
		return []string{input}
	}

	log.Printf("[*] Loaded %d URLs from %s", len(urls), input)
	return urls
}

// initConfig reads in config file and ENV variables if set.
func initConfig() {