from flask import Blueprint, jsonify, request
from peewee import SQL, fn
from database import Message, Member

from datetime import datetime

group_stat_api = Blueprint('group_stat_api', __name__)

@group_stat_api.route("/chat_activity", methods=["GET"])
def chat_activity():
    from_date = request.args.get("from_date")
    to_date = request.args.get("to_date")

    date_format = request.args.get("date_format")
    group_by = request.args.get("group_by")

    if not date_format:
        date_format = "%Y-%m-%d"

    if not group_by:
        group_by = "day"

    dates = []
    counts = []

    if from_date and to_date:
        from_date = datetime.strptime(from_date, date_format)
        to_date = datetime.strptime(to_date, date_format)
        
        query = (Message
                 .select(fn.date_trunc(group_by, Message.timestamp).alias('range'), fn.count(Message.id).alias('count'))
                 .where((Message.timestamp >= from_date) & (Message.timestamp <= to_date))
                 .group_by(fn.date_trunc(group_by, Message.timestamp))
                 .order_by(SQL("range"))
                 )

        for group in query:
            dates.append(group.range.strftime("%Y-%m-%d"))
            counts.append(group.count)

    return jsonify({
        "date": dates,
        "counts": counts
    })