from chalicelib.utils import pg_client
from chalicelib.core import license
import requests


def process_data(data):
    return {
        'edition': license.EDITION,
        'tracking': data["opt_out"],
        'version': data["version_number"],
        'user_id': data["tenant_key"],
        'tenant_key': data["tenant_key"],
        'owner_email': None if data["opt_out"] else data["email"],
        'organization_name': None if data["opt_out"] else data["name"],
        'users_count': data["t_users"],
        'projects_count': data["t_projects"],
        'sessions_count': data["t_sessions"],
        'integrations_count': data["t_integrations"]
    }


def compute():
    with pg_client.PostgresClient(long_query=True) as cur:
        cur.execute(
            f"""UPDATE public.tenants
                SET t_integrations = COALESCE((SELECT COUNT(DISTINCT provider)
                                               FROM public.integrations
                                                        INNER JOIN public.projects USING (project_id)
                                               WHERE projects.tenant_id = all_tenants.tenant_id) +
                                              (SELECT COUNT(*)
                                               FROM public.webhooks
                                               WHERE webhooks.tenant_id = all_tenants.tenant_id
                                                 AND type = 'slack') +
                                              (SELECT COUNT(*)
                                               FROM public.jira_cloud
                                                        INNER JOIN public.users USING (user_id)
                                               WHERE users.tenant_id = all_tenants.tenant_id), 0),
                    t_projects=COALESCE((SELECT COUNT(*)
                                         FROM public.projects
                                         WHERE deleted_at ISNULL
                                           AND projects.tenant_id = all_tenants.tenant_id), 0),
                    t_sessions=t_sessions + COALESCE((SELECT COUNT(*)
                                                      FROM public.sessions INNER JOIN public.projects USING (project_id)
                                                      WHERE projects.tenant_id = all_tenants.tenant_id
                                                        AND start_ts >= (SELECT last_telemetry FROM tenants)
                                                        AND start_ts <=CAST(EXTRACT(epoch FROM date_trunc('day', now())) * 1000 AS BIGINT)), 0),
                    t_users=COALESCE((SELECT COUNT(*)
                                      FROM public.users
                                      WHERE deleted_at ISNULL
                                        AND users.tenant_id = all_tenants.tenant_id), 0),
                    last_telemetry=CAST(EXTRACT(epoch FROM date_trunc('day', now())) * 1000 AS BIGINT)
                FROM (SELECT tenant_id
                      FROM public.tenants
                     ) AS all_tenants
                WHERE tenants.tenant_id = all_tenants.tenant_id
                RETURNING name,t_integrations,t_projects,t_sessions,t_users,tenant_key,opt_out,
                    (SELECT openreplay_version()) AS version_number,
                    (SELECT email FROM public.users WHERE role = 'owner' AND users.tenant_id=tenants.tenant_id LIMIT 1);"""
        )
        data = cur.fetchall()
        if len(data) > 0:
            requests.post('https://api.openreplay.com/os/telemetry',
                          json={"stats": [process_data(d) for d in data]})


def new_client(tenant_id):
    with pg_client.PostgresClient() as cur:
        cur.execute(
            cur.mogrify(f"""SELECT *, openreplay_version() AS version_number,
                            (SELECT email FROM public.users WHERE tenant_id=%(tenant_id)s AND role='owner' LIMIT 1) AS email
                            FROM public.tenants 
                            WHERE tenant_id=%(tenant_id)s
                            LIMIT 1;""", {"tenant_id": tenant_id}))
        data = cur.fetchone()
        requests.post('https://api.openreplay.com/os/signup', json=process_data(data))