volume_linux.go

//go:build !windows && !darwin
// +build !windows,!darwin

package volume

import (
	"errors"
	"log"
	"os/exec"
	"regexp"
	"strconv"
	"strings"
)

var (
	useAmixer bool
)

const (
//	outputdevice string = "Master"
//	outputdevice string = "Speaker" // Modified from Master to Speaker for talkkonnect for raspberry pi
)

func init() {
	if _, err := exec.LookPath("pactl"); err != nil {
		useAmixer = true
	}
}

func cmdEnv() []string {
	return []string{"LANG=C", "LC_ALL=C"}
}

func getVolumeCmd(outputdevice string) []string {
	if useAmixer {
		return []string{"amixer", "-M", "get", outputdevice}
	}
	return []string{"pactl", "list", "sinks"}
}

var volumePattern = regexp.MustCompile(`\d+%`)

func parseRXVolume(out string) (int, error) {
	lines := strings.Split(out, "\t")
	for _, line := range lines {
		s := strings.TrimLeft(line, " \t")
		if useAmixer && (strings.Contains(s, "Playback") || strings.Contains(s, "Front Left:")) && strings.Contains(s, "%") || !useAmixer && strings.HasPrefix(s, "Volume:") {
			volumeStr := volumePattern.FindString(s)
			return strconv.Atoi(volumeStr[:len(volumeStr)-1])
		}
	}
	return 0, errors.New("no volume found")
}

func parseTXVolume(in string) (int, error) {
	lines := strings.Split(in, "\t")
	for _, line := range lines {
		s := strings.TrimLeft(line, " \t")
		if useAmixer && (strings.Contains(s, "Capture")) {
			volumeStr := volumePattern.FindString(s)
			return strconv.Atoi(volumeStr[:len(volumeStr)-1])
		}
	}
	return 0, errors.New("no volume found")
}

func setVolumeCmd(volume int, outputdevice string) []string {
	if useAmixer {
		return []string{"amixer", "set", outputdevice, strconv.Itoa(volume) + "%"}
	} else if _, err := strconv.Atoi(outputdevice); err == nil {
		return []string{"pactl", "set-sink-volume", outputdevice, strconv.Itoa(volume) + "%"}
	}
	return []string{"pactl", "set-sink-volume", "0", strconv.Itoa(volume) + "%"}
}

func increaseVolumeCmd(diff int, outputdevice string) []string {
	var OrigVolume int
	var err error
	OrigVolume, err = GetVolume(outputdevice)
	if err != nil {
		log.Println("error: Cannot Get Volume of Current Output Device")
		return nil
	}

	if diff > 0 {
		log.Printf("debug: Changing Volume From %v increase by %v dB step on %v\n", OrigVolume, diff, outputdevice)
		return []string{"amixer", "sset", "-q", outputdevice, strconv.Itoa(abs(diff)) + "db+"}
	}

	if diff < 0 {
		log.Printf("debug: Changing Volume From %v decrease by %v dB step on %v\n", OrigVolume, diff, outputdevice)
		return []string{"amixer", "sset", "-q", outputdevice, strconv.Itoa(abs(diff)) + "db-"}
	}

	return nil
}

func getMutedCmd(outputdevice string) []string {
	if useAmixer {
		return []string{"amixer", "sget", outputdevice}
	}
	return []string{"pactl", "list", "sinks"}
}

func parseRXMuted(out string) (bool, error) {
	lines := strings.Split(out, "\t")
	for _, line := range lines {
		s := strings.TrimLeft(line, "\t")
		if (useAmixer && strings.Contains(s, "Playback")) && (strings.Contains(s, "on") || strings.Contains(s, "off")) {
			if strings.Contains(s, "[off]") || strings.Contains(s, "off") || strings.Contains(s, "yes") {
				return true, nil
			}
			if strings.Contains(s, "[on]") || strings.Contains(s, "on") || strings.Contains(s, "no") {
				return false, nil
			}
		}
		if !useAmixer && strings.HasPrefix(s, "Mute: ") {
			if strings.Contains(s, "[off]") || strings.Contains(s, "off") || strings.Contains(s, "yes") {
				return true, nil
			}
			if strings.Contains(s, "[on]") || strings.Contains(s, "on") || strings.Contains(s, "no") {
				return false, nil
			}
		}
	}
	return false, errors.New("no muted information found")
}

func muteCmd(outputdevice string) []string {
	if useAmixer {
		return []string{"amixer", "set", outputdevice, "mute"}
	} else if _, err := strconv.Atoi(outputdevice); err == nil {
		return []string{"pactl", "set-sink-mute", outputdevice, "1"}
	}
	return []string{"pactl", "set-sink-mute", "0", "1"}
}

func unmuteCmd(outputdevice string) []string {
	if useAmixer {
		return []string{"amixer", "set", outputdevice, "unmute"}
	} else if _, err := strconv.Atoi(outputdevice); err == nil {
		return []string{"pactl", "set-sink-mute", outputdevice, "0"}
	}
	return []string{"pactl", "set-sink-mute", "0", "0"}
}

func abs(x int) int {
	if x < 0 {
		return -x
	}
	return x
}