package targets

import (
	"fmt"
	"git.veenee.ru/veenee/iplistget/common"
	"net"
	"os"
)

func init() {
	common.RegisterTarget(File{FileName: "iplistget.txt"})
}

type File struct {
	FileName string
}

func (s File) Set(filename string, iplist []net.IPNet) error {
	if filename != "" {
		s.FileName = filename
	}

	file, err := os.Create(s.FileName)
	if err != nil {
		return err
	}
	defer file.Close()

	for ind := range iplist {
		if _, err := file.WriteString(fmt.Sprintln(iplist[ind].String())); err != nil {
			return err
		}
	}

	return nil
}