ai-malware.go

package main

import (
	"flag"
	"fmt"
	"log"

	"ai-malware/template"  // Use module path instead of relative path
)

func main() {
	// Print cool green banner with AI robot icon and text "AI-Malware"
	  banner := `
 █████╗ ██╗     ███╗   ███╗ █████╗ ██╗      ██╗    ██╗ █████╗ ██████╗ ███████╗
██╔══██╗██║     ████╗ ████║██╔══██╗██║      ██║    ██║██╔══██╗██╔══██╗██╔════╝
███████║██║     ██╔████╔██║███████║██║      ██║ █╗ ██║███████║██████╔╝█████╗  
██╔══██║██║ ══  ██║╚██╔╝██║██╔══██║██║      ██║███╗██║██╔══██║██╔██══╝██╔══╝  
██║  ██║██║     ██║ ╚═╝ ██║██║  ██║███████╗ ╚███╔███╔╝██║  ██║██║  ██ ███████╗
╚═╝  ╚═╝╚═╝     ╚═╝     ╚═╝╚═╝  ╚═╝╚══════╝  ╚══╝╚══╝ ╚═╝  ╚═╝╚═╝  ╚═╝╚══════╝
[🤖 AI-Powered]
@github/Elmerikh
`
	fmt.Print(banner)

	ip := flag.String("ip", "127.0.0.1", "(optional) Target IP address,works with the default prompt to generate a reverse shell")
	port := flag.Int("port", 8080, "(optional) Target port ,works with the default prompt to generate a reverse shell")
	custom := flag.String("custom", "", "(optional) Custom prompt for code generation")
	testMode := flag.Bool("test", false, "(optional) Run in test mode , to test prompts and see ai output")
	system := flag.String("system", "", "(optional) Override the default system prompt")

	flag.Parse()

	result := template.GenerateCode(*ip, *port, *custom, *system, *testMode)
	if result["success"].(bool) {
		if *testMode {
			fmt.Println("\nTest mode: PowerShell script generated")
		} else {
			fmt.Println("\nCompiled mode: Executable created")
			fmt.Println("Run shell.exe to execute the code")
		}
	} else {
		log.Printf("Error: %s", result["message"])
	}
	fmt.Printf("Result: %s\n", result["message"])
}