Add Module::getOrInsertGlobal(string, llvm::Type)

Author: Lotes

include/IR/Module.h

@@ -48,6 +48,8 @@ class Module : public Napi::ObjectWrap<Module> {
 
     Napi::Value getGlobalVariable(const Napi::CallbackInfo &info);
 
+    Napi::Value getOrInsertGlobal(const Napi::CallbackInfo &info);
+
     void addModuleFlag(const Napi::CallbackInfo &info);
 
     Napi::Value empty(const Napi::CallbackInfo &info);

include/Util/ErrMsg.h

@@ -24,6 +24,7 @@ namespace ErrMsg {
             constexpr const char *getFunction = "Module.getFunction needs to be called with: (name: string)";
             constexpr const char *getOrInsertFunction = "Module.getOrInsertFunction needs to be called with: (name: string, fnType: FunctionType)";
             constexpr const char *getGlobalVariable = "Module.getGlobalVariable needs to be called with: (name: string, allowInternal?: boolean)";
+            constexpr const char *getOrInsertGlobal = "Module.getOrInsertGlobal needs to be called with: (name: string, type: Type)";
             constexpr const char *addModuleFlag = "Module.addModuleFlag needs to be called with (behavior: number, key: string, value: number)"
                                                   "\n\t - limit: behavior should belong to [1, 7]";
         }

llvm-bindings.d.ts

@@ -227,6 +227,8 @@ declare namespace llvm {
 
         public getGlobalVariable(name: string, allowInternal?: boolean): GlobalVariable | null;
 
+        public getOrInsertGlobal(name: string, varType: Type): FunctionCallee;
+
         public addModuleFlag(behavior: number, key: string, value: number): void;
 
         public empty(): boolean;

src/IR/Module.cpp

@@ -31,6 +31,7 @@ void Module::Init(Napi::Env env, Napi::Object &exports) {
             InstanceMethod("getFunction", &Module::getFunction),
             InstanceMethod("getOrInsertFunction", &Module::getOrInsertFunction),
             InstanceMethod("getGlobalVariable", &Module::getGlobalVariable),
+            InstanceMethod("getOrInsertGlobal", &Module::getOrInsertGlobal),
             InstanceMethod("addModuleFlag", &Module::addModuleFlag),
             InstanceMethod("empty", &Module::empty),
             InstanceMethod("print", &Module::print)
@@ -198,6 +199,17 @@ Napi::Value Module::getGlobalVariable(const Napi::CallbackInfo &info) {
     throw Napi::TypeError::New(env, ErrMsg::Class::Module::getGlobalVariable);
 }
 
+Napi::Value Module::getOrInsertGlobal(const Napi::CallbackInfo &info) {
+    Napi::Env env = info.Env();
+    if (info.Length() == 2 && info[0].IsString() && Type::IsClassOf(info[1])) {
+        std::string functionName = info[0].As<Napi::String>();
+        llvm::Type *type = Type::Extract(info[1]);
+        llvm::Constant* glob = module->getOrInsertGlobal(functionName, type);
+        return GlobalVariable::New(env, dyn_cast_or_null<llvm::GlobalVariable>(glob));
+    }
+    throw Napi::TypeError::New(env, ErrMsg::Class::Module::getOrInsertGlobal);
+}
+
 void Module::addModuleFlag(const Napi::CallbackInfo &info) {
     Napi::Env env = info.Env();
     if (info.Length() == 3 && info[0].IsNumber() && info[1].IsString() && info[2].IsNumber()) {

tests/IR/Module.spec.ts

@@ -359,4 +359,13 @@ describe('Test Module', () => {
             expect(module.print()).toMatchSnapshot();
         });
     });
+
+    describe('Global variables', () => {
+        test('Global variable declared', () => {
+            const context = new llvm.LLVMContext();
+            const module = new llvm.Module(FileName, context);
+            module.getOrInsertGlobal('HomerSimpson', llvm.Type.getInt32Ty(context));
+            expect(module.print()).toContain('@HomerSimpson');
+        });
+    });
 });