Handle '==' and '!=' for 'when' directive

1 files changed, 11 insertions, 2 deletions

diff --git a/src/common/ast.odin b/src/common/ast.odin
index d32b32f..c8e2da5 100644
--- a/src/common/ast.odin
+++ b/src/common/ast.odin
@@ -314,11 +314,20 @@ collect_globals :: proc(
 				}
 
 				if ident != nil && implicit != nil {
-					allowed :=
-						ident.name == "ODIN_OS" &&
+					allowed := false
+
+					if binary.op.text == "==" {
+						allowed = ident.name == "ODIN_OS" &&
 							implicit.field.name == fmt.tprint(ODIN_OS) ||
 						ident.name == "ODIN_ARCH" &&
 							implicit.field.name == fmt.tprint(ODIN_ARCH)
+					} else if binary.op.text == "!=" {
+						allowed = ident.name == "ODIN_OS" &&
+							implicit.field.name != fmt.tprint(ODIN_OS) ||
+						ident.name == "ODIN_ARCH" &&
+							implicit.field.name != fmt.tprint(ODIN_ARCH)
+					}
+
 					if allowed {
 						if block, ok := when_decl.body.derived.(^ast.Block_Stmt);
 						   ok {