robogg133/packets
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

started to use cobra ( need to test this version)

Browse Source
This commit is contained in:
roboogg133
2025-08-12 17:55:57 -03:00
parent 45135b45a7
commit 331dba55f1
1 changed files with 176 additions and 161 deletions
Download Patch File Download Diff File Expand all files Collapse all files

209
cmd/packets/main.go
View File

@@ -28,6 +28,7 @@ import (
"github.com/BurntSushi/toml" "github.com/BurntSushi/toml"
"github.com/klauspost/compress/zstd" "github.com/klauspost/compress/zstd"
"github.com/schollz/progressbar/v3" "github.com/schollz/progressbar/v3"
"github.com/spf13/cobra"
lua "github.com/yuin/gopher-lua" lua "github.com/yuin/gopher-lua"
"golang.org/x/net/ipv4" "golang.org/x/net/ipv4"
_ "modernc.org/sqlite" _ "modernc.org/sqlite"
@@ -103,56 +104,21 @@ var serialPass uint
var cfg ConfigTOML var cfg ConfigTOML
var PacketsDir string var PacketsDir string
func main() { var isUpgrade bool
var upgradeHelper string
PacketsDir = internal.PacketsPackageDir() var rootCmd = &cobra.Command{Use: "packets"}
_, err := os.Stat(filepath.Join(PacketsDir, "config.toml")) var installCmd = &cobra.Command{
if os.IsNotExist(err) { Use: "install {package} [packages...]",
fmt.Println("can't find config.toml, generating a default one") Short: "Install a package",
Args: cobra.MinimumNArgs(1),
os.MkdirAll(PacketsDir, 0644) Run: func(cmd *cobra.Command, args []string) {
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 { if os.Getuid() != 0 {
fmt.Println("please, run as root") fmt.Println("please, run as root")
return return
} }
if len(os.Args) < 3 {
fmt.Println("usage: packets install <name>")
return
}
db, err := sql.Open("sqlite", filepath.Join(PacketsDir, "index.db")) db, err := sql.Open("sqlite", filepath.Join(PacketsDir, "index.db"))
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)
@@ -160,7 +126,7 @@ func main() {
} }
defer db.Close() defer db.Close()
nameToQuery := os.Args[2] nameToQuery := args[0]
var exist bool var exist bool
db.QueryRow("SELECT EXISTS(SELECT 1 FROM packages WHERE realname = ? LIMIT 1)", nameToQuery).Scan(&exist) db.QueryRow("SELECT EXISTS(SELECT 1 FROM packages WHERE realname = ? LIMIT 1)", nameToQuery).Scan(&exist)
if exist { if exist {
@@ -215,20 +181,18 @@ func main() {
QueryInstall(pkgs[choice].Realname) QueryInstall(pkgs[choice].Realname)
return return
} }
},
case "serve":
if os.Getuid() != 0 {
fmt.Println("please, run as root")
return
} }
if len(os.Args) < 3 { var serve = &cobra.Command{
fmt.Println("usage: packets serve <option>\navaiable options: init, stop") Use: "serve",
return Short: "Start or stops the packets daemon",
} }
switch os.Args[2] {
case "init":
var serveInit = &cobra.Command{
Use: "init",
Short: "Starts the packets daemon",
Run: func(cmd *cobra.Command, args []string) {
var sockets [2]string var sockets [2]string
sockets[0] = filepath.Join(PacketsDir, "udpsocket") sockets[0] = filepath.Join(PacketsDir, "udpsocket")
sockets[1] = filepath.Join(PacketsDir, "httpsocket") sockets[1] = filepath.Join(PacketsDir, "httpsocket")
@@ -242,10 +206,12 @@ func main() {
} }
} }
return },
}
case "stop": var serveStop = &cobra.Command{
Use: "stop",
Short: "Stops the packets daemon",
Run: func(cmd *cobra.Command, args []string) {
var pidfiles [2]string var pidfiles [2]string
pidfiles[0] = filepath.Join(PacketsDir, "http.pid") pidfiles[0] = filepath.Join(PacketsDir, "http.pid")
pidfiles[1] = filepath.Join(PacketsDir, "udp.pid") pidfiles[1] = filepath.Join(PacketsDir, "udp.pid")
@@ -259,17 +225,19 @@ func main() {
pid, _ := strconv.Atoi(string(data)) pid, _ := strconv.Atoi(string(data))
syscall.Kill(pid, syscall.SIGTERM) syscall.Kill(pid, syscall.SIGTERM)
} }
return },
default:
return
} }
case "sync":
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 { if os.Getuid() != 0 {
fmt.Println("please, run as root") fmt.Println("please, run as root")
return return
} }
if len(os.Args) < 3 { if len(args) == 0 {
fmt.Println("Starting to sync with https://servidordomal.fun/index.db") fmt.Println("Starting to sync with https://servidordomal.fun/index.db")
if err := Sync("https://servidordomal.fun/index.db"); err != nil { if err := Sync("https://servidordomal.fun/index.db"); err != nil {
fmt.Println("failed to sync with https://servidordomal.fun/index.db : ", err) fmt.Println("failed to sync with https://servidordomal.fun/index.db : ", err)
@@ -279,7 +247,7 @@ func main() {
return return
} }
syncurl := os.Args[2] syncurl := args[0]
fmt.Printf("Starting to sync with %s\n", syncurl) fmt.Printf("Starting to sync with %s\n", syncurl)
if err := Sync(syncurl); err != nil { if err := Sync(syncurl); err != nil {
@@ -287,45 +255,47 @@ func main() {
return return
} }
fmt.Println("Sucessifully sync!") fmt.Println("Sucessifully sync!")
return },
}
case "remove": 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 { if os.Getuid() != 0 {
fmt.Println("please, run as root") fmt.Println("please, run as root")
return return
} }
for _, realname := range args {
if len(os.Args) < 3 { err := Unninstall(realname)
fmt.Println("usage: packets remove <package-name>")
return
}
err := Unninstall(os.Args[2])
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)
return return
} }
return }
case "list": },
}
var listCmd = &cobra.Command{
Use: "list",
Short: "List all installed packages",
Run: func(cmd *cobra.Command, args []string) {
if err := ListPackets(); err != nil { if err := ListPackets(); err != nil {
return return
} }
return },
case "upgrade":
if os.Getuid() != 0 {
fmt.Println("please, run as root")
return
} }
if len(os.Args) < 3 { var upgradeCmd = &cobra.Command{
fmt.Println("usage: packets upgrade <realname>") Use: "upgrade [packages...]",
return Short: "Upgrade package",
Run: func(cmd *cobra.Command, args []string) {
if len(args) == 0 {
log.Fatal("Please insert mannualy the package realname to upgrade this command isn't done")
} }
og_realname := os.Args[2] for _, og_realname := range args {
db, err := sql.Open("sqlite", filepath.Join(PacketsDir, "index.db")) db, err := sql.Open("sqlite", filepath.Join(PacketsDir, "index.db"))
if err != nil { if err != nil {
@@ -365,12 +335,57 @@ func main() {
} }
fmt.Println("founded an upgrade") fmt.Println("founded an upgrade")
upgradeHelper = neo_realname
QueryInstall(neo_realname) QueryInstall(neo_realname)
default:
fmt.Printf(" %s it's not a command\n", cmd)
return
} }
},
}
func main() {
// ABOUT CONFIG.TOML
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)
}
// COMMANDS
rootCmd.AddCommand(installCmd)
rootCmd.AddCommand(serve)
serve.AddCommand(serveInit)
serve.AddCommand(serveStop)
rootCmd.AddCommand(syncCmd)
rootCmd.AddCommand(removeCmd)
rootCmd.AddCommand(upgradeCmd)
rootCmd.AddCommand(listCmd)
rootCmd.Execute()
} }
func Install(packagepath string, serial uint) error { func Install(packagepath string, serial uint) error {
@@ -697,8 +712,8 @@ func GetPackageByMirror(mirror string, realname string) error {
return err return err
} }
if os.Args[1] == "upgrade" { if isUpgrade {
if err := Upgrade(filepath.Join(cfg.Config.CacheDir, filename), os.Args[2], serialPass); err != nil { if err := Upgrade(filepath.Join(cfg.Config.CacheDir, filename), upgradeHelper, serialPass); err != nil {
return err return err
} }
return nil return nil
@@ -790,14 +805,14 @@ func QueryInstall(realname string) {
u, _ := url.Parse(mirrors) u, _ := url.Parse(mirrors)
filename := path.Base(u.Path) filename := path.Base(u.Path)
fmt.Println(":: Checking in cache dir") fmt.Println("Checking in cache dir")
if CheckDownloaded(filename) { if CheckDownloaded(filename) {
err := Validate(filename, realname) err := Validate(filename, realname)
if err != nil { if err != nil {
return return
} }
if os.Args[1] == "upgrade" { if isUpgrade {
if err := Upgrade(filepath.Join(cfg.Config.CacheDir, filename), os.Args[2], serialPass); err != nil { if err := Upgrade(filepath.Join(cfg.Config.CacheDir, filename), upgradeHelper, serialPass); err != nil {
log.Fatal(err) log.Fatal(err)
return return
} }
@@ -840,8 +855,8 @@ func QueryInstall(realname string) {
if err != nil { if err != nil {
continue continue
} else { } else {
if os.Args[1] == "upgrade" { if isUpgrade {
if err := Upgrade(filepath.Join(cfg.Config.CacheDir, filename), os.Args[2], serialPass); err != nil { if err := Upgrade(filepath.Join(cfg.Config.CacheDir, filename), upgradeHelper, serialPass); err != nil {
log.Fatal(err) log.Fatal(err)
return return
} }