From 8a28b0690b638536a070fd39d5b7b5a93626b3d2 Mon Sep 17 00:00:00 2001
From: tux <t@tux.rs>
Date: Tue, 4 Mar 2025 19:56:51 +0530
Subject: [PATCH] feat: rename client cmd to tcp

---
 cmd/client.go | 37 -------------------------------------
 cmd/server.go |  1 -
 cmd/tcp.go    | 33 +++++++++++++++++++++++++++++++++
 3 files changed, 33 insertions(+), 38 deletions(-)
 delete mode 100644 cmd/client.go
 create mode 100644 cmd/tcp.go

diff --git a/cmd/client.go b/cmd/client.go
deleted file mode 100644
index 6d85337..0000000
--- a/cmd/client.go
+++ /dev/null
@@ -1,37 +0,0 @@
-/*
-Copyright © 2024 tux <0xtux@pm.me>
-*/
-
-package cmd
-
-import (
-	"github.com/spf13/cobra"
-	"github.com/tuxdotrs/trok/internal/client"
-)
-
-// clientCmd represents the local command
-var clientCmd = &cobra.Command{
-	Use:   "client",
-	Short: "Initiates a local proxy to the remote server",
-	Long:  "Initiates a local proxy to the remote server",
-	Run: func(cmd *cobra.Command, args []string) {
-		serverAddr, err := cmd.Flags().GetString("serverAddr")
-		if err != nil {
-			panic(err)
-		}
-
-		localAddr, err := cmd.Flags().GetString("localAddr")
-		if err != nil {
-			panic(err)
-		}
-
-		client.Start(serverAddr, localAddr)
-	},
-}
-
-func init() {
-	rootCmd.AddCommand(clientCmd)
-
-	clientCmd.Flags().StringP("serverAddr", "s", "trok.cloud:1337", "Remote server address")
-	clientCmd.Flags().StringP("localAddr", "a", "0.0.0.0:80", "Local addr to expose")
-}
diff --git a/cmd/server.go b/cmd/server.go
index acecd1a..de987e4 100644
--- a/cmd/server.go
+++ b/cmd/server.go
@@ -27,5 +27,4 @@ func init() {
 	rootCmd.AddCommand(serverCmd)
 
 	serverCmd.Flags().StringP("addr", "a", "0.0.0.0:1337", "Addr for the server to listen on")
-	clientCmd.MarkFlagRequired("addr")
 }
diff --git a/cmd/tcp.go b/cmd/tcp.go
new file mode 100644
index 0000000..3beb78c
--- /dev/null
+++ b/cmd/tcp.go
@@ -0,0 +1,33 @@
+/*
+Copyright © 2024 tux <0xtux@pm.me>
+*/
+
+package cmd
+
+import (
+	"fmt"
+
+	"github.com/spf13/cobra"
+	"github.com/tuxdotrs/trok/internal/client"
+)
+
+var tcpCmd = &cobra.Command{
+	Use:   "tcp [port]",
+	Short: "Start TCP proxy",
+	Args:  cobra.ExactArgs(1),
+	Run: func(cmd *cobra.Command, args []string) {
+		port := args[0]
+
+		serverAddr, err := cmd.Flags().GetString("server")
+		if err != nil {
+			panic(err)
+		}
+
+		client.Start(serverAddr, fmt.Sprintf(":%s", port))
+	},
+}
+
+func init() {
+	rootCmd.AddCommand(tcpCmd)
+	tcpCmd.Flags().StringP("server", "s", "trok.cloud:1337", "Remote server address")
+}