refactor: enhance WriteLine function and update main to allow dynamic URL input

Author: 9dl

helper/interface.go

@@ -53,9 +53,13 @@ func WriteLine(option, value string, newLine, padding bool) {
 	fmt.Print(optionColor)
 	fmt.Print(gray("] "))
 
-	if newLine {
-		fmt.Println(white(value))
-	} else {
+	if option == ">" {
 		fmt.Print(white(value))
+	} else {
+		if newLine {
+			fmt.Println(white(value))
+		} else {
+			fmt.Print(white(value))
+		}
 	}
 }

main.go

@@ -11,9 +11,17 @@ import (
 )
 
 func main() {
-	urls := []string{
-		"http://localhost:5000/product?id=1",
-	}
+
+	urls := []string{} // ToDo: Add File Upload
+
+	helper.ClearScreen()
+	helper.WriteLine("!", "GoInjection", true, false)
+	helper.WriteLine("!", "Developed By Fourier for Axion Security", true, false)
+	helper.WriteLine("!", "https://github.com/Axion-Security/GoInjection", true, false)
+	helper.WriteLine(">", "URL: ", true, false)
+	var input string
+	fmt.Scanln(&input)
+	urls = append(urls, input)
 
 	for _, u := range urls {
 		cleanedURLs, err := helper.GetUrls(u)