ver 0.1
This commit is contained in:
parent
80c8b0c16c
commit
7caccb7273
33
main.odin
Normal file
33
main.odin
Normal file
@ -0,0 +1,33 @@
|
|||||||
|
package flag
|
||||||
|
|
||||||
|
import "core:fmt"
|
||||||
|
import "myflag"
|
||||||
|
|
||||||
|
main :: proc(){
|
||||||
|
/*
|
||||||
|
arena: vmem.Arena
|
||||||
|
arena_err := vmem.arena_init_growing(&arena)
|
||||||
|
ensure(arena_err == nil)
|
||||||
|
arena_alloc := vmem.arena_allocator(&arena)
|
||||||
|
defer vmem.arena_destroy(&arena) //clean
|
||||||
|
|
||||||
|
myflag.init_custom_allocator(arena_alloc)
|
||||||
|
defer myflag.destroy()
|
||||||
|
*/
|
||||||
|
|
||||||
|
defer myflag.destroy()
|
||||||
|
isRunning: bool
|
||||||
|
isNumber: int
|
||||||
|
isString: string
|
||||||
|
|
||||||
|
myflag.BoolVar(&isRunning, "isRunning", true, "A select piece of code is running!")
|
||||||
|
myflag.IntVar(&isNumber, "isNumber", 3434, "A number!")
|
||||||
|
myflag.StringVar(&isString, "isString", "Hellope!", "A string!")
|
||||||
|
|
||||||
|
myflag.parse_flags()
|
||||||
|
fmt.println("-------MAIN--------")
|
||||||
|
fmt.println(isRunning)
|
||||||
|
fmt.println(isNumber)
|
||||||
|
fmt.println(isString)
|
||||||
|
fmt.println("-------MAIN--------")
|
||||||
|
}
|
||||||
145
myflag/flag.odin
Normal file
145
myflag/flag.odin
Normal file
@ -0,0 +1,145 @@
|
|||||||
|
package myflag
|
||||||
|
|
||||||
|
import "core:strconv"
|
||||||
|
import "core:strings"
|
||||||
|
import "core:os"
|
||||||
|
import "core:fmt"
|
||||||
|
import "base:runtime"
|
||||||
|
|
||||||
|
Flag_Value_Ptr :: union{
|
||||||
|
^bool,
|
||||||
|
^int,
|
||||||
|
^string,
|
||||||
|
}
|
||||||
|
|
||||||
|
Flag :: struct {
|
||||||
|
name: string,
|
||||||
|
value: Flag_Value_Ptr,
|
||||||
|
usage: string,
|
||||||
|
parsed: bool, // sck: flag to check if it has been parsed
|
||||||
|
}
|
||||||
|
|
||||||
|
all_flags: [dynamic]Flag
|
||||||
|
|
||||||
|
|
||||||
|
// sck: Used to create a customer allocator
|
||||||
|
init_custom_allocator :: proc(allocator: runtime.Allocator){
|
||||||
|
// sck: make sure to destory the old allocations
|
||||||
|
destroy()
|
||||||
|
all_flags = make([dynamic]Flag, allocator = allocator)
|
||||||
|
}
|
||||||
|
|
||||||
|
destroy :: proc(){
|
||||||
|
delete(all_flags)
|
||||||
|
all_flags = {} // sck: needed?
|
||||||
|
}
|
||||||
|
|
||||||
|
print_flags :: proc(){
|
||||||
|
for f in all_flags{
|
||||||
|
fmt.printfln("%s: %v: %s", f.name, f.value, f.usage)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
print_usage :: proc(){
|
||||||
|
fmt.println("\tUsage: ")
|
||||||
|
for f in all_flags {
|
||||||
|
switch v in f.value {
|
||||||
|
case ^bool:
|
||||||
|
fmt.printfln("\t-%s:\n\t\t%s (default: %v)\n",
|
||||||
|
f.name, f.usage, v^)
|
||||||
|
|
||||||
|
case ^int:
|
||||||
|
fmt.printfln("\t-%s:\n\t\t%s (default: %d)\n",
|
||||||
|
f.name, f.usage, v^)
|
||||||
|
|
||||||
|
case ^string:
|
||||||
|
fmt.printfln("\t-%s:\n\t\t%s (default: %s)\n",
|
||||||
|
f.name, f.usage, v^)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
os.exit(0)
|
||||||
|
}
|
||||||
|
|
||||||
|
BoolVar :: proc(ptr: ^bool, name: string, default: bool, usage: string) {
|
||||||
|
ptr^ = default
|
||||||
|
|
||||||
|
append(&all_flags, Flag{
|
||||||
|
name = name,
|
||||||
|
value = ptr,
|
||||||
|
usage = usage,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
IntVar :: proc(ptr: ^int, name: string, default: int, usage: string) {
|
||||||
|
ptr^ = default
|
||||||
|
|
||||||
|
append(&all_flags, Flag{
|
||||||
|
name = name,
|
||||||
|
value = ptr,
|
||||||
|
usage = usage,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
StringVar :: proc(ptr: ^string, name: string, default: string, usage: string) {
|
||||||
|
ptr^ = default
|
||||||
|
|
||||||
|
append(&all_flags, Flag{
|
||||||
|
name = name,
|
||||||
|
value = ptr,
|
||||||
|
usage = usage,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
parse_flags :: proc (){
|
||||||
|
if len(os.args) < 2 { print_usage() }
|
||||||
|
|
||||||
|
for &a in os.args {
|
||||||
|
if a == "-help" || a == "--help"{ print_usage() }
|
||||||
|
if a == os.args[0] { continue }
|
||||||
|
|
||||||
|
for &f in all_flags {
|
||||||
|
a = strings.trim_prefix(a, "-")
|
||||||
|
|
||||||
|
name := a
|
||||||
|
value: string
|
||||||
|
|
||||||
|
split := strings.index_any(a, "=")
|
||||||
|
if split >= 0 {
|
||||||
|
name = a[:split]
|
||||||
|
value = a[split+1:]
|
||||||
|
}
|
||||||
|
|
||||||
|
if name == f.name && !f.parsed{
|
||||||
|
f.parsed = true
|
||||||
|
switch &v in f.value {
|
||||||
|
case ^bool:
|
||||||
|
parsed, ok := strconv.parse_bool(value)
|
||||||
|
if !ok {
|
||||||
|
//error
|
||||||
|
}
|
||||||
|
v^ = parsed
|
||||||
|
|
||||||
|
case ^int:
|
||||||
|
parsed, ok := strconv.parse_int(value)
|
||||||
|
if !ok {
|
||||||
|
fmt.fprintfln(os.stderr, "[ERROR] Failed to parsed flag %s:%s", f.name, ok)
|
||||||
|
}
|
||||||
|
v^ = parsed
|
||||||
|
|
||||||
|
case ^string:
|
||||||
|
// TODO: validate string
|
||||||
|
v^ = value
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
fmt.println("[INFO] parsed flags: ")
|
||||||
|
for f in all_flags{
|
||||||
|
if f.parsed == true{
|
||||||
|
fmt.println(f.name)
|
||||||
|
fmt.println(f.value)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
*/
|
||||||
|
}
|
||||||
Loading…
x
Reference in New Issue
Block a user