import requests
import json
import datetime
import streamlit as st
from itertools import zip_longest
import os
import seaborn as sns
from matplotlib.backends.backend_agg import RendererAgg
import matplotlib.pyplot as plt
import matplotlib.dates as mdates

_lock = RendererAgg.lock


def basic_info():
    config = dict()
    config["access_token"] = st.secrets["access_token"]
    config['instagram_account_id'] = st.secrets.get("instagram_account_id", "")
    config["version"] = 'v16.0'
    config["graph_domain"] = 'https://graph.facebook.com/'
    config["endpoint_base"] = config["graph_domain"] + config["version"] + '/'
    return config


def InstaApiCall(url, params, request_type):
    if request_type == 'POST':
        req = requests.post(url, params)
    else:
        req = requests.get(url, params)
    res = dict()
    res["url"] = url
    res["endpoint_params"] = params
    res["endpoint_params_pretty"] = json.dumps(params, indent=4)
    res["json_data"] = json.loads(req.content)
    res["json_data_pretty"] = json.dumps(res["json_data"], indent=4)
    return res


def getUserMedia(params, pagingUrl=''):
    Params = dict()
    Params['fields'] = 'id,caption,media_type,media_url,permalink,thumbnail_url,timestamp,username,like_count,comments_count'
    Params['access_token'] = params['access_token']

    if not params['endpoint_base']:
        return None

    if pagingUrl == '':
        url = params['endpoint_base'] + params['instagram_account_id'] + '/media'
    else:
        url = pagingUrl

    return InstaApiCall(url, Params, 'GET')


def getUser(params):
    Params = dict()
    Params['fields'] = 'followers_count'
    Params['access_token'] = params['access_token']

    if not params['endpoint_base']:
        return None

    url = params['endpoint_base'] + params['instagram_account_id']

    return InstaApiCall(url, Params, 'GET')


def saveCount(count, filename):
    with open(filename, 'w') as f:
        json.dump(count, f, indent=4)


def getCount(filename):
    try:
        with open(filename, 'r') as f:
            return json.load(f)
    except (FileNotFoundError, json.decoder.JSONDecodeError):
        return {}


st.set_page_config(layout="wide")
params = basic_info()

count_filename = "count.json"

if not params['instagram_account_id']:
    st.write('.envファイルでinstagram_account_idを確認')
else:
    response = getUserMedia(params)
    user_response = getUser(params)
    if not response or not user_response:
        st.write('.envファイルでaccess_tokenを確認')
    else:
        posts = response['json_data']['data'][::-1]
        user_data = user_response['json_data']
        followers_count = user_data.get('followers_count', 0)

        NUM_COLUMNS = 6
        MAX_WIDTH = 1000
        BOX_WIDTH = int(MAX_WIDTH / NUM_COLUMNS)
        BOX_HEIGHT = 400

        yesterday = (datetime.datetime.now(datetime.timezone(datetime.timedelta(hours=9))) - datetime.timedelta(days=1)).strftime('%Y-%m-%d')
        follower_diff = followers_count - getCount(count_filename).get(yesterday, {}).get('followers_count', followers_count)
        st.markdown(f"<h4 style='font-size:1.2em;'>Follower: {followers_count} ({'+' if follower_diff >= 0 else ''}{follower_diff})</h4>", unsafe_allow_html=True)

        show_description = st.checkbox("キャプションを表示")
        show_summary_chart = st.checkbox("サマリーグラフを表示")
        show_count_chart = st.checkbox("いいね／コメント数グラフを表示")

        posts.reverse()
        post_groups = [list(filter(None, group)) for group in zip_longest(*[iter(posts)] * NUM_COLUMNS)]

        count = getCount(count_filename)
        today = datetime.datetime.now(datetime.timezone(datetime.timedelta(hours=9))).strftime('%Y-%m-%d')

        if today not in count:
            count[today] = {}

        count[today]['followers_count'] = followers_count

        if datetime.datetime.now(datetime.timezone(datetime.timedelta(hours=9))).strftime('%H:%M') == '23:59':
            count[yesterday] = count[today]

        max_like_diff = 0
        max_comment_diff = 0
        for post_group in post_groups:
            for post in post_group:
                like_count_diff = post['like_count'] - count.get(yesterday, {}).get(post['id'], {}).get('like_count', post['like_count'])
                comment_count_diff = post['comments_count'] - count.get(yesterday, {}).get(post['id'], {}).get('comments_count', post['comments_count'])
                max_like_diff = max(like_count_diff, max_like_diff)
                max_comment_diff = max(comment_count_diff, max_comment_diff)

        if show_summary_chart:
            with _lock:
                fig, ax1 = plt.subplots()
                ax2 = ax1.twinx()
                ax1.xaxis.set_major_formatter(mdates.DateFormatter('%m/%d'))

                followers_data = [c.get('followers_count', 0) for _, c in count.items()]
                like_data = [count[date].get(post['id'], {}).get('like_count', 0) - count.get(yesterday, {}).get(post['id'], {}).get('like_count', count[date].get(post['id'], {}).get('like_count', 0)) for post in post_group for date, c in count.items()]
                comment_data = [count[date].get(post['id'], {}).get('comments_count', 0) - count.get(yesterday, {}).get(post['id'], {}).get('comments_count', count[date].get(post['id'], {}).get('comments_count', 0)) for post in post_group for date, c in count.items()]

                ax1.plot([datetime.datetime.strptime(k, '%Y-%m-%d') for k in count], followers_data, color="cyan", label="followers_count")
                ax1.plot([datetime.datetime.strptime(k, '%Y-%m-%d') for k in count], like_data, color="orange", label='like_count')
                ax2.plot([datetime.datetime.strptime(k, '%Y-%m-%d') for k in count], comment_data, color="green", label='comments_count')

                ax1.set_ylabel('follower_count, like_count')
                ax2.set_ylabel('comment_count')

                ax1.legend(loc="upper left")
                ax2.legend(loc="upper right")

                st.pyplot(fig)

        for post_group in post_groups:
            with st.container():
                columns = st.columns(NUM_COLUMNS)
                for i, post in enumerate(post_group):
                    with columns[i]:
                        st.image(post['media_url'], width=BOX_WIDTH, use_column_width=True)
                        st.write(f"{datetime.datetime.strptime(post['timestamp'], '%Y-%m-%dT%H:%M:%S%z').astimezone(datetime.timezone(datetime.timedelta(hours=9))).strftime('%Y-%m-%d %H:%M:%S')}")
                        like_count_diff = post['like_count'] - count.get(yesterday, {}).get(post['id'], {}).get('like_count', post['like_count'])
                        comment_count_diff = post['comments_count'] - count.get(yesterday, {}).get(post['id'], {}).get('comments_count', post['comments_count'])
                        st.markdown(
                            f"👍: {post['like_count']} <span style='{'' if like_count_diff != max_like_diff or max_like_diff == 0 else 'color:green;'}'>({'+' if like_count_diff >= 0 else ''}{like_count_diff})</span>"
                            f"\n💬: {post['comments_count']} <span style='{'' if comment_count_diff != max_comment_diff or max_comment_diff == 0 else 'color:green;'}'>({'+' if comment_count_diff >= 0 else ''}{comment_count_diff})</span>",
                            unsafe_allow_html=True)
                        caption = post['caption']
                        if caption is not None:
                            caption = caption.strip()
                            if "[Description]" in caption:
                                caption = caption.split("[Description]")[1].lstrip()
                            if "[Tags]" in caption:
                                caption = caption.split("[Tags]")[0].rstrip()
                            caption = caption.replace("#", "")
                            caption = caption.replace("[model]", "👗")
                            caption = caption.replace("[Equip]", "📷")
                            caption = caption.replace("[Develop]", "🖨")
                            if show_description:
                                st.write(caption or "No caption provided")
                            else:
                                st.write(caption[:0] if caption is not None and len(caption) > 50 else caption or "No caption provided")

                        if show_count_chart:
                            with _lock:
                                fig, ax1 = plt.subplots()
                                ax2 = ax1.twinx()
                                ax1.xaxis.set_major_formatter(mdates.DateFormatter('%m/%d'))

                                like_data = [count[date].get(post['id'], {}).get('like_count', 0) - count.get(yesterday, {}).get(post['id'], {}).get('like_count', count[date].get(post['id'], {}).get('like_count', 0)) for date, c in count.items()]
                                comment_data = [count[date].get(post['id'], {}).get('comments_count', 0) - count.get(yesterday, {}).get(post['id'], {}).get('comments_count', count[date].get(post['id'], {}).get('comments_count', 0)) for date, c in count.items()]

                                ax1.plot([datetime.datetime.strptime(k, '%Y-%m-%d') for k in count], like_data, color="orange", label='like_count')
                                ax2.plot([datetime.datetime.strptime(k, '%Y-%m-%d') for k in count], comment_data, color="green", label='comments_count')

                                ax1.set_ylabel('like_count')
                                ax2.set_ylabel('comment_count')

                                ax1.legend(loc="upper left")
                                ax2.legend(loc="upper right")

                                st.pyplot(fig)

                        count[today][post['id']] = {'like_count': post['like_count'], 'comments_count': post['comments_count']}
        saveCount(count, count_filename)