// Copyright 2021 The Chromium OS Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

// package servo_lib provides servo-related variables, such as dut-controls
// to be run before and after flashing, and a programmer argument.
package servo_lib

import "fmt"

type ServoConfig struct {
	dut_on     [][]string
	dut_off    [][]string
	programmer string
}

type UnsupportedServoError struct {
	servoType ServoType
	board     string
}

func (m *UnsupportedServoError) Error() string {
	return fmt.Sprintf("board %v does not support servo %v", m.board, m.servoType.string)
}

func GetServoConfig(board, servoSerial string, servoType ServoType) (*ServoConfig, error) {
	dut_on := [][]string{{"cpu_fw_spi:on"}}
	dut_off := [][]string{{"cpu_fw_spi:off"}}
	programmer := ""
	if servoType.IsV2() {
		programmer = fmt.Sprintf("ft2232_spi:type=google-servo-v2,serial=%v", servoSerial)
	} else if servoType.IsMicro() || servoType.IsC2D2() {
		programmer = fmt.Sprintf("raiden_debug_spi:serial=%v", servoSerial)
	} else if servoType.IsCCD() {
		dut_on = nil
		dut_off = nil
		programmer = fmt.Sprintf("raiden_debug_spi:target=AP,serial=%v", servoSerial)
	} else {
		return nil, &UnsupportedServoError{servoType, board}
	}
	return &ServoConfig{dut_on, dut_off, programmer}, nil
}