Merge pull request #12 from DawnNights/master

添加简易在线运行代码功能

main.go

@@ -12,6 +12,7 @@ import (
 	_ "github.com/Yiwen-Chan/ZeroBot-Plugin/github"
 	_ "github.com/Yiwen-Chan/ZeroBot-Plugin/manager"
 	_ "github.com/Yiwen-Chan/ZeroBot-Plugin/music"
+	_ "github.com/Yiwen-Chan/ZeroBot-Plugin/runcode"
 	//_ "github.com/Yiwen-Chan/ZeroBot-Plugin/setutime"
 )
 

runcode/code_runner.go

@@ -0,0 +1,142 @@
+package runcode
+
+import (
+	"encoding/json"
+	"fmt"
+	zero "github.com/wdvxdr1123/ZeroBot"
+	"io/ioutil"
+	"net/http"
+	"net/url"
+	"strings"
+	"time"
+)
+
+func init()  {
+	runAllow := true
+	runTypes := map[string][2]string{
+		"Py2": {"0","py"},
+		"Ruby": {"1","rb"},
+		"PHP": {"3","php"},
+		"Go": {"6","go"},
+		"C": {"7","c"},
+		"C++": {"7","cpp"},
+		"Java": {"8","java"},
+		"Rust": {"9","rs"},
+		"C#": {"10","cs"},
+		"Perl": {"14","pl"},
+		"Python": {"15","py3"},
+		"Swift": {"16","swift"},
+		"Lua": {"17","lua"},
+	}
+
+	zero.OnCommand("runList").Handle(func(ctx *zero.Ctx) {
+		ctx.Send(`[使用说明]
+Run 语种<<<
+代码块
+>>>
+[支持语种]
+Go || Python || Java || C/C++ || C# || Lua
+Rust || PHP || Perl || Ruby || Swift || Py2`)
+	})
+
+	zero.OnCommand("runOpen").Handle(func(ctx *zero.Ctx) {
+		if ctx.Event.UserID == 213864964{
+			runAllow = true
+			ctx.Send(fmt.Sprintf(
+				"[CQ:at,qq=%d]在线运行代码功能已启用",
+				ctx.Event.UserID,
+			))
+		}
+	})
+
+	zero.OnCommand("runClose").Handle(func(ctx *zero.Ctx) {
+		if ctx.Event.UserID == 213864964{
+			runAllow = false
+			ctx.Send(fmt.Sprintf(
+				"[CQ:at,qq=%d]在线运行代码功能已禁用",
+				ctx.Event.UserID,
+			))
+		}
+	})
+
+	zero.OnRegex("(?is:Run (.+?)<<<(.+?)>>>)").Handle(func(ctx *zero.Ctx) {
+		if runAllow==false{
+			ctx.Send(fmt.Sprintf(
+				"[CQ:at,qq=%d]在线运行代码功能已被禁用",
+				ctx.Event.UserID,
+				))
+			return
+		}
+		getType := ctx.State["regex_matched"].([]string)[1]
+		if runType,exist:=runTypes[getType];exist{
+			println("正在尝试执行",getType,"代码块")
+			getCode := ctx.State["regex_matched"].([]string)[2]
+			getCode = strings.Replace(getCode,"&#91;","[",-1)
+			getCode = strings.Replace(getCode,"&#93;","]",-1)
+
+			res := runCode(getCode,runType)
+			if res["errors"] == "\n\n"{
+				ctx.Send(fmt.Sprintf(
+					"[CQ:at,qq=%d]本次%s语言代码执行结果如下:\n%s",
+					ctx.Event.UserID,
+					getType,
+					res["output"][:len(res["output"])-1],
+				))
+			}else {
+				ctx.Send(fmt.Sprintf(
+					"[CQ:at,qq=%d]本次%s语言代码执行失败:%s",
+					ctx.Event.UserID,
+					getType,
+					res["errors"],
+					))
+			}
+		}else {
+			ctx.Send(fmt.Sprintf(
+				"[CQ:at,qq=%d][%s]语言不是受支持的编程语种呢~",
+				ctx.Event.UserID,
+				getType,
+				))
+		}
+	})
+}
+
+func runCode(code string,runType [2]string) map[string]string {
+	//对菜鸟api发送数据并返回结果
+	result := map[string]string{}
+	api := "https://tool.runoob.com/compile2.php"
+	headers := map[string]string{
+		"Content-Type": "application/x-www-form-urlencoded; charset=UTF-8",
+		"Referer": "https://c.runoob.com/",
+		"User-Agent": "Mozilla/5.0 (Windows NT 6.1; Win64; x64; rv:87.0) Gecko/20100101 Firefox/87.0",
+	}
+	data := map[string]string{
+		"code": code,
+		"token": "4381fe197827ec87cbac9552f14ec62a",
+		"stdin": "",
+		"language": runType[0],
+		"fileext": runType[1],
+	}
+	json.Unmarshal(netPost(api,data,headers),&result)
+	return result
+}
+
+func netPost(api string,data map[string]string,headers map[string]string) []byte {
+	//发送POST请求获取返回数据
+	client := &http.Client{
+		Timeout: time.Duration(6 * time.Second),
+	}
+
+	param := url.Values{}
+	for key,value := range data{
+		param.Set(key,value)
+	}
+
+	request,_ := http.NewRequest("POST",api,strings.NewReader(param.Encode()))
+	for key,value := range headers{
+		request.Header.Add(key,value)
+	}
+	res,_ := client.Do(request)
+	defer res.Body.Close()
+	result,_ := ioutil.ReadAll(res.Body)
+	return result
+}