Merge pull request #50 from issamoxix/feature/create-run-refactor-function

feat(tools): create RunRefactor function to handle refactor command processing

Author: issamoxix

main.go

@@ -50,6 +50,7 @@ func main() {
 				"    -c, c <arg>              	    : Get a prompt response based on the given argument.\n" +
 				"    login, -login, --login <token>  : Login to the application.\n" +
 				"    -l, l, live, --live       	    : Get a live prompt response.\n" +
+				"    refactor, r, -r, --refactor	    : Run the refactor command.\n" +
 				"    help, h, -h, --help      	    : Show this help message.")
 			return
 
@@ -80,6 +81,10 @@ func main() {
 			fmt.Println("Login successful! Token stored. ", token)
 			return
 
+		case "refactor", "r", "-r", "--refactor":
+			tools.RunRefactor()
+			return
+
 		default:
 			fmt.Printf("Unknown command: %s\nUse \"comit help\" to see available commands", os.Args[1])
 			return

services/openai/ai.go

@@ -21,6 +21,10 @@ type commitResponse struct {
 	Message []string `json:"message"`
 }
 
+type refactorResponse struct {
+	Message string `json:"message"`
+}
+
 type RequestData struct {
 	Code string `json:"code"`
 }
@@ -45,6 +49,44 @@ type BranchSelector interface {
 	SelectBranchMessage(messages []string, context string) error
 }
 
+func GetRefactorMessage(content string, token string) string {
+	var url = utils.ComitURL + "/refactore" + "?token=" + token
+	payload := RequestData{
+		Code: content,
+	}
+
+	jsonData, err := json.Marshal(payload)
+	if err != nil {
+		return "Error: " + err.Error()
+	}
+
+	resp, err := http.Post(url, "application/json", bytes.NewBuffer(jsonData))
+
+	if err != nil {
+		return "Error: " + err.Error()
+	}
+
+	if resp.StatusCode != 200 {
+		return "Error: " + resp.Status
+	}
+
+	defer resp.Body.Close()
+
+	body, err := io.ReadAll(resp.Body)
+	if err != nil {
+		return "Error: " + err.Error()
+	}
+
+	var data refactorResponse
+	if err := json.Unmarshal(body, &data); err != nil {
+		return "Error: " + err.Error()
+	}
+
+	lines := strings.Split(data.Message, "\n")
+	PretterPromptResponse(lines)
+	return "Ok"
+}
+
 func GetCommitMessage(content string, selector CommitMessageSelector, token string) string {
 	var url = utils.ComitURL + "/commit" + "?token=" + token
 	payload := RequestData{

services/utils/config.go

@@ -1,6 +1,6 @@
 package utils
 
-var Version = "0.9"
+var Version = "0.9.1"
 
 var (
 	UpdateLink = "https://github.com/issamoxix/Comit/releases/download/%s/%s"

services/utils/tools/commit_helper.go

@@ -131,6 +131,23 @@ func RunCommit() {
 	}
 }
 
+func RunRefactor() {
+	output, err := CheckStage()
+	if err != nil {
+		fmt.Println("Error:", err)
+		return
+	}
+	token, _ := auth.GetToken()
+	if token == "" {
+		token = "default"
+	}
+	messageStatus := ai.GetRefactorMessage(output, token)
+	if messageStatus != "Ok" {
+		fmt.Println("Something went wrong please try again")
+		return
+	}
+}
+
 func allEmpty(messages []string) bool {
 	for _, msg := range messages {
 		if msg != "" {