import matplotlib.pyplot as plt
import matplotlib.dates as mdates
from matplotlib.ticker import FormatStrFormatter, StrMethodFormatter
import numpy as np
import pandas as pd
import pdb

SMALL_SIZE = 6
MEDIUM_SIZE = 10
BIGGER_SIZE = 12

plt.rc('font', size=MEDIUM_SIZE)          # controls default text sizes
plt.rc('axes', titlesize=MEDIUM_SIZE)     # fontsize of the axes title
plt.rc('axes', labelsize=MEDIUM_SIZE)    # fontsize of the x and y labels
plt.rc('xtick', labelsize=SMALL_SIZE)    # fontsize of the tick labels
plt.rc('ytick', labelsize=MEDIUM_SIZE)    # fontsize of the tick labels
plt.rc('legend', fontsize=MEDIUM_SIZE)    # legend fontsize
plt.rc('figure', titlesize=BIGGER_SIZE)  # fontsize of the figure title
plt.style.use('seaborn-v0_8-dark')

mdates.set_epoch('2000-01-01T00:00:00')

class Graphit():

    name = 'fediverse data graph library'

    def __init__(self, servers=None, servers_max=None, users=None, users_max=None, mau=None, mau_max=None, global_day=None, global_servers=None, global_users=None, global_mau=None):

        self.servers = servers
        self.servers_max = servers_max
        self.users = users
        self.users_max = users_max
        self.mau = mau
        self.mau_max = mau_max
        self.global_day = global_day
        self.global_servers = global_servers
        self.global_users = global_users
        self.global_mau = global_mau

    def graph(self):

        # generate graphs

        plt.plot([-6, -5, -4, -3, -2, -1, 0], [self.servers[6], self.servers[5], self.servers[4], self.servers[3], self.servers[2], self.servers[1], self.servers[0]], marker='o', color='mediumseagreen')

        plt.plot([-6, -5, -4, -3, -2, -1, 0], [self.servers_max, self.servers_max, self.servers_max, self.servers_max, self.servers_max, self.servers_max, self.servers_max], color='red')

        plt.title('fediverse: total alive servers (max: ' + str(f"{self.servers_max:,}" + ')'), loc='right', color='blue')

        plt.xlabel('Last seven days')

        plt.ylabel('fediverse alive servers')

        plt.grid(visible=True)

        plt.legend(('servers', 'max'), shadow=True, loc=(0.01, 1.00), handlelength=1.5, fontsize=10)

        plt.savefig('servers.png')

        plt.close()

        plt.plot([-6, -5, -4, -3, -2, -1, 0], [self.users[6], self.users[5], self.users[4], self.users[3], self.users[2], self.users[1], self.users[0]], marker='o', color='royalblue')

        plt.plot([-6, -5, -4, -3, -2, -1, 0], [self.users_max, self.users_max, self.users_max, self.users_max, self.users_max, self.users_max, self.users_max], color='red')

        plt.title('fediverse: total users (max: ' + str(f"{self.users_max:,}" + ')'), loc='right', color='royalblue')

        plt.legend(('users', 'max'), shadow=True, loc=(0.01, 0.80), handlelength=1.5, fontsize=10)

        plt.xlabel('Last seven days')

        plt.ylabel('users')

        plt.grid(visible=True)

        plt.savefig('users.png')

        plt.close()

        plt.plot([-6, -5, -4, -3, -2, -1, 0], [self.mau[6], self.mau[5], self.mau[4], self.mau[3], self.mau[2], self.mau[1], self.mau[0]], marker='o', color='royalblue')

        plt.plot([-6, -5, -4, -3, -2, -1, 0], [self.mau_max, self.mau_max, self.mau_max, self.mau_max, self.mau_max, self.mau_max, self.mau_max], color='red')

        plt.title('fediverse: total MAU (max: ' + str(f"{self.mau_max:,}" + ')'), loc='right', color='royalblue')

        plt.legend(('mau', 'max'), shadow=True, loc=(0.01, 0.80), handlelength=1.5, fontsize=10)

        plt.xlabel('Last seven days')

        plt.ylabel('MAU')

        plt.grid(visible=True)

        plt.savefig('mau.png')

        plt.close()

    def generate(self):

        df = pd.DataFrame(
                {'date': np.array(self.global_day),
                'servers': np.array(self.global_servers),
                'users': np.array(self.global_users),
                'mau': np.array(self.global_mau)})

        df['date'] = pd.to_datetime(df['date'])

        # servers

        fig, ax = plt.subplots()

        server_min = 0

        ax.fill_between(df.date, server_min, df.servers, alpha=0.7, label='Registered', color='darkgreen')

        ax.yaxis.set_major_formatter(StrMethodFormatter('{x:,.0f}'))

        plt.tick_params(rotation=45)

        ax.set_title("fediverse's total servers")

        ax.set_xlabel('date')

        ax.set_ylabel('servers')

        ax.grid(visible=True)

        ax.label_outer()

        plt.savefig('app/graphs/global_servers.png')

        # users

        fig, ax = plt.subplots()

        users_min = 0

        ax.fill_between(df.date, users_min, df.users, alpha=0.7, label='Registered', color='blue')

        ax.yaxis.set_major_formatter(StrMethodFormatter('{x:,.0f}'))

        plt.tick_params(rotation=45)

        ax.set_title("fediverse's registered users")

        ax.set_xlabel('date')

        ax.set_ylabel('users')

        ax.grid(visible=True)

        plt.savefig('app/graphs/global_users.png')

        # mau

        fig, ax = plt.subplots()

        ax.plot(df.date, df.mau, label='MAU', color='blue')

        ax.yaxis.set_major_formatter(StrMethodFormatter('{x:,.0f}'))

        plt.tick_params(rotation=45)

        ax.set_title("fediverse's Monthly Active Users")
        
        ax.set_xlabel('date')

        ax.set_ylabel('users')

        ax.grid(visible=True)

        plt.savefig('app/graphs/global_mau.png')

        plt.close()