diff --git a/cmd/http/httpsocket.go b/cmd/http/httpsocket.go deleted file mode 100644 index fa88314..0000000 --- a/cmd/http/httpsocket.go +++ /dev/null @@ -1,37 +0,0 @@ -package main - -import ( - "fmt" - "log" - "net/http" - "os" - "path/filepath" - - "packets/internal" - - "github.com/BurntSushi/toml" -) - -type ConfigTOML struct { - Config struct { - DefaultHttpPort int `toml:"httpPort"` - DefaultCacheDir string `toml:"cacheDir"` - } `toml:"Config"` -} - -func main() { - - internal.PacketsPackageDir() - var cfg ConfigTOML - toml.Decode(filepath.Join(internal.PacketsPackageDir(), "config.toml"), &cfg) - - pid := os.Getpid() - if err := os.WriteFile(filepath.Join(internal.PacketsPackageDir(), "http.pid"), []byte(fmt.Sprint(pid)), 0644); err != nil { - fmt.Println("error saving subprocess pid", err) - } - - fs := http.FileServer(http.Dir(cfg.Config.DefaultCacheDir)) - http.Handle("/", fs) - - log.Fatal(http.ListenAndServe(fmt.Sprintf(":%d", cfg.Config.DefaultHttpPort), nil)) -} diff --git a/cmd/packets/main.go b/cmd/packets/main.go deleted file mode 100644 index 517d27c..0000000 --- a/cmd/packets/main.go +++ /dev/null @@ -1,1852 +0,0 @@ -//go:build linux - -package main - -import ( - "archive/tar" - "bufio" - "crypto/sha256" - "database/sql" - "encoding/hex" - "errors" - "fmt" - "io" - "io/fs" - "log" - "net" - "net/http" - "net/url" - "os" - "os/exec" - "packets/internal" - "path" - "path/filepath" - "runtime" - "strconv" - "strings" - "syscall" - "time" - - "github.com/BurntSushi/toml" - "github.com/klauspost/compress/zstd" - "github.com/schollz/progressbar/v3" - "github.com/spf13/cobra" - lua "github.com/yuin/gopher-lua" - "golang.org/x/net/ipv4" - _ "modernc.org/sqlite" -) - -type UpgradeHelpAuto struct { - OgRealname string - NeoRealname string -} - -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 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"` -} - -// errors - -var ErrNotInstalled = errors.New("this package isn't installed") - -var serialPass uint -var cfg ConfigTOML -var PacketsDir string - -var isUpgrade bool -var upgradeHelper string - -var Unsafe bool - -var rootCmd = &cobra.Command{Use: "packets"} - -var installCmd = &cobra.Command{ - Use: "install {package} [packages...]", - Short: "Install a package", - Args: cobra.MinimumNArgs(1), - Run: func(cmd *cobra.Command, args []string) { - if os.Getuid() != 0 { - fmt.Println("please, run as root") - return - } - - db, err := sql.Open("sqlite", filepath.Join(PacketsDir, "index.db")) - if err != nil { - log.Fatal(err) - return - } - defer db.Close() - - nameToQuery := args[0] - 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(":: Founded %d versions for (%s)\n Select 1 to install\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.Print(">> ") - fmt.Fscan(bufio.NewReader(os.Stdin), &choice) - if choice > len(pkgs) || choice < 0 { - fmt.Println("invalid option") - return - } - - QueryInstall(pkgs[choice].Realname) - return - } - }, -} - -var serve = &cobra.Command{ - Use: "serve", - Short: "Start or stops the packets daemon", -} - -var serveInit = &cobra.Command{ - Use: "init", - Short: "Starts the packets daemon", - Run: func(cmd *cobra.Command, args []string) { - 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) - } - - } - }, -} -var serveStop = &cobra.Command{ - Use: "stop", - Short: "Stops the packets daemon", - Run: func(cmd *cobra.Command, args []string) { - 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) - } - }, -} - -var syncCmd = &cobra.Command{ - Use: "sync [url]", - Short: "Syncronizes with an remote index.db, and check if the data dir is changed", - Run: func(cmd *cobra.Command, args []string) { - if os.Getuid() != 0 { - fmt.Println("please, run as root") - return - } - - if len(args) == 0 { - fmt.Println("Starting to sync with https://servidordomal.fun/index.db") - if err := Sync("https://servidordomal.fun/index.db"); err != nil { - fmt.Println("failed to sync with https://servidordomal.fun/index.db : ", err) - return - } - fmt.Println("Sucessifully sync!") - return - } - - syncurl := args[0] - - 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!") - }, -} - -var removeCmd = &cobra.Command{ - Use: "remove {package} [packages...]", - Short: "Remove packages by the package realname", - Run: func(cmd *cobra.Command, args []string) { - if os.Getuid() != 0 { - fmt.Println("please, run as root") - return - } - for _, realname := range args { - - err := Unninstall(realname) - if err != nil { - if err == ErrNotInstalled { - fmt.Println(err.Error()) - } else { - log.Fatal(err) - } - return - } - } - }, -} -var listCmd = &cobra.Command{ - Use: "list", - Short: "List all installed packages", - Run: func(cmd *cobra.Command, args []string) { - if err := ListPackets(); err != nil { - return - } - }, -} - -var searchCmd = &cobra.Command{ - Use: "search [query name]", - Short: "List all packages in index.db", - Args: cobra.MaximumNArgs(1), - Run: func(cmd *cobra.Command, args []string) { - if err := Search(cmd, args); err != nil { - log.Fatal(err) - } - }, -} - -var upgradeCmd = &cobra.Command{ - Use: "upgrade [packages...]", - Short: "Upgrade package", - Run: func(cmd *cobra.Command, args []string) { - - isUpgrade = true - if len(args) == 0 { - if err := AutoUpgrade(); err != nil { - log.Fatal(err) - } - return - } - - for _, og_realname := range args { - - 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 { - if err == sql.ErrNoRows { - fmt.Printf("error: cant find %s\n", og_realname) - os.Exit(1) - } else { - log.Fatal(err) - } - } - - 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 { - if err == sql.ErrNoRows { - fmt.Printf("error: cant find %s\n", og_realname) - os.Exit(1) - } else { - log.Fatal(err) - } - } - - 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 an upgrade") - upgradeHelper = og_realname - QueryInstall(neo_realname) - } - }, -} - -var infoCmd = &cobra.Command{ - Use: "info {name}", - Short: "Get technical package information", - Args: cobra.MinimumNArgs(1), - Run: func(cmd *cobra.Command, args []string) { - - _, err := os.Stat(filepath.Join(PacketsDir, "index.db")) - if err != nil { - 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() - - var name, version, hash, description, mirrorsRaw, family string - var dependenciesRaw *string - var dependencies, mirrors []string - var serial int - - if err := db.QueryRow("SELECT name, version, hash, description, dependencies, mirrors, family, serial FROM packages WHERE realname = ?", args[0]).Scan(&name, &version, &hash, &description, &dependenciesRaw, &mirrorsRaw, &family, &serial); err != nil { - if err == sql.ErrNoRows { - fmt.Printf("cant find any result for %s\n", args[0]) - os.Exit(1) - } else { - log.Fatal(err) - } - - } - - if dependenciesRaw != nil { - dependencies = strings.Fields(*dependenciesRaw) - } else { - dependencies = []string{} - } - mirrors = strings.Fields(mirrorsRaw) - - fmt.Printf(":: Package %s : %s\n:: (%s)\n %s\n Dependencies: %v\n Mirrors: %v\n ---------------------------------\n Family: %s\n Serial: %d\n SHA256sum: %s\n", args[0], version, name, description, dependencies, mirrors, family, serial, hash) - }, -} - -func main() { - - if runtime.GOOS == "linux" { - if os.Geteuid() != 0 { - fmt.Println("Please run packets as root\nThis can change in future versions but currently it is required.") - return - } - } - - // ABOUT CONFIG.TOML - PacketsDir = internal.PacketsPackageDir() - _, err := os.Stat(filepath.Join(PacketsDir, "config.toml")) - if err != nil { - 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") - fmt.Println("Operation Sucess!") - } - } - - _, err = toml.DecodeFile(filepath.Join(PacketsDir, "config.toml"), &cfg) - if err != nil { - log.Fatal(err) - } - - // COMMANDS - rootCmd.AddCommand(serve) - - serve.AddCommand(serveInit) - serve.AddCommand(serveStop) - - rootCmd.AddCommand(installCmd) - installCmd.Flags().BoolVar(&Unsafe, "unsafe", false, "Execute lua script without sandbox") - - rootCmd.AddCommand(syncCmd) - rootCmd.AddCommand(removeCmd) - removeCmd.Flags().BoolVar(&Unsafe, "unsafe", false, "Execute lua script without sandbox") - - rootCmd.AddCommand(upgradeCmd) - upgradeCmd.Flags().BoolVar(&Unsafe, "unsafe", false, "Execute lua script without sandbox") - - rootCmd.AddCommand(listCmd) - rootCmd.AddCommand(searchCmd) - - rootCmd.AddCommand(infoCmd) - - rootCmd.Execute() - -} - -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? [Y/n]\n", cfg.Config.LastDataDir, cfg.Config.DataDir) - - var answer string - - fmt.Print(">> ") - 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") - os.Remove(cfg.Config.LastDataDir) - bar.Finish() - } - } - - f, err := os.Open(packagepath) - if err != nil { - return err - } - defer f.Close() - - zs, err := zstd.NewReader(f) - if err != nil { - return err - } - defer zs.Close() - - tr := tar.NewReader(zs) - - bar := progressbar.NewOptions64( - -1, - progressbar.OptionSetDescription("[2/2] Unpacking ..."), - progressbar.OptionSetWriter(os.Stdout), - progressbar.OptionClearOnFinish(), - ) - 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 - } - - n, err := io.Copy(out, tr) - out.Close() - if err != nil { - return err - } - bar.Add64(n) - - err = os.Chmod(absPath, os.FileMode(hdr.Mode)) - if err != nil { - return err - } - default: - - } - } - db, err := sql.Open("sqlite", filepath.Join(PacketsDir, "index.db")) - if err != nil { - log.Fatal(err) - return err - } - defer db.Close() - - var familyUuid string - - if err := db.QueryRow("SELECT family FROM packages WHERE realname = ?", name).Scan(&familyUuid); err != nil { - return err - } - - manifest.Info.Family = familyUuid - manifest.Info.Serial = serialPass - - file, err := os.OpenFile(filepath.Join(cfg.Config.DataDir, name, "manifest.toml"), os.O_WRONLY|os.O_TRUNC, 0644) - if err != nil { - return err - } - defer file.Close() - - encoder := toml.NewEncoder(file) - - encoder.Encode(&manifest) - - L := lua.NewState() - defer L.Close() - - if !Unsafe { - internal.SandboxDir = filepath.Join(cfg.Config.DataDir, name) - osObject := L.GetGlobal("os").(*lua.LTable) - ioObject := L.GetGlobal("io").(*lua.LTable) - - L.SetGlobal("package", lua.LNil) - L.SetGlobal("require", lua.LNil) - - 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)) - - // Packets build functions - build := L.NewTable() - - L.SetField(build, "requirements", L.NewFunction(internal.CompileRequirements)) - L.SetField(build, "compile", L.NewFunction(internal.LuaCompile)) - - L.SetGlobal("build", build) - - 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)) - osObject.RawSetString("mkdir", L.NewFunction(internal.LMkdir)) - - 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(filepath.Join(cfg.Config.DataDir, name, 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 { - if err := os.Remove(filepath.Join(cfg.Config.CacheDir, filename)); err != nil { - if os.IsNotExist(err) { - return fmt.Errorf("failed to download, status code not 200OK") - } else { - 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 isUpgrade { - if err := Upgrade(filepath.Join(cfg.Config.CacheDir, filename), upgradeHelper, 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 err != nil { - 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 { - - log.Fatal(err) - - } - 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 - } - defer simplecheck.Close() - - var mirrors string - var serial uint - var family string - - ResolvDependencies(realname) - - err = db.QueryRow("SELECT mirrors, serial, family FROM packages WHERE realname = ?", realname).Scan(&mirrors, &serial, &family) - if err != nil { - log.Fatal(err) - } - - if !isUpgrade { - - var recognize string - - err = simplecheck.QueryRow("SELECT realname FROM packages WHERE family = ?", family).Scan(&recognize) - if err != nil { - if err == sql.ErrNoRows { - } - } - if err == nil { - fmt.Println(":: Older package found, upgrading.") - isUpgrade = true - upgradeHelper = recognize - } - } - - if !strings.Contains(mirrors, " ") { - u, _ := url.Parse(mirrors) - filename := path.Base(u.Path) - - fmt.Println("Checking in cache dir") - if CheckDownloaded(filename) { - err := Validate(filename, realname) - if err != nil { - return - } - if isUpgrade { - if err := Upgrade(filepath.Join(cfg.Config.CacheDir, filename), upgradeHelper, 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\n", filename) - if CheckDownloaded(filename) { - err := Validate(filename, realname) - if err != nil { - continue - } else { - if isUpgrade { - if err := Upgrade(filepath.Join(cfg.Config.CacheDir, filename), upgradeHelper, 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 err != nil { - if os.IsNotExist(err) { - return false - } else { - return true - } - } - 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...") - - err := os.Remove(filepath.Join(cfg.Config.CacheDir, filename)) - if err != nil { - return err - } - QueryInstall(realname) - return nil - } - 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 err != nil { - 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.Print(">> ") - fmt.Scan(&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") - - return nil - - 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") - os.Remove(cfg.Config.LastDataDir) - bar.Finish() - - return nil - 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 - } - } - - 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 ErrNotInstalled - } - fmt.Printf(":: Sure you will remove %s ? [Y/n] ", realname) - var answer string - fmt.Print(">> ") - fmt.Scan(&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() - - if !Unsafe { - internal.SandboxDir = filepath.Join(cfg.Config.DataDir, realname) - osObject := L.GetGlobal("os").(*lua.LTable) - ioObject := L.GetGlobal("io").(*lua.LTable) - - L.SetGlobal("package", lua.LNil) - L.SetGlobal("require", lua.LNil) - - 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(path.Join(cfg.Config.DataDir, realname, manifest.Hooks.Remove))) - L.SetGlobal("data_dir", lua.LString(filepath.Join(cfg.Config.DataDir, realname, "data"))) - - L.SetGlobal("path_join", L.NewFunction(internal.Ljoin)) - - // Packets build functions - build := L.NewTable() - - L.SetField(build, "requirements", L.NewFunction(internal.CompileRequirements)) - L.SetField(build, "compile", L.NewFunction(internal.LuaCompile)) - - L.SetGlobal("build", build) - - 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)) - osObject.RawSetString("mkdir", L.NewFunction(internal.LMkdir)) - - 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("stderr", lua.LNil) - ioObject.RawSetString("stdin", lua.LNil) - ioObject.RawSetString("lines", lua.LNil) - ioObject.RawSetString("open", L.NewFunction(internal.SafeOpen)) - - } - - if err := L.DoFile(filepath.Join(cfg.Config.DataDir, realname, 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 ErrNotInstalled - } - - 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? [Y/n]\n", cfg.Config.LastDataDir, cfg.Config.DataDir) - - var answer string - fmt.Print(">> ") - 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") - 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), - progressbar.OptionClearOnFinish(), - ) - - 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)) - - dbx, err := sql.Open("sqlite", filepath.Join(PacketsDir, "index.db")) - if err != nil { - log.Fatal(err) - return err - } - defer dbx.Close() - - var familyUuid string - - if err := dbx.QueryRow("SELECT family FROM packages WHERE realname = ?", name).Scan(&familyUuid); err != nil { - return err - } - - manifest.Info.Family = familyUuid - manifest.Info.Serial = serialPass - - file, err := os.OpenFile(filepath.Join(cfg.Config.DataDir, name, "manifest.toml"), os.O_WRONLY|os.O_TRUNC, 0644) - if err != nil { - return err - } - defer file.Close() - - encoder := toml.NewEncoder(file) - - encoder.Encode(&manifest) - - L := lua.NewState() - defer L.Close() - - if !Unsafe { - internal.SandboxDir = filepath.Join(cfg.Config.DataDir, name) - osObject := L.GetGlobal("os").(*lua.LTable) - ioObject := L.GetGlobal("io").(*lua.LTable) - - L.SetGlobal("package", lua.LNil) - L.SetGlobal("require", lua.LNil) - - 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)) - - // Packets build functions - build := L.NewTable() - - L.SetField(build, "requirements", L.NewFunction(internal.CompileRequirements)) - L.SetField(build, "compile", L.NewFunction(internal.LuaCompile)) - - L.SetGlobal("build", build) - - 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)) - osObject.RawSetString("mkdir", L.NewFunction(internal.LMkdir)) - - 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("stderr", lua.LNil) - ioObject.RawSetString("stdin", lua.LNil) - ioObject.RawSetString("lines", lua.LNil) - ioObject.RawSetString("open", L.NewFunction(internal.SafeOpen)) - } - - if err := L.DoFile(filepath.Join(cfg.Config.DataDir, name, manifest.Hooks.Install)); err != nil { - log.Panic(err) - } - - fmt.Printf("\nPackage %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, - } - - _, 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 AutoUpgrade() error { - - index, err := sql.Open("sqlite", filepath.Join(PacketsDir, "index.db")) - if err != nil { - return err - } - defer index.Close() - - installed, err := sql.Open("sqlite", filepath.Join(PacketsDir, "installed.db")) - if err != nil { - return err - } - defer installed.Close() - - rows, err := installed.Query("SELECT family, serial, realname FROM packages") - if err != nil { - return err - } - defer rows.Close() - - var family, realname, ogRealname string - var serial, maxserial int - - var list []UpgradeHelpAuto - - for rows.Next() { - rows.Scan(&family, &serial, &ogRealname) - if err := index.QueryRow("SELECT serial, realname FROM packages WHERE serial = (SELECT MAX(serial) FROM packages WHERE family = ?) AND family = ?", family, family).Scan(&maxserial, &realname); err != nil { - return err - } - - if maxserial > serial { - brinco := &UpgradeHelpAuto{ - OgRealname: ogRealname, - NeoRealname: realname, - } - list = append(list, *brinco) - } - } - - for _, v := range list { - upgradeHelper = v.OgRealname - QueryInstall(v.NeoRealname) - } - - return nil -} - -func Search(cmd *cobra.Command, args []string) error { - db, err := sql.Open("sqlite", filepath.Join(PacketsDir, "index.db")) - if err != nil { - return err - } - defer db.Close() - - var name, realname, version, description string - var dependenciesRaw *string - var arrayDepedencies []string - - if len(args) == 0 { - - rows, err := db.Query("SELECT name, realname, version, description, dependencies FROM packages") - if err != nil { - return err - } - defer rows.Close() - - for rows.Next() { - if err := rows.Scan(&name, &realname, &version, &description, &dependenciesRaw); err != nil { - return err - } - - if dependenciesRaw != nil { - arrayDepedencies = strings.Fields(*dependenciesRaw) - } else { - arrayDepedencies = []string{} - } - fmt.Printf(":: Package %s : %s\n:: (%s)\n %s\n Dependencies: %v\n\n", realname, version, name, description, arrayDepedencies) - } - } else { - rows, err := db.Query("SELECT name, realname, version, description, dependencies FROM packages WHERE name = ?", args[0]) - if err != nil { - return err - } - defer rows.Close() - - for rows.Next() { - if err := rows.Scan(&name, &realname, &version, &description, &dependenciesRaw); err != nil { - return err - } - - if dependenciesRaw != nil { - arrayDepedencies = strings.Fields(*dependenciesRaw) - } else { - arrayDepedencies = []string{} - } - fmt.Printf(":: Package %s : %s\n:: (%s)\n %s\n Dependencies: %v\n\n", realname, version, name, description, arrayDepedencies) - } - - } - return nil -} - -/* -TODO: - asyncronous install with goroutines - packets search -*/ diff --git a/cmd/udp/udpsocket.go b/cmd/udp/udpsocket.go deleted file mode 100644 index 901baa6..0000000 --- a/cmd/udp/udpsocket.go +++ /dev/null @@ -1,65 +0,0 @@ -package main - -import ( - "fmt" - "log" - "net" - "os" - "packets/internal" - "path/filepath" - "strings" - - "github.com/BurntSushi/toml" -) - -type ConfigTOML struct { - Config struct { - HttpPort int `toml:"httpPort"` - CacheDir string `toml:"cacheDir"` - } `toml:"Config"` -} - -var cfg ConfigTOML - -func CheckDownloaded(filename string) bool { - - _, err := os.Stat(filepath.Join(cfg.Config.CacheDir)) - if os.IsNotExist(err) { - return false - } else { - return true - } - -} - -func main() { - pid := os.Getpid() - if err := os.WriteFile(filepath.Join(internal.PacketsPackageDir(), "udp.pid"), []byte(fmt.Sprint(pid)), 0644); err != nil { - fmt.Println("error saving subprocess pid", err) - } - toml.Decode(filepath.Join(internal.PacketsPackageDir(), "config.toml"), &cfg) - - addr := net.UDPAddr{IP: net.IPv4zero, Port: 1333} - conn, err := net.ListenUDP("udp", &addr) - if err != nil { - log.Fatal(err) - } - defer conn.Close() - buf := make([]byte, 1500) - - for { - n, remote, err := conn.ReadFromUDP(buf) - if err != nil { - log.Println("error creating udp socket", err) - } - msg := string(buf[:n]) - if !strings.HasPrefix(msg, "Q:") { - continue - } - filename := strings.TrimPrefix(msg, "Q:") - if CheckDownloaded(filename) { - reply := fmt.Sprintf("H:%s:%d", filename, cfg.Config.HttpPort) - conn.WriteToUDP([]byte(reply), remote) - } - } -}