import os
from datetime import datetime
from argparse import ArgumentParser

parser = ArgumentParser()
parser.add_argument("dir", type=str, help="Publication ID")
parser.add_argument("num", type=int, help="Publication number")
args = parser.parse_args()

basilic_path = os.path.join("/net", "servers", "www-sop", "teams", "reves", "Basilic")
print("INFO: Using basilic path -", basilic_path)
year = datetime.now().year

year_dir = os.path.join(basilic_path, str(year))
pub_dir = os.path.join(year_dir, args.dir)

cont = "f"
while cont.lower() != 'y' and cont.lower() != 'n':
    cont = input(f"INFO: Following directory will be created: {pub_dir}. Continue? [y/n]: ")
    if cont.lower() == 'n':
        exit()

# Create publication directory
os.makedirs(pub_dir, exist_ok=True, mode=0o775)

# Create index.php for year directory if it doesn't exist
year_dir_index_path = os.path.join(year_dir, "index.php")
if not os.path.exists(year_dir_index_path):
    print(f"INFO: Creating {year_dir_index_path} with permission 664")
    open(year_dir_index_path, "w").close()
    os.chmod(year_dir_index_path, 0o664)

# Create thumbs and .thumbs in publication directory
os.makedirs(os.path.join(pub_dir, "thumbs"), exist_ok=True, mode=0o775)
os.makedirs(os.path.join(pub_dir, ".thumbs"), exist_ok=True, mode=0o775)

# Write index.php for publication directory
pub_dir_index_tmpl = f'<?php $id={str(args.num)}; include("../../gdpubli.php"); ?>'
with open(os.path.join(pub_dir, "index.php"), "w") as f:
    f.write(pub_dir_index_tmpl)
os.chmod(os.path.join(pub_dir, "index.php"), 0o664)