path: root/src/background.ts

import * as gapi from './gapi';
import { MsgType, Msg } from './msg';
import { Duration, TrackedPeriod, TrackedPeriodFlat } from './duration';
import moment from 'moment';
import { GraphData, getGraphData } from './graph';
import { PatternEntry, PatternEntryFlat } from './pattern';

let mainPatterns: PatternEntry[] = [];
let analyzePatterns: PatternEntry[] = [];
let calendars: {[id: string]: gapi.GCalendarMeta} = {};
let calData: {[id: string]: gapi.GCalendar} = {};
let config = {
    trackedPeriods: [
        new TrackedPeriod('Today', Duration.days(1), Duration.days(0)),
        new TrackedPeriod('Yesterday', Duration.days(2), Duration.days(1)),
        new TrackedPeriod('This Week', Duration.weeks(1), Duration.weeks(0)),
        new TrackedPeriod('This Month', Duration.months(1), Duration.months(0))] as TrackedPeriod[],
    overrideNewTab: false
};
let mainGraphData: GraphData[] = [];
let dirtyMetadata = false;
let dirtyCalData = false;
let loadPromise: Promise<void> = null;

enum ChromeError {
    storageGetError = "storageGetError",
    storageSetError = "storageSetError"
}

const chromeStorageGet = (keys: string[]): Promise<any> => (
    new Promise(resolver => chrome.storage.local.get(keys, items => {
        if (chrome.runtime.lastError) throw ChromeError.storageGetError;
        resolver(items);
    }))
);

const chromeStorageSet = (obj: {[key: string]: any}): Promise<void> => (
    new Promise(resolver => chrome.storage.local.set(obj, () => {
        if (chrome.runtime.lastError) throw ChromeError.storageSetError;
        resolver();
    }))
);

async function loadMetadata() {
    try {
        let items = await chromeStorageGet(['calendars', 'config', 'mainPatterns', 'analyzePatterns']);
        if (!items.hasOwnProperty('config'))
            console.log("no saved metadata");
        else
        {
            config = {
                trackedPeriods: items.config.trackedPeriods.map((p: TrackedPeriodFlat) => TrackedPeriod.inflate(p)),
                overrideNewTab: items.config.overrideNewTab
            };
            calendars = items.calendars;
            mainPatterns = items.mainPatterns.map((p: PatternEntryFlat) => PatternEntry.inflate(p));
            analyzePatterns = items.analyzePatterns.map((p: PatternEntryFlat) => PatternEntry.inflate(p));
            console.log('metadata loaded');
        }
    } catch (_) {
        console.error("error while loading saved metadata");
    }
}

async function saveMetadata() {
    await chromeStorageSet({
        calendars,
        config: {
            trackedPeriods: config.trackedPeriods.map(p => p.deflate()),
            overrideNewTab: config.overrideNewTab
        },
        mainPatterns: mainPatterns.map(p => p.deflate()),
        analyzePatterns: analyzePatterns.map(p => p.deflate())
    });
    console.log('metadata saved');
}

async function loadCachedCals() {
    try {
        let items = await chromeStorageGet(['calData']);
        if (!items.hasOwnProperty('calData'))
            console.log("no cached cals");
        else
        {
            let calDataFlat: {[id: string]: gapi.GCalendarFlat} = items.calData;
            console.log(calDataFlat);
            for (let id in calDataFlat) {
                calData[id] = gapi.GCalendar.inflate(calDataFlat[id]);
            }
            console.log("cached cals loaded");
        }
    } catch (e) {
        console.log(e);
        console.error("error while loading cached cals");
    }
}

async function saveCachedCals() {
    let calDataFlat: {[id: string]: gapi.GCalendarFlat} = {};
    for (let id in calData) {
        if (!(calendars.hasOwnProperty(id) &&
            calendars[id].enabled)) continue;
        calDataFlat[id] = calData[id].deflate();
    }
    try {
        await chromeStorageSet({ calData: calDataFlat });
        console.log('cached data saved');
    } catch (_) {
        console.log("failed to save cached data");
    }
}

function getCalData(id: string) {
    if (!calData.hasOwnProperty(id))
        calData[id] = new gapi.GCalendar(id, calendars[id].name);
    return calData[id];
}

async function getCalEvents(id: string, start: Date, end: Date) {
    let gcal = getCalData(id);
    try {
        let res = await gcal.getEvents(new Date(start), new Date(end));
        dirtyCalData = res.changed;
        return res.events;
    } catch(err) {
        console.log(`cannot load calendar ${id}`, err);
        calendars[id].enabled = false;
        return [];
    }
}

function updateMainGraphData() {
    console.log('refreshing graph data');
    console.log(mainGraphData);
    let pms = [];
    for (let i = 0; i < config.trackedPeriods.length; i++)
    {
        let p = config.trackedPeriods[i];
        let startD = p.start.toMoment();
        let endD = p.end.toMoment();
        if (!(startD && endD)) return;
        let start = moment().add(1, 'days').startOf('day');
        if (endD.valueOf() == 0) {
            switch (p.start.unit) {
                case 'days': start = moment().add(1, 'days').startOf('day'); break;
                case 'weeks': start = moment().add(1, 'weeks').startOf('isoWeek'); break;
                case 'months': start = moment().add(1, 'months').startOf('month'); break;
                default:
            }
        }
        let end = start.clone();
        start.subtract(startD);
        end.subtract(endD);
        pms.push(getGraphData(
            start.toDate(), end.toDate(), mainPatterns, calendars,
            getCalEvents
        ).then(r => {
            mainGraphData[i] = {
                    name: p.name,
                    start: start.toDate(),
                    end: end.toDate(),
                    data: r.patternGraphData
            };
        }));
    }
    return Promise.all(pms);
}

async function pollSync() {
    console.log('poll');
    /* sync all enabled calendars */
    let pms = [];
    for (let id in calendars) {
        if (!calendars[id].enabled) continue;
        pms.push(getCalData(id).sync().catch(err => {
            console.log(`cannot sync calendar ${id}`, err);
            calendars[id].enabled = false;
        }));
    }
    await Promise.all(pms);
    /* update the tracked graph data */
    await updateMainGraphData();
    pms = [];
    /* save the storage if state is changed */
    if (dirtyMetadata)
        pms.push(saveMetadata().then(() => dirtyMetadata = false));
    if (dirtyCalData)
        pms.push(saveCachedCals().then(() => dirtyCalData = false));
    await Promise.all(pms);
    /* setup the next loop */
    return new Promise(resolver => (
        window.setTimeout(() => { resolver(); pollSync();}, 10000)
    ));
}

function handleMsg(port: chrome.runtime.Port) {
    console.assert(port.name == 'main');
    port.onMessage.addListener(_msg => {
        let msg = Msg.inflate<any>(_msg);
        console.log(msg);
        switch (msg.opt) {
        case MsgType.updatePatterns: {
            let patterns = msg.data.