mirror of
https://github.com/Gouryella/drip.git
synced 2026-02-24 05:10:43 +00:00
c67cba6625
- Explicitly included golang.org/x/sys v0.38.0 in go.mod and removed indirect references - Optimized process liveness detection logic on Windows platforms using the golang.org/x/sys/windows package - Used OpenProcess and GetExitCodeProcess instead of Signal detection methods to improve accuracy - Updated command-line output and added an ASCII banner to improve user experience - Added blank lines to the github.com/spf13/cobra package to standardize import format
102 lines
2.6 KiB
Go
102 lines
2.6 KiB
Go
package cli
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"drip/internal/client/cli/ui"
|
|
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var (
|
|
// Version information
|
|
Version = "dev"
|
|
GitCommit = "unknown"
|
|
BuildTime = "unknown"
|
|
versionPlain bool
|
|
|
|
// Global flags
|
|
serverURL string
|
|
authToken string
|
|
verbose bool
|
|
insecure bool
|
|
)
|
|
|
|
var rootCmd = &cobra.Command{
|
|
Use: "drip",
|
|
Short: "Drip - Fast and secure tunnels to localhost",
|
|
Long: `Drip - High-performance tunneling service with TCP over TLS 1.3
|
|
|
|
Expose your local services to the internet securely and easily.
|
|
|
|
Configuration:
|
|
First time: Run 'drip config init' to set up server and token
|
|
Subsequent: Just run 'drip http <port>' or 'drip tcp <port>'
|
|
|
|
Examples:
|
|
drip config init # Set up configuration
|
|
drip http 3000 # HTTP tunnel
|
|
drip tcp 5432 # PostgreSQL tunnel
|
|
drip http 8080 --subdomain myapp # Custom subdomain
|
|
|
|
Features:
|
|
✓ TCP over TLS 1.3 (secure and fast)
|
|
✓ HTTP and TCP tunnel support
|
|
✓ Auto-save configuration
|
|
✓ Custom subdomains
|
|
✓ Authentication via token`,
|
|
}
|
|
|
|
func init() {
|
|
rootCmd.PersistentFlags().StringVarP(&serverURL, "server", "s", "", "Server address (e.g., tunnel.example.com:443)")
|
|
rootCmd.PersistentFlags().StringVarP(&authToken, "token", "t", "", "Authentication token")
|
|
rootCmd.PersistentFlags().BoolVarP(&verbose, "verbose", "v", false, "Verbose output")
|
|
rootCmd.PersistentFlags().BoolVarP(&insecure, "insecure", "k", false, "Skip TLS verification (testing only, NOT recommended)")
|
|
|
|
versionCmd.Flags().BoolVar(&versionPlain, "short", false, "Print version information without styling")
|
|
|
|
rootCmd.AddCommand(versionCmd)
|
|
// http and tcp commands are added in their respective init() functions
|
|
// config command is added in config.go init() function
|
|
}
|
|
|
|
var versionCmd = &cobra.Command{
|
|
Use: "version",
|
|
Short: "Print version information",
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
if versionPlain {
|
|
fmt.Printf("Version: %s\nGit Commit: %s\nBuild Time: %s\n", Version, GitCommit, BuildTime)
|
|
return
|
|
}
|
|
|
|
banner := `
|
|
____ _
|
|
| _ \ _ __(_)_ __
|
|
| | | | '__| | '_ \
|
|
| |_| | | | | |_) |
|
|
|____/|_| |_| .__/
|
|
|_|
|
|
D R I P
|
|
`
|
|
fmt.Println(ui.Info(
|
|
banner,
|
|
"",
|
|
ui.KeyValue("Version", Version),
|
|
ui.KeyValue("Git Commit", GitCommit),
|
|
ui.KeyValue("Build Time", BuildTime),
|
|
))
|
|
},
|
|
}
|
|
|
|
// Execute runs the root command
|
|
func Execute() error {
|
|
return rootCmd.Execute()
|
|
}
|
|
|
|
// SetVersion sets the version information
|
|
func SetVersion(version, commit, buildTime string) {
|
|
Version = version
|
|
GitCommit = commit
|
|
BuildTime = buildTime
|
|
}
|