import ppci
from . import Parser, TypeChecker, Analyzer, CodeGenerator
from . astprinter import AstPrinter
import glob

class Builder:
    """ 
        Generates IR-code from c3 source.
        Reports errors to the diagnostics system
    """
    def __init__(self, diag):
        self.diag = diag
        self.parser = Parser(diag)
        self.tc = TypeChecker(diag)
        self.al = Analyzer(diag)
        self.cg = CodeGenerator()
        self.packages = {}

    def getPackage(self, pname):
        """ package provider for use when analyzing """
        if pname in self.packages:
            return self.packages[pname]
        else:
            # Try to lookup package from file
            fns = glob.glob('./*/{}.c3'.format(pname))
            if fns:
                with open(fns[0]) as f:
                    src = f.read()
                self.build(src)
            if pname in self.packages:
                return self.packages[pname]

    def parse(self, src):
        pkg = self.parser.parseSource(src)
        if not pkg:
            return

        # TODO: merge the two below?
        #AstPrinter().printAst(pkg)
        if not self.al.analyzePackage(pkg, self):
            return
        if not self.tc.checkPackage(pkg):
            return

        # Store for later use:
        self.packages[pkg.name] = pkg
        return pkg

    def build(self, src):
        """ Create IR-code from sources """
        pkg = self.parse(src)

        # Only return ircode when everything is OK
        if pkg:
            return self.cg.gencode(pkg)