about summary refs log tree commit diff
path: root/pkg/bot/cmds.go
blob: ec0a943fdc3b5ea75fe8950c9463d3fa5a036c52 (plain)
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
92
93
94
95
96
97
98
99
package bot

import (
	"fmt"
	"jinx/pkg/discord/events"
	"jinx/pkg/lang/compiler"
	"jinx/pkg/lang/parser"
	"jinx/pkg/lang/scanner"
	"jinx/pkg/lang/vm"
	"jinx/pkg/lang/vm/text"
	"strings"
)

type Cmd func(b *Bot, content string, msg events.Message) error

var (
	nameToCmd = map[string]Cmd{
		"ping": pingCmd,
		"vm":   vmCmd,
		"lang": langCmd,
	}
)

func pingCmd(b *Bot, content string, msg events.Message) error {
	return b.client.SendMessage(msg.ChannelID, "pong")
}

func vmCmd(b *Bot, content string, msg events.Message) error {
	src := strings.Split(content, "```")[1]

	comp := text.NewCompiler(strings.NewReader(src))
	bc, err := comp.Compile()
	if err != nil {
		b.logger.Error().Err(err).Msg("error compiling code")
		return err
	}

	vm := vm.New(&bc)

	if err := vm.Run(); err != nil {
		return err
	}

	res, err := vm.GetResult()
	if err != nil {
		return err
	}

	b.logger.Debug().Msg("executed code")

	return b.client.SendMessage(msg.ChannelID, res)
}

func langCmd(b *Bot, content string, msg events.Message) error {
	src := strings.Split(content, "```")[1]

	scanner := scanner.New(strings.NewReader(src))
	tokens, err := scanner.Scan()
	if err != nil {
		return err
	}

	parser := parser.New(tokens)
	program, err := parser.Parse()
	if err != nil {
		return err
	}

	comp := compiler.New(program)
	bc, err := comp.Compile()
	if err != nil {
		b.logger.Error().Err(err).Msg("error compiling code")
		return err
	}

	b.client.SendMessage(msg.ChannelID, "successfully compiled code")

	decomp := text.NewDecompiler(bc)
	decompRes := decomp.Decompile()

	b.client.SendMessage(msg.ChannelID, fmt.Sprintf("resulting decompiled bytecode:\n```\n%s\n```", decompRes))

	vm := vm.New(&bc)

	b.client.SendMessage(msg.ChannelID, "executing code...")

	if err := vm.Run(); err != nil {
		return err
	}

	res, err := vm.GetResult()
	if err != nil {
		return err
	}

	b.logger.Debug().Msg("executed code")

	return b.client.SendMessage(msg.ChannelID, res)
}