#!/usr/bin/python
#
# Copyright 2015 Michael Sparks
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
from pyxie.parsing.lexer import lexer
from pyxie.parsing.grammar import parse
from pyxie.model.transform import ast_to_cst
from pyxie.codegen.simple_cpp import C_Program, source, makefile_tmpl

import os
import pprint
import time

# Parse testing

if 0:
    testdir = "test-data/progs/"
    for testfile in os.listdir(testdir):
        if not testfile.endswith(".pyxie"):
            # Suitable for parse testing, but not compiling.
            continue
        print "_______________________________________________________________"
        print "TEST PARSING", testfile
        data = open(os.path.join(testdir,testfile)).read()
        x = parse(data, lexer)
        pprint.pprint(x)
# ast_to_cst
# Compilation Tests
testdir = "test-data"
testprogs_dir = os.path.join(testdir, "progs")

rootdir = os.getcwd()
for testfile in os.listdir(testprogs_dir):
    pyxie.codegen.simple_cpp. source = []
    os.chdir(rootdir)
    if not testfile.endswith(".pyxie"):
        # Suitable for parse testing, but not compiling.
        continue
    print "_______________________________________________________________"
    print "TEST COMPILING", testfile
    data = open(os.path.join(testprogs_dir,testfile)).read()
    AST = parse(data, lexer)
    pprint.pprint(AST)
    cname = testfile.replace(".pyxie", "")
    CST = ast_to_cst(cname, AST)
    print "CST:", pprint.pformat(CST)

    program = C_Program.fromjson(CST)
    program.generate()

    print source

    allresults_dir = os.path.join(testdir, "genprogs")
    
    thisresult_dir = os.path.join(allresults_dir, testfile)
    
    pprint.pprint(source, width=200)

    now = int(time.time())

    print "BUILDING PROGRAM", thisresult_dir
    try:
        os.mkdir(thisresult_dir)
    except OSError as e:
       if e.errno != 17: # Directory exists
           raise
    f = open(os.path.join(thisresult_dir,program.name+".c"), "w")
    for line in source:
        f.write(line)
        f.write("\n")
    f.close()

    makefile = makefile_tmpl % {"filename": program.name }
    f = open(os.path.join(thisresult_dir ,"Makefile"), "w")
    f.write(makefile)
    f.close()

    os.chdir(thisresult_dir)
    os.system("make")
