packets/cmd/packets/main.go

//go:build linux

package main

import (
	"archive/tar"
	"bufio"
	"crypto/sha256"
	"database/sql"
	"encoding/hex"
	"fmt"
	"io"
	"io/fs"
	"log"
	"net"
	"net/http"
	"net/url"
	"os"
	"os/exec"
	"packets/internal"
	"path"
	"path/filepath"
	"strconv"
	"strings"
	"syscall"
	"time"

	"github.com/BurntSushi/toml"
	"github.com/klauspost/compress/zstd"
	"github.com/schollz/progressbar/v3"
	lua "github.com/yuin/gopher-lua"
	"golang.org/x/net/ipv4"
	_ "modernc.org/sqlite"
)

type ConfigTOML struct {
	Config struct {
		HttpPort           int    `toml:"httpPort"`
		CacheDir           string `toml:"cacheDir"`
		AutoDeleteCacheDir bool   `toml:"dayToDeleteCacheDir"`
		DaysToDelete       int    `toml:"daysToDelete"`
		DataDir            string `toml:"dataDir"`
		BinDir             string `toml:"binDir"`
		LastDataDir        string `toml:"lastDataDir"`
	} `toml:"Config"`
}

type IndexTOML struct {
	Name        string    `toml:"name"`
	Version     string    `toml:"version"`
	Author      string    `toml:"author"`
	Description string    `toml:"description"`
	CreatedAt   time.Time `toml:"createdAt"`
}

type CountingReader struct {
	R     io.Reader
	Total int64
}

func (c *CountingReader) Read(p []byte) (int, error) {
	n, err := c.R.Read(p)
	c.Total += int64(n)
	return n, err
}

type Installed struct {
	Realname     string
	Version      string
	Dependencies []string
	Family       string
	Serial       uint
}

type Peer struct {
	IP   net.IP
	Port int
}

type Quer1 struct {
	Realname    string
	Version     string
	Description string
}

type Manifest struct {
	Info struct {
		Name         string   `toml:"name"`
		Version      string   `toml:"version"`
		Description  string   `toml:"description"`
		Dependencies []string `toml:"dependencies"`
		Author       string   `toml:"author"`
		Family       string   `toml:"family"`
		Serial       uint     `toml:"serial"`
	} `toml:"Info"`
	Hooks struct {
		Install string `toml:"install"`
		Remove  string `toml:"remove"`
	} `toml:"Hooks"`
}

var serialPass uint
var cfg ConfigTOML
var PacketsDir string

func main() {

	PacketsDir = internal.PacketsPackageDir()

	_, err := os.Stat(filepath.Join(PacketsDir, "config.toml"))
	if os.IsNotExist(err) {
		fmt.Println("can't find config.toml, generating a default one")

		os.MkdirAll(PacketsDir, 0644)
		file, err := os.Create(filepath.Join(PacketsDir, "config.toml"))
		if err != nil {
			log.Fatal(err)
		}
		defer file.Close()

		cfg := internal.DefaultConfigTOML()

		encoder := toml.NewEncoder(file)

		if err := encoder.Encode(cfg); err != nil {
			log.Fatal(err)
		}
		file.WriteString("\n\n# BE CAREFULL CHANGING BIN_DIR, BECAUSE THE BINARIES DON'T MOVE AUTOMATICALLY\n#NEVER CHANGE lastDataDir\n")
		fmt.Println("Operation Sucess!")
	}

	_, err = toml.DecodeFile(filepath.Join(PacketsDir, "config.toml"), &cfg)
	if err != nil {
		log.Fatal(err)
	}

	if len(os.Args) < 2 {
		fmt.Println("invalid syntax")
		return
	}

	cmd := os.Args[1]

	switch cmd {
	case "install":
		if os.Getuid() != 0 {
			fmt.Println("please, run as root")
			return
		}

		if len(os.Args) < 3 {
			fmt.Println("usage: packets install <name>")
			return
		}

		db, err := sql.Open("sqlite", filepath.Join(PacketsDir, "index.db"))
		if err != nil {
			log.Fatal(err)
			return
		}
		defer db.Close()

		nameToQuery := os.Args[2]
		var exist bool
		db.QueryRow("SELECT EXISTS(SELECT 1 FROM packages WHERE realname = ?  LIMIT 1)", nameToQuery).Scan(&exist)
		if exist {
			QueryInstall(nameToQuery)
			return
		}

		rows, err := db.Query("SELECT realname, version, description FROM packages WHERE name = ?", nameToQuery)
		if err != nil {
			if strings.Contains(err.Error(), "file is not a database (26)") {
				fmt.Println("index.db corrupted")
				return
			}
			log.Panic(err)
			return
		}

		defer rows.Close()

		var pkgs []Quer1
		for rows.Next() {
			var q Quer1
			if err := rows.Scan(&q.Realname, &q.Version, &q.Description); err != nil {
				log.Fatal(err)
			}
			pkgs = append(pkgs, q)
		}
		switch len(pkgs) {
		case 0:
			fmt.Printf("can't find any results for %s\n", nameToQuery)
			return
		case 1:
			fmt.Printf("Founded 1 package for %s \n", nameToQuery)

			fmt.Printf("Downloading %s \n", pkgs[0].Realname)
			QueryInstall(pkgs[0].Realname)
			return

		default:
			fmt.Printf("Found %d versions of %s\n Select 1\n", len(pkgs), nameToQuery)
			for i, q := range pkgs {
				fmt.Printf("[%d] %s : %s\n %s\n", i, q.Realname, q.Version, q.Description)
			}
			var choice int

			fmt.Fscan(bufio.NewReader(os.Stdin), &choice)
			if choice > len(pkgs) || choice < 0 {
				fmt.Println("invalid option")
				return
			}

			QueryInstall(pkgs[choice].Realname)
			return
		}

	case "serve":
		if os.Getuid() != 0 {
			fmt.Println("please, run as root")
			return
		}

		if len(os.Args) < 3 {
			fmt.Println("usage: packets serve <option>\navaiable options: init, stop")
			return
		}
		switch os.Args[2] {
		case "init":

			var sockets [2]string
			sockets[0] = filepath.Join(PacketsDir, "udpsocket")
			sockets[1] = filepath.Join(PacketsDir, "httpsocket")

			for _, v := range sockets {
				abs, _ := filepath.Abs(v)
				cmd := exec.Command(abs)
				cmd.Stdout, cmd.Stderr = os.Stdout, os.Stderr
				if err := cmd.Start(); err != nil {
					log.Fatalf("failed to start %s: %v", v, err)
				}

			}
			return

		case "stop":

			var pidfiles [2]string
			pidfiles[0] = filepath.Join(PacketsDir, "http.pid")
			pidfiles[1] = filepath.Join(PacketsDir, "udp.pid")

			for _, v := range pidfiles {
				data, err := os.ReadFile(v)
				if err != nil {
					fmt.Println("cant read PID:", err)
					return
				}
				pid, _ := strconv.Atoi(string(data))
				syscall.Kill(pid, syscall.SIGTERM)
			}
			return
		default:
			return
		}
	case "sync":
		if os.Getuid() != 0 {
			fmt.Println("please, run as root")
			return
		}

		if len(os.Args) < 3 {
			fmt.Println("Starting to sync with https://servidordomal.fun/mirror/index.db")
			if err := Sync("https://servidordomal.fun/mirror/index.db"); err != nil {
				fmt.Println("failed to sync with https://servidordomal.fun/mirror/index.db : ", err)
				return
			}
			fmt.Println("Sucessifully sync!")
			return
		}

		syncurl := os.Args[2]

		fmt.Printf("Starting to sync with %s\n", syncurl)
		if err := Sync(syncurl); err != nil {
			fmt.Printf("failed to sync with %s : %e ", syncurl, err)
			return
		}
		fmt.Println("Sucessifully sync!")
		return

	case "remove":
		if os.Getuid() != 0 {
			fmt.Println("please, run as root")
			return
		}

		if len(os.Args) < 3 {
			fmt.Println("usage: packets remove <package-name>")
			return
		}

		err := Unninstall(os.Args[2])
		if err != nil {
			log.Fatal(err)
			return
		}
		return
	case "list":

		if err := ListPackets(); err != nil {
			return
		}
		return

	case "upgrade":

		if os.Getuid() != 0 {
			fmt.Println("please, run as root")
			return
		}

		if len(os.Args) < 3 {
			fmt.Println("usage: packets upgrade <realname>")
			return
		}

		og_realname := os.Args[2]

		db, err := sql.Open("sqlite", filepath.Join(PacketsDir, "index.db"))
		if err != nil {
			log.Fatal(err)
			return
		}
		defer db.Close()

		idb, err := sql.Open("sqlite", filepath.Join(PacketsDir, "installed.db"))
		if err != nil {
			log.Fatal(err)
			return
		}
		defer idb.Close()

		var family string
		if err := idb.QueryRow("SELECT family FROM packages WHERE realname = ?", og_realname).Scan(&family); err != nil {
			log.Fatal("line 239", err)
			return
		}

		var neo_realname string

		if err := db.QueryRow("SELECT realname FROM packages WHERE family = ? ORDER BY serial DESC LIMIT 1", family).Scan(&neo_realname); err != nil {
			log.Fatal("line 245", err)
			return
		}

		if neo_realname == og_realname {
			fmt.Printf("%s is up to date!\n", og_realname)
			return
		}

		if err := db.QueryRow("SELECT serial FROM packages WHERE family = ? ORDER BY serial DESC LIMIT 1", family).Scan(&serialPass); err != nil {
			log.Fatal("line 255", err)
			return
		}

		fmt.Println("founded upgrade")
		QueryInstall(neo_realname)

	default:
		fmt.Printf(" %s it's not a command\n", cmd)
		return
	}
}

func Install(packagepath string, serial uint) error {

	manifest, err := internal.ManifestReadXZ(packagepath)
	if err != nil {
		log.Panic(err)
	}

	name := manifest.Info.Name

	var destDir = filepath.Join(cfg.Config.DataDir, name)

	if cfg.Config.LastDataDir != cfg.Config.DataDir {
		fmt.Printf("Ooops... Data directory has been changed from (%s), to (%s), do you want to cancel the installation and Sync first?\n y/n? ", cfg.Config.LastDataDir, cfg.Config.DataDir)

		var answer string

		fmt.Scan(&answer)

		if answer == "y" || answer == "Y" {
			if err := os.MkdirAll(cfg.Config.DataDir, 0755); err != nil {
				return err
			}

			bar := progressbar.NewOptions64(-1,
				progressbar.OptionSetDescription("Moving ..."),
			)

			err := filepath.WalkDir(cfg.Config.LastDataDir, func(last string, d fs.DirEntry, walkErr error) error {
				if walkErr != nil {
					return walkErr
				}

				if last == cfg.Config.LastDataDir {
					return nil
				}

				rel, err := filepath.Rel(cfg.Config.LastDataDir, last)
				if err != nil {
					return err
				}
				dest := filepath.Join(cfg.Config.DataDir, rel)

				if d.IsDir() {
					return os.MkdirAll(dest, 0755)
				}

				return os.Rename(last, dest)

			})
			if err != nil {
				return err
			}

			f, err := os.OpenFile(filepath.Join(PacketsDir, "config.toml"), os.O_WRONLY, 0644)
			if err != nil {
				return err
			}

			cfg.Config.LastDataDir = cfg.Config.DataDir

			encoder := toml.NewEncoder(f)

			err = encoder.Encode(cfg)

			if err != nil {
				bar.Finish()
				return err
			}

			f.WriteString("\n\n# BE CAREFULL CHANGING BIN_DIR, BECAUSE THE BINARIES DON'T MOVE AUTOMATICALLY\n#NEVER CHANGE lastDataDir\n")
			os.Remove(cfg.Config.LastDataDir)
			bar.Finish()
		}
	}

	f, err := os.Open(packagepath)
	if err != nil {
		return err
	}
	defer f.Close()

	stat, _ := f.Stat()

	totalsize := stat.Size()

	counter := &CountingReader{R: f}

	zs, err := zstd.NewReader(f)
	if err != nil {
		return err
	}

	tr := tar.NewReader(zs)

	bar := progressbar.NewOptions64(
		totalsize,
		progressbar.OptionSetDescription("[2/2] Unpacking ..."),
		progressbar.OptionSetWriter(os.Stdout),
	)
	for {
		hdr, err := tr.Next()
		if err == io.EOF {
			break
		}
		if err != nil {
			return err
		}

		rel := filepath.Clean(hdr.Name)

		if rel == ".." || strings.HasPrefix(rel, ".."+string(os.PathSeparator)) {
			continue
		}

		if err := os.MkdirAll(destDir, 0755); err != nil {
			return err
		}

		absPath := filepath.Join(destDir, rel)

		switch hdr.Typeflag {

		case tar.TypeDir:
			err = os.MkdirAll(absPath, os.FileMode(hdr.Mode))

			if err != nil {
				return err
			}

		case tar.TypeReg:
			err = os.MkdirAll(filepath.Dir(absPath), 0755)
			if err != nil {
				return err
			}

			out, err := os.Create(absPath)
			if err != nil {
				return err
			}

			_, err = io.Copy(io.MultiWriter(out), tr)
			out.Close()
			if err != nil {
				return err
			}

			bar.Set(int(counter.Total))

			err = os.Chmod(absPath, os.FileMode(hdr.Mode))
			if err != nil {
				return err
			}
		default:

		}
	}

	L := lua.NewState()
	defer L.Close()

	osObject := L.GetGlobal("os").(*lua.LTable)
	ioObject := L.GetGlobal("io").(*lua.LTable)

	L.SetGlobal("packets_package_dir", lua.LString(cfg.Config.DataDir))
	L.SetGlobal("packets_bin_dir", lua.LString(cfg.Config.BinDir))
	L.SetGlobal("script", lua.LString(manifest.Hooks.Install))
	L.SetGlobal("data_dir", lua.LString(filepath.Join(cfg.Config.DataDir, name, "data")))

	L.SetGlobal("path_join", L.NewFunction(internal.Ljoin))

	osObject.RawSetString("execute", lua.LNil)
	osObject.RawSetString("exit", lua.LNil)
	osObject.RawSetString("getenv", lua.LNil)

	osObject.RawSetString("remove", L.NewFunction(internal.SafeRemove))
	osObject.RawSetString("rename", L.NewFunction(internal.SafeRename))
	osObject.RawSetString("copy", L.NewFunction(internal.SafeCopy))
	osObject.RawSetString("symlink", L.NewFunction(internal.SymbolicLua))

	ioObject.RawSetString("input", lua.LNil)
	ioObject.RawSetString("output", lua.LNil)
	ioObject.RawSetString("popen", lua.LNil)
	ioObject.RawSetString("tmpfile", lua.LNil)
	ioObject.RawSetString("stdout", lua.LNil)
	ioObject.RawSetString("stdeer", lua.LNil)
	ioObject.RawSetString("stdin", lua.LNil)
	ioObject.RawSetString("lines", lua.LNil)
	ioObject.RawSetString("open", L.NewFunction(internal.SafeOpen))

	if err := L.DoFile(manifest.Hooks.Install); err != nil {
		log.Panic(err)
	}

	bar.Finish()
	fmt.Printf("Package %s fully installed\n", name)

	var insert = Installed{
		Realname:     manifest.Info.Name,
		Version:      manifest.Info.Version,
		Dependencies: manifest.Info.Dependencies,
		Family:       manifest.Info.Family,
		Serial:       manifest.Info.Serial,
	}

	if err := AddToInstalledDB(insert); err != nil {
		log.Fatal(err)
		return err
	}
	return nil
}

func GetPackageByMirror(mirror string, realname string) error {

	db, err := sql.Open("sqlite", filepath.Join(PacketsDir, "index.db"))
	if err != nil {
		log.Fatal(err)
		return err
	}
	defer db.Close()

	var serial uint

	if err := db.QueryRow("SELECT serial FROM packages WHERE realname = ?", realname).Scan(&serial); err != nil {
		log.Fatal(err)
	}

	u, _ := url.Parse(mirror)
	filename := path.Base(u.Path)

	resp, err := http.Get(mirror)
	if err != nil {
		log.Panic("error doing get request, do you really have an internet connection?")
		return err
	}

	var domain = mirror
	var link bool

	if cont := strings.Contains(mirror, "https"); cont {
		link = true
		domain = strings.Replace(mirror, "https", "", 1)
	} else {
		link = true
		domain = strings.Replace(mirror, "http", "", 1)
	}
	if link {

		domain = strings.Replace(domain, "://", "", 1)
		slice := strings.SplitN(domain, "/", 2)

		domain = slice[0]
	}

	bar := progressbar.NewOptions64(resp.ContentLength,
		progressbar.OptionSetDescription(fmt.Sprintf("[1/2] Downloading from %s ...", domain)),
		progressbar.OptionShowBytes(true),
		progressbar.OptionSetPredictTime(true),
		progressbar.OptionShowCount(),
		progressbar.OptionClearOnFinish(),
	)

	defer resp.Body.Close()

	if resp.StatusCode != 200 {
		err := os.Remove(filepath.Join(cfg.Config.CacheDir, filename))
		if os.IsNotExist(err) {
			return fmt.Errorf("failed to download, status code not 200OK")
		} else if err != nil {
			return err
		}
		return fmt.Errorf("failed to download, status code not 200OK")
	}

	if err := os.MkdirAll(cfg.Config.CacheDir, 0755); err != nil {
		log.Fatal("error creating file for package ", err)
		return err
	}

	out, err := os.Create(filepath.Join(cfg.Config.CacheDir, filename))
	if err != nil {
		log.Fatal("error creating package ", err)
		return err
	}
	defer out.Close()

	_, err = io.Copy(io.MultiWriter(out, bar), resp.Body)
	if err != nil {
		err := os.Remove(filepath.Join(cfg.Config.CacheDir, filename))
		if err != nil {
			return err
		}
		return err
	}
	bar.Finish()

	err = Validate(filename, realname)
	if err != nil {
		return err
	}

	if os.Args[1] == "upgrade" {
		if err := Upgrade(filepath.Join(cfg.Config.CacheDir, filename), os.Args[2], serialPass); err != nil {
			return err
		}
		return nil
	}
	err = Install(filepath.Join(cfg.Config.CacheDir, filename), serial)
	if err != nil {
		return err
	}
	return nil

}
func ResolvDependencies(realname string) {

	db, err := sql.Open("sqlite", filepath.Join(PacketsDir, "index.db"))
	if err != nil {
		log.Fatal(err)
	}
	defer db.Close()

	var dependencies *string

	err = db.QueryRow("SELECT dependencies FROM packages WHERE realname = ?", realname).Scan(&dependencies)
	if err != nil {
		log.Panic(err)
		return
	}

	if dependencies == nil {
		return
	}

	dependencie := strings.Fields(*dependencies)

	for _, v := range dependencie {
		err := AlredySatisfied(v)
		if err != nil {
			fmt.Printf("error installing %v : %s", v, err.Error())
			continue
		}
		QueryInstall(v)
	}
}

func QueryInstall(realname string) {

	_, err := os.Stat(filepath.Join(PacketsDir, "index.db"))
	if os.IsNotExist(err) {
		fmt.Println("cant find index.db, please use sync first")
	}

	db, err := sql.Open("sqlite", filepath.Join(PacketsDir, "index.db"))
	if err != nil {
		log.Fatal("cant find index.db, please use sync first")
	}
	defer db.Close()

	simplecheck, err := sql.Open("sqlite", filepath.Join(PacketsDir, "installed.db"))
	if err == nil {

		var exist bool
		simplecheck.QueryRow("SELECT EXISTS(SELECT 1 FROM packages WHERE realname = ?  LIMIT 1)", realname).Scan(&exist)

		if exist {
			fmt.Println("Alredy installed!")
			simplecheck.Close()
			return
		}

		simplecheck.Close()
	}

	var mirrors string

	ResolvDependencies(realname)

	err = db.QueryRow("SELECT mirrors FROM packages WHERE realname = ?", realname).Scan(&mirrors)
	if err != nil {
		log.Panic(err)
		return
	}
	var serial uint
	err = db.QueryRow("SELECT serial FROM packages WHERE realname = ?", realname).Scan(&serial)
	if err != nil {
		log.Panic(err)
		return
	}

	if !strings.Contains(mirrors, " ") {
		u, _ := url.Parse(mirrors)
		filename := path.Base(u.Path)

		fmt.Println("Checking if the package exists")
		if CheckDownloaded(filename) {
			err := Validate(filename, realname)
			if err != nil {
				return
			}
			if os.Args[1] == "upgrade" {
				if err := Upgrade(filepath.Join(cfg.Config.CacheDir, filename), os.Args[2], serialPass); err != nil {
					log.Fatal(err)
					return
				}
				return
			}
			Install(filepath.Join(cfg.Config.CacheDir, filename), serial)
			return

		}
		fmt.Println("Asking in LAN for the package")
		peers := AskLAN(filename)
		answers := len(peers)
		if answers != 0 {
			for _, p := range peers {
				fmt.Printf("Downloading from %s\n", p.IP)
				if err := GetPackageByMirror(fmt.Sprintf("http://%s:%d/%s", p.IP, p.Port, filename), realname); err != nil {
					log.Println(err)
					break
				}
				fmt.Printf("Download failed!\n")
			}
		}
		if err := GetPackageByMirror(mirrors, realname); err != nil {
			log.Println(err)
			return
		}

	} else {

		fmt.Println("A mirror list was found")
		mirrorlist := strings.Fields(mirrors)

		for _, v := range mirrorlist {
			u, _ := url.Parse(v)
			filename := path.Base(u.Path)

			fmt.Printf("Checking for %s", filename)
			if CheckDownloaded(filename) {
				err := Validate(filename, realname)
				if err != nil {
					continue
				} else {
					if os.Args[1] == "upgrade" {
						if err := Upgrade(filepath.Join(cfg.Config.CacheDir, filename), os.Args[2], serialPass); err != nil {
							log.Fatal(err)
							return
						}
						break
					}
					Install(filepath.Join(cfg.Config.CacheDir, filename), serial)
					break
				}
			}
			fmt.Println("Checking for package in LAN")
			peers := AskLAN(filename)
			answers := len(peers)
			if answers != 0 {
				for _, p := range peers {
					fmt.Printf("Downloading from %s\n", v)
					if err := GetPackageByMirror(fmt.Sprintf("http://%s:%d/%s", p.IP, p.Port, filename), realname); err == nil {
						break
					}
					fmt.Printf("Failed!\n")
				}
			}
			fmt.Printf("Downloading from %s\n", v)
			if err := GetPackageByMirror(v, realname); err != nil {
				log.Println(err)
				break
			}

		}
	}

}

func CheckDownloaded(filename string) bool {

	_, err := os.Stat(filepath.Join(cfg.Config.CacheDir, filename))
	if os.IsNotExist(err) {
		return false
	} else {
		return true
	}

}

func Validate(filename string, realname string) error {

	db, err := sql.Open("sqlite", filepath.Join(PacketsDir, "index.db"))
	if err != nil {
		log.Fatal(err)
		return err
	}
	defer db.Close()

	downloaded, err := os.Open(filepath.Join(cfg.Config.CacheDir, filename))
	if err != nil {
		log.Fatal("error reading new file")
		return err
	}
	defer downloaded.Close()

	h := sha256.New()
	if _, err := io.Copy(h, downloaded); err != nil {
		fmt.Println("error doing sha256sum")
		return err
	}

	sum := h.Sum(nil)
	hashString := hex.EncodeToString(sum)

	var hashStringDB string

	err = db.QueryRow("SELECT hash FROM packages WHERE realname = ?", realname).Scan(&hashStringDB)
	if err != nil {
		log.Panic(err)
		return err
	}

	if hashString != hashStringDB {
		fmt.Println("tampered package, removing it...\nplease run the command again")

		err := os.Remove(filepath.Join(cfg.Config.CacheDir, filename))
		if err != nil {
			return err
		}
		return fmt.Errorf("the package isn't safe, alredy removed")
	}
	return nil
}

func AskLAN(filename string) []Peer {
	var peers []Peer
	query := []byte("Q:" + filename)

	pc, err := net.ListenPacket("udp", ":0")
	if err != nil {
		log.Panic("error starting udp socket:", err)
	}
	defer pc.Close()

	if pconn := ipv4.NewPacketConn(pc); pconn != nil {
		_ = pconn.SetTTL(1)
	}

	ifaces, _ := net.Interfaces()
	for _, ifc := range ifaces {
		if ifc.Flags&net.FlagUp == 0 || ifc.Flags&net.FlagLoopback != 0 {
			continue
		}

		addrs, _ := ifc.Addrs()
		for _, a := range addrs {
			ipnet, ok := a.(*net.IPNet)
			if !ok || ipnet.IP.To4() == nil {
				continue
			}

			bcast := broadcastAddr(ipnet.IP.To4(), ipnet.Mask)
			dst := &net.UDPAddr{IP: bcast, Port: 1333}

			_, err = pc.WriteTo(query, dst)
			if err != nil {
				log.Printf("[%s] can't send to  %s: %v", ifc.Name, bcast, err)
			}
		}
	}
	_ = pc.SetDeadline(time.Now().Add(2 * time.Second))
	buf := make([]byte, 1500)

	for {
		n, addr, err := pc.ReadFrom(buf)
		if err != nil {
			break
		}
		msg := string(buf[:n])

		if strings.HasPrefix(msg, "H:"+filename) {
			parts := strings.Split(msg, ":")
			port, _ := strconv.Atoi(parts[2])
			fmt.Printf("%s have the package\n", addr)
			peers = append(peers, Peer{IP: addr.(*net.UDPAddr).IP, Port: port})
		}
	}
	return peers
}

func broadcastAddr(ip net.IP, mask net.IPMask) net.IP {
	b := make(net.IP, len(ip))
	for i := range ip {
		b[i] = ip[i] | ^mask[i]
	}
	return b
}

func Sync(url string) error {
	resp, err := http.Get(url)
	if err != nil {
		return err
	}
	defer resp.Body.Close()
	_, err = os.Stat(filepath.Join(PacketsDir, "index.db"))

	if os.IsNotExist(err) {
		os.MkdirAll(PacketsDir, 0755)
	}
	f, err := os.Create(filepath.Join(PacketsDir, "index.db"))
	if err != nil {
		return err
	}

	if _, err = io.Copy(f, resp.Body); err != nil {
		return err
	}

	if cfg.Config.LastDataDir != cfg.Config.DataDir {
		fmt.Printf("Ooops... Data directory has been changed on %s do you want to move the packages from (%s), to (%s)\n", filepath.Join(PacketsDir, "config.toml"), cfg.Config.LastDataDir, cfg.Config.DataDir)
		fmt.Println("What you want to do?")
		fmt.Println("[y] Yes [n] No, [x] Ignore it and stop to show this message (not recommended)")
	}
	var answer string
	fmt.Scanln(&answer)

	switch answer {
	case "n":
		fmt.Println("Just ignoring...")

	case "x":
		f, err := os.OpenFile(filepath.Join(PacketsDir, "config.toml"), os.O_WRONLY, 0644)
		if err != nil {
			return err
		}

		cfg.Config.LastDataDir = cfg.Config.DataDir

		encoder := toml.NewEncoder(f)

		err = encoder.Encode(cfg)
		if err != nil {
			return err
		}
		f.WriteString("\n\n# BE CAREFULL CHANGING BIN_DIR, BECAUSE THE BINARIES DON'T MOVE AUTOMATICALLY\n#NEVER CHANGE lastDataDir\n")
		os.Remove(cfg.Config.LastDataDir)

	case "y":
		if err := os.MkdirAll(cfg.Config.DataDir, 0755); err != nil {
			return err
		}

		bar := progressbar.NewOptions64(-1,
			progressbar.OptionSetDescription("Moving ..."),
		)

		err := filepath.WalkDir(cfg.Config.LastDataDir, func(last string, d fs.DirEntry, walkErr error) error {
			if walkErr != nil {
				return walkErr
			}

			if last == cfg.Config.LastDataDir {
				return nil
			}

			rel, err := filepath.Rel(cfg.Config.LastDataDir, last)
			if err != nil {
				return err
			}
			dest := filepath.Join(cfg.Config.DataDir, rel)

			if d.IsDir() {
				return os.MkdirAll(dest, 0755)
			}

			return os.Rename(last, dest)

		})
		if err != nil {
			return err
		}

		f, err := os.OpenFile(filepath.Join(PacketsDir, "config.toml"), os.O_WRONLY, 0644)
		if err != nil {
			return err
		}

		cfg.Config.LastDataDir = cfg.Config.DataDir

		encoder := toml.NewEncoder(f)

		err = encoder.Encode(cfg)
		if err != nil {
			bar.Finish()
			return err
		}
		f.WriteString("\n\n# BE CAREFULL CHANGING BIN_DIR, BECAUSE THE BINARIES DON'T MOVE AUTOMATICALLY\n#NEVER CHANGE lastDataDir\n")
		os.Remove(cfg.Config.LastDataDir)
		bar.Finish()

	default:
		if err := os.MkdirAll(cfg.Config.DataDir, 0755); err != nil {
			return err
		}

		bar := progressbar.NewOptions64(-1,
			progressbar.OptionSetDescription("Moving ..."),
		)

		err := filepath.WalkDir(cfg.Config.LastDataDir, func(last string, d fs.DirEntry, walkErr error) error {
			if walkErr != nil {
				return walkErr
			}

			if last == cfg.Config.LastDataDir {
				return nil
			}

			rel, err := filepath.Rel(cfg.Config.LastDataDir, last)
			if err != nil {
				return err
			}
			dest := filepath.Join(cfg.Config.DataDir, rel)

			if d.IsDir() {
				return os.MkdirAll(dest, 0755)
			}

			return os.Rename(last, dest)

		})
		if err != nil {
			return err
		}
		bar.Finish()

	}

	return nil
}

func AddToInstalledDB(insert Installed) error {
	db, err := sql.Open("sqlite", filepath.Join(PacketsDir, "installed.db"))
	if err != nil {
		log.Fatal(err)
		return err
	}
	defer db.Close()

	_, err = db.Exec("CREATE TABLE IF NOT EXISTS packages (realname TEXT NOT NULL UNIQUE PRIMARY KEY, version TEXT NOT NULL, dependencies TEXT, name TEXT, family TEXT NOT NULL, serial INTEGER)")
	if err != nil {
		return err
	}

	if len(insert.Dependencies) == 0 {
		_, err = db.Exec("INSERT INTO packages (realname, version, family, serial) VALUES (?, ?, ?, ?)", insert.Realname, insert.Version, insert.Family, insert.Serial)
		if err != nil {
			return err
		}
		return nil
	}
	var query string
	for _, v := range insert.Dependencies {

		query = query + v + " "
	}

	query = query[:len(query)-1]

	_, err = db.Exec("INSERT INTO packages (realname, version, dependencies, family, serial) VALUES (?, ?, ?, ?, ?)", insert.Realname, insert.Version, query, insert.Family, insert.Serial)
	if err != nil {
		return err
	}

	return nil
}

func Unninstall(realname string) error {
	db, err := sql.Open("sqlite", filepath.Join(PacketsDir, "installed.db"))
	if err != nil {
		log.Fatal(err)
		return err
	}
	defer db.Close()

	var exist bool

	err = db.QueryRow("SELECT EXISTS(SELECT 1 FROM packages WHERE realname = ?  LIMIT 1)", realname).Scan(&exist)
	if err != nil {
		return err
	}

	if !exist {
		return fmt.Errorf("this package isn't installed")
	}
	fmt.Printf("Sure you will remove %s ? [Y/n] ", realname)
	var answer string
	fmt.Scanf("%s", &answer)

	if answer != "y" && answer != "Y" {
		return fmt.Errorf("operation cancelled")
	}

	var manifest Manifest

	toml.DecodeFile(filepath.Join(cfg.Config.DataDir, realname, "manifest.toml"), &manifest)

	L := lua.NewState()
	defer L.Close()

	osObject := L.GetGlobal("os").(*lua.LTable)
	ioObject := L.GetGlobal("io").(*lua.LTable)

	L.SetGlobal("packets_package_dir", lua.LString(cfg.Config.DataDir))
	L.SetGlobal("packets_bin_dir", lua.LString(cfg.Config.BinDir))
	L.SetGlobal("script", lua.LString(manifest.Hooks.Remove))
	L.SetGlobal("data_dir", lua.LString(filepath.Join(cfg.Config.DataDir, realname, "data")))

	L.SetGlobal("path_join", L.NewFunction(internal.Ljoin))

	osObject.RawSetString("execute", lua.LNil)
	osObject.RawSetString("exit", lua.LNil)
	osObject.RawSetString("getenv", lua.LNil)

	osObject.RawSetString("remove", L.NewFunction(internal.SafeRemove))
	osObject.RawSetString("rename", L.NewFunction(internal.SafeRename))
	osObject.RawSetString("copy", L.NewFunction(internal.SafeCopy))
	osObject.RawSetString("symlink", L.NewFunction(internal.SymbolicLua))

	ioObject.RawSetString("input", lua.LNil)
	ioObject.RawSetString("output", lua.LNil)
	ioObject.RawSetString("popen", lua.LNil)
	ioObject.RawSetString("tmpfile", lua.LNil)
	ioObject.RawSetString("stdout", lua.LNil)
	ioObject.RawSetString("stdeer", lua.LNil)
	ioObject.RawSetString("stdin", lua.LNil)
	ioObject.RawSetString("lines", lua.LNil)
	ioObject.RawSetString("open", L.NewFunction(internal.SafeOpen))

	if err := L.DoFile(manifest.Hooks.Remove); err != nil {
		log.Panic(err)
	}

	if err := os.RemoveAll(filepath.Join(cfg.Config.DataDir, realname)); err != nil {
		return err
	}

	_, err = db.Exec("DELETE FROM packages WHERE realname =  ?", realname)
	if err != nil {
		return err
	}

	fmt.Println("Sucessifully removed")
	return nil
}

func AlredySatisfied(realname string) error {

	db, err := sql.Open("sqlite", filepath.Join(PacketsDir, "installed.db"))
	if err != nil {
		log.Fatal(err)
		return err
	}
	defer db.Close()

	var exist bool

	err = db.QueryRow("SELECT EXISTS(SELECT 1 FROM packages WHERE realname = ?  LIMIT 1)", realname).Scan(&exist)
	if err != nil {
		return err
	}

	if !exist {
		return nil
	}
	return fmt.Errorf("conflict")
}

func ListPackets() error {

	db, err := sql.Open("sqlite", filepath.Join(PacketsDir, "installed.db"))
	if err != nil {
		log.Fatal(err)
		return err
	}
	defer db.Close()

	rows, err := db.Query("SELECT realname, version FROM packages")
	if err != nil {
		return err
	}

	var realname string
	var version string
	defer rows.Close()

	fmt.Println("Installed packages:")
	for rows.Next() {
		rows.Scan(&realname, &version)
		fmt.Printf("%s	%s\n", realname, version)
	}
	return nil
}

func Upgrade(packagepath string, og_realname string, serial uint) error {

	db, err := sql.Open("sqlite", filepath.Join(PacketsDir, "installed.db"))
	if err != nil {
		log.Fatal(err)
		return err
	}
	defer db.Close()

	var exist bool

	err = db.QueryRow("SELECT EXISTS(SELECT 1 FROM packages WHERE realname = ?  LIMIT 1)", og_realname).Scan(&exist)
	if err != nil {
		return err
	}

	if !exist {
		return fmt.Errorf("this package isn't installed")
	}

	if cfg.Config.LastDataDir != cfg.Config.DataDir {
		fmt.Printf("Ooops... Data directory has been changed from (%s), to (%s), do you want to cancel the installation and Sync first?\n y/n? ", cfg.Config.LastDataDir, cfg.Config.DataDir)

		var answer string

		fmt.Scan(&answer)

		if answer == "y" || answer == "Y" {
			if err := os.MkdirAll(cfg.Config.DataDir, 0755); err != nil {
				return err
			}

			bar := progressbar.NewOptions64(-1,
				progressbar.OptionSetDescription("Moving ..."),
			)

			err := filepath.WalkDir(cfg.Config.LastDataDir, func(last string, d fs.DirEntry, walkErr error) error {
				if walkErr != nil {
					return walkErr
				}

				if last == cfg.Config.LastDataDir {
					return nil
				}

				rel, err := filepath.Rel(cfg.Config.LastDataDir, last)
				if err != nil {
					return err
				}
				dest := filepath.Join(cfg.Config.DataDir, rel)

				if d.IsDir() {
					return os.MkdirAll(dest, 0755)
				}

				return os.Rename(last, dest)

			})

			if err != nil {
				return err
			}

			f, err := os.OpenFile(filepath.Join(PacketsDir, "config.toml"), os.O_WRONLY, 0644)
			if err != nil {
				return err
			}

			cfg.Config.LastDataDir = cfg.Config.DataDir

			encoder := toml.NewEncoder(f)

			err = encoder.Encode(cfg)

			if err != nil {
				bar.Finish()
				return err
			}

			f.WriteString("\n\n# BE CAREFULL CHANGING BIN_DIR, BECAUSE THE BINARIES DON'T MOVE AUTOMATICALLY\n#NEVER CHANGE lastDataDir\n")
			os.Remove(cfg.Config.LastDataDir)
			bar.Finish()
		}
	}

	manifest, err := internal.ManifestReadXZ(packagepath)
	if err != nil {
		log.Panic(err)
	}

	name := manifest.Info.Name

	fmt.Printf("Unpacking (%s) above (%s)\n", name, og_realname)

	var destDir = filepath.Join(cfg.Config.DataDir, og_realname)

	f, err := os.Open(packagepath)
	if err != nil {
		return err
	}
	stats, _ := f.Stat()
	totalSize := stats.Size()
	defer f.Close()

	counter := &CountingReader{R: f}

	zs, err := zstd.NewReader(f)
	if err != nil {
		return err
	}

	tr := tar.NewReader(zs)

	bar := progressbar.NewOptions64(
		totalSize,
		progressbar.OptionSetDescription("[2/2] Upgrading ..."),
		progressbar.OptionSetWriter(os.Stdout),
	)

	for {
		hdr, err := tr.Next()
		if err == io.EOF {
			break
		}
		if err != nil {
			return err
		}

		rel := filepath.Clean(hdr.Name)

		if rel == ".." || strings.HasPrefix(rel, ".."+string(os.PathSeparator)) {
			continue
		}

		if err := os.MkdirAll(destDir, 0755); err != nil {
			return err
		}

		absPath := filepath.Join(destDir, rel)

		switch hdr.Typeflag {

		case tar.TypeDir:
			err = os.MkdirAll(absPath, os.FileMode(hdr.Mode))

			if err != nil {
				return err
			}

		case tar.TypeReg:
			err = os.MkdirAll(filepath.Dir(absPath), 0755)
			if err != nil {
				return err
			}

			out, err := os.Create(absPath)
			if err != nil {
				return err
			}
			_, err = io.Copy(out, tr)
			out.Close()
			if err != nil {
				return err
			}

			bar.Set(int(counter.Total))

			err = os.Chmod(absPath, os.FileMode(hdr.Mode))
			if err != nil {
				return err
			}
		default:

		}

	}
	bar.Finish()

	os.Rename(destDir, filepath.Join(cfg.Config.DataDir, name))

	L := lua.NewState()
	defer L.Close()

	osObject := L.GetGlobal("os").(*lua.LTable)
	ioObject := L.GetGlobal("io").(*lua.LTable)

	L.SetGlobal("packets_package_dir", lua.LString(cfg.Config.DataDir))
	L.SetGlobal("packets_bin_dir", lua.LString(cfg.Config.BinDir))
	L.SetGlobal("script", lua.LString(manifest.Hooks.Install))
	L.SetGlobal("data_dir", lua.LString(filepath.Join(cfg.Config.DataDir, name, "data")))

	L.SetGlobal("path_join", L.NewFunction(internal.Ljoin))

	osObject.RawSetString("execute", lua.LNil)
	osObject.RawSetString("exit", lua.LNil)
	osObject.RawSetString("getenv", lua.LNil)

	osObject.RawSetString("remove", L.NewFunction(internal.SafeRemove))
	osObject.RawSetString("rename", L.NewFunction(internal.SafeRename))
	osObject.RawSetString("copy", L.NewFunction(internal.SafeCopy))
	osObject.RawSetString("symlink", L.NewFunction(internal.SymbolicLua))

	ioObject.RawSetString("input", lua.LNil)
	ioObject.RawSetString("output", lua.LNil)
	ioObject.RawSetString("popen", lua.LNil)
	ioObject.RawSetString("tmpfile", lua.LNil)
	ioObject.RawSetString("stdout", lua.LNil)
	ioObject.RawSetString("stdeer", lua.LNil)
	ioObject.RawSetString("stdin", lua.LNil)
	ioObject.RawSetString("lines", lua.LNil)
	ioObject.RawSetString("open", L.NewFunction(internal.SafeOpen))

	if err := L.DoFile(manifest.Hooks.Install); err != nil {
		log.Panic(err)
	}

	fmt.Printf("Package %s fully installed", name)

	var insert = Installed{
		Realname:     manifest.Info.Name,
		Version:      manifest.Info.Version,
		Dependencies: manifest.Info.Dependencies,
		Family:       manifest.Info.Family,
		Serial:       manifest.Info.Serial,
	}

	_, err = db.Exec("DELETE FROM packages WHERE realname =  ?", og_realname)
	if err != nil {
		return err
	}

	if err := AddToInstalledDB(insert); err != nil {
		return err
	}
	return nil
}

func SearchUpgrades(name string) error {

	db, err := sql.Open("sqlite", filepath.Join(PacketsDir, "index.db"))
	if err != nil {
		log.Fatal(err)
		return err
	}
	defer db.Close()

	return nil
}