// 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

import "strings"

type ServoType struct {
	string
}

func NewServoType(servo_type string) ServoType {
	return ServoType{servo_type}
}

func (s *ServoType) IsV2() bool {
	return strings.Contains(s.string, "servo_v2")
}
func (s *ServoType) IsV4() bool {
	return strings.Contains(s.string, "servo_v4")
}
func (s *ServoType) IsC2D2() bool {
	return strings.Contains(s.string, "c2d2")
}
func (s *ServoType) IsCCD() bool {
	return strings.Contains(s.string, "ccd")
}
func (s *ServoType) IsMicro() bool {
	return strings.Contains(s.string, "servo_micro")
}

func (s *ServoType) GetSerialNumberOption() string {
	if s.IsV4() && s.IsCCD() {
		return "ccd_serialname"
	}
	if s.IsV4() && s.IsMicro() {
		return "servo_micro_serialname"
	}
	return "serialname"
}