Clean up

decode.odin

@@ -1,16 +1,14 @@
 package mustache
 
 import "base:runtime"
-import "core:mem"
-
 import "core:reflect"
 import "core:strings"
-import "core:log"
+import "core:mem"
 import "core:fmt"
 
-decode :: proc( v: any, key: string ) -> any {
+import "core:log"
-	if key ==  "." do return v
 
+decode :: proc( v: any, key: string ) -> any {
 	v := reflect.any_base(v)
 	ti := type_info_of(v.id)
 
@@ -33,7 +31,7 @@ decode :: proc( v: any, key: string ) -> any {
 		newkey_0 := newkey[0]
 		newkey_1 := "."
 
-		if len(newkey) == 2 {
+		if len(newkey) != 2 {
 			newkey_0 = newkey[0][:len(newkey[0])-1]
 			newkey_1 = newkey[1]
 		}
@@ -79,6 +77,10 @@ decode :: proc( v: any, key: string ) -> any {
 		return ""
 	}
 
+	if key ==  "." {
+		return  v
+	}
+
 	return  ""
 }
 
@@ -86,25 +88,47 @@ decode_string :: proc( v: any, key: string ) -> string {
 	return fmt.tprintf("%v", decode(v, key))
 }
 
-section :: proc( r: ^strings.Reader, v: any, key: string ) -> string {
+@private 
+mustache_section :: proc( r: ^strings.Reader, v: any, key: string, inv: bool = false ) -> string {
+	tmp := mustache(r, v, key)
+
+	if inv {
+		return tmp
+	}
+
+	defer delete(tmp)
+	return ""
+}
+
+section :: proc( r: ^strings.Reader, v: any, key: string, inv: bool = false ) -> string {
 	save := r.i
 
 	t := reflect.any_base(decode(v, key))
 	ti := type_info_of(t.id)
 
-	ret : string= ""
 	#partial switch i in runtime.type_info_base(ti).variant {
 	case runtime.Type_Info_Slice:
+		ret : string= ""
 		for i in 0..<reflect.length(t) {
 			elem :=reflect.index(t, i);
-
 			strings.reader_seek(r, save, .Start)
-			tmp := mustache(r, elem, key)
+
+			tmp := mustache_section(r, elem, key, !inv)
 			defer delete(tmp)
 
 			ret = strings.concatenate({ret, tmp})
 		}
+		return ret
+	case runtime.Type_Info_Boolean:
+		b, _ := reflect.as_bool(t)
+
+		return mustache_section(r, t, key, b~inv)
 	}
 
-	return ret
+	if t == nil {
+		return mustache_section(r, t, key, inv)
+	}
+
+	return mustache_section(r, t, key, !inv)
 }
+

mustache.odin

@@ -21,13 +21,13 @@ A new string with all placeholders replaced by their corresponding values from
 the dictionary. If a key is missing in the dictionary, the placeholder is 
 replaces with an empty string.
 */
-mustache_string :: proc(fmt: string, v: any ) -> string {
+mustache_string :: proc(fmt: string, v: any , end_block: string = "") -> string {
 	r : strings.Reader
 	strings.reader_init(&r, fmt)
-	return  mustache(&r, v, "")
+	return  mustache(&r, v, end_block)
 }
 
-mustache_reader :: proc(r: ^strings.Reader, v: any, end_block: string ) -> string {
+mustache_reader :: proc(r: ^strings.Reader, v: any, end_block: string = "" ) -> string {
 	/*
 	template works as a state machine, it manipulates `b` (returned string)
 	and `key` (placeholder string), according to  the states. No error
@@ -82,6 +82,8 @@ mustache_reader :: proc(r: ^strings.Reader, v: any, end_block: string ) -> strin
 					}
 				case '#':
 					strings.write_string(&b, section(r, v, skey[1:]) )
+				case '^':
+					strings.write_string(&b, section(r, v, skey[1:], true) )
 				case:
 					strings.write_string(&b, decode_string(v, skey) )
 				}