feat: rewrite the storage plugin in golang

Refs #6814
This commit is contained in:
Jose Diaz-Gonzalez
2026-01-08 01:51:28 -05:00
parent b99e0b303d
commit 8af070ac00
26 changed files with 911 additions and 375 deletions

View File

@@ -0,0 +1,60 @@
package main
import (
"flag"
"fmt"
"os"
"strconv"
"strings"
"github.com/dokku/dokku/plugins/common"
)
const (
helpHeader = `Usage: dokku storage[:COMMAND]
Manage mounted volumes
Additional commands:`
helpContent = `
storage:ensure-directory [--chown option] <directory>, Creates a persistent storage directory in the recommended storage path
storage:list <app> [--format text|json], List bind mounts for app's container(s) (host:container)
storage:mount <app> <host-dir:container-dir>, Create a new bind mount
storage:report [<app>] [<flag>], Displays a storage report for one or more apps
storage:unmount <app> <host-dir:container-dir>, Remove an existing bind mount`
)
func main() {
flag.Usage = usage
flag.Parse()
cmd := flag.Arg(0)
switch cmd {
case "storage":
usage()
case "storage:help":
usage()
case "help":
result, err := common.CallExecCommand(common.ExecCommandInput{
Command: "ps",
Args: []string{"-o", "command=", strconv.Itoa(os.Getppid())},
})
if err == nil && strings.Contains(result.StdoutContents(), "--all") {
fmt.Println(helpContent)
} else {
fmt.Print("\n storage, Manage mounted volumes\n")
}
default:
dokkuNotImplementExitCode, err := strconv.Atoi(os.Getenv("DOKKU_NOT_IMPLEMENTED_EXIT"))
if err != nil {
fmt.Println("failed to retrieve DOKKU_NOT_IMPLEMENTED_EXIT environment variable")
dokkuNotImplementExitCode = 10
}
os.Exit(dokkuNotImplementExitCode)
}
}
func usage() {
common.CommandUsage(helpHeader, helpContent)
}

View File

@@ -0,0 +1,72 @@
package main
import (
"fmt"
"os"
"strings"
"github.com/dokku/dokku/plugins/common"
"github.com/dokku/dokku/plugins/storage"
flag "github.com/spf13/pflag"
)
func main() {
parts := strings.Split(os.Args[0], "/")
subcommand := parts[len(parts)-1]
var err error
switch subcommand {
case "default":
err = storage.CommandHelp()
case "ensure-directory":
args := flag.NewFlagSet("storage:ensure-directory", flag.ExitOnError)
chown := args.String("chown", "herokuish", "--chown: chown option (herokuish, heroku, paketo, root, false)")
args.Parse(os.Args[2:])
directory := args.Arg(0)
err = storage.CommandEnsureDirectory(directory, *chown)
case "list":
args := flag.NewFlagSet("storage:list", flag.ExitOnError)
format := args.String("format", "text", "--format: output format (text, json)")
args.Parse(os.Args[2:])
appName := args.Arg(0)
err = storage.CommandList(appName, *format)
case "mount":
args := flag.NewFlagSet("storage:mount", flag.ExitOnError)
args.Parse(os.Args[2:])
appName := args.Arg(0)
mountPath := args.Arg(1)
err = storage.CommandMount(appName, mountPath)
case "report":
args := flag.NewFlagSet("storage:report", flag.ExitOnError)
format := args.String("format", "stdout", "--format: output format (stdout, json)")
args.Parse(os.Args[2:])
osArgs, infoFlag, parseErr := common.ParseReportArgs("storage", args.Args())
if parseErr != nil {
err = parseErr
} else {
appName := ""
if len(osArgs) > 0 {
appName = osArgs[0]
}
if *format == "stdout" {
err = storage.CommandReport(appName, infoFlag)
} else {
err = storage.CommandReportSingleApp(appName, infoFlag, *format)
}
}
case "unmount":
args := flag.NewFlagSet("storage:unmount", flag.ExitOnError)
args.Parse(os.Args[2:])
appName := args.Arg(0)
mountPath := args.Arg(1)
err = storage.CommandUnmount(appName, mountPath)
default:
err = fmt.Errorf("Invalid plugin subcommand call: %s", subcommand)
}
if err != nil {
common.LogFailWithError(err)
}
}

View File

@@ -0,0 +1,34 @@
package main
import (
"flag"
"fmt"
"os"
"strings"
"github.com/dokku/dokku/plugins/common"
"github.com/dokku/dokku/plugins/storage"
)
func main() {
parts := strings.Split(os.Args[0], "/")
trigger := parts[len(parts)-1]
flag.Parse()
var err error
switch trigger {
case "install":
err = storage.TriggerInstall()
case "storage-list":
appName := flag.Arg(0)
phase := flag.Arg(1)
format := flag.Arg(2)
err = storage.TriggerStorageList(appName, phase, format)
default:
err = fmt.Errorf("Invalid plugin trigger call: %s", trigger)
}
if err != nil {
common.LogFailWithError(err)
}
}