package wolfx

import (
	"encoding/json"
	"fmt"
	"hayai/config"
	"hayai/constants"
	"hayai/seismo"
	"hayai/utils"
	"log"
	"net/url"
	"os"
	"time"

	"embed"

	"github.com/faiface/beep/speaker"
	"github.com/faiface/beep/wav"
	"github.com/gen2brain/beeep"
	"github.com/gorilla/websocket"
	"github.com/sqweek/dialog"
)

//go:embed assets/alert-sat.wav
var alertSoundFile embed.FS

type TypeMessage struct {
	Type string
}

type Issue struct {
	Source string
	Status string
}

type Accuracy struct {
	Epicenter string
	Depth     string
	Magnitude string
}

type MaxIntChange struct {
	String string
	Reason string
}

type WarnArea struct {
	Chiiki  string
	Shindo1 string
	Shindo2 string
	Time    string
	Type    string
	Arrive  string
}

type JMAEEW struct {
	Type          string
	Title         string
	CodeType      string
	Issue         Issue
	EventID       string
	Serial        int
	AnnouncedTime string
	OriginTime    string
	Hypocenter    string
	Latitude      float64
	Longitude     float64
	Magunitude    float64
	Depth         int
	MaxIntensity  string
	Accuracy      Accuracy
	MaxIntChange  MaxIntChange
	WarnArea      []WarnArea
	IsSea         bool
	IsTraining    bool
	IsAssumption  bool
	IsWarn        bool
	IsFinal       bool
	IsCancel      bool
	OriginalText  string
	Pond          string
}

func Listen() {
	// Init sound
	alertSound, err := alertSoundFile.Open("assets/alert-sat.wav")
	utils.CheckError(err)
	defer alertSound.Close()
	streamer, format, err := wav.Decode(alertSound)
	utils.CheckError(err)
	defer streamer.Close()
	// Listen
	u := url.URL{Scheme: "wss", Host: constants.WSHost, Path: "/all_eew"}
	log.Printf("connecting to %s", u.String())

	c, _, err := websocket.DefaultDialer.Dial(u.String(), nil)
	if err != nil {
		log.Fatal("dial:", err)
	}
	defer c.Close()

	done := make(chan struct{})

	defer close(done)

	log.Printf("connected.")
	for {
		_, message, err := c.ReadMessage()
		utils.CheckError(err)
		var typeMessage TypeMessage
		if config.Config.TestWarning {
			message = constants.TestMessage
		}
		json.Unmarshal(message, &typeMessage)
		if typeMessage.Type == "jma_eew" {
			log.Printf("recv: %s", message)
			var jmaeew JMAEEW
			json.Unmarshal(message, &jmaeew)
			if !jmaeew.IsWarn && config.Config.OnlyWarnings {
				continue
			}
			equivalentMagnitude := seismo.CalculateEquivalentMagnitude(jmaeew.Magunitude, jmaeew.Latitude, jmaeew.Longitude, config.Config.Latitude, config.Config.Longitude)
			if config.Config.IssueWarningAtAnyMagnitude || config.Config.IssueWarningAtEquivalentMagnitude < equivalentMagnitude {
				alert_body := fmt.Sprintf("Epicenter: %s\nMagnitude %0.1f\nApproximately magnitude %0.1f at your location\nStrong shaking is expected soon.\nStay calm and seek shelter nearby.\n\nOrigin time: %s JST\nAnnouncement time: %s JST\nDepth: %dkm\nCoordinates: %0.1f, %0.1f\n\nSource: %s\nStatus: %s\n\n%s", jmaeew.Hypocenter, jmaeew.Magunitude, equivalentMagnitude, jmaeew.OriginTime, jmaeew.AnnouncedTime, jmaeew.Depth, jmaeew.Latitude, jmaeew.Longitude, jmaeew.Issue.Source, jmaeew.Issue.Status, jmaeew.OriginalText)
				if config.Config.IssuePopup {
					go dialog.Message("%s", alert_body).Title("Early Earthquake Warning!").Info()
				}

				if config.Config.IssueNotification {
					homedir, err := os.UserHomeDir()
					utils.CheckError(err)
					err = beeep.Notify("Early Earthquake Warning!", alert_body, homedir+"/.icons/actions/scalable/dialog-warning.svg")
					utils.CheckError(err)
				}

				if config.Config.OpenWebPages {
					utils.OpenURL(constants.OpenURLA)
					utils.OpenURL(constants.OpenURLB)
				}

				if config.Config.IssueWarningSound {
					speaker.Init(format.SampleRate, format.SampleRate.N(time.Second/10))
					streamer.Seek(0)
					speaker.Play(streamer)
					for i := 0; i < 10; i++ {
						time.Sleep(2 * time.Second)
						streamer.Seek(0)
					}
					speaker.Close()
				}
			}
		}
	}

}