__init__.py

# Copyright 2016 Mycroft AI, Inc.
#
# This file is part of Mycroft Core.
#
# Mycroft Core is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Mycroft Core is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Mycroft Core.  If not, see <http://www.gnu.org/licenses/>.

from adapt.intent import IntentBuilder
from mycroft.skills.core import MycroftSkill

import datetime
import re
from os import listdir
from os.path import dirname
import random
from mycroft.skills.audioservice import AudioService
from mycroft.util import play_wav

__author__ = 'jarbas'


class EasterEggsSkill(MycroftSkill):
    def __init__(self):
        super(EasterEggsSkill, self).__init__()

    def initialize(self):
        stardate_intent = IntentBuilder("StardateIntent").\
            require("StardateKeyword").build()
        self.register_intent(stardate_intent,
                             self.handle_stardate_intent)

        intent = IntentBuilder("PodBayDoorsIntent"). \
            require("PodBayDoorsKeyword").build()
        self.register_intent(intent,
                             self.handle_pod_intent)

        intent = IntentBuilder("LanguagesYouSpeakIntent"). \
            require("LanguagesYouSpeakKeyword").build()
        self.register_intent(intent,
                             self.handle_number_of_languages_intent)

        intent = IntentBuilder("RoboticsLawsIntent"). \
            require("RoboticsKeyword").require("LawKeyword")\
            .optionally("LawOfRobotics").build()
        self.register_intent(intent,
                             self.handle_robotic_laws_intent)

        intent = IntentBuilder("rock_paper_scissors_lizard_spock_Intent"). \
            require("rock_paper_scissors_lizard_spock_Keyword").build()
        self.register_intent(intent,
                             self.handle_rock_paper_scissors_lizard_spock_intent)

        intent = IntentBuilder("PortalIntent"). \
            require("PortalKeyword").build()
        self.register_intent(intent,
                             self.handle_portal_intent)

        intent = IntentBuilder("DukeNukemIntent"). \
            require("DukeNukemKeyword").build()
        self.register_intent(intent,
                             self.handle_dukenukem_intent)

        intent = IntentBuilder("HALIntent"). \
            require("HALKeyword").build()
        self.register_intent(intent,
                             self.handle_hal_intent)

        intent = IntentBuilder("BenderIntent"). \
            require("BenderKeyword").build()
        self.register_intent(intent,
                             self.handle_bender_intent)

        intent = IntentBuilder("ArnoldIntent"). \
            require("ArnoldKeyword").build()
        self.register_intent(intent,
                             self.handle_arnold_intent)

        intent = IntentBuilder("GladosIntent"). \
            require("GladosKeyword").build()
        self.register_intent(intent,
                             self.handle_glados_intent)

        intent = IntentBuilder("HolidayIntent"). \
            require("HolidayKeyword").build()
        self.register_intent(intent,
                             self.handle_holiday_intent)

        self.audio_service = AudioService(self.bus)

    def handle_stardate_intent(self, message):
        sd = Stardate().toStardate()
        self.speak_dialog("stardate", {"stardate": sd})

    def handle_pod_intent(self, message):
        self.speak_dialog("pod")

    def handle_robotic_laws_intent(self, message):
        law = str(message.data.get("LawOfRobotics", "all"))
        if law == "1":
            self.speak_dialog("rule1")
        elif law == "2":
            self.speak_dialog("rule2")
        elif law == "3":
            self.speak_dialog("rule3")
        else:
            self.speak_dialog("rule1")
            self.speak_dialog("rule2")
            self.speak_dialog("rule3")

    def handle_rock_paper_scissors_lizard_spock_intent(self, message):
        self.speak_dialog("rock_paper_scissors_lizard_spock")

    def handle_number_of_languages_intent(self, message):
        self.speak_dialog("languages")

    def handle_portal_intent(self, message):
        path = dirname(__file__)+"/sounds/portal"
        files = [mp3 for mp3 in listdir(path) if
                 ".mp3" in mp3]
        if len(files):
            mp3 =  path + "/" + random.choice(files)
            self.audio_service.play(mp3)
        else:
            self.speak_dialog("bad_file")

    def handle_hal_intent(self, message):
        path = dirname(__file__) + "/sounds/hal"
        files = [mp3 for mp3 in listdir(path) if ".mp3" in mp3]
        if len(files):
            mp3 =  path + "/" + random.choice(files)
            self.audio_service.play(mp3)
        else:
            self.speak_dialog("bad_file")

    def handle_dukenukem_intent(self, message):
        path = dirname(__file__) + "/sounds/dukenukem"
        files = [wav for wav in listdir(path) if ".wav" in wav]
        if len(files):
            wav = path + "/" + random.choice(files)
            play_wav(wav)
        else:
            self.speak_dialog("bad_file")

    def handle_arnold_intent(self, message):
        path = dirname(__file__) + "/sounds/arnold"
        files = [wav for wav in listdir(path) if ".wav" in wav]
        if len(files):
            wav = path + "/" + random.choice(files)
            play_wav(wav)
        else:
            self.speak_dialog("bad_file")

    def handle_bender_intent(self, message):
        path = dirname(__file__) + "/sounds/bender"
        files = [mp3 for mp3 in listdir(path) if ".mp3" in mp3]
        if len(files):
            mp3 = path + "/" + random.choice(files)
            self.audio_service.play(mp3)
        else:
            self.speak_dialog("bad_file")

    def handle_glados_intent(self, message):
        path = dirname(__file__) + "/sounds/glados"
        files = [mp3 for mp3 in listdir(path) if ".mp3" in mp3]
        if len(files):
            mp3 = path + "/" + random.choice(files)
            self.audio_service.play(mp3)
        else:
            self.speak_dialog("bad_file")

    def handle_holiday_intent(self, message):
        path = dirname(__file__) + "/sounds/holiday"
        files = [mp3 for mp3 in listdir(path) if ".mp3" in mp3]
        if len(files):
            mp3 = path + "/" + random.choice(files)
            self.audio_service.play(mp3)
        else:
            self.speak_dialog("bad_file")

    def stop(self):
        pass


def create_skill():
    return EasterEggsSkill()


# https://www.trekterest.com/stardatefaq.php
class Stardate():
    # Definitions to help with leap years.
    nrmdays = [31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
    lyrdays = [31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
    dayseconds = 86400
    sdreg = re.compile(r"^\[(-{0,1}\d+)\](\d+)\.{1}(\d+)$")
    datetimereg = re.compile(
        r"^(\d{4})-(\d{1,2})-(\d{1,2}) (\d{1,2}):(\d{1,2}):(\d{1,2})$")

    # The epoch for stardates, 2162-01-04, is 789294 (0xc0b2e) days after
    # the internal epoch.  This is 789294*86400 (0xc0b2e*0x15180) ==
    # 68195001600 (0xfe0bd2500) seconds.
    ufpepoch = 0xfe0bd2500

    # The epoch for TNG-style stardates, 2323-01-01, is 848094 (0xcf0de)
    # days after the internal epoch.  This is 73275321600 (0x110f8cad00)
    # seconds.
    tngepoch = 0x110f8cad00

    def gleapyear(self, y):
        return (not y % 4) and (y % 100 or not y % 400)

    def gdays(self, y):
        return self.lyrdays if self.gleapyear(y) else self.nrmdays

    def __init__(self):
        pass

    def toStardate(self, date=None):
        S, F = 0, 0
        if not date:
            date = self.getcurdate()
        date = list(re.findall(self.datetimereg, date)[0])
        date = [int(i) for i in date]
        S = self.gregin(date)

        isneg = True
        nissue, integer, frac = 0, 0, 0

        if S >= self.tngepoch:
            return self.toTngStardate(S, F)

        if S < self.ufpepoch:
            # negative stardate
            diff = self.ufpepoch - S
            nsecs = 2000 * self.dayseconds - 1 - (
            diff % (2000 * self.dayseconds))
            isneg = True
            nissue = 1 + ((diff / (2000 * self.dayseconds)) & 0xffffffff)
            integer = nsecs / (self.dayseconds / 5)
            frac = (((nsecs % (self.dayseconds / 5)) << 32) | F) * 50
        elif S < self.tngepoch:
            # positive stardate
            diff = S - self.ufpepoch
            nsecs = diff % (2000 * self.dayseconds)
            isneg = False
            nissue = (diff / (2000 * self.dayseconds)) & 0xffffffff

            if nissue < 19 or (
                            nissue == 19 and nsecs < (
                        7340 * (self.dayseconds / 5))):
                # TOS era
                integer = nsecs / (self.dayseconds / 5)
                frac = (((nsecs % (self.dayseconds / 5)) << 32) | F) * 50
            else:
                # film era
                nsecs += (nissue - 19) * 2000 * self.dayseconds
                nissue = 19
                nsecs -= 7340 * (self.dayseconds / 5)
                if nsecs >= 5000 * self.dayseconds:
                    # late film era
                    nsecs -= 5000 * self.dayseconds
                    integer = 7840 + nsecs / (self.dayseconds * 2)
                    if integer >= 10000:
                        integer -= 10000
                        nissue += 1
                    frac = (((nsecs % (self.dayseconds * 2)) << 32) | F) * 5
                else:
                    # early film era
                    integer = 7340 + nsecs / (self.dayseconds * 10)
                    frac = (((nsecs % (self.dayseconds * 10)) << 32) | F)

        ret = "[" + ("-" if isneg else "") + str(nissue) + "]" + str(
            integer).zfill(4)
        frac = ((((frac * 125) / 108) >> 32) & 0xffffffff)  # round
        ret += "." + str(frac)
        return ret

    def toTngStardate(self, S=0, F=0):
        diff = S - self.tngepoch
        # 1 issue is 86400*146097/4 seconds long, which just fits in 32 bits.
        nissue = 21 + diff / ((self.dayseconds / 4) * 146097)
        nsecs = diff % ((self.dayseconds / 4) * 146097)
        # 1 unit is (86400*146097/4)/100000 seconds, which isn't even.
        # It cancels to 27*146097/125.  For a six-figure fraction,
        # divide that by 1000000.
        h = nsecs * 125000000
        l = F * 125000000
        h = h + ((l >> 32) & 0xffffffff)
        h = h / (27 * 146097)
        ret = "[%d]%05d" % (nissue, ((h / 1000000) & 0xffffffff))
        ret += ".%06d" % (h % 1000000)
        return ret;

    def getcurdate(self):
        return datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")

    def gregin(self, date=None):
        y, m, d, H, M, S = date

        cycle = y % 400

        low = (y == 0)
        if low:
            y = 399
        else:
            y = y - 1

        t = y * 365
        t = t - y / 100
        t = t + y / 400
        t = t + y / 4

        n = 2 + d - 1
        m -= 1
        while m > 0:
            n += self.gdays(cycle)[m]
            m -= 1

        t = t + n
        if low:
            t = t - 146097
        t = t * self.dayseconds

        retS = t + H * 3600 + M * 60 + S

        return retS

    def fromStardate(self, stardate):
        nineteen = 19
        twenty = 20
        S, F = 0, 0

        sd = re.findall(self.sdreg, stardate)
        if not len(sd):
            print ("Invalid stardate format")
            return
        sd = sd[0]
        nissue = int(sd[0])
        isneg = nissue < 0
        nissue = abs(nissue)
        integer = int(sd[1])
        frac = int(sd[2] + '0' * (6 - len(sd[2])))

        if (integer > 99999) or \
                (not isneg and nissue == twenty and integer > 5005) or \
                ((isneg or nissue < twenty) and integer > 9999):
            print ("Integer part is out of range")
            return

        if isneg or nissue <= twenty:
            # Pre-TNG stardate
            if not isneg:
                # There are two changes in stardate rate to handle:
                #      up to [19]7340      0.2 days/unit
                # [19]7340 to [19]7840     10   days/unit
                # [19]7840 to [20]5006      2   days/unit
                # we scale to the first of these.

                fiddle = False
                if nissue == twenty:
                    nissue = nineteen
                    integer += 10000
                    fiddle = True
                elif nissue == nineteen and integer >= 7340:
                    fiddle = True

                if fiddle:
                    # We have a stardate in the range [19]7340 to [19]15006.  First
                    # we scale it to match the prior rate, so this range changes to
                    # 7340 to 390640.
                    integer = 7340 + ((integer - 7340) * 50) + frac / (
                        1000000 / 50)
                    frac = (frac * 50) % 1000000

                    # Next, if the stardate is greater than what was originally
                    # [19]7840 (now represented as 32340), it is in the 2 days/unit
                    # range, so scale it back again.  The range affected, 32340 to
                    # 390640, changes to 32340 to 104000.
                    if integer >= 32340:
                        frac = frac / 5 + (integer % 5) * (1000000 / 5)
                        integer = 32340 + (integer - 32340) / 5

                S = self.ufpepoch + nissue * 2000 * self.dayseconds

            else:
                # Negative stardate.  In order to avoid underflow in some cases, we
                # actually calculate a date one issue (2000 days) too late, and
                # then subtract that much as the last stage.
                S = self.ufpepoch - (nissue - 1) * 2000 * self.dayseconds

            S = S + (self.dayseconds / 5) * integer

            # frac is scaled such that it is in the range 0-999999, and a value
            # of 1000000 would represent 86400/5 seconds.  We want to put frac
            # in the top half of a uint64, multiply by 86400/5 and divide by
            # 1000000, in order to leave the uint64 containing (top half) a
            # number of seconds and (bottom half) a fraction.  In order to
            # avoid overflow, this scaling is cancelled down to a multiply by
            #  54 and a divide by 3125.
            f = (frac << 32) * 54
            f = (f + 3124) / 3125
            S = S + ((f >> 32) & 0xffffffff)
            F = f & 0xffffffff

            if isneg:
                # Subtract off the issue that was added above.
                S = S - 2000 * self.dayseconds

        else:
            # TNG stardate
            nissue = nissue - 21

            # Each issue is 86400*146097/4 seconds long.
            S = self.tngepoch + nissue * (self.dayseconds / 4) * 146097

            # 1 unit is (86400*146097/4)/100000 seconds, which isn't even.
            # It cancels to 27146097/125.
            t = integer * 1000000
            t = t + frac
            t = t * 27 * 146097
            S = S + t / 125000000

            t = (t % 125000000) << 32
            t = (t + 124999999) / 125000000
            F = t & 0xffffffff

        return self.calout(S, F)

    def calout(self, S=0, F=0):
        tod = S % self.dayseconds
        days = S / self.dayseconds

        # We need the days number to be days since an xx01.01.01 to get the
        # leap year cycle right.  For the Julian calendar, it is already
        # so (0001=01=01).  But for the Gregorian calendar, the epoch is
        # 0000-12-30, so we must add on 400 years minus 2 days.  The year
        # number gets corrected below.
        days = days + 146095

        # Approximate the year number, underestimating but only by a limited
        # amount.  days/366 is a first approximation, but it goes out by 1
        # day every non-leap year, and so will be a full year out after 366
        # non-leap years.  In the Julian calendar, we get 366 non-leap years
        # every 488 years, so adding (days/366)/487 corrects for this.  In
        # the Gregorian calendar, it is not so simple: we get 400 years
        # every 146097 days, and then add on days/366 within that set of 400
        # years.
        year = (days / 146097) * 400 + (days % 146097) / 366

        # We then adjust the number of days remaining to match this
        # approximation of the year.  Note that this approximation
        # will never be more than two years off the correct date,
        # so the number of days left no longer needs to be stored
        # in a uint64.
        days = (days + year / 100) - (year / 400)
        days = days - (year * 365 + year / 4)

        # Now correct the year to an actual year number (see notes above).
        year = year - 399

        return self.docalout(year % 400, year, days & 0xffffffff, tod)

    def docalout(self, cycle, year, ndays, tod):
        nmonth = 0
        # Walk through the months, fixing the year, and as a side effect
        # calculating the month number and day of the month.
        while ndays >= self.gdays(cycle)[nmonth]:
            ndays -= self.gdays(cycle)[nmonth]
            nmonth += 1
            if nmonth == 12:
                nmonth = 0
                year += 1
                cycle += 1

        ndays += 1
        nmonth += 1
        # Now sort out the time of day.
        hr = tod / 3600
        tod %= 3600
        minut = tod / 60
        sec = tod % 60

        return "%d-%02d-%02d %d:%d:%d" % (year, nmonth, ndays, hr, minut, sec)