// Copyright (C) 2015-2017 Mikael Berthe <mikael@lilotux.net>. All rights reserved.

// Use of this source code is governed by the MIT license,

// which can be found in the LICENSE file.

//

// Thanks to atbrask's blog post <http://www.atbrask.dk/?p=98> for the

// protocol details.

// gobm65 is a Beurer BM65 Blood Pressure Monitor CLI reader.

package main

// Installation:

//

// % go get hg.lilotux.net/golang/mikael/gobm65

//

// Examples:

//

// Get help:

// % gobm65 --help

//

// Get records and display the average:

// % gobm65 --average

// ... display more statistics:

// % gobm65 --stats

// ... also display World Health Organization classification:

// % gobm65 --stats --class

//

// Display all records but reduce the list so that close records are replaced

// with their average:

// % gobm65 --reduce

// Display the latest 3 records with the average:

// % gobm65 -l 3 --average

// Display all records since a specific date:

// % gobm65 --since "2016-06-01"

// Display all records before a specific date:

// % gobm65 --to-date "2016-06-30"

// Display all records of the last 7 days:

// % gobm65 --since "$(date "+%F" -d "7 days ago")"

//

// Display statistics for morning records:

// % gobm65 --from-time 06:00 --to-time 12:00 --stats

// One can invert times to get night data:

// % gobm65 --from-time 21:00 --to-time 09:00

//

// Display the last/first 10 records in JSON:

// % gobm65 -l 10 --format json

//

// Save the records to a JSON file:

// % gobm65 -o data_u2.json

//

// Read a JSON file and display average of the last 3 records:

// % gobm65 -i data_u2.json -l 3 --average

// % gobm65 -i data_u2.json -l 3 --stats

// Read a JSON file, merge with device records, and save to another file:

// % gobm65 -i data_u2.json --merge -o data_u2-new.json

//

// Data from several JSON files can be merged, files are separated with a ';':

// % gobm65 -i "data_u0.json;data_u1.json;data_u2.json"

import (

	"encoding/json"

	"fmt"

	"io"

	"io/ioutil"

	"log"

	"math"

	"os"

	"sort"

	"strings"

	"time"

	flag "github.com/spf13/pflag"

	"github.com/tarm/serial"

)

type measurement struct {

	Header    int

	Systolic  int

	Diastolic int

	Pulse     int

	Month     int

	Day       int

	Hour      int

	Minute    int

	Year      int

}

type simpleTime struct {

	hour, minute int

}

// World Heath Organization blood pressure classification

const (

	BPOptimal              = iota // < 120,80:  Optimal

	BPNormal                      // < 130,85:  Normal

	BPHighNormal                  // < 140,90:  High-Normal

	BPMildHypertension            // < 160,100: Grade 1 Mild Hypertension

	BPModerateHypertension        // < 180,110: Grade 2 Moderate Hypertension

	BPSevereHypertension          // >=180,110: Grade 3 Severe Hypertension

)

// Special cases that do not fit in the previous classification

const (

	// >=140, <90: Isolated Systolic Hypertension

	IsolatedSystolicHypertension = 1

)

// WHOPressureClassification contains the World Health Organization blood

// pressure categories

var WHOPressureClassification = []string{

	"Optimal",

	"Normal",

	"High-Normal",

	"Mild Hypertension",

	"Moderate Hypertension",

	"Severe Hypertension",

}

// WHOPressureFlag is an array of special cases

var WHOPressureFlag = []string{

	"",

	"Isolated Systolic Hypertension",

}

func getData(s io.ReadWriteCloser, buf []byte, size int) (int, error) {

	t := 0

	b := buf

	for t < size {

		n, err := s.Read(b[t:])

		if err != nil {

			log.Fatal(err) // XXX

			return t, err

		}

		//log.Printf("(%d bytes) %q\n", n, b[t:t+1])

		t += n

	}

	return t, nil

}

func fetchData(dev string) (items []measurement, err error) {

	c := &serial.Config{Name: dev, Baud: 4800}

	var s *serial.Port

	s, err = serial.OpenPort(c)

	if err != nil {

		return items, err

	}

	// =================== Handshake =====================

	q := []byte("\xaa")

	//log.Printf("Query: %q\n", q)

	log.Println("Starting handshake...")

	n, err := s.Write(q)

	if err != nil {

		return items, err

	}

	buf := make([]byte, 128)

	n, err = getData(s, buf, 1)

	if err != nil {

		return items, err

	}

	if n == 1 && buf[0] == '\x55' {

		log.Println("Handshake successful.")

	} else {

		log.Printf("(%d bytes) %q\n", n, buf[:n])

		s.Close()

		return items, fmt.Errorf("handshake failed")

	}

	// =================== Desc =====================

	q = []byte("\xa4")

	//log.Printf("Query: %q\n", q)

	log.Println("Requesting device description...")

	n, err = s.Write(q)

	if err != nil {

		return items, err

	}

	n, err = getData(s, buf, 32)

	log.Printf("DESC> %q\n", buf[:n])

	// =================== Count =====================

	q = []byte("\xa2")

	//log.Printf("Query: %q\n", q)

	log.Println("Requesting data counter...")

	n, err = s.Write(q)

	if err != nil {

		return items, err

	}

	n, err = getData(s, buf, 1)

	if err != nil {

		return items, err

	}

	var nRecords int

	if n == 1 {

		log.Printf("%d item(s) available.", buf[0])

		nRecords = int(buf[0])

	} else {

		log.Printf("(%d bytes) %q\n", n, buf[:n])

		return items, fmt.Errorf("no measurement found")

	}

	// =================== Records =====================

	for i := 0; i < nRecords; i++ {

		q = []byte{'\xa3', uint8(i + 1)}

		//log.Printf("Query: %q\n", q)

		//log.Printf("Requesting measurement %d...", i+1)

		n, err = s.Write(q)

		if err != nil {

			return items, err

		}

		n, err = getData(s, buf, 9)

		//log.Printf("DESC> %q\n", buf[:n])

		var data measurement

		data.Header = int(buf[0])

		data.Systolic = int(buf[1]) + 25

		data.Diastolic = int(buf[2]) + 25

		data.Pulse = int(buf[3])

		data.Month = int(buf[4])

		data.Day = int(buf[5])

		data.Hour = int(buf[6])

		data.Minute = int(buf[7])

		data.Year = int(buf[8]) + 2000

		items = append(items, data)

	}

	s.Close()

	return mergeItems(items, []measurement{}), nil

}

func loadFromJSONFile(filename string) (items []measurement, err error) {

	data, err := ioutil.ReadFile(filename)

	if err != nil {

		return items, err

	}

	err = json.Unmarshal(data, &items)

	return items, err

}

func loadFromJSONFiles(files string) (items []measurement, err error) {

	filenames := strings.Split(files, ";")

	for _, f := range filenames {

		if f == "" {

			continue

		}

		records, err := loadFromJSONFile(f)

		if err != nil {

			return items, err

		}

		items = mergeItems(records, items)

	}

	return

}

func mergeItems(newItems, oldItems []measurement) []measurement {

	var result []measurement

	// Sort method: isLater returns true if mi's date is later or

	// equal to mj's date.

	isLater := func(mi, mj measurement) bool {

		switch {

		case mi.Year < mj.Year:

			return false

		case mi.Year > mj.Year:

			return true

		case mi.Month < mj.Month:

			return false

		case mi.Month > mj.Month:

			return true

		case mi.Day < mj.Day:

			return false

		case mi.Day > mj.Day:

			return true

		case mi.Hour < mj.Hour:

			return false

		case mi.Hour > mj.Hour:

			return true

		case mi.Minute < mj.Minute:

			return false

		default:

			return true

		}

	}

	// Note that sort.Slice was introduced in go 1.8

	sort.Slice(oldItems, func(i, j int) bool {

		return isLater(oldItems[i], oldItems[j])

	})

	sort.Slice(newItems, func(i, j int) bool {

		return isLater(newItems[i], newItems[j])

	})

	// insertIfMissing inserts a measurement into a sorted slice

	insertIfMissing := func(l []measurement, m measurement) []measurement {

		var later bool

		var i int

		for i = range l {

			later = isLater(l[i], m)

			if !later {

				break

			}

			if l[i] == m { // Duplicate

				return l

			}

		}

		if later {

			return append(l, m)

		}

		return append(l[:i], append([]measurement{m}, l[i:]...)...)

	}

	for _, item := range newItems {

		result = insertIfMissing(result, item)

	}

	for _, item := range oldItems {

		result = insertIfMissing(result, item)

	}

	return result

}

func parseDate(dateStr string) (date time.Time, err error) {

	if dateStr == "" {

		return

	}

	var yy, mm, dd, h, m, s int

	n, e := fmt.Sscanf(dateStr, "%d-%d-%d %d:%d:%d", &yy, &mm, &dd, &h, &m, &s)

	if e != nil && n < 3 {

		err = e

		return

	}

	if n < 6 {

		log.Printf("Date parsed with only %d fields\n", n)

	}

	date = time.Date(yy, time.Month(mm), dd, h, m, s, 0, time.Local)

	return

}

func parseTime(timeStr string) (t simpleTime, err error) {

	_, err = fmt.Sscanf(timeStr, "%d:%d", &t.hour, &t.minute)

	return

}

// diffTime returns the duration between two (decreasing) sorted measurements

func diffTime(a, b measurement) time.Duration {

	dateA := time.Date(a.Year, time.Month(a.Month), a.Day, a.Hour, a.Minute, 0, 0, time.Local)

	dateB := time.Date(b.Year, time.Month(b.Month), b.Day, b.Hour, b.Minute, 0, 0, time.Local)

	return dateA.Sub(dateB)

}

func average(items []measurement) (measurement, error) {

	var avgMeasure measurement

	var avgCount int

	for _, data := range items {

		avgMeasure.Systolic += data.Systolic

		avgMeasure.Diastolic += data.Diastolic

		avgMeasure.Pulse += data.Pulse

		avgCount++

	}

	roundDivision := func(a, b int) int {

		return int(0.5 + float64(a)/float64(b))

	}

	if avgCount == 0 {

		return avgMeasure, fmt.Errorf("cannot compute average: empty set")

	}

	avgMeasure.Systolic = roundDivision(avgMeasure.Systolic, avgCount)

	avgMeasure.Diastolic = roundDivision(avgMeasure.Diastolic, avgCount)

	avgMeasure.Pulse = roundDivision(avgMeasure.Pulse, avgCount)

	return avgMeasure, nil

}

func intMedian(numbers []int) int {

	middle := len(numbers) / 2

	med := numbers[middle]

	if len(numbers)%2 == 0 {

		med = (med + numbers[middle-1]) / 2

	}

	return med