aboutsummaryrefslogtreecommitdiff
path: root/lib/commands/commands.tm
blob: 8a1310426802adb89d004809201601d9fdc306dc (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
# Functions for running system commands

use ./commands.c

extern run_command : func(exe:Text, args:[Text], env:{Text=Text}, input:[Byte]?, output:&[Byte]?, error:&[Byte]? -> Int32)
extern command_by_line : func(exe:Text, args:[Text], env:{Text=Text} -> func(->Text?)?)

enum ExitType(Exited(status:Int32), Signaled(signal:Int32), Failed)
    func succeeded(e:ExitType -> Bool)
        when e is Exited(status) return (status == 0)
        else return no

    func or_fail(e:ExitType, message:Text?=none)
        if not e.succeeded()
            fail(message or "Program failed: $e")

struct ProgramResult(output:[Byte], errors:[Byte], exit_type:ExitType)
    func or_fail(r:ProgramResult, message:Text?=none -> ProgramResult)
        when r.exit_type is Exited(status)
            if status != 0
                fail(message or "Program failed: $r")
        else fail(message or "Program failed: $r")
        return r

    func output_text(r:ProgramResult, trim_newline=yes -> Text?)
        when r.exit_type is Exited(status)
            if status == 0
                if text := Text.from_bytes(r.output)
                    if trim_newline
                        text = text.without_suffix("\n")
                    return text
        else return none
        return none

    func error_text(r:ProgramResult -> Text?)
        when r.exit_type is Exited(status)
            if status == 0
                return Text.from_bytes(r.errors)
        else return none
        return none

    func succeeded(r:ProgramResult -> Bool)
        when r.exit_type is Exited(status)
            return (status == 0)
        else
            return no

struct Command(command:Text, args:[Text]=[], env:{Text=Text}={})
    func from_path(path:Path, args:[Text]=[], env:{Text=Text}={} -> Command)
        return Command(Text(path), args, env)

    func result(command:Command, input="", input_bytes:[Byte]=[] -> ProgramResult)
        if input.length > 0
            (&input_bytes).insert_all(input.bytes())

        output : [Byte]
        errors : [Byte]
        status := run_command(command.command, command.args, command.env, input_bytes, &output, &errors)

        if C_code:Bool(WIFEXITED(@status))
            return ProgramResult(output, errors, ExitType.Exited(C_code:Int32(WEXITSTATUS(@status))))

        if C_code:Bool(WIFSIGNALED(@status))
            return ProgramResult(output, errors, ExitType.Signaled(C_code:Int32(WTERMSIG(@status))))

        return ProgramResult(output, errors, ExitType.Failed)

    func run(command:Command, -> ExitType)
        status := run_command(command.command, command.args, command.env, none, none, none)

        if C_code:Bool(WIFEXITED(@status))
            return ExitType.Exited(C_code:Int32(WEXITSTATUS(@status)))

        if C_code:Bool(WIFSIGNALED(@status))
            return ExitType.Signaled(C_code:Int32(WTERMSIG(@status)))

        return ExitType.Failed

    func get_output(command:Command, input="", trim_newline=yes -> Text?)
        return command.result(input=input).output_text(trim_newline=trim_newline)

    func get_output_bytes(command:Command, input="", input_bytes:[Byte]=[] -> [Byte]?)
        result := command.result(input=input, input_bytes=input_bytes)
        when result.exit_type is Exited(status)
            if status == 0 return result.output
            return none
        else return none

    func by_line(command:Command -> func(->Text?)?)
        return command_by_line(command.command, command.args, command.env)