adventOfCode/adventLaunch.lua at main · remyCases/adventOfCode · 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
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
#!/usr/bin/env lua

-- Copyright (C) 2024 Rémy Cases
-- See LICENSE file for extended copyright information.
-- This file is part of adventOfCode project from https://github.com/remyCases/adventOfCode.

local utils = require 'utils.argparser'

-- locals
local os_name = package.config:sub(1,1) == "\\" and "win" or "unix"
local lang = 0
local i = 1
local earlyEnd = false
local index = 1
local maxLang = 7

-- tables
local langFlag = {
	["cobol"] = 1,
	["cob"] = 1,
	["c99"] = 2,
	["c"] = 2,
	["nim"] = 3,
	["nimlang"] = 3,
	["rust"] = 4,
	["zig"] = 5,
	["py"] = 6,
	["python"] = 6,
	["lua"] = 7,
}

local cmdFlag = {
	[1] = {"./build/%d/bin/mainCob %02d%d", "[[cobol]]"},
	[2] = {"./build/%d/bin/mainC99 %d %d", "[[c99]]"},
	[3] = {"./build/%d/bin/mainNim --day %d --part %d", "[[nim]]"},
	[4] = {"./build/%d/bin/mainRust --day %d --part %d", "[[rust]]"},
	[5] = {"./build/%d/bin/mainZig --day %d --part %d", "[[zig]]"},
	[6] = {"./build/%d/bin/mainPy --day %d --part %d", "[[python]]"},
	[7] = {"./build/%d/bin/mainLua --day %d --part %d", "[[lua]]"},
}

local function addFlag (x)
	if not langFlag[x] then
		print(string.format("Given language [[%s]] is not supported (yet?).", x))
	else
		lang = lang + (1 << (langFlag[x] - 1))
	end
end

-- arguments for parser
utils.addArgs("h", "help", "b")
utils.addArgs("r", "rebuild", "b")
utils.addArgs("d", "day", nil)
utils.addArgs("p", "part", nil)
utils.addArgs(1, "lang", "+")
utils.addArgs("y", "year", nil)
local args = {}

-- help function
local function help ()
	utils.helpArgs()
end

-- parsing arguments

while arg[i] do
	local key, val = utils.parseArg(arg[i], i)
	if key then
		if key == "lang" then
			addFlag(val)
		else
			args[key] = val
		end
	end
	i = i + 1
end

if args.help then
	help()
	goto endParse
end

-- test if all arguments were given
if not args.year then
	print("No year given, end of program.")
	earlyEnd = true
end
if not args.day then
	print("No day given, end of program.")
	earlyEnd = true
end
if not args.part then
	print("No part given, end of program.")
	earlyEnd = true
end

if earlyEnd then
	goto endParse
end

if args.rebuild then
	os.execute("make build_all")
end

if lang == 0 then
	lang = (1 << maxLang) - 1
end

-- apply arguments
while lang > 0 do
	if (lang&1) == 1 then
		local res = cmdFlag[index]
		local cmd = res[1]
		if os_name == "win" then cmd = cmd:gsub("/", "\\") end
		local lng = res[2]
		io.write(lng .. ":\t")
		local parsed_cmd = string.format(cmd, args.year, args.day, args.part)
	    local handle = io.popen(parsed_cmd)
		if not handle then goto nextParse end
		local output = handle:read("*a")
		local rc = {handle:close()}
		if (rc[1] == true) then -- error handling
			io.write(output)
		end
	end
	::nextParse::
	index = index + 1
	lang = lang >> 1
end

::endParse::