from rocketchat_API.rocketchat import RocketChat
import os
import json
from datetime import datetime
from monthdelta import monthdelta

def main():
    print("Start chat info generation")

    crapauds_total = 0
    crapauds_recent = 0
    channels_total = 0
    channels_recent = 0
    channels_list = []
    messages_total = 0
    messages_recent = 0

    recent_date = datetime.now() - monthdelta()

    rocket = RocketChat(None, None, auth_token= os.environ['ROCKETCHAT_AUTH'], user_id= os.environ['ROCKETCHAT_USERID'], server_url=os.environ['ROCKETCHAT_SERVER'])

    print("Check users")
    users = rocket.users_list().json()

    crapauds_total = users["total"]
    
    print("Check channels")
    channels = getAllChannels(rocket)
    channels_total = len(channels)
    
    users = []
    for channel in channels:
        messages_total += int(channel["msgs"])
        date = channel["lm"]
        channel_date = datetime.strptime(date, "%Y-%m-%dT%H:%M:%S.%fZ")
        if  channel_date > recent_date:
            channels_recent += 1
            
            print("Check messages for channels {}".format(channel['name']))
            messages = rocket.channels_history(channel["_id"], oldest= recent_date, count= 10000).json()
            if messages["success"]:
                nbMessages = len(messages["messages"])
                messages_recent += nbMessages
                channels_list.append((channel["name"], nbMessages))
                users.extend(map(lambda message: message["u"]["_id"], messages["messages"]))
            else:
                print("Error : {}".format(messages["error"]))

    # Get the channels with the most number of message
    channels_list.sort(key=lambda channel: channel[1], reverse= True)
    channels_recentlist = list(map(lambda channel: channel[0], channels_list))

    crapauds_recent = len(set(users))

    info = {
        "crapauds": {
            "total": crapauds_total,
            "recent": crapauds_recent
        },
        "canaux": {
            "total": channels_total,
            "recent": channels_recent,
            # Get only the 10 first channels
            "liste": channels_recentlist[0:10]
        },
        "messages": {
            "total": messages_total,
            "recent": messages_recent
        }
    }

    save(info)
    print("End chat info generation")

def save(info):
    # Récupération du répertoire racine du repo
    rootFolder = os.path.join(os.path.dirname(__file__), '..')
    # Répertoire pour stocker le fichier de sortie
    dataFolder = os.path.join(rootFolder, 'site', 'data')

    statsFilePath = os.path.abspath(
        os.path.join(dataFolder, 'chat.json'))
    with open(statsFilePath, "w") as file_write:
        json.dump(info, file_write)

def getAllChannels(rocket):
  index = 0
  allChannels = []
  while True:
    channels = rocket.channels_list(offset = index).json()
      
    allChannels.extend([ channel for channel in channels['channels'] if 'archived' not in channel])
    if channels['count'] + channels['offset'] >= channels['total']:
      break
    index += channels['count']
  return allChannels

def getAllUsers(rocket):
  index = 0
  allUsers = []
  while True:
    users = rocket.users_list(offset = index).json()
      
    allUsers.extend(users["users"])
    if users['count'] + users['offset'] >= users['total']:
        break
    index += users['count']
  return allUsers

def getAllMessages(rocket, roomid, begindate):
    index = 0
    allMessages = []
    while True:
        messages = rocket.channels_history(roomid, offset = index, oldest= begindate).json()
      
        allMessages.extend(messages["m"])
        if messages['count'] + messages['offset'] >= messages['total']:
            break
        index += messages['count']
    return allMessages

if __name__ == "__main__":
    main()