import configparser
from imaplib import IMAP4
import ssl
import email
import os
from glob import glob
import sys
from gpxpy import gpx

from sqlalchemy import create_engine, text
from sqlalchemy.engine import Connection
import gpxpy


config = configparser.ConfigParser()
config.read('config.ini')

db = create_engine(f"postgresql://{config['db']['username']}:{config['db']['password']}@{config['db']['host']}/{config['db']['database']}").connect()

mail = IMAP4(host=config['mail']['host'])
fitotrack_msg_filter = '(OR SUBJECT "fitotrack" SUBJECT "Fitotrack" SUBJECT "FITOTRACK" SUBJECT "FitoTrack")'


def init_database():
    with open('init.sql') as f:
        db.execute('\n'.join(f.readlines()))


def get_gpx_files_from_mail():
    try:
        os.mkdir('gpx_files')
    except FileExistsError:
        pass
    mail.starttls(ssl.create_default_context())
    mail.login(config['mail']['username'], config['mail']['password'])

    mail.create(config['mail']['mailbox_dir'])
    mail.select()
    _, ids = mail.search(None, fitotrack_msg_filter)
    ids = ids[0].split()
    for i in ids:
        _, fetched = mail.fetch(i, '(RFC822)')
        email_message = email.message_from_bytes(fetched[0][1])
        for part in email_message.walk():
            if part.get_content_maintype() == 'multipart' or part.get_content_disposition() is None:
                continue
            filename = part.get_filename()

            if filename and not os.path.exists(f'gpx_files/{filename}'):
                with open(os.path.join('gpx_files', filename), 'wb') as f:
                    f.write(part.get_payload(decode=True))
    
    mail.logout()


def process_gpx_files(tx: Connection, owner: str):
    for filepath in glob('gpx_files/*.gpx'):
        filename = os.path.split(filepath)[-1]
        if list(db.execute(text('select exists(select from training where filename = :filename)'), dict(filename=filename)))[0][0]:
            continue
        with open(filepath) as f:
            gpx_file = gpxpy.parse(f)
            if gpx_file.creator != 'FitoTrack':
                raise ValueError('gpx file not generated by the FitoTrack app')
            training_id = list(db.execute(
                """insert into training (owner, filename, medium, description, moving_time, stopped_time, moving_distance, stopped_distance) values 
                    (?, ?, ?, ?, ?, ?, ?, ?) returning id""",
                (owner,
                filename,
                "",
                gpx_file.description,
                gpx_file.get_moving_data().moving_time,
                gpx_file.get_moving_data().stopped_time,
                gpx_file.get_moving_data().moving_distance,
                gpx_file.get_moving_data().stopped_distance,)
            ))[0][0]
            for track in gpx_file.tracks:
                for segment in track.segments:
                    for point in segment.points:
                        db.execute("""
                            insert into training_data (training_id, t, lat, lon, speed, elevation)
                                values (?, ?, ?, ?, ?, ?)
                            """,
                            (training_id,
                            point.time,
                            point.latitude,
                            point.longitude,
                            point.speed,
                            point.elevation,)
                        )


def main(owner: str):
    init_database()
    get_gpx_files_from_mail()
    db.transaction(process_gpx_files, owner)


if __name__ == '__main__':
    try:
        main(sys.argv[1])
    except IndexError:
        print('Run the script with "python main.py OWNER_NAME"')
    except (KeyboardInterrupt, EOFError):
        pass