Starting to do config file and change packets dir to /etc/packets
This commit is contained in:
@@ -11,6 +11,7 @@ require (
|
|||||||
)
|
)
|
||||||
|
|
||||||
require (
|
require (
|
||||||
|
github.com/BurntSushi/toml v1.5.0 // indirect
|
||||||
github.com/dustin/go-humanize v1.0.1 // indirect
|
github.com/dustin/go-humanize v1.0.1 // indirect
|
||||||
github.com/google/uuid v1.6.0 // indirect
|
github.com/google/uuid v1.6.0 // indirect
|
||||||
github.com/mattn/go-isatty v0.0.20 // indirect
|
github.com/mattn/go-isatty v0.0.20 // indirect
|
||||||
|
|||||||
@@ -1,3 +1,5 @@
|
|||||||
|
github.com/BurntSushi/toml v1.5.0 h1:W5quZX/G/csjUnuI8SUYlsHs9M38FC7znL0lIO+DvMg=
|
||||||
|
github.com/BurntSushi/toml v1.5.0/go.mod h1:ukJfTF/6rtPPRCnwkur4qwRxa8vTRFBF0uk2lLoLwho=
|
||||||
github.com/chengxilo/virtualterm v1.0.4 h1:Z6IpERbRVlfB8WkOmtbHiDbBANU7cimRIof7mk9/PwM=
|
github.com/chengxilo/virtualterm v1.0.4 h1:Z6IpERbRVlfB8WkOmtbHiDbBANU7cimRIof7mk9/PwM=
|
||||||
github.com/chengxilo/virtualterm v1.0.4/go.mod h1:DyxxBZz/x1iqJjFxTFcr6/x+jSpqN0iwWCOK1q10rlY=
|
github.com/chengxilo/virtualterm v1.0.4/go.mod h1:DyxxBZz/x1iqJjFxTFcr6/x+jSpqN0iwWCOK1q10rlY=
|
||||||
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
||||||
|
|||||||
@@ -1,3 +1,5 @@
|
|||||||
module httpsocket
|
module httpsocket
|
||||||
|
|
||||||
go 1.24.4
|
go 1.24.4
|
||||||
|
|
||||||
|
require github.com/BurntSushi/toml v1.5.0 // indirect
|
||||||
|
|||||||
2
src/httpsocket/go.sum
Normal file
2
src/httpsocket/go.sum
Normal file
@@ -0,0 +1,2 @@
|
|||||||
|
github.com/BurntSushi/toml v1.5.0 h1:W5quZX/G/csjUnuI8SUYlsHs9M38FC7znL0lIO+DvMg=
|
||||||
|
github.com/BurntSushi/toml v1.5.0/go.mod h1:ukJfTF/6rtPPRCnwkur4qwRxa8vTRFBF0uk2lLoLwho=
|
||||||
@@ -5,16 +5,28 @@ import (
|
|||||||
"log"
|
"log"
|
||||||
"net/http"
|
"net/http"
|
||||||
"os"
|
"os"
|
||||||
|
|
||||||
|
"github.com/BurntSushi/toml"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
type ConfigTOML struct {
|
||||||
|
Config struct {
|
||||||
|
DefaultHttpPort int `toml:"defaultHttpPort"`
|
||||||
|
DefaultCacheDir string `toml:"defaultCacheDir"`
|
||||||
|
} `toml:"Config"`
|
||||||
|
}
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
|
var cfg ConfigTOML
|
||||||
|
toml.Decode("opt/packets/packets/config.toml", &cfg)
|
||||||
|
|
||||||
pid := os.Getpid()
|
pid := os.Getpid()
|
||||||
if err := os.WriteFile("/opt/packets/packets/http.pid", []byte(fmt.Sprint(pid)), 0644); err != nil {
|
if err := os.WriteFile("./http.pid", []byte(fmt.Sprint(pid)), 0644); err != nil {
|
||||||
fmt.Println("error saving subprocess pid", err)
|
fmt.Println("error saving subprocess pid", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
fs := http.FileServer(http.Dir("/var/cache/packets"))
|
fs := http.FileServer(http.Dir(cfg.Config.DefaultCacheDir))
|
||||||
http.Handle("/", fs)
|
http.Handle("/", fs)
|
||||||
|
|
||||||
log.Fatal(http.ListenAndServe(fmt.Sprintf(":%d", 9123), nil))
|
log.Fatal(http.ListenAndServe(fmt.Sprintf(":%d", cfg.Config.DefaultHttpPort), nil))
|
||||||
}
|
}
|
||||||
|
|||||||
52
src/main.go
52
src/main.go
@@ -24,6 +24,7 @@ import (
|
|||||||
"syscall"
|
"syscall"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
|
"github.com/BurntSushi/toml"
|
||||||
"github.com/schollz/progressbar/v3"
|
"github.com/schollz/progressbar/v3"
|
||||||
"golang.org/x/net/ipv4"
|
"golang.org/x/net/ipv4"
|
||||||
_ "modernc.org/sqlite"
|
_ "modernc.org/sqlite"
|
||||||
@@ -31,6 +32,24 @@ import (
|
|||||||
"github.com/ulikunitz/xz"
|
"github.com/ulikunitz/xz"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
type ConfigTOML struct {
|
||||||
|
Config struct {
|
||||||
|
DefaultHttpPort int `toml:"defaultHttpPort"`
|
||||||
|
DefaultCacheDir string `toml:"defaultCacheDir"`
|
||||||
|
AutoDeleteCacheDir bool `toml:"dutoDeleteCacheDir"`
|
||||||
|
DaysToDelete int `toml:"daysToDelete"`
|
||||||
|
DataDir string `toml:"dataDir"`
|
||||||
|
} `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 {
|
type CountingReader struct {
|
||||||
R io.Reader
|
R io.Reader
|
||||||
Total int64
|
Total int64
|
||||||
@@ -71,10 +90,23 @@ type Manifest struct {
|
|||||||
Serial uint `json:"serial"`
|
Serial uint `json:"serial"`
|
||||||
}
|
}
|
||||||
|
|
||||||
var Serialpass uint
|
var serialPass uint
|
||||||
|
var cfg ConfigTOML
|
||||||
|
var PacketsDir string
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
|
|
||||||
|
out, _ := exec.Command("uname", "-s").Output()
|
||||||
|
if uname := strings.TrimSpace(string(out)); uname == "OpenTTY" {
|
||||||
|
PacketsDir = "/mnt"
|
||||||
|
} else {
|
||||||
|
PacketsDir = "/etc/packets"
|
||||||
|
}
|
||||||
|
_, err := toml.DecodeFile(filepath.Join(PacketsDir, "config.toml"), &cfg)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
if len(os.Args) < 2 {
|
if len(os.Args) < 2 {
|
||||||
fmt.Println("invalid syntax")
|
fmt.Println("invalid syntax")
|
||||||
return
|
return
|
||||||
@@ -94,7 +126,7 @@ func main() {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
db, err := sql.Open("sqlite", "/opt/packets/packets/index.db")
|
db, err := sql.Open("sqlite", filepath.Join(PacketsDir, "index.db"))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatal(err)
|
log.Fatal(err)
|
||||||
return
|
return
|
||||||
@@ -171,8 +203,8 @@ func main() {
|
|||||||
case "init":
|
case "init":
|
||||||
|
|
||||||
var sockets [2]string
|
var sockets [2]string
|
||||||
sockets[0] = "/opt/packets/packets/udpsocket"
|
sockets[0] = filepath.Join(PacketsDir, "udpsocket")
|
||||||
sockets[1] = "/opt/packets/packets/httpsocket"
|
sockets[1] = filepath.Join(PacketsDir, "httpsocket")
|
||||||
|
|
||||||
for _, v := range sockets {
|
for _, v := range sockets {
|
||||||
abs, _ := filepath.Abs(v)
|
abs, _ := filepath.Abs(v)
|
||||||
@@ -188,8 +220,8 @@ func main() {
|
|||||||
case "stop":
|
case "stop":
|
||||||
|
|
||||||
var pidfiles [2]string
|
var pidfiles [2]string
|
||||||
pidfiles[0] = "/opt/packets/packets/http.pid"
|
pidfiles[0] = filepath.Join(PacketsDir, "http.pid")
|
||||||
pidfiles[1] = "/opt/packets/packets/udp.pid"
|
pidfiles[1] = filepath.Join(PacketsDir, "udp.pid")
|
||||||
|
|
||||||
for _, v := range pidfiles {
|
for _, v := range pidfiles {
|
||||||
data, err := os.ReadFile(v)
|
data, err := os.ReadFile(v)
|
||||||
@@ -300,7 +332,7 @@ func main() {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
if err := db.QueryRow("SELECT serial FROM packages WHERE family = ? ORDER BY serial DESC LIMIT 1", family).Scan(&Serialpass); err != nil {
|
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)
|
log.Fatal("line 255", err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -580,7 +612,7 @@ func GetPackageByMirror(mirror string, realname string) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if os.Args[1] == "upgrade" {
|
if os.Args[1] == "upgrade" {
|
||||||
if err := Upgrade(fmt.Sprintf("/var/cache/packets/%s", filename), os.Args[2], Serialpass); err != nil {
|
if err := Upgrade(fmt.Sprintf("/var/cache/packets/%s", filename), os.Args[2], serialPass); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
@@ -679,7 +711,7 @@ func QueryInstall(realname string) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
if os.Args[1] == "upgrade" {
|
if os.Args[1] == "upgrade" {
|
||||||
if err := Upgrade(fmt.Sprintf("/var/cache/packets/%s", filename), os.Args[2], Serialpass); err != nil {
|
if err := Upgrade(fmt.Sprintf("/var/cache/packets/%s", filename), os.Args[2], serialPass); err != nil {
|
||||||
log.Fatal(err)
|
log.Fatal(err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -723,7 +755,7 @@ func QueryInstall(realname string) {
|
|||||||
continue
|
continue
|
||||||
} else {
|
} else {
|
||||||
if os.Args[1] == "upgrade" {
|
if os.Args[1] == "upgrade" {
|
||||||
if err := Upgrade(fmt.Sprintf("/var/cache/packets/%s", filename), os.Args[2], Serialpass); err != nil {
|
if err := Upgrade(fmt.Sprintf("/var/cache/packets/%s", filename), os.Args[2], serialPass); err != nil {
|
||||||
log.Fatal(err)
|
log.Fatal(err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user