__filename__="happening.py"
__author__="Bob Mottram"
__license__="AGPL3+"
__version__="1.1.0"
__maintainer__="Bob Mottram"
__email__="bob@freedombone.net"
__status__="Production"

import json
import time
import os
from datetime import datetime

from utils import loadJson
from utils import locatePost
from utils import daysInMonth
from utils import mergeDicts

def isHappeningEvent(tag: {}) -> bool:
    """Is this tag an Event or Place ActivityStreams type?
    """
    if not tag.get('type'):
        return False
    if tag['type']!='Event' and tag['type']!='Place':
        return False
    return True

def isHappeningPost(postJsonObject: {}) -> bool:
    """Is this a post with tags?
    """
    if not postJsonObject:
        return False
    if not postJsonObject.get('object'):
        return False
    if not isinstance(postJsonObject['object'], dict):
        return False
    if not postJsonObject['object'].get('tag'):
        return False
    return True

def getTodaysEvents(baseDir: str,nickname: str,domain: str, \
                    currYear=None,currMonthNumber=None, \
                    currDayOfMonth=None) -> {}:
    """Retrieves calendar events for today
    Returns a dictionary of lists containing Event and Place activities
    """
    now=datetime.now()
    if not currYear:
        year=now.year
    else:
        year=currYear
    if not currMonthNumber:
        monthNumber=now.month
    else:
        monthNumber=currMonthNumber
    if not currDayOfMonth:
        dayNumber=now.day
    else:
        dayNumber=currDayOfMonth

    calendarFilename= \
        baseDir+'/accounts/'+nickname+'@'+domain+ \
        '/calendar/'+str(year)+'/'+str(monthNumber)+'.txt'
    events={}
    if not os.path.isfile(calendarFilename):
        return events

    calendarPostIds=[]
    recreateEventsFile=False
    with open(calendarFilename,'r') as eventsFile: 
        for postId in eventsFile:
            postId=postId.replace('\n','')
            postFilename=locatePost(baseDir,nickname,domain,postId)
            if not postFilename:
                recreateEventsFile=True
                continue

            postJsonObject=loadJson(postFilename)
            if not isHappeningPost(postJsonObject):
                continue

            postEvent=[]
            dayOfMonth=None
            for tag in postJsonObject['object']['tag']:
                if not isHappeningEvent(tag):
                    continue
                # this tag is an event or a place
                if tag['type']=='Event':
                    # tag is an event
                    if not tag.get('startTime'):
                        continue
                    eventTime= \
                        datetime.strptime(tag['startTime'], \
                                          "%Y-%m-%dT%H:%M:%S%z")
                    if int(eventTime.strftime("%Y"))==year and \
                       int(eventTime.strftime("%m"))==monthNumber and \
                       int(eventTime.strftime("%d"))==dayNumber:
                        dayOfMonth=str(int(eventTime.strftime("%d")))
                        if '#statuses#' in postId:
                            # link to the id so that the event can be
                            # easily deleted
                            tag['postId']=postId.split('#statuses#')[1]
                        postEvent.append(tag)
                else:
                    # tag is a place
                    postEvent.append(tag)
            if postEvent and dayOfMonth:
                calendarPostIds.append(postId)
                if not events.get(dayOfMonth):
                    events[dayOfMonth]=[]
                events[dayOfMonth].append(postEvent)

    # if some posts have been deleted then regenerate the calendar file
    if recreateEventsFile:
        calendarFile=open(calendarFilename, "w")
        for postId in calendarPostIds:
            calendarFile.write(postId+'\n')
        calendarFile.close()

    return events

def todaysEventsCheck(baseDir: str,nickname: str,domain: str) -> bool:
    """Are there calendar events today?
    """
    now=datetime.now()
    year=now.year
    monthNumber=now.month
    dayNumber=now.day

    calendarFilename= \
        baseDir+'/accounts/'+nickname+'@'+domain+ \
        '/calendar/'+str(year)+'/'+str(monthNumber)+'.txt'
    if not os.path.isfile(calendarFilename):
        return False

    eventsExist=False
    with open(calendarFilename,'r') as eventsFile: 
        for postId in eventsFile:
            postId=postId.replace('\n','')
            postFilename=locatePost(baseDir,nickname,domain,postId)
            if not postFilename:
                continue

            postJsonObject=loadJson(postFilename)
            if not isHappeningPost(postJsonObject):
                continue

            for tag in postJsonObject['object']['tag']:
                if not isHappeningEvent(tag):
                    continue
                # this tag is an event or a place
                if tag['type']!='Event':
                    continue
                # tag is an event
                if not tag.get('startTime'):
                    continue
                eventTime= \
                    datetime.strptime(tag['startTime'], \
                                      "%Y-%m-%dT%H:%M:%S%z")
                if int(eventTime.strftime("%Y"))==year and \
                   int(eventTime.strftime("%m"))==monthNumber and \
                   int(eventTime.strftime("%d"))==dayNumber:
                    eventsExist=True
                    break

    return eventsExist

def thisWeeksEventsCheck(baseDir: str,nickname: str,domain: str) -> bool:
    """Are there calendar events this week?
    """
    now=datetime.now()
    year=now.year
    monthNumber=now.month
    dayNumber=now.day

    calendarFilename= \
        baseDir+'/accounts/'+nickname+'@'+domain+ \
        '/calendar/'+str(year)+'/'+str(monthNumber)+'.txt'
    if not os.path.isfile(calendarFilename):
        return False

    eventsExist=False
    with open(calendarFilename,'r') as eventsFile: 
        for postId in eventsFile:
            postId=postId.replace('\n','')
            postFilename=locatePost(baseDir,nickname,domain,postId)
            if not postFilename:
                continue

            postJsonObject=loadJson(postFilename)
            if not isHappeningPost(postJsonObject):
                continue

            for tag in postJsonObject['object']['tag']:
                if not isHappeningEvent(tag):
                    continue
                # this tag is an event or a place
                if tag['type']!='Event':
                    continue
                # tag is an event
                if not tag.get('startTime'):
                    continue
                eventTime= \
                    datetime.strptime(tag['startTime'], \
                                      "%Y-%m-%dT%H:%M:%S%z")
                if int(eventTime.strftime("%Y"))==year and \
                   int(eventTime.strftime("%m"))==monthNumber and \
                   (int(eventTime.strftime("%d"))>dayNumber and \
                    int(eventTime.strftime("%d"))<=dayNumber+6):
                    eventsExist=True                            
                    break

    return eventsExist

def getThisWeeksEvents(baseDir: str,nickname: str,domain: str) -> {}:
    """Retrieves calendar events for this week
    Returns a dictionary indexed by day number of lists containing
    Event and Place activities
    Note: currently not used but could be with a weekly calendar screen
    """
    now=datetime.now()
    year=now.year
    monthNumber=now.month
    dayNumber=now.day

    calendarFilename= \
        baseDir+'/accounts/'+nickname+'@'+domain+ \
        '/calendar/'+str(year)+'/'+str(monthNumber)+'.txt'

    events={}
    if not os.path.isfile(calendarFilename):
        return events

    calendarPostIds=[]
    recreateEventsFile=False
    with open(calendarFilename,'r') as eventsFile: 
        for postId in eventsFile:
            postId=postId.replace('\n','')
            postFilename=locatePost(baseDir,nickname,domain,postId)
            if not postFilename:
                recreateEventsFile=True
                continue

            postJsonObject=loadJson(postFilename)
            if not isHappeningPost(postJsonObject):
                continue

            postEvent=[]
            dayOfMonth=None
            weekDayIndex=None
            for tag in postJsonObject['object']['tag']:
                if not isHappeningEvent(tag):
                    continue
                # this tag is an event or a place
                if tag['type']=='Event':
                    # tag is an event
                    if not tag.get('startTime'):
                        continue
                    eventTime= \
                        datetime.strptime(tag['startTime'], \
                                          "%Y-%m-%dT%H:%M:%S%z")
                    if int(eventTime.strftime("%Y"))==year and \
                       int(eventTime.strftime("%m"))==monthNumber and \
                       (int(eventTime.strftime("%d"))>=dayNumber and \
                        int(eventTime.strftime("%d"))<=dayNumber+6):
                        dayOfMonth=str(int(eventTime.strftime("%d")))
                        weekDayIndex=dayOfMonth-dayNumber
                        postEvent.append(tag)
                else:
                    # tag is a place
                    postEvent.append(tag)
            if postEvent and weekDayIndex:
                calendarPostIds.append(postId)
                if not events.get(dayOfMonth):
                    events[weekDayIndex]=[]
                events[dayOfMonth].append(postEvent)

    # if some posts have been deleted then regenerate the calendar file
    if recreateEventsFile:
        calendarFile=open(calendarFilename, "w")
        for postId in calendarPostIds:
            calendarFile.write(postId+'\n')
        calendarFile.close()

    lastDayOfMonth=daysInMonth(year,monthNumber)
    if dayNumber+6 > lastDayOfMonth:
        monthNumber+=1
        if monthNumber>12:
            monthNumber=1
            year+=1
        for d in range(1,dayNumber+6-lastDayOfMonth):
            dailyEvents= \
                getTodaysEvents(baseDir,nickname,domain, \
                                year,monthNumber,d)
            if dailyEvents:
                if dailyEvents.get(d):
                    newEvents={}
                    newEvents[d+(7-(dayNumber+6-lastDayOfMonth))]= \
                        dailyEvents[d]
                    events=mergeDicts(events,newEvents)

    return events

def getCalendarEvents(baseDir: str,nickname: str,domain: str, \
                      year: int,monthNumber: int) -> {}:
    """Retrieves calendar events
    Returns a dictionary indexed by day number of lists containing
    Event and Place activities
    """
    calendarFilename= \
        baseDir+'/accounts/'+nickname+'@'+domain+ \
        '/calendar/'+str(year)+'/'+str(monthNumber)+'.txt'

    events={}
    if not os.path.isfile(calendarFilename):
        return events

    calendarPostIds=[]
    recreateEventsFile=False
    with open(calendarFilename,'r') as eventsFile: 
        for postId in eventsFile:
            postId=postId.replace('\n','')
            postFilename=locatePost(baseDir,nickname,domain,postId)
            if not postFilename:
                recreateEventsFile=True
                continue

            postJsonObject=loadJson(postFilename)
            if not isHappeningPost(postJsonObject):
                continue

            postEvent=[]
            dayOfMonth=None
            for tag in postJsonObject['object']['tag']:
                if not isHappeningEvent(tag):
                    continue
                # this tag is an event or a place
                if tag['type']=='Event':
                    # tag is an event
                    if not tag.get('startTime'):
                        continue
                    eventTime= \
                        datetime.strptime(tag['startTime'], \
                                          "%Y-%m-%dT%H:%M:%S%z")
                    if int(eventTime.strftime("%Y"))==year and \
                       int(eventTime.strftime("%m"))==monthNumber:
                        dayOfMonth=str(int(eventTime.strftime("%d")))
                        postEvent.append(tag)
                else:
                    # tag is a place
                    postEvent.append(tag)

            if postEvent and dayOfMonth:
                calendarPostIds.append(postId)
                if not events.get(dayOfMonth):
                    events[dayOfMonth]=[]
                events[dayOfMonth].append(postEvent)

    # if some posts have been deleted then regenerate the calendar file
    if recreateEventsFile:
        calendarFile=open(calendarFilename, "w")
        for postId in calendarPostIds:
            calendarFile.write(postId+'\n')
        calendarFile.close()
    
    return events

def removeCalendarEvent(baseDir: str,nickname: str,domain: str, \
                        year: int,monthNumber: int,messageId: str) -> None:
    """Removes a calendar event
    """
    calendarFilename= \
        baseDir+'/accounts/'+nickname+'@'+domain+ \
        '/calendar/'+str(year)+'/'+str(monthNumber)+'.txt'
    if not os.path.isfile(calendarFilename):
        return
    if '/' in messageId:
        messageId=messageId.replace('/','#')
    if messageId not in open(calendarFilename).read():
        return
    lines=None
    with open(calendarFilename, "r") as f:
        lines=f.readlines()
    if not lines:
        return
    with open(calendarFilename, "w+") as f:
        for line in lines:
            if messageId not in line:
                f.write(line)