Add `#any_int` directive to replace `auto_cast` uses on parameters.

5 files changed, 46 insertions, 5 deletions

diff --git a/src/check_expr.cpp b/src/check_expr.cpp
index cb9fce7b6..b185dd080 100644
--- a/src/check_expr.cpp
+++ b/src/check_expr.cpp
@@ -4485,6 +4485,10 @@ CALL_ARGUMENT_CHECKER(check_call_arguments_internal) {
 					bool ok = false;
 					if (e->flags & EntityFlag_AutoCast) {
 						ok = check_is_castable_to(c, &o, t);
+					} else if (e->flags & EntityFlag_AnyInt) {
+						if (is_type_integer(t)) {
+							ok = check_is_castable_to(c, &o, t);
+						}
 					}
 					if (ok) {
 						s = assign_score_function(MAXIMUM_TYPE_DISTANCE);
@@ -8730,6 +8734,12 @@ gbString write_expr_to_string(gbString str, Ast *node, bool shorthand) {
 		if (f->flags&FieldFlag_auto_cast) {
 			str = gb_string_appendc(str, "auto_cast ");
 		}
+		if (f->flags&FieldFlag_any_int) {
+			str = gb_string_appendc(str, "#any_int ");
+		}
+		if (f->flags&FieldFlag_const) {
+			str = gb_string_appendc(str, "#const ");
+		}
 
 		for_array(i, f->names) {
 			Ast *name = f->names[i];
diff --git a/src/check_type.cpp b/src/check_type.cpp
index ab3004320..d2a216c93 100644
--- a/src/check_type.cpp
+++ b/src/check_type.cpp
@@ -1526,6 +1526,10 @@ Type *check_get_params(CheckerContext *ctx, Scope *scope, Ast *_params, bool *is
 					error(name, "'#const' can only be applied to variable fields");
 					p->flags &= ~FieldFlag_const;
 				}
+				if (p->flags&FieldFlag_any_int) {
+					error(name, "'#const' can only be applied to variable fields");
+					p->flags &= ~FieldFlag_any_int;
+				}
 
 				param = alloc_entity_type_name(scope, name->Ident.token, type, EntityState_Resolved);
 				param->TypeName.is_type_alias = true;
@@ -1572,6 +1576,12 @@ Type *check_get_params(CheckerContext *ctx, Scope *scope, Ast *_params, bool *is
 							if (!check_is_castable_to(ctx, &op, type)) {
 								ok = false;
 							}
+						} else if (p->flags&FieldFlag_any_int) {
+							if (!is_type_integer(op.type) || !is_type_integer(type)) {
+								ok = false;
+							} else if (!check_is_castable_to(ctx, &op, type)) {
+								ok = false;
+							}
 						}
 						if (!ok) {
 							success = false;
@@ -1609,6 +1619,10 @@ Type *check_get_params(CheckerContext *ctx, Scope *scope, Ast *_params, bool *is
 						error(name, "'auto_cast' can only be applied to variable fields");
 						p->flags &= ~FieldFlag_auto_cast;
 					}
+					if (p->flags&FieldFlag_any_int) {
+						error(name, "'#any_int' can only be applied to variable fields");
+						p->flags &= ~FieldFlag_any_int;
+					}
 					if (p->flags&FieldFlag_const) {
 						error(name, "'#const' can only be applied to variable fields");
 						p->flags &= ~FieldFlag_const;
@@ -1632,6 +1646,14 @@ Type *check_get_params(CheckerContext *ctx, Scope *scope, Ast *_params, bool *is
 			if (p->flags&FieldFlag_auto_cast) {
 				param->flags |= EntityFlag_AutoCast;
 			}
+			if (p->flags&FieldFlag_any_int) {
+				if (!is_type_integer(param->type)) {
+					gbString str = type_to_string(param->type);
+					error(name, "A parameter with '#any_int' must be an integer, got %s", str);
+					gb_string_free(str);
+				}
+				param->flags |= EntityFlag_AnyInt;
+			}
 			if (p->flags&FieldFlag_const) {
 				param->flags |= EntityFlag_ConstInput;
 			}
diff --git a/src/entity.cpp b/src/entity.cpp
index cd9d3a88b..e7b888365 100644
--- a/src/entity.cpp
+++ b/src/entity.cpp
@@ -62,11 +62,12 @@ enum EntityFlag : u64 {
 
 	EntityFlag_CVarArg       = 1ull<<22,
 	EntityFlag_AutoCast      = 1ull<<23,
+	EntityFlag_AnyInt        = 1ull<<24,
 
-	EntityFlag_Disabled      = 1ull<<24,
-	EntityFlag_Cold          = 1ull<<25, // procedure is rarely called
+	EntityFlag_Disabled      = 1ull<<25,
+	EntityFlag_Cold          = 1ull<<26, // procedure is rarely called
 
-	EntityFlag_Lazy          = 1ull<<26, // Lazily type checked
+	EntityFlag_Lazy          = 1ull<<27, // Lazily type checked
 
 	EntityFlag_Test          = 1ull<<30,
 
diff --git a/src/parser.cpp b/src/parser.cpp
index bdc34e4a1..2c2f609c5 100644
--- a/src/parser.cpp
+++ b/src/parser.cpp
@@ -3402,6 +3402,7 @@ enum FieldPrefixKind {
 	FieldPrefix_no_alias,
 	FieldPrefix_c_var_arg,
 	FieldPrefix_auto_cast,
+	FieldPrefix_any_int,
 };
 
 FieldPrefixKind is_token_field_prefix(AstFile *f) {
@@ -3425,6 +3426,8 @@ FieldPrefixKind is_token_field_prefix(AstFile *f) {
 				return FieldPrefix_c_var_arg;
 			} else if (f->curr_token.string == "const") {
 				return FieldPrefix_const;
+			} else if (f->curr_token.string == "any_int") {
+				return FieldPrefix_any_int;
 			}
 			break;
 		}
@@ -3440,6 +3443,7 @@ u32 parse_field_prefixes(AstFile *f) {
 	i32 c_vararg_count  = 0;
 	i32 auto_cast_count = 0;
 	i32 const_count     = 0;
+	i32 any_int_count   = 0;
 
 	for (;;) {
 		FieldPrefixKind kind = is_token_field_prefix(f);
@@ -3457,7 +3461,8 @@ u32 parse_field_prefixes(AstFile *f) {
 		case FieldPrefix_no_alias:  no_alias_count  += 1; advance_token(f); break;
 		case FieldPrefix_c_var_arg: c_vararg_count  += 1; advance_token(f); break;
 		case FieldPrefix_auto_cast: auto_cast_count += 1; advance_token(f); break;
-		case FieldPrefix_const:     const_count += 1; advance_token(f); break;
+		case FieldPrefix_const:     const_count     += 1; advance_token(f); break;
+		case FieldPrefix_any_int:   any_int_count   += 1; advance_token(f); break;
 		}
 	}
 	if (using_count     > 1) syntax_error(f->curr_token, "Multiple 'using' in this field list");
@@ -3465,6 +3470,7 @@ u32 parse_field_prefixes(AstFile *f) {
 	if (c_vararg_count  > 1) syntax_error(f->curr_token, "Multiple '#c_vararg' in this field list");
 	if (auto_cast_count > 1) syntax_error(f->curr_token, "Multiple 'auto_cast' in this field list");
 	if (const_count > 1)     syntax_error(f->curr_token, "Multiple '#const' in this field list");
+	if (any_int_count > 1)   syntax_error(f->curr_token, "Multiple '#any_int' in this field list");
 
 
 	u32 field_flags = 0;
@@ -3472,7 +3478,8 @@ u32 parse_field_prefixes(AstFile *f) {
 	if (no_alias_count  > 0) field_flags |= FieldFlag_no_alias;
 	if (c_vararg_count  > 0) field_flags |= FieldFlag_c_vararg;
 	if (auto_cast_count > 0) field_flags |= FieldFlag_auto_cast;
-	if (const_count > 0)     field_flags |= FieldFlag_const;
+	if (const_count     > 0) field_flags |= FieldFlag_const;
+	if (any_int_count  > 0)  field_flags |= FieldFlag_any_int;
 	return field_flags;
 }
 
diff --git a/src/parser.hpp b/src/parser.hpp
index 397b43f44..7e86297cc 100644
--- a/src/parser.hpp
+++ b/src/parser.hpp
@@ -275,6 +275,7 @@ enum FieldFlag {
 	FieldFlag_c_vararg  = 1<<3,
 	FieldFlag_auto_cast = 1<<4,
 	FieldFlag_const     = 1<<5,
+	FieldFlag_any_int   = 1<<6,
 
 	FieldFlag_Tags = 1<<10,