// Copyright 2018 The Periph Authors. All rights reserved.
// Use of this source code is governed under the Apache License, Version 2.0
// that can be found in the LICENSE file.

package rainbowhat

import (
	"periph.io/x/conn/v3/gpio"
	"periph.io/x/conn/v3/i2c/i2creg"
	"periph.io/x/conn/v3/spi/spireg"
	"github.com/GermanBionicSystems/devices/v3/apa102"
	"github.com/GermanBionicSystems/devices/v3/bmxx80"
	"github.com/GermanBionicSystems/devices/v3/ht16k33"
	"periph.io/x/host/v3/rpi"
)

// Dev represents a Rainbow HAT  (https://shop.pimoroni.com/products/rainbow-hat-for-android-things)
type Dev struct {
	ledstrip *apa102.Dev
	bmp280   *bmxx80.Dev
	display  *ht16k33.Display
	buttonA  gpio.PinIn
	buttonB  gpio.PinIn
	buttonC  gpio.PinIn
	ledR     gpio.PinOut
	ledG     gpio.PinOut
	ledB     gpio.PinOut
	buzzer   gpio.PinOut
	servo    gpio.PinOut
}

// NewRainbowHat returns a rainbowhat driver.
func NewRainbowHat(ao *apa102.Opts) (*Dev, error) {
	i2cPort, err := i2creg.Open("/dev/i2c-1")
	if err != nil {
		return nil, err
	}

	spiPort, err := spireg.Open("/dev/spidev0.0")
	if err != nil {
		return nil, err
	}

	bmp280, err := bmxx80.NewI2C(i2cPort, 0x77, &bmxx80.DefaultOpts)
	if err != nil {
		return nil, err
	}

	display, err := ht16k33.NewAlphaNumericDisplay(i2cPort, ht16k33.I2CAddr)
	if err != nil {
		return nil, err
	}

	opts := *ao
	opts.NumPixels = 7
	ledstrip, err := apa102.New(spiPort, &opts)
	if err != nil {
		return nil, err
	}

	dev := &Dev{
		ledstrip: ledstrip,
		bmp280:   bmp280,
		display:  display,
		buttonA:  rpi.P1_40, // GPIO21
		buttonB:  rpi.P1_38, // GPIO20
		buttonC:  rpi.P1_36, // GPIO16
		ledR:     rpi.P1_31, // GPIO06
		ledG:     rpi.P1_35, // GPIO19
		ledB:     rpi.P1_37, // GPIO26
		buzzer:   rpi.P1_33, // PWM1
		servo:    rpi.P1_32, // PWM0
	}

	if err := dev.buttonA.In(gpio.PullUp, gpio.BothEdges); err != nil {
		return nil, err
	}

	if err := dev.buttonB.In(gpio.PullUp, gpio.BothEdges); err != nil {
		return nil, err
	}

	if err := dev.buttonC.In(gpio.PullUp, gpio.BothEdges); err != nil {
		return nil, err
	}

	return dev, nil
}

// GetLedStrip returns apa102.Dev seven addressable led strip.
func (d *Dev) GetLedStrip() *apa102.Dev {
	return d.ledstrip
}

// GetBmp280 returns bmxx80.Dev handler.
func (d *Dev) GetBmp280() *bmxx80.Dev {
	return d.bmp280
}

// GetDisplay returns ht16k33.Display with four alphanumeric digits.
func (d *Dev) GetDisplay() *ht16k33.Display {
	return d.display
}

// GetButtonA returns gpio.PinIn corresponding to the A capacitive button.
func (d *Dev) GetButtonA() gpio.PinIn {
	return d.buttonA
}

// GetButtonB returns gpio.PinIn corresponding to the B capacitive button.
func (d *Dev) GetButtonB() gpio.PinIn {
	return d.buttonB
}

// GetButtonC returns gpio.PinIn corresponding to the C capacitive button.
func (d *Dev) GetButtonC() gpio.PinIn {
	return d.buttonC
}

// GetLedR returns gpio.PinOut corresponding to the red LED.
func (d *Dev) GetLedR() gpio.PinOut {
	return d.ledR
}

// GetLedG returns gpio.PinOut corresponding to the green LED.
func (d *Dev) GetLedG() gpio.PinOut {
	return d.ledG
}

// GetLedB returns gpio.PinOut corresponding to the blue LED.
func (d *Dev) GetLedB() gpio.PinOut {
	return d.ledB
}

// GetBuzzer returns gpio.PinOut corresponding to the buzzer pin.
func (d *Dev) GetBuzzer() gpio.PinOut {
	return d.buzzer
}

// GetServo returns gpio.PinOut corresponding to the servo pin.
func (d *Dev) GetServo() gpio.PinOut {
	return d.servo
}

// Halt all internal devices.
func (d *Dev) Halt() error {
	if err := d.bmp280.Halt(); err != nil {
		return err
	}

	if err := d.ledstrip.Halt(); err != nil {
		return err
	}

	if err := d.display.Halt(); err != nil {
		return err
	}

	if err := d.ledR.Halt(); err != nil {
		return err
	}

	if err := d.ledG.Halt(); err != nil {
		return err
	}

	if err := d.ledB.Halt(); err != nil {
		return err
	}

	if err := d.buttonA.Halt(); err != nil {
		return err
	}

	if err := d.buttonB.Halt(); err != nil {
		return err
	}

	return d.buttonC.Halt()
}