Feat #45: Add CLI commands to manage the DB
This commit is contained in:
parent
51f9be1486
commit
94a5fbfc51
4 changed files with 153 additions and 0 deletions
65
polyculeconnect/cmd/db/connect.go
Normal file
65
polyculeconnect/cmd/db/connect.go
Normal file
|
@ -0,0 +1,65 @@
|
|||
package db
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"os/exec"
|
||||
"syscall"
|
||||
|
||||
"git.faercol.me/faercol/polyculeconnect/polyculeconnect/cmd/utils"
|
||||
"git.faercol.me/faercol/polyculeconnect/polyculeconnect/config"
|
||||
"github.com/spf13/cobra"
|
||||
)
|
||||
|
||||
// connectCmd represents the db connect command
|
||||
var connectCmd = &cobra.Command{
|
||||
Use: "connect",
|
||||
Short: "Connect to the database",
|
||||
Long: `Connect to the database.`,
|
||||
Run: func(cmd *cobra.Command, args []string) {
|
||||
conf := utils.InitConfig("")
|
||||
if err := connectToDB(conf); err != nil {
|
||||
utils.Failf("Failed to connect to DB: %s", err.Error())
|
||||
}
|
||||
},
|
||||
}
|
||||
|
||||
func connectSQLite(conf *config.StorageConfig) error {
|
||||
path, err := exec.LookPath("sqlite3")
|
||||
if err != nil {
|
||||
if errors.Is(err, exec.ErrNotFound) {
|
||||
return errors.New("sqlite3 not installed")
|
||||
}
|
||||
return fmt.Errorf("failed to find sqlite3 executable: %w", err)
|
||||
}
|
||||
|
||||
if err := syscall.Exec(path, []string{path, conf.File}, nil); err != nil {
|
||||
return fmt.Errorf("failed to run sqlite3 command: %w", err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func connectToDB(conf *config.AppConfig) error {
|
||||
switch conf.StorageType {
|
||||
case string(config.Memory):
|
||||
return errors.New("no DB associated with memory storage")
|
||||
case string(config.SQLite):
|
||||
return connectSQLite(conf.StorageConfig)
|
||||
default:
|
||||
return fmt.Errorf("unsupported storage type %q", conf.StorageType)
|
||||
}
|
||||
}
|
||||
|
||||
func init() {
|
||||
dbCmd.AddCommand(connectCmd)
|
||||
|
||||
// Here you will define your flags and configuration settings.
|
||||
|
||||
// Cobra supports Persistent Flags which will work for this command
|
||||
// and all subcommands, e.g.:
|
||||
// dbCmd.PersistentFlags().String("foo", "", "A help for foo")
|
||||
|
||||
// Cobra supports local flags which will only run when this command
|
||||
// is called directly, e.g.:
|
||||
// dbCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
|
||||
}
|
27
polyculeconnect/cmd/db/db.go
Normal file
27
polyculeconnect/cmd/db/db.go
Normal file
|
@ -0,0 +1,27 @@
|
|||
package db
|
||||
|
||||
import (
|
||||
"git.faercol.me/faercol/polyculeconnect/polyculeconnect/cmd"
|
||||
"github.com/spf13/cobra"
|
||||
)
|
||||
|
||||
// dbCmd represents the db command
|
||||
var dbCmd = &cobra.Command{
|
||||
Use: "db",
|
||||
Short: "Manage the database",
|
||||
Long: `Manage the database.`,
|
||||
}
|
||||
|
||||
func init() {
|
||||
cmd.RootCmd.AddCommand(dbCmd)
|
||||
|
||||
// Here you will define your flags and configuration settings.
|
||||
|
||||
// Cobra supports Persistent Flags which will work for this command
|
||||
// and all subcommands, e.g.:
|
||||
// dbCmd.PersistentFlags().String("foo", "", "A help for foo")
|
||||
|
||||
// Cobra supports local flags which will only run when this command
|
||||
// is called directly, e.g.:
|
||||
// dbCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
|
||||
}
|
60
polyculeconnect/cmd/db/destroy.go
Normal file
60
polyculeconnect/cmd/db/destroy.go
Normal file
|
@ -0,0 +1,60 @@
|
|||
package db
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"os"
|
||||
|
||||
"git.faercol.me/faercol/polyculeconnect/polyculeconnect/cmd/utils"
|
||||
"git.faercol.me/faercol/polyculeconnect/polyculeconnect/config"
|
||||
"github.com/spf13/cobra"
|
||||
)
|
||||
|
||||
// destroyCmd represents the db destroy command
|
||||
var destroyCmd = &cobra.Command{
|
||||
Use: "destroy",
|
||||
Short: "Completely delete the current database",
|
||||
Long: `Delete the current database.`,
|
||||
Run: func(cmd *cobra.Command, args []string) {
|
||||
conf := utils.InitConfig("")
|
||||
if err := deleteDB(conf); err != nil {
|
||||
utils.Failf("Failed to connect to DB: %s", err.Error())
|
||||
}
|
||||
fmt.Println("DB deleted")
|
||||
},
|
||||
}
|
||||
|
||||
func deleteSqliteDB(path string) error {
|
||||
if err := os.Remove(path); err != nil {
|
||||
if errors.Is(err, os.ErrNotExist) { // if the file has already been deleted we don't want to fail here
|
||||
return nil
|
||||
}
|
||||
return fmt.Errorf("failed to delete SQLite file: %w", err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func deleteDB(conf *config.AppConfig) error {
|
||||
switch conf.StorageType {
|
||||
case string(config.Memory):
|
||||
return errors.New("no DB to delete in memory mode")
|
||||
case string(config.SQLite):
|
||||
return deleteSqliteDB(conf.StorageConfig.File)
|
||||
default:
|
||||
return fmt.Errorf("unsupported storage type %q", conf.StorageType)
|
||||
}
|
||||
}
|
||||
|
||||
func init() {
|
||||
dbCmd.AddCommand(destroyCmd)
|
||||
|
||||
// Here you will define your flags and configuration settings.
|
||||
|
||||
// Cobra supports Persistent Flags which will work for this command
|
||||
// and all subcommands, e.g.:
|
||||
// dbCmd.PersistentFlags().String("foo", "", "A help for foo")
|
||||
|
||||
// Cobra supports local flags which will only run when this command
|
||||
// is called directly, e.g.:
|
||||
// dbCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
|
||||
}
|
|
@ -4,6 +4,7 @@ import (
|
|||
"git.faercol.me/faercol/polyculeconnect/polyculeconnect/cmd"
|
||||
_ "git.faercol.me/faercol/polyculeconnect/polyculeconnect/cmd/app"
|
||||
_ "git.faercol.me/faercol/polyculeconnect/polyculeconnect/cmd/backend"
|
||||
_ "git.faercol.me/faercol/polyculeconnect/polyculeconnect/cmd/db"
|
||||
_ "git.faercol.me/faercol/polyculeconnect/polyculeconnect/cmd/serve"
|
||||
)
|
||||
|
||||
|
|
Loading…
Reference in a new issue