import ir
from . import astnodes

def genModule(pkg):
   m = ir.Module(pkg.name)
   for s in pkg.scope:
      print(s)
      if type(s) is astnodes.Variable:
         genGlobal(m, s)
   return m

def genGlobal(m, var):
   v = ir.Value()
   v.name = var.name
   m.Globals.append(v)

class CodeGenerator:
   """ Generates intermediate code """
   def gencode(self, ast):
      print('Code generator')
      assert type(ast) is astnodes.Package
      return genModule(ast)