diff options
| -rw-r--r-- | piston.py | 117 |
1 files changed, 117 insertions, 0 deletions
diff --git a/piston.py b/piston.py new file mode 100644 index 0000000..3b8504d --- /dev/null +++ b/piston.py @@ -0,0 +1,117 @@ +# +# Piston commands +# +# Copyright © 2023 by luk3yx +# +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU Affero General Public License as +# published by the Free Software Foundation, either version 3 of the +# License, or (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU Affero General Public License for more details. +# +# You should have received a copy of the GNU Affero General Public License +# along with this program. If not, see <https://www.gnu.org/licenses/>. +# + +import requests + +PISTON_URL = 'https://emkc.org/api/v2/piston/execute' + + +@register_command('sh', 'bash') +def sh(irc, hostmask, is_admin, args): + res = requests.post(PISTON_URL, json={ + 'language': 'bash', + 'version': '*', + 'files': [{'name': 'main.sh', 'content': args[1]}] + }) + output = res.json()['run']['output'].rstrip('\n') + irc.msg(args[0], f'{hostmask[0]}: {output or "(no output)"}') + + +LUA_CODE = r""" +local f = assert(io.open("code.lua", "rb")) +local code = f:read("*a") +f:close() +local f = load("return " .. code) +if f then + print(f()) +else + dofile("code.lua") +end +""" + + +@register_command('lua') +def lua(irc, hostmask, is_admin, args): + res = requests.post(PISTON_URL, json={ + 'language': 'lua', + 'version': '*', + 'files': [ + {'name': 'init.lua', 'content': LUA_CODE}, + {'name': 'code.lua', 'content': args[1]} + ] + }) + output = res.json()['run']['output'].rstrip('\n') + irc.msg(args[0], f'{hostmask[0]}: {output or "(no output)"}') + + +PYTHON_CODE = r""" +import ast, builtins + +with open('code.py', 'r') as f: + code = f.read() + + +# Automatically import libraries where possible +class _BuiltinsWrapper(dict): + __slots__ = () + def __missing__(self, key): + try: + return __import__(key) + except ImportError: + raise NameError(f'name {key!r} is not defined') + + +env = {'__builtins__': _BuiltinsWrapper(builtins.__dict__), 'code': code} + + +try: + ast.parse(code, mode='eval') +except SyntaxError: + exec(code, env) +else: + print(eval(code, env)) +""" + + +@register_command('py', 'py3', 'python', 'python3') +def py3(irc, hostmask, is_admin, args): + res = requests.post(PISTON_URL, json={ + 'language': 'python', + 'version': '3', + 'files': [ + {'name': 'main.py', 'content': PYTHON_CODE}, + {'name': 'code.py', 'content': args[1]} + ] + }) + run = res.json()['run'] + output = run['output'].rstrip('\n') + if run['code'] == 1: + output = output.rsplit('\n', 1)[-1] + irc.msg(args[0], f'{hostmask[0]}: {output or "(no output)"}') + + +@register_command('hs', 'haskell') +def hs(irc, hostmask, is_admin, args): + res = requests.post(PISTON_URL, json={ + 'language': 'haskell', + 'version': '*', + 'files': [{'name': 'main.hs', 'content': args[1]}] + }) + output = res.json()['run']['output'].rstrip('\n') + irc.msg(args[0], f'{hostmask[0]}: {output or "(no output)"}') |