Merge pull request #3 from Zhou-Shilin/feature-ui

Add GUI support

codes/Demo/pom.xml

@@ -0,0 +1,54 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project xmlns="http://maven.apache.org/POM/4.0.0"
+ xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+ xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
+ <modelVersion>4.0.0</modelVersion>
+
+ <groupId>org.cubegpt</groupId>
+ <artifactId>Demo</artifactId>
+ <version>1.0-SNAPSHOT</version>
+ <packaging>jar</packaging>
+
+ <dependencies>
+ <dependency>
+ <groupId>org.bukkit</groupId>
+ <artifactId>bukkit</artifactId>
+ <version>1.13.2-R0.1-SNAPSHOT</version>
+ <scope>provided</scope>
+ </dependency>
+ </dependencies>
+
+ <build>
+ <sourceDirectory>src/main/java</sourceDirectory>
+ <resources>
+ <resource>
+ <directory>src/main/resources</directory>
+ <filtering>true</filtering>
+ </resource>
+ </resources>
+ <plugins>
+ <plugin>
+ <groupId>org.apache.maven.plugins</groupId>
+ <artifactId>maven-compiler-plugin</artifactId>
+ <version>3.8.1</version>
+ <configuration>
+ <source>1.8</source>
+ <target>1.8</target>
+ </configuration>
+ </plugin>
+ <plugin>
+ <groupId>org.apache.maven.plugins</groupId>
+ <artifactId>maven-jar-plugin</artifactId>
+ <configuration>
+ <archive>
+ <manifest>
+ <addClasspath>true</addClasspath>
+ <classpathPrefix>lib/</classpathPrefix>
+ <mainClass>org.cubegpt._54672e99.Main</mainClass>
+ </manifest>
+ </archive>
+ </configuration>
+ </plugin>
+ </plugins>
+ </build>
+</project>

codes/Demo/src/main/java/org/cubegpt/54672e99/Main.java

@@ -0,0 +1,30 @@
+package org.cubegpt._54672e99;
+
+import org.bukkit.Bukkit;
+import org.bukkit.plugin.java.JavaPlugin;
+import org.bukkit.event.Listener;
+import org.bukkit.event.EventHandler;
+import org.bukkit.event.player.PlayerJoinEvent;
+import org.bukkit.command.CommandExecutor;
+import org.bukkit.command.Command;
+import org.bukkit.command.CommandSender;
+import org.bukkit.entity.Player;
+
+public class Main extends JavaPlugin implements Listener {
+
+ @Override
+ public void onEnable() {
+ getServer().getPluginManager().registerEvents(this, this);
+ Bukkit.getConsoleSender().sendMessage("Plugin enabled!");
+ }
+
+ @EventHandler
+ public void onPlayerJoin(PlayerJoinEvent event) {
+ event.getPlayer().sendMessage("hello");
+ }
+
+ @Override
+ public void onDisable() {
+ Bukkit.getConsoleSender().sendMessage("Plugin disabled!");
+ }
+}

codes/Demo/src/main/resources/config.yml

@@ -0,0 +1 @@
+greeting-message: 'hello'

codes/Demo/src/main/resources/plugin.yml

@@ -0,0 +1,10 @@
+name: Demo
+version: 1.0
+main: org.cubegpt._54672e99.Main
+api-version: 1.13
+author: CubeGPT
+dependencies: []
+commands:
+ demo:
+ description: Sends a greeting message to everyone.
+ usage: /<command>

codes/Demo/target/classes/config.yml

@@ -0,0 +1 @@
+greeting-message: 'hello'

codes/Demo/target/classes/plugin.yml

@@ -0,0 +1,10 @@
+name: Demo
+version: 1.0
+main: org.cubegpt._54672e99.Main
+api-version: 1.13
+author: CubeGPT
+dependencies: []
+commands:
+ demo:
+ description: Sends a greeting message to everyone.
+ usage: /<command>

codes/Demo/target/maven-archiver/pom.properties

@@ -0,0 +1,3 @@
+artifactId=Demo
+groupId=org.cubegpt
+version=1.0-SNAPSHOT

codes/Demo/target/maven-status/maven-compiler-plugin/compile/default-compile/createdFiles.lst

@@ -0,0 +1 @@
+org\cubegpt\_54672e99\Main.class

codes/Demo/target/maven-status/maven-compiler-plugin/compile/default-compile/inputFiles.lst

@@ -0,0 +1 @@
+D:\zhousl\BukkitGPT\BukkitGPT-v3\codes\Demo\src\main\java\org\cubegpt\54672e99\Main.java

core.py

@@ -23,6 +23,10 @@ def initialize():
  locale.setlocale(locale.LC_ALL, 'en_US.UTF-8')
  logger(f"Launch. Software version {config.VERSION_NUMBER}, platform {sys.platform}")
 
+ if "gpt-3.5" in config.GENERATION_MODEL and config.BYPASS_NO_GPT35_FOR_GENERATION_LIMIT is False:
+ print("gpt-3.5 writes bugs *all the time* and is not recommended for code generation. Switching to gpt-4.")
+ config.edit_config("GENERATION_MODEL", config.GENERATION_MODEL.replace("gpt-3.5", "gpt-4"))
+
 def askgpt(system_prompt: str, user_prompt: str, model_name: str, disable_json_mode: bool = False, image_url: str = None):
  """
  Interacts with ChatGPT using the specified prompts.

ui.py

@@ -9,7 +9,9 @@
 from log_writer import logger
 import config
 import core
+import build
 
+#---------- Functions ----------#
 def open_config(args: dict):
  """
  Opens the config file.
@@ -94,22 +96,134 @@ def raise_error(args: dict):
  """
  raise Exception("This is a test error.")
 
-root = CreateQGUI(title="CubeAgents",
- tab_names=["Generate", "Settings", "DevTools"]
+#---------- Generate Function ----------#
+def generate(args: dict):
+ """
+ Generates the plugin.
+
+ Args:
+ args (dict): A dictionary containing the arguments.
+
+ Returns:
+ bool: Always True.
+ """
+ global error_msg, pkg_id_path
+
+ # Get user inputs
+ name = args["PluginName"].get()
+ description = args["PluginDescription"].get()
+
+ artifact_name = name.replace(" ", "")
+ package_id = f"org.cubegpt.{uuid.uuid4().hex[:8]}"
+
+ pkg_id_path = ""
+ for id in package_id.split("."):
+ pkg_id_path += id + "/"
+
+ logger(f"user_input -> name: {name}")
+ logger(f"user_input -> description: {description}")
+ logger(f"random_generate -> package_id: {package_id}")
+ logger(f"str_path -> pkg_id_path: {pkg_id_path}")
+
+ print("Generating plugin...")
+
+ codes = core.askgpt(config.SYS_GEN.replace("%ARTIFACT_NAME%", artifact_name).replace("%PKG_ID_LST%", pkg_id_path), config.USR_GEN.replace("%DESCRIPTION", description), config.GENERATION_MODEL)
+ logger(f"codes: {codes}")
+
+ core.response_to_action(codes)
+
+ print("Code generated. Building now...")
+
+ result = build.build_plugin(artifact_name)
+ if "BUILD SUCCESS" in result:
+ print(f"Build complete. Find your plugin at 'codes/{artifact_name}/target/{artifact_name}.jar'")
+ elif "Compilation failure":
+ error_msg = result
+ print("Build failed. To pass the error to ChatGPT && let it fix, jump to the Fixing page and click the Fix button.")
+ else:
+ print("Unknown error. Please check the logs && send the log to @BaimoQilin on discord.")
+
+
+ return True
+
+def fix(args: dict):
+ """
+ Fixes the error.
+
+ Args:
+ args (dict): A dictionary containing the arguments.
+
+ Returns:
+ bool: Always True.
+ """
+ artifact_name = args["PluginName"].get()
+
+ print("Passing the error to ChatGPT...")
+
+ files = [f"codes/{artifact_name}/src/main/java/{pkg_id_path}Main.java",
+ f"codes/{artifact_name}/src/main/resources/plugin.yml",
+ f"codes/{artifact_name}/src/main/resources/config.yml",
+ f"codes/{artifact_name}/pom.xml"]
+
+ ids = ["main_java",
+ "plugin_yml",
+ "config_yml",
+ "pom_xml"]
+
+ main_java = None
+ plugin_yml = None
+ config_yml = None
+ pom_xml = None
+
+ for file in files:
+ with open(file, "r") as f:
+ code = f.read()
+ id = ids[files.index(file)]
+ globals()[id] = code
+
+ print("Generating...")
+ codes = core.askgpt(config.SYS_FIX.replace("%ARTIFACT_NAME%", str(artifact_name)), config.USR_FIX.replace("%MAIN_JAVA%", str(main_java)).replace("%PLUGIN_YML%", str(plugin_yml)).replace("%CONFIG_YML%", str(config_yml)).replace("%POM_XML%", str(pom_xml)).replave("%PKG_ID_LST%", pkg_id_path).replace("%P_ERROR_MSG%", str(error_msg)), config.FIXING_MODEL)
+
+ shutil.rmtree(f"codes/{artifact_name}")
+ core.response_to_action(codes)
+
+ print("Code generated. Building now...")
+
+ result = build.build_plugin(artifact_name)
+
+ if "BUILD SUCCESS" in result:
+ print(f"Build complete. Find your plugin at 'codes/{artifact_name}/target/{artifact_name}.jar'")
+ else:
+ print("Build failed again. Please check the logs && send the log to @BaimoQilin on discord.")
+
+ return True
+
+#---------- Main Program ----------#
+
+root = CreateQGUI(title="BukkitGPT-v3",
+ tab_names=["Generate", "Fixing", "Settings", "DevTools"]
  )
+error_msg = None
 
 logger("Starting program.")
 
 # Initialize Core
 core.initialize()
 
+print("BukkitGPT v3 beta console running")
+
 # Banner
-root.add_banner_tool(GitHub("https://github.com/CubeGPT/CubeAgents"))
+root.add_banner_tool(GitHub("https://github.com/CubeGPT/BukkitGPT-v3"))
 
 # Generate Page
-# Codes here.
+root.add_notebook_tool(InputBox(name="PluginName", default="ExamplePlugin", label_info="Plugin Name"))
+root.add_notebook_tool(InputBox(name="PluginDescription", default="Send msg 'hello' to every joined player.", label_info="Plugin Description"))
+
+root.add_notebook_tool(RunButton(bind_func=generate, name="Generate", text="Generate Plugin", checked_text="Generating...", tab_index=0))
 
-# More pages here.
+# Fixing Page #
+root.add_notebook_tool(Label(name="Fixing_DESCRIPTION", text="This is a fixing page. If the build fails, click the Fix button to fix the error in the LATEST build.", tab_index=1))
+root.add_notebook_tool(RunButton(bind_func=fix, name="Fix", text="Fix", checked_text="Fixing...", tab_index=1))
 
 # Settings Page
 root.add_notebook_tool(InputBox(name="API_KEY", default=config.API_KEY, label_info="API Key", tab_index=2))
@@ -132,7 +246,7 @@ def raise_error(args: dict):
 # Sidebar
 root.set_navigation_about(author="CubeGPT Team",
  version=config.VERSION_NUMBER,
- github_url="https://github.com/CubeGPT/CubeAgents")
+ github_url="https://github.com/CubeGPT/BukkitGPT-v3")