package main

import (
	"flag"
	"fmt"
	"git.e3t.cc/e3team/pancheri"
	"gopkg.in/yaml.v2"
	"io"
	"os"
	"strings"
	"time"
)

func main() {
	hostsPath := flag.String("hostsfile", "", "Hosts file to convert to a dnsbl")
	voidIp := flag.String("voidip", "0.0.0.0", "IP that this hosts file uses as a void IP")
	printUsage := flag.Bool("help", false, "Print command line usage")

	flag.Parse()

	if *printUsage {
		flag.Usage()
		os.Exit(0)
	}

	f, err := os.Open(*hostsPath)
	if err != nil {
		fmt.Printf("error opening hosts file: %s", err)
		os.Exit(1)
	}

	buf := new(strings.Builder)
	_, err = io.Copy(buf, f)
	if err != nil {
		fmt.Printf("error reading hosts file: %s", err)
		os.Exit(1)
	}

	err = f.Close()
	if err != nil {
		fmt.Printf("error closing hosts file: %s", err)
		os.Exit(1)
	}

	split := strings.Split(buf.String(), "\n")

	blackhole := pancheri.BlackholeFile{
		DenyDomains: *new([]string),
	}

	for _, line := range split {
		if strings.HasPrefix(line, "#") {
			continue
		}
		if line == "" {
			continue
		}
		if !strings.HasPrefix(line, *voidIp) {
			continue
		}

		lineSplit := strings.Split(line, " ")

		if strings.Join(lineSplit[1:], " ") == *voidIp {
			continue
		}

		blackhole.DenyDomains = append(blackhole.DenyDomains, strings.Join(lineSplit[1:], " "))
	}

	marshaled, err := yaml.Marshal(blackhole)

	if err != nil {
		fmt.Printf("error saving blacklist file: %s", err)
		os.Exit(1)
	}

	fmt.Printf("# Compiled by pancheri-compile at %s\n", time.Now().Format(time.RFC3339))
	fmt.Printf("# %d hosts loaded from %s\n", len(blackhole.DenyDomains), *hostsPath)
	fmt.Printf("%s", marshaled)
}