feat: support ffi_export attribute

quartz/ast.qz

@@ -107,6 +107,7 @@ struct Function {
  }?,
  no_allocation: bool,
  is_test: bool,
+ ffi_export: string?,
 }
 
 module Function {

quartz/generator.qz

@@ -581,6 +581,7 @@ module Generator {
  }]](),
  result_type: self.main_signature_result!,
  body: start_body,
+ ffi_export: nil,
  });
 
  self.start();
@@ -638,8 +639,17 @@ module Generator {
  self.main_signature_result = func.result_type?;
  }
 
+ if func.ffi_export != nil {
+ self.start();
+ self.write("export");
+ self.write(derive::to_string(func.ffi_export!));
+ self.write(format("(func ${})", func.name));
+ self.end();
+ }
+
  self.start();
  self.write("func");
+
  self.write("$".concat(func.name));
 
  for i in 0..func.params.length {

quartz/ir.qz

@@ -194,6 +194,7 @@ struct IrFunc {
  }],
  result_type: IrType,
  body: vec[IrTerm],
+ ffi_export: string?,
 }
 
 struct IrCall {

quartz/ir_code_gen.qz

@@ -257,6 +257,7 @@ module IrCodeGenerator {
  },
  },
  }),
+ ..nil,
  },
  });
 
@@ -302,6 +303,7 @@ module IrCodeGenerator {
  t_address: true,
  },
  body: body,
+ ..nil,
  },
  });
  }
@@ -629,6 +631,7 @@ module IrCodeGenerator {
  t_nil: true,
  },
  body: body,
+ ..nil,
  },
  };
  }
@@ -745,6 +748,7 @@ module IrCodeGenerator {
  result_type: IrType {
  t_nil: true,
  },
+ ..nil,
  },
  });
 
@@ -807,6 +811,7 @@ module IrCodeGenerator {
  params: params,
  result_type: IrType::new(func.result_type),
  body: body,
+ ffi_export: func.ffi_export,
  };
  }
 

quartz/parser.qz

@@ -391,6 +391,7 @@ module Parser {
  variadic: params.variadic,
  no_allocation: false,
  is_test: false,
+ ffi_export: nil,
  },
  location: Location {
  start: start_token.location.start,

quartz/preprocessor.qz

@@ -149,6 +149,21 @@ module Preprocessor {
  }
 
  next.t_func!.is_test = true;
+ } else if attr.name.data == "ffi_export" {
+ let next = m.decls.at(index + 1).data;
+ if next.t_func == nil {
+ return _ or error::new(
+ "expected function, but got {}".format(derive::to_string(next)),
+ );
+ }
+ if attr.args.length == 0 {
+ return _ or error::new("expected ffi_export");
+ }
+ if attr.args.at(0).data.t_string == nil {
+ return _ or error::new("expected string");
+ }
+
+ next.t_func!.ffi_export = attr.args.at(0).data.t_string!.value?;
  } else {
  return _ or error::new("unknown attribute");
  }