import ir
from . import astnodes

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

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

def genFunction(m, fnc):
   f = ir.Function()
   m.Globals.append(f)

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