# -*- coding: utf-8 -*-

"""
    Code based on:

https://github.com/mvillalba/python-ant/blob/develop/demos/ant.core/03-basicchannel.py

    in the python-ant repository and

https://github.com/tomwardill/developerhealth

    by Tom Wardill and

https://www.johannesbader.ch/2014/06/track-your-heartrate-on-raspberry-pi-with-ant/

    by Johannes Bader
"""
import sys
import time
from ant.core import driver, node, event, message, log
from ant.core.constants import CHANNEL_TYPE_TWOWAY_RECEIVE, TIMEOUT_NEVER

import json

class HeartRateMonitor(event.EventCallback):

    DEFAULT_SERIAL = "/dev/ttyUSB0"
    DEFAULT_NETKEY = 'B9A521FBBD72C345'.decode('hex')

    def __init__(self, call_func, serial = None, netkey = None):
        self.call_func = call_func
        self.serial = serial or HeartRateMonitor.DEFAULT_SERIAL
        self.netkey = netkey or HeartRateMonitor.DEFAULT_NETKEY
        self.antnode = None
        self.channel = None

    def start(self):
        #print("starting node")
        self._start_antnode()
        self._setup_channel()
        self.channel.registerCallback(self)
        #print("start listening for hr events")

    def stop(self):
        if self.channel:
            self.channel.close()
            self.channel.unassign()
        if self.antnode:
            self.antnode.stop()

    def __enter__(self):
        return self

    def __exit__(self, type_, value, traceback):
        self.stop()

    def _start_antnode(self):
        stick = driver.USB2Driver(self.serial)
        self.antnode = node.Node(stick)
        self.antnode.start()

    def _setup_channel(self):
        key = node.NetworkKey('N:ANT+', self.netkey)
        self.antnode.setNetworkKey(0, key)
        self.channel = self.antnode.getFreeChannel()
        self.channel.name = 'C:HRM'
        self.channel.assign('N:ANT+', CHANNEL_TYPE_TWOWAY_RECEIVE)
        self.channel.setID(120, 0, 0)
        self.channel.setSearchTimeout(TIMEOUT_NEVER)
        self.channel.setPeriod(8070)
        self.channel.setFrequency(57)
        self.channel.open()

    def process(self, msg):
        time.sleep(5)  # 5秒の停止
        if isinstance(msg, message.ChannelBroadcastDataMessage):
            #print("heart rate is {}".format(ord(msg.payload[-1])))
            heart_rate = int(ord(msg.payload[-1]))
            self.call_func(heart_rate)

def my_print(data):
    print("heart rate is {}".format(data))

    # 心拍数をJSONファイルに書き込む    
    data_json = {"heart_rate": data}
    file = open("heart_rate.json", "w")
    json.dump(data_json , file)


if __name__ == '__main__':
    hrm = HeartRateMonitor(my_print)
    print("monitor start")
    hrm.start()
    print("monitor start listening")
    print("press Ctrl-C to stop this script")
    while True:
        try:
            time.sleep(1)
        except KeyboardInterrupt:
            print("monitor stop")
            hrm.stop()
            sys.exit(0)

{
    "interactionModel": {
        "languageModel": {
            "invocationName": "ジャービス",
            "intents": [
                {
                    "name": "AMAZON.CancelIntent",
                    "samples": []
                },
                {
                    "name": "AMAZON.HelpIntent",
                    "samples": []
                },
                {
                    "name": "AMAZON.StopIntent",
                    "samples": []
                },
                {
                    "name": "AMAZON.PauseIntent",
                    "samples": []
                },
                {
                    "name": "AMAZON.ResumeIntent",
                    "samples": []
                },
                {
                    "name": "musicStart",
                    "slots": [],
                    "samples": [
                        "今の気分におすすめの音楽をかけて"
                    ]
                }
            ],
            "types": []
        }
    }
}

# -*- coding: utf-8 -*-

from flask import Flask, render_template
from flask_ask import Ask, question, statement, audio
import json

app = Flask(__name__)
ask = Ask(app, '/')


# 起動時の対応
@ask.launch
def launched():
    return question('ジャービスです。いかがなさいましたか？')

# 音楽を再生
@ask.intent('musicStart')
def music_start():
    # 心拍のload
    file2 = open("heart_rate.json", "r")
    json_dict = json.load(file2)
    heart_rate = json_dict["heart_rate"]

    # 心拍の異常値処理
    if heart_rate > 200:
        heart_rate = 0

    # 条件分岐
    mean_heart_rate = 70
    if heart_rate > mean_heart_rate:
        out_index = 0
    else:
        out_index = 1

   # 出力の文章を作成
    speech_0 = '心拍数は{}です。'.format(heart_rate)

    speech_1 = [
        '普段の平均より高いです。落ち着いた曲を再生します', 
        '普段の平均より低いです。テンションが上がる曲を再生します'
    ]

    speech = speech_0 + speech_1[out_index]

    # 音楽URL
    stream_url = [
        "※音楽1のURLをここに書きます※", 
        "※音楽2のURLをここに書きます※"
    ]

    return audio(speech).play(stream_url[out_index])


@ask.intent('AMAZON.PauseIntent')
def stop():
    return audio("Jarvisを終了します").stop()


@ask.intent('AMAZON.StopIntent')
def stop():
    return audio("Jarvisを終了します").stop()


@ask.session_ended
def session_ended():
    return "{}", 200

if __name__ == '__main__':
    app.run(host="127.0.0.1", port=5000)