piston-telegram/runCommand.go at master · Vrganj/piston-telegram · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
package main

import (
	"strings"

	"gopkg.in/tucnak/telebot.v2"
)

func getLanguage(text string) string {
	index := strings.Index(text, " ")

	if index == -1 {
		return strings.ToLower(text)
	}

	return strings.ToLower(text[:index])
}

func getSource(text string) string {
	index := strings.Index(text, " ")

	if index == -1 {
		return ""
	}

	return text[index+1:]
}

func validateLanguage(language string) bool {
	languages := []string{
		"awk",
		"bash",
		"brainfuck", "bf",
		"c",
		"cpp", "c++",
		"csharp", "cs", "c#",
		"deno", "denojs", "denots",
		"elixir", "exs",
		"emacs", "elisp", "el",
		"go",
		"haskell", "hs",
		"java",
		"jelly",
		"julia", "jl",
		"kotlin",
		"lua",
		"nasm", "asm",
		"nasm64", "asm64",
		"nim",
		"node", "javascript", "js",
		"perl", "pl",
		"php",
		"python2",
		"python3", "python",
		"paradoc",
		"ruby",
		"rust",
		"swift",
		"typescript", "ts",
	}

	for _, l := range languages {
		if l == language {
			return true
		}
	}

	return false
}
func runCommand(message *telebot.Message, bot *telebot.Bot) {
	text := strings.TrimSpace(message.Text[4:])
	language := getLanguage(text)
	source := getSource(text)

	if language == "" {
		bot.Send(message.Sender, "Provide a language")
		return
	}

	if !validateLanguage(language) {
		bot.Send(message.Sender, "Invalid language provided")
		return
	}

	if source == "" {
		bot.Send(message.Sender, "Provide source code")
		return
	}

	execute(message, bot, language, source)
}