from classes.config import Config

from pathlib import Path
from io import BytesIO

import datetime
import time

from gnupg import GPG

def log_string(input_string, log_level="INFO"):
    now = datetime.datetime.now()
    print(f"[{now.strftime('%Y-%m-%d %H:%M:%S.%f')}][{log_level}] {input_string}")

if __name__ == "__main__":
    config = Config.from_file(Path(__file__).parent / "settings.ini")

    for directory in config.directories:
        log_string(f"Start processing {directory.name}")
        source = Path(directory.source)
        
        for newfile in source.iterdir():
            if newfile.is_file() and (time.time() - newfile.stat().st_mtime > 20):
                log_string(f"Found file {newfile.name}")
                try:
                    try:
                        raw = newfile.read_text(encoding="utf-8").encode("utf-8")
                    except UnicodeDecodeError:
                        raw = newfile.read_text(encoding="Windows-1252").encode("utf-8")

                    if config.server.theirkey and config.server.ourkey:
                        log_string(f"Encrypting for {config.server.theirkey}...")
                        encrypted = GPG().encrypt(raw, config.server.theirkey, sign=config.server.ourkey, always_trust=True).data
                    else:
                        log_string("Not encrypting. Do not use in production.", "WARNING")
                        encrypted = raw.encode()

                    upfl = BytesIO(encrypted)
                    uppath = Path(config.server.inpath) / f"{newfile.name}.pgp"

                    with config.server.get_sftp_client() as sftp:
                        log_string(f"Uploading to {uppath}...")
                        sftp.putfo(upfl, str(uppath))

                    if directory.sourcebackup:
                        log_string(f"Backing up...")
                        newfile.rename(Path(directory.sourcebackup) / newfile.name)
                    else:
                        log_string(f"Deleting input file...")
                        newfile.unlink()
                        log_string("Completed processing {newfile.name}")

                except Exception as e:
                    log_string(f"Something went wrong uploading file {newfile.name} from {directory.name}: {e}", "ERROR")

    try:
        log_string("Start downloading files from server")
        with config.server.get_sftp_client() as sftp:
            for response in sftp.listdir(config.server.outpath):
                rpath = str(Path(config.server.outpath) / response)
                if (time.time() - sftp.stat(rpath).st_mtime < 20):
                    continue

                log_string(f"Found file {response}, downloading...")

                encrypted = sftp.open(rpath).read()

                if config.server.ourkey:
                    log_string("Decrypting...")
                    decrypted = GPG().decrypt(encrypted).data.decode()
                else:
                    log_string("Not decrypting. Do not use in production.", "WARN")
                    decrypted = str(encrypted)

                founddir = None

                for directory in config.directories:
                    if directory.regex.search(response).group():
                        log_string(f"File seems to belong to {directory.name}")
                        founddir = directory
                        break

                    if not founddir:
                        founddir = directory

                outpath = Path(directory.destination) / response

                if outpath.suffix == ".pgp":
                    outpath = outpath.with_suffix("")

                assert not outpath.exists()

                outpath.write_text(decrypted)

                if founddir.destinationbackup:
                    log_string("Backing up...")
                    (Path(founddir.destinationbackup) / outpath.name).write_text(decrypted)

                log_string(f"Deleting file {response} from server...")
                sftp.remove(rpath)

                log_string(f"Done processing {response}")

    except Exception as e:
        log_string(f"Something went wrong downloading files from the server: {e}", "ERROR")