import sys, os, shutil
import main

def rename_drop_sql(site_name):
    files = os.walk(f"sites/{site_name}/sql/drop")

    sql_files = []
    for file in files:
        sql_files = file[2]
        
    for file_name in sql_files:
        words = None
        with open(f"sites/{site_name}/sql/drop/{file_name}", "r") as file:
            words = file.read()
            words = words.replace("%sitename%", site_name)
        
        with open(f"sites/{site_name}/sql/drop/{file_name}", "w") as file:
            file.write(words)

def rename_create_sql(site_name):
    files = os.walk(f"sites/{site_name}/sql/create")

    sql_files = []
    for file in files:
        sql_files = file[2]
        
    for file_name in sql_files:
        words = None
        with open(f"sites/{site_name}/sql/create/{file_name}", "r") as file:
            words = file.read()
            words = words.replace("%sitename%", site_name)
        
        with open(f"sites/{site_name}/sql/create/{file_name}", "w") as file:
            file.write(words)


def create():
    site_name = input("Site Name: ")
    site_owner = input("Site Owner: ")
    email = input("Contact Email: ")
    

    if not os.path.exists(f"sites/{site_name}"):
        print(f"Creating {site_name} site...")
        os.mkdir(f"sites/{site_name}")
        
        print(f"Creating sql tables files...")
        shutil.copytree(f"sites/default/sql", f"sites/{site_name}/sql")
        rename_create_sql(site_name)
        rename_drop_sql(site_name)

        with open(f"sites/{site_name}/site.ini", "w+") as config:
            config.write(f"[site]\n")
            config.write(f"site_name={site_name}\n")
            config.write(f"site_owner={site_owner}\n")
            config.write(f"email={email}\n")
        
        print(f"Site {site_name} config created!")
        print(f"Site {site_name} created!")

if __name__ == "__main__":
    if len(sys.argv) > 1:
        func_name = sys.argv[1]

        if func_name == "create_site":
            create()

        if func_name == "propagate":
            main.create_site(sys.argv[2])
        
        if func_name == "delete":
            main.delete_site(sys.argv[2])
            shutil.rmtree(f"sites/{sys.argv[2]}")