Merge pull request #1 from Zhou-Shilin/master

feat: Basic features

Author: Zhou-Shilin

build.py

@@ -0,0 +1,28 @@
+from subprocess import Popen, PIPE, STDOUT
+from log_writer import logger
+
+def build_plugin(artifact_name):
+    project_path = f"codes/{artifact_name}"
+    build_command = ["cd", project_path, "&&", "mvn", "-V", "-B", "clean", "package", "--file", "pom.xml"]
+    
+    process = Popen(build_command, stdout=PIPE, stderr=STDOUT, shell=True)
+
+    def log_subprocess_output(pipe):
+        output = ""
+        for line in iter(pipe.readline, b''):
+            str_line = str(line)
+            output += str_line
+            logger(f"building -> {str_line}")
+        return output
+
+    with process.stdout:
+        output = log_subprocess_output(process.stdout)
+
+    process.wait()
+
+    return output
+
+if __name__ == "__main__":
+    result = build_plugin("ExamplePlugin2")
+    print(result)
+    print(type(result))

codes/ExamplePlugin4/pom.xml

@@ -0,0 +1,40 @@
+<?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>ExamplePlugin4</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>
+            </resource>
+        </resources>
+        <plugins>
+            <plugin>
+                <groupId>org.apache.maven.plugins</groupId>
+                <artifactId>maven-compiler-plugin</artifactId>
+                <version>3.8.0</version>
+                <configuration>
+                    <source>1.8</source>
+                    <target>1.8</target>
+                </configuration>
+            </plugin>
+        </plugins>
+    </build>
+</project>

codes/ExamplePlugin4/src/main/java/org/cubegpt/188eba63/Main.java

@@ -0,0 +1,20 @@
+package org.cubegpt._188eba63;
+
+import org.bukkit.Bukkit;
+import org.bukkit.event.EventHandler;
+import org.bukkit.event.Listener;
+import org.bukkit.event.player.PlayerJoinEvent;
+import org.bukkit.plugin.java.JavaPlugin;
+
+public class Main extends JavaPlugin implements Listener {
+
+    @Override
+    public void onEnable() {
+        Bukkit.getServer().getPluginManager().registerEvents(this, this);
+    }
+
+    @EventHandler
+    public void onPlayerJoin(PlayerJoinEvent event) {
+        event.getPlayer().sendMessage("hello");
+    }
+}

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

@@ -0,0 +1 @@
+message: "hello"

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

@@ -0,0 +1,5 @@
+name: ExamplePlugin4
+version: 1.0
+main: org.cubegpt._188eba63.Main
+api-version: 1.13
+author: CubeGPT

codes/ExamplePlugin4/target/ExamplePlugin4-1.0-SNAPSHOT.jar

@@ -0,0 +1 @@
+message: "hello"

codes/ExamplePlugin4/target/classes/config.yml

@@ -0,0 +1,5 @@
+name: ExamplePlugin4
+version: 1.0
+main: org.cubegpt._188eba63.Main
+api-version: 1.13
+author: CubeGPT

codes/ExamplePlugin4/target/classes/org/cubegpt/_188eba63/Main.class

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

codes/ExamplePlugin4/target/classes/plugin.yml

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

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

@@ -0,0 +1 @@
+D:\zhousl\BukkitGPT\BukkitGPT-v3\codes\ExamplePlugin4\src\main\java\org\cubegpt\188eba63\Main.java

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

@@ -8,4 +8,44 @@
 API_KEY: "" # Free API Key with GPT-4 access: https://github.com/CubeGPT/.github/discussions/1
 BASE_URL: "https://api.openai.com/v1/chat/completions"
 
-# More config options here.
+GENERATION_MODEL: "gpt-4-turbo-2024-04-09"
+
+# DEVELOPER SETTINGS #
+VERSION_NUMBER: "Beta 1.0"
+
+# PROMPT SETTINGS #
+# If you don't know what it is, please don't touch it. Be sure to backup before editing.
+
+## Code Generation ##
+SYS_GEN: | 
+  You're a minecraft bukkit plugin coder AI. Game Version: 1.13.2 (1.13.2-R0.1-SNAPSHOT)
+  Write the code & choose a artifact name for the following files with the infomation which is also provided by the user:
+  %WORKING_PATH%/Main.java
+  src/main/resources/plugin.yml
+  src/main/resources/config.yml
+  pom.xml
+  Response in json format:
+  {
+      \"codes\": [
+          {
+              \"file\": \"codes/%ARTIFACT_NAME%/src/main/java/%PKG_ID_LST%Main.java\",
+              \"code\": \"package ...;\\nimport org.bukkit.Bukkit;\\npublic class Main extends JavaPlugin implements CommandExecutor {\\n... (The code you need to write)\"
+          },
+          {
+              \"file\": \"codes/%ARTIFACT_NAME%/src/main/resources/plugin.yml\",
+              \"code\": \"name: ...\\nversion: ...\\n...\"
+          },
+          {
+              \"file\": \"codes/%ARTIFACT_NAME%/src/main/resources/config.yml\",
+              \"code\": \"...\"
+          },
+          {
+              \"file\": \"codes/%ARTIFACT_NAME%/pom.xml\",
+              \"code\": \"...\"
+          }
+      ]
+  }
+  You should never response anything else. Never use Markdown format. Use \n for line feed, and never forget to use \ before ". Never write uncompeleted codes, such as leave a comment that says "// Your codes here" or "// Uncompeleted".
+
+USR_GEN: | 
+  %DESCRIPTION%

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

@@ -4,11 +4,58 @@
 from log_writer import logger
 import core
 import config
+import build
 
 if __name__ == "__main__":
     core.initialize()
 
-    # Codes here.
+    print("BukkitGPT v3 beta console running")
+
+    # Get user inputs
+    name = input("Enter the plugin name: ")
+    description = input("Enter the plugin description: ")
+    
+    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":
+        print("Build failed. Passing the error to ChatGPT and let it to fix it?")
+        fix = input("Y/n: ")
+        if fix == "n":
+            print("Exiting...")
+            sys.exit(0)
+        else:
+            print("Fixing is a uncompleted feature.")
+            print("Restart the program and retype the name & description, etc. to try again.")
+            print("Exiting...")
+            sys.exit(0)
+    else:
+        print("Unknown error. Please check the logs && send the log to @BaimoQilin on discord.")
+        print("Exiting...")
+        sys.exit(0)
+
 
 else:
     print("Error: Please run console.py as the main program instead of importing it from another program.")

config.yaml

@@ -1,8 +1,9 @@
 from openai import OpenAI
-import mcschematic
+import chardet
 import sys
 import json
 import locale
+import os
 
 from log_writer import logger
 import config
@@ -83,7 +84,74 @@ def askgpt(system_prompt: str, user_prompt: str, model_name: str, disable_json_m
     logger(f"askgpt: extracted reply {assistant_reply}")
     return assistant_reply
 
-# Core codes here.
+def response_to_action(msg):
+    """
+    Converts a response from ChatGPT to an action.
+
+    Args:
+        msg (str): The response from ChatGPT.
+
+    Returns:
+        str: The action to take.
+    """
+    text = json.loads(msg)
+
+    codes = text["codes"]
+
+    for section in codes:
+        file = section["file"]
+        code = section["code"]
+
+        paths = file.split("/")
+
+        # Join the list elements to form a path
+        path = os.path.join(*paths)
+
+        # Get the directory path and the file name
+        dir_path, file_name = os.path.split(path)
+
+        # Create directories, if they don't exist
+        try:
+            os.makedirs(dir_path, exist_ok=True)
+        except FileNotFoundError:
+            pass
+
+        # Create the file
+        with open(path, 'w') as f:
+            f.write(code)  # Write an empty string to the file
+
+def mixed_decode(text: str):
+    """
+    Decode a mixed text containing both normal text and a byte sequence.
+
+    Args:
+        text (str): The mixed text to be decoded.
+
+    Returns:
+        str: The decoded text, where the byte sequence has been converted to its corresponding characters.
+
+    """
+    # Split the normal text and the byte sequence
+    # Assuming the byte sequence is everything after the last colon and space ": "
+    try:
+        normal_text, byte_text = text.rsplit(": ", 1)
+    except (TypeError, ValueError):
+        # The text only contains normal text
+        return text
+
+    # Convert the byte sequence to actual bytes
+    byte_sequence = byte_text.encode('latin1')  # latin1 encoding maps byte values directly to unicode code points
+
+    # Detect the encoding of the byte sequence
+    detected_encoding = chardet.detect(byte_sequence)
+    encoding = detected_encoding['encoding']
+
+    # Decode the byte sequence
+    decoded_text = byte_sequence.decode(encoding)
+
+    # Combine the normal text with the decoded byte sequence
+    final_text = normal_text + ": " + decoded_text
+    return final_text
 
 if __name__ == "__main__":
     print("This script is not meant to be run directly. Please run console.py instead.")