package main

import (
	"context"
	"fmt"
	"log"
	"os"

	"github.com/FlowingSPDG/streamdeck"
)

type Settings struct {
	Counter    int    `json:"counter"`
	ButtonText string `json:"buttonText"`
}

func main() {
	ctx := context.Background()
	params, err := streamdeck.ParseRegistrationParams(os.Args)
	if err != nil {
		log.Fatal(err)
	}

	client := streamdeck.NewClient(ctx, params)
	setup(client)
	
	if err := client.Run(ctx); err != nil {
		log.Fatal(err)
	}
}

func setup(client *streamdeck.Client) {
	action := client.Action("com.example.myaction")

	// Traditional event handling
	action.RegisterHandler(streamdeck.WillAppear, func(ctx context.Context, client *streamdeck.Client, event streamdeck.Event) error {
		var p streamdeck.WillAppearPayload[Settings]
		if err := event.UnmarshalPayload(&p); err != nil {
			return fmt.Errorf("failed to unmarshal WillAppear payload: %w", err)
		}
		
		// Handle the payload
		return client.SetTitle(ctx, p.Settings.ButtonText, streamdeck.HardwareAndSoftware)
	})

	// Type-safe event handling with automatic unmarshaling
	streamdeck.RegisterTypedHandler(action, streamdeck.KeyDown, func(ctx context.Context, client *streamdeck.Client, p streamdeck.KeyDownPayload[Settings]) error {
		// Payload is automatically unmarshaled and typed
		p.Settings.Counter++
		return client.SetSettings(ctx, p.Settings)
	})
}

// Define your settings type
type MySettings struct {
	Counter int    `json:"counter"`
	Text    string `json:"text"`
}

// Define your message types for Property Inspector communication
type PropertyInspectorMessage struct {
	Action string `json:"action"`
}

type ResetCompleteResponse struct {
	Action string `json:"action"`
}

// Register typed handlers for different events using intuitive method names
streamdeck.OnWillAppear(action, func(ctx context.Context, client *streamdeck.Client, p streamdeck.WillAppearPayload[MySettings]) error {
	// p.Settings is automatically typed as MySettings
	return client.SetTitle(ctx, p.Settings.Text, streamdeck.HardwareAndSoftware)
})

streamdeck.OnKeyDown(action, func(ctx context.Context, client *streamdeck.Client, p streamdeck.KeyDownPayload[MySettings]) error {
	// p.Settings is automatically typed as MySettings
	p.Settings.Counter++
	return client.SetSettings(ctx, p.Settings)
})

streamdeck.OnDidReceiveSettings(action, func(ctx context.Context, client *streamdeck.Client, p streamdeck.DidReceiveSettingsPayload[MySettings]) error {
	// p.Settings is automatically typed as MySettings
	return nil
})

// Fully typed Property Inspector communication
streamdeck.OnSendToPlugin(action, func(ctx context.Context, client *streamdeck.Client, payload PropertyInspectorMessage) error {
	switch payload.Action {
	case "reset":
		return client.SendToPropertyInspector(ctx, ResetCompleteResponse{
			Action: "resetComplete",
		})
	}
	return nil
})

action.RegisterHandler(streamdeck.WillAppear, func(ctx context.Context, client *streamdeck.Client, event streamdeck.Event) error {
	var p streamdeck.WillAppearPayload[Settings]
	if err := event.UnmarshalPayload(&p); err != nil {
		return fmt.Errorf("failed to unmarshal WillAppear payload: %w", err)
	}
	
	// Handle the payload
	return nil
})