#! /usr/bin/env python




from __future__ import annotations

from IPython import embed
from IPython.terminal.ipapp import load_default_config



import liblktlang
import liblktlang as lkt


HEADER = """
--
-- liblktlang playground
--

The file(s) passed as argument have been put into the u variable, or units if
there are multiple.
""".strip()


class Playground(liblktlang.App):

    def __init__(self, args=None):
        self.parsing_errors = []
        """
        List of parsing errors found during App initialization.
        """

        super().__init__(args)

    def add_arguments(self):
        self.parser.add_argument(
            '-i', '--input-script', type=str, default='',
            help="Script to execute when playground has loaded the units"
        )
        super(Playground, self).add_arguments()

    def on_parsing_errors(self, unit):
        self.parsing_errors += [
            unit.format_gnu_diagnostic(d) for d in unit.diagnostics
        ]

    def main(self):
        print(HEADER)
        if self.parsing_errors:
            print("")
            print("Note that some units have parsing errors:")
            print("")
            for line in self.parsing_errors:
                print(line)

        c = load_default_config()

        if self.args.input_script:
            execfile(self.args.input_script)

        # Put useful values in local variables, so that they're easily
        # accessible from embed.
        units = self.units
        ctx = self.ctx
        u = self.u

        embed(config=c, display_banner=False)


if __name__ == '__main__':
    Playground.run()
