Feat/wish

.github/workflows/go.yml

@@ -35,5 +35,5 @@ jobs:
         run: go test -v ./...
 
       - name: Build
-        run: go build -o tgcom ./cmd/tgcom
+        run: go build -o tgcom
 

cmd/root.go

@@ -0,0 +1,214 @@
+package cmd
+
+import (
+	"fmt"
+	"log"
+	"os"
+	"os/exec"
+	"runtime"
+	"strings"
+
+	tea "github.com/charmbracelet/bubbletea"
+	"github.com/dyne/tgcom/utils/modfile"
+	"github.com/dyne/tgcom/utils/tui"
+	"github.com/dyne/tgcom/utils/tui/modelutils"
+	"github.com/spf13/cobra"
+	"github.com/spf13/pflag"
+)
+
+var (
+	FileToRead string
+	inputFlag  modfile.Config
+	remotePath string
+	Tui        bool
+)
+
+var rootCmd = &cobra.Command{
+	Use:   "tgcom",
+	Short: "tgcom is a tool that allows users to comment or uncomment pieces of code",
+	Long: `tgcom is a CLI library written in Go that allows users to
+	comment or uncomment pieces of code. It supports many different
+	languages including Go, C, Java, Python, Bash, and many others...`,
+	Run: func(cmd *cobra.Command, args []string) {
+		if remotePath != "" {
+			executeRemoteCommand(remotePath)
+			return
+		}
+
+		if noFlagsGiven(cmd) {
+			customUsageFunc(cmd)
+			os.Exit(1)
+		}
+		ReadFlags(cmd)
+	},
+}
+
+func Execute() {
+	err := rootCmd.Execute()
+	if err != nil {
+		os.Exit(1)
+	}
+}
+
+func init() {
+	rootCmd.SetHelpFunc(customHelpFunc)
+	rootCmd.SetUsageFunc(customUsageFunc)
+
+	rootCmd.PersistentFlags().StringVarP(&FileToRead, "file", "f", "", "pass argument to the flag and will modify the file content")
+	rootCmd.PersistentFlags().StringVarP(&inputFlag.LineNum, "line", "l", "", "pass argument to line flag and will modify the line in the specified range")
+	rootCmd.PersistentFlags().BoolVarP(&inputFlag.DryRun, "dry-run", "d", false, "pass argument to dry-run flag and will print the result")
+	rootCmd.PersistentFlags().StringVarP(&inputFlag.Action, "action", "a", "toggle", "pass argument to action to comment/uncomment/toggle some lines")
+	rootCmd.PersistentFlags().StringVarP(&inputFlag.StartLabel, "start-label", "s", "", "pass argument to start-label to modify lines after start-label")
+	rootCmd.PersistentFlags().StringVarP(&inputFlag.EndLabel, "end-label", "e", "", "pass argument to end-label to modify lines up to end-label")
+	rootCmd.PersistentFlags().StringVarP(&inputFlag.Lang, "language", "L", "", "pass argument to language to specify the language of the input code")
+	rootCmd.PersistentFlags().StringVarP(&remotePath, "remote", "w", "", "pass remote user, host, and directory in the format user@host:/path/to/directory")
+	rootCmd.PersistentFlags().BoolVarP(&Tui, "tui", "t", false, "run the terminal user interface")
+	// Mark flags based on command name
+	rootCmd.PersistentPreRunE = func(cmd *cobra.Command, args []string) error {
+		if cmd.Name() != "server" {
+			cmd.MarkFlagsRequiredTogether("start-label", "end-label")
+			cmd.MarkFlagsMutuallyExclusive("line", "start-label")
+			cmd.MarkFlagsMutuallyExclusive("line", "end-label")
+			cmd.MarkFlagsOneRequired("file", "language", "remote", "tui")
+			cmd.MarkFlagsMutuallyExclusive("file", "language")
+		}
+		return nil
+	}
+
+	// Register server command
+	rootCmd.AddCommand(serverCmd)
+}
+
+func noFlagsGiven(cmd *cobra.Command) bool {
+	hasFlags := false
+	cmd.Flags().VisitAll(func(f *pflag.Flag) {
+		if f.Changed {
+			hasFlags = true
+		}
+	})
+	return !hasFlags
+}
+
+func ReadFlags(cmd *cobra.Command) {
+	if Tui {
+		currentDir, err := os.Getwd()
+		if err != nil {
+			fmt.Fprintf(os.Stderr, "Error getting current working directory: %v\n", err)
+			os.Exit(1)
+		}
+
+		// Initialize your model with the current directory
+		model := tui.Model{
+			State:         "FileSelection",
+			FilesSelector: modelutils.InitialModel(currentDir, 20),
+		}
+		clearScreen()
+		// Bubble Tea program
+		p := tea.NewProgram(model)
+
+		// Start the program
+		if _, err := p.Run(); err != nil {
+			os.Exit(1)
+		}
+		clearScreen()
+	} else {
+
+		if strings.Contains(FileToRead, ",") {
+			if cmd.Flags().Changed("line") {
+				fmt.Println("Warning: when passing multiple files to flag -f, don't use -l flag")
+			}
+			if cmd.Flags().Changed("start-label") && cmd.Flags().Changed("end-label") {
+				fileInfo := strings.Split(FileToRead, ",")
+				for i := 0; i < len(fileInfo); i++ {
+					inputFlag.Filename = fileInfo[i]
+					if err := modfile.ChangeFile(inputFlag); err != nil {
+						log.Fatal(err)
+					}
+				}
+			} else {
+				fileInfo := strings.Split(FileToRead, ",")
+				for i := 0; i < len(fileInfo); i++ {
+					if strings.Contains(fileInfo[i], ":") {
+						parts := strings.Split(fileInfo[i], ":")
+						if len(parts) != 2 {
+							log.Fatalf("invalid syntax. Use 'File:lines'")
+						}
+						inputFlag.Filename = parts[0]
+						inputFlag.LineNum = parts[1]
+						if err := modfile.ChangeFile(inputFlag); err != nil {
+							log.Fatal(err)
+						}
+					} else {
+						log.Fatalf("invalid syntax. Use 'File:lines'")
+					}
+				}
+			}
+		} else {
+			if cmd.Flags().Changed("line") || cmd.Flags().Changed("start-label") && cmd.Flags().Changed("end-label") {
+				inputFlag.Filename = FileToRead
+				if err := modfile.ChangeFile(inputFlag); err != nil {
+					log.Fatal(err)
+				}
+			} else {
+				log.Fatalf("Not specified what you want to modify: add -l flag or -s and -e flags")
+			}
+		}
+	}
+}
+
+func customHelpFunc(cmd *cobra.Command, args []string) {
+	fmt.Println("Tgcom CLI Application")
+	fmt.Println()
+	fmt.Println(cmd.Long)
+	fmt.Println()
+	fmt.Println("Usage:")
+	fmt.Println("  tgcom [flags]")
+	fmt.Println()
+	fmt.Println("Flags:")
+	cmd.Flags().VisitAll(func(flag *pflag.Flag) {
+		if flag.Name == "action" {
+			fmt.Printf("  -%s, --%s: %s (default: %s)\n", flag.Shorthand, flag.Name, flag.Usage, flag.DefValue)
+		} else {
+			fmt.Printf("  -%s, --%s: %s\n", flag.Shorthand, flag.Name, flag.Usage)
+		}
+	})
+	fmt.Println()
+	fmt.Println("Supported Languages:")
+	for lang := range modfile.CommentChars {
+		fmt.Println(lang)
+	}
+	fmt.Println()
+	fmt.Println("Examples:")
+	fmt.Println("  # Toggle comments on lines 1-5 in example.go")
+	fmt.Println("  tgcom -f example.go -l 1-5 -a toggle")
+	fmt.Println()
+	fmt.Println("  # Dry run: show the changes without modifying the file")
+	fmt.Println("  tgcom -f example.go -s START -e END -a toggle -d")
+}
+
+func customUsageFunc(cmd *cobra.Command) error {
+	fmt.Println(cmd.Short)
+	fmt.Println("Usage:")
+	fmt.Printf("  %s\n", cmd.UseLine())
+	fmt.Println()
+	fmt.Println("Flags:")
+	cmd.Flags().VisitAll(func(flag *pflag.Flag) {
+		if flag.Name == "action" {
+			fmt.Printf("  -%s, --%s: %s (default: %s)\n", flag.Shorthand, flag.Name, flag.Usage, flag.DefValue)
+		} else {
+			fmt.Printf("  -%s, --%s: %s\n", flag.Shorthand, flag.Name, flag.Usage)
+		}
+	})
+	return nil
+}
+func clearScreen() {
+	var cmd *exec.Cmd
+	switch runtime.GOOS {
+	case "windows":
+		cmd = exec.Command("cmd", "/c", "cls")
+	default:
+		cmd = exec.Command("clear")
+	}
+	cmd.Stdout = os.Stdout
+	cmd.Run()
+}

cmd/server.go

@@ -0,0 +1,125 @@
+package cmd
+
+import (
+	"fmt"
+	"io"
+	"log"
+	"os"
+	"os/exec"
+	"os/signal"
+	"strings"
+	"syscall"
+	"time"
+
+	"github.com/creack/pty"
+	"github.com/dyne/tgcom/utils/server"
+	"github.com/spf13/cobra"
+	"golang.org/x/term"
+)
+
+// serverCmd represents the server command
+var serverCmd = &cobra.Command{
+	Use:   "server",
+	Short: "Start the SSH server",
+	Long:  `Start the SSH server that allows remote interactions with tgcom.`,
+	Run: func(cmd *cobra.Command, args []string) {
+		server.StartServer()
+	},
+}
+
+func init() {
+	// Register the server command
+	rootCmd.AddCommand(serverCmd)
+}
+
+func executeRemoteCommand(remotePath string) {
+	parts := strings.SplitN(remotePath, "@", 2)
+	if len(parts) != 2 {
+		fmt.Println("Invalid format. Usage: tgcom -w user@remote:/path/folder")
+		os.Exit(1)
+	}
+
+	userHost := parts[0]
+	pathParts := strings.SplitN(parts[1], ":", 2)
+	if len(pathParts) != 2 {
+		fmt.Println("Invalid format. Usage: tgcom -w user@remote:/path/folder")
+		os.Exit(1)
+	}
+
+	host := pathParts[0]
+	dir := pathParts[1]
+
+	sshCmd := "ssh"
+	sshArgs := []string{"-t", "-p", "2222", userHost + "@" + host, "tgcom", dir}
+
+	// Start SSH command with PTY
+	if err := startSSHWithPTY(sshCmd, sshArgs); err != nil {
+		log.Fatalf("Error starting SSH with PTY: %v", err)
+	}
+}
+
+func startSSHWithPTY(cmd string, args []string) error {
+	// Create SSH command
+	sshCommand := exec.Command(cmd, args...)
+
+	// Start PTY
+	ptmx, err := pty.Start(sshCommand)
+	if err != nil {
+		return fmt.Errorf("failed to start PTY: %w", err)
+	}
+	defer ptmx.Close()
+
+	// Set terminal attributes
+	oldState, err := term.MakeRaw(int(os.Stdin.Fd()))
+	if err != nil {
+		return fmt.Errorf("failed to make terminal raw: %w", err)
+	}
+	defer term.Restore(int(os.Stdin.Fd()), oldState)
+
+	// Resize PTY to current terminal size
+	if err := resizePTY(ptmx); err != nil {
+		return fmt.Errorf("failed to resize PTY: %w", err)
+	}
+
+	// Forward input to PTY
+	go func() {
+		_, _ = io.Copy(ptmx, os.Stdin)
+	}()
+
+	// Forward output from PTY
+	go func() {
+		_, _ = io.Copy(os.Stdout, ptmx)
+	}()
+
+	// Handle PTY signals and resize
+	go func() {
+		ch := make(chan os.Signal, 1)
+		signal.Notify(ch, syscall.SIGWINCH)
+		for range ch {
+			if err := resizePTY(ptmx); err != nil {
+				log.Printf("Error resizing PTY: %v", err)
+			}
+		}
+	}()
+
+	// Wait for SSH command to finish
+	if err := sshCommand.Wait(); err != nil {
+		return fmt.Errorf("SSH command failed: %w", err)
+	}
+
+	// Wait a bit before exiting to ensure all output is processed
+	time.Sleep(100 * time.Millisecond)
+
+	return nil
+}
+
+func resizePTY(ptmx *os.File) error {
+	size, err := pty.GetsizeFull(os.Stdin)
+	if err != nil {
+		return fmt.Errorf("failed to get terminal size: %w", err)
+	}
+	if err := pty.Setsize(ptmx, size); err != nil {
+		return fmt.Errorf("failed to set terminal size: %w", err)
+	}
+	return nil
+}