import json
import os
from datetime import datetime, timezone

import frontmatter

from .extract import extract_html, extract_media
from .utils import DateTimeEncoder

OLD_DATE = "2016-03-05 22:22:00.350000"
EXPORT_DEST = "../discoursio-web/data/"
parentDir = "/".join(os.getcwd().split("/")[:-1])
contentDir = parentDir + "/discoursio-web/content/"
ts = datetime.now(tz=timezone.utc)


def get_metadata(r):
    authors = []
    for a in r["authors"]:
        authors.append(
            {  # a short version for public listings
                "slug": a.slug or "discours",
                "name": a.name or "Дискурс",
                "userpic": a.userpic or "https://discours.io/static/img/discours.png",
            }
        )
    metadata = {}
    metadata["title"] = r.get("title", "").replace("{", "(").replace("}", ")")
    metadata["authors"] = authors
    metadata["createdAt"] = r.get("createdAt", ts)
    metadata["layout"] = r["layout"]
    metadata["topics"] = [topic for topic in r["topics"]]
    metadata["topics"].sort()
    if r.get("cover", False):
        metadata["cover"] = r.get("cover")
    return metadata


def export_mdx(r):
    # print('[export] mdx %s' % r['slug'])
    content = ""
    metadata = get_metadata(r)
    content = frontmatter.dumps(frontmatter.Post(r["body"], **metadata))
    ext = "mdx"
    filepath = contentDir + r["slug"]
    bc = bytes(content, "utf-8").decode("utf-8", "ignore")
    open(filepath + "." + ext, "w").write(bc)


def export_body(shout, storage):
    entry = storage["content_items"]["by_oid"][shout["oid"]]
    if entry:
        body = extract_html(entry)
        media = extract_media(entry)
        shout["body"] = body  # prepare_html_body(entry)  # prepare_md_body(entry)
        shout["media"] = media
        export_mdx(shout)
        print("[export] html for %s" % shout["slug"])
        open(contentDir + shout["slug"] + ".html", "w").write(body)
    else:
        raise Exception("no content_items entry found")


def export_slug(slug, storage):
    shout = storage["shouts"]["by_slug"][slug]
    shout = storage["shouts"]["by_slug"].get(slug)
    assert shout, "[export] no shout found by slug: %s " % slug
    author = shout["authors"][0]
    assert author, "[export] no author error"
    export_body(shout, storage)


def export_email_subscriptions():
    email_subscriptions_data = json.loads(open("migration/data/email_subscriptions.json").read())
    for data in email_subscriptions_data:
        # TODO: migrate to mailgun list manually
        # migrate_email_subscription(data)
        pass
    print("[migration] " + str(len(email_subscriptions_data)) + " email subscriptions exported")


def export_shouts(storage):
    # update what was just migrated or load json again
    if len(storage["users"]["by_slugs"].keys()) == 0:
        storage["users"]["by_slugs"] = json.loads(open(EXPORT_DEST + "authors.json").read())
        print("[migration] " + str(len(storage["users"]["by_slugs"].keys())) + " exported authors ")
    if len(storage["shouts"]["by_slugs"].keys()) == 0:
        storage["shouts"]["by_slugs"] = json.loads(open(EXPORT_DEST + "articles.json").read())
        print(
            "[migration] " + str(len(storage["shouts"]["by_slugs"].keys())) + " exported articles "
        )
    for slug in storage["shouts"]["by_slugs"].keys():
        export_slug(slug, storage)


def export_json(export_articles={}, export_authors={}, export_topics={}, export_comments={}):
    open(EXPORT_DEST + "authors.json", "w").write(
        json.dumps(
            export_authors,
            cls=DateTimeEncoder,
            indent=4,
            sort_keys=True,
            ensure_ascii=False,
        )
    )
    print("[migration] " + str(len(export_authors.items())) + " authors exported")
    open(EXPORT_DEST + "topics.json", "w").write(
        json.dumps(
            export_topics,
            cls=DateTimeEncoder,
            indent=4,
            sort_keys=True,
            ensure_ascii=False,
        )
    )
    print("[migration] " + str(len(export_topics.keys())) + " topics exported")

    open(EXPORT_DEST + "articles.json", "w").write(
        json.dumps(
            export_articles,
            cls=DateTimeEncoder,
            indent=4,
            sort_keys=True,
            ensure_ascii=False,
        )
    )
    print("[migration] " + str(len(export_articles.items())) + " articles exported")
    open(EXPORT_DEST + "comments.json", "w").write(
        json.dumps(
            export_comments,
            cls=DateTimeEncoder,
            indent=4,
            sort_keys=True,
            ensure_ascii=False,
        )
    )
    print("[migration] " + str(len(export_comments.items())) + " exported articles with comments")