import sys
import os

if len(sys.argv) <= 1:
    print("Missing structure name. Add argument to script.")
    sys.exit()

title = str(sys.argv[1])

arguments = {'title': title, "o": "{", "c": "}"}


os.mkdir(title)
os.mkdir(title+"/Controllers")
os.mkdir(title+"/Presenters")
os.mkdir(title+"/Views")

os.chdir(title+"/Controllers")
# 1.ViewControllerFile
viewControllerFile = open(title+"ViewController.swift","w")

viewController = """import UIKit

final class {title}ViewController: UIViewController {o}

    lazy var presenter = {title}Presenter(view: self)

    override func viewDidLoad() {o}
        super.viewDidLoad()
    {c}
{c}
extension {title}ViewController: {title}View {o}
{c}
""".format(**arguments)

viewControllerFile.write(viewController)
viewControllerFile.close

os.chdir("..")
os.chdir("Presenters/")

# 2.PresenterFile
presenterFile = open(title+"Presenter.swift","w")

presenter = """
final class {title}Presenter {o}

    unowned var view: {title}View

    init(view: {title}View) {o}
        self.view = view
    {c}
{c}
""".format(**arguments)

presenterFile.write(presenter)
presenterFile.close

os.chdir("..")
os.chdir("Views/")
# 3.ViewFile
viewFile = open(title+"View.swift","w")

view = """
protocol {title}View: class {o}

{c}
""".format(**arguments)

viewFile.write(view)
viewFile.close