lite-shell/src/commands/_if.hpp at main · Serious-senpai/lite-shell · 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
92
93
94
95
96
#pragma once

#include <all.hpp>

class _IfCommand : public liteshell::BaseCommand
{
public:
    _IfCommand()
        : liteshell::BaseCommand(
              "_if",
              "Hidden command",
              "",
              liteshell::CommandConstraint(
                  "x", "", true,
                  "operator", "", true,
                  "y", "", true,
                  "true", "", true,
                  "false", "", true)
                  .add_option("-m", "", false))
    {
    }

    DWORD run(const liteshell::Context &context)
    {
        auto first = context.get("x"), op = context.get("operator"), second = context.get("y");
        bool result;
        if (context.present.count("-m"))
        {
            auto f = context.client->get_environment()->eval_ll(first),
                 s = context.client->get_environment()->eval_ll(second);
            if (op == "==")
            {
                result = f == s;
            }
            else if (op == "!=")
            {
                result = f != s;
            }
            else if (op == "<")
            {
                result = f < s;
            }
            else if (op == ">")
            {
                result = f > s;
            }
            else if (op == "<=")
            {
                result = f <= s;
            }
            else if (op == ">=")
            {
                result = f >= s;
            }
            else
            {
                throw std::invalid_argument("Invalid operator");
            }
        }
        else
        {
            if (op == "==")
            {
                result = first == second;
            }
            else if (op == "!=")
            {
                result = first != second;
            }
            else if (op == "<")
            {
                result = first < second;
            }
            else if (op == ">")
            {
                result = first > second;
            }
            else if (op == "<=")
            {
                result = first <= second;
            }
            else if (op == ">=")
            {
                result = first >= second;
            }
            else
            {
                throw std::invalid_argument("Invalid operator");
            }
        }

        context.client->get_stream()->jump(context.get(result ? "true" : "false"));

        return 0;
    }
};