101 lines
1.7 KiB
Go
101 lines
1.7 KiB
Go
package updates
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"runtime"
|
|
|
|
"github.com/Safing/portbase/database"
|
|
"github.com/Safing/portbase/modules"
|
|
)
|
|
|
|
var (
|
|
updateStoragePath string
|
|
)
|
|
|
|
func init() {
|
|
modules.Register("updates", prep, start, nil, "global", "database")
|
|
}
|
|
|
|
func prep() error {
|
|
updateStoragePath = filepath.Join(database.GetDatabaseRoot(), "updates")
|
|
|
|
err := checkUpdateDirs()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func start() error {
|
|
err := initUpdateStatusHook()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = ReloadLatest()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
go updater()
|
|
return nil
|
|
}
|
|
|
|
func stop() error {
|
|
return os.RemoveAll(filepath.Join(updateStoragePath, "tmp"))
|
|
}
|
|
|
|
func checkUpdateDirs() error {
|
|
// all
|
|
err := checkDir(filepath.Join(updateStoragePath, "all"))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// os_platform
|
|
err = checkDir(filepath.Join(updateStoragePath, fmt.Sprintf("%s_%s", runtime.GOOS, runtime.GOARCH)))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// tmp
|
|
err = checkDir(filepath.Join(updateStoragePath, "tmp"))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func checkDir(dirPath string) error {
|
|
f, err := os.Stat(dirPath)
|
|
if err == nil {
|
|
// file exists
|
|
if f.IsDir() {
|
|
return nil
|
|
}
|
|
err = os.Remove(dirPath)
|
|
if err != nil {
|
|
return fmt.Errorf("could not remove file %s to place dir: %s", dirPath, err)
|
|
}
|
|
err = os.MkdirAll(dirPath, 0755)
|
|
if err != nil {
|
|
return fmt.Errorf("could not create dir %s: %s", dirPath, err)
|
|
}
|
|
return nil
|
|
}
|
|
// file does not exist
|
|
if os.IsNotExist(err) {
|
|
err = os.MkdirAll(dirPath, 0755)
|
|
if err != nil {
|
|
return fmt.Errorf("could not create dir %s: %s", dirPath, err)
|
|
}
|
|
return nil
|
|
}
|
|
// other error
|
|
return fmt.Errorf("failed to access %s: %s", dirPath, err)
|
|
}
|