mirror of
https://github.com/cirruslabs/orchard.git
synced 2026-10-08 07:51:33 +02:00
Nikolay Edigaryev
10f56bb5e3
* proxy.Connections(): handle "use of closed network connection" error * Controller: less strict timeouts that work nicely for WebSockets * Worker: only attempt connect to the gRPC once our UID is known * Introduce "orchard ssh" and "orchard vnc" commands * Worker: prevent context leak by moving logic into a separate function * Fix linter errors * Port forwarding integration test * Check for "uname -mo" output
61 lines
1.6 KiB
Go
61 lines
1.6 KiB
Go
package command
|
|
|
|
import (
|
|
"github.com/cirruslabs/orchard/internal/command/context"
|
|
"github.com/cirruslabs/orchard/internal/command/controller"
|
|
"github.com/cirruslabs/orchard/internal/command/create"
|
|
deletepkg "github.com/cirruslabs/orchard/internal/command/deletecmd"
|
|
"github.com/cirruslabs/orchard/internal/command/dev"
|
|
"github.com/cirruslabs/orchard/internal/command/get"
|
|
"github.com/cirruslabs/orchard/internal/command/list"
|
|
"github.com/cirruslabs/orchard/internal/command/portforward"
|
|
"github.com/cirruslabs/orchard/internal/command/ssh"
|
|
"github.com/cirruslabs/orchard/internal/command/stop"
|
|
"github.com/cirruslabs/orchard/internal/command/vnc"
|
|
"github.com/cirruslabs/orchard/internal/command/worker"
|
|
"github.com/cirruslabs/orchard/internal/version"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
func NewRootCmd() *cobra.Command {
|
|
command := &cobra.Command{
|
|
Use: "orchard",
|
|
SilenceUsage: true,
|
|
SilenceErrors: true,
|
|
Version: version.FullVersion,
|
|
}
|
|
|
|
addGroupedCommands(command, "Working With Resources:",
|
|
create.NewCommand(),
|
|
get.NewCommand(),
|
|
list.NewCommand(),
|
|
portforward.NewCommand(),
|
|
ssh.NewCommand(),
|
|
vnc.NewCommand(),
|
|
deletepkg.NewCommand(),
|
|
stop.NewCommand(),
|
|
)
|
|
|
|
addGroupedCommands(command, "Administrative Tasks:",
|
|
context.NewCommand(),
|
|
controller.NewCommand(),
|
|
worker.NewCommand(),
|
|
dev.NewCommand(),
|
|
)
|
|
|
|
return command
|
|
}
|
|
|
|
func addGroupedCommands(parent *cobra.Command, title string, commands ...*cobra.Command) {
|
|
group := &cobra.Group{
|
|
ID: title,
|
|
Title: title,
|
|
}
|
|
parent.AddGroup(group)
|
|
|
|
for _, command := range commands {
|
|
command.GroupID = group.ID
|
|
parent.AddCommand(command)
|
|
}
|
|
}
|