import argparse

import ifcopenshell

from ifc4d.msp2ifc import MSP2Ifc
from ifc4d.p6xer2ifc import P6XER2Ifc
from ifc4d.p62ifc import P62Ifc
from ifc4d.pp2ifc import PP2Ifc

parser = argparse.ArgumentParser()
parser.add_argument("-f", "--file", action="store", type=str, required=True, help="schedule file name to be parsed")
parser.add_argument(
    "-s", "--schedule", action="store", required=True, type=str, help="file format as xer, p6xml, mspxml, pp"
)
parser.add_argument(
    "-i", "--ifcfile", action="store", required=False, type=str, help='ifc file name as string e.g. "file.ifc"'
)
parser.add_argument(
    "-o", "--output", action="store", required=True, type=str, help='ifc file name as string e.g. "file.ifc"'
)
args = parser.parse_args()


def get_file():
    ifcfile = None
    if args.ifcfile:
        ifcfile = ifcopenshell.open(args.ifcfile)
    elif args.output:
        ifc = ifcopenshell.file(schema="IFC4")
        ifc.create_entity("IfcWorkPlan")
        ifc.create_entity("IfcProject")
        ifc.write(args.output)
        ifcfile = ifcopenshell.open(args.output)
    else:
        ifcfile = None
    return ifcfile


if not args.ifcfile:
    print("You need to provide an ifc file to add schedule")
    print("Examples python ifc4d -i model.ifc -s xer -f schedule.xer -o newifc.ifc")
elif not args.output:
    print("an output file is required to save changes")
    print("python ifc4d -o newfile.ifc -s xer -f schedule.xer")

elif args.schedule == "xer":
    p6xer = P6XER2Ifc()
    p6xer.xer = args.file
    p6xer.output = args.output
    ifcfile = get_file()
    if ifcfile:
        p6xer.file = ifcfile
        p6xer.execute()
    else:
        raise Exception("No files provided for output")
elif args.schedule == "mspxml":
    msp = MSP2Ifc()
    msp.xml = args.file()
    msp.output = args.output
    ifcfile = get_file()
    if ifcfile:
        msp.file = ifcfile
        msp.execute()
elif args.schedule == "p6xml":
    p6xml = P62Ifc()
    p6xml.output = args.output
    p6xml.xml = args.file
    ifcfile = get_file()
    if ifcfile:
        p6xml.file = ifcopenshell.open(args.ifcfile)
        p6xml.execute()
elif args.schedule == "pp":
    pp = PP2Ifc()
    pp.output = args.output
    pp.pp = args.file
    ifcfile = get_file()
    if ifcfile:
        pp.file = ifcopenshell.open(args.ifcfile)
        pp.execute()
else:
    print("schedule type you selected is not implemented at the moment")