mirror of
https://github.com/AvengeMedia/DankMaterialShell.git
synced 2025-12-06 05:25:41 -05:00
106 lines
2.1 KiB
Go
106 lines
2.1 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/AvengeMedia/DankMaterialShell/core/internal/log"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var dpmsCmd = &cobra.Command{
|
|
Use: "dpms",
|
|
Short: "Control display power management",
|
|
}
|
|
|
|
var dpmsOnCmd = &cobra.Command{
|
|
Use: "on [output]",
|
|
Short: "Turn display(s) on",
|
|
Args: cobra.MaximumNArgs(1),
|
|
ValidArgsFunction: func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
|
|
if len(args) != 0 {
|
|
return nil, cobra.ShellCompDirectiveNoFileComp
|
|
}
|
|
return getDPMSOutputs(), cobra.ShellCompDirectiveNoFileComp
|
|
},
|
|
Run: runDPMSOn,
|
|
}
|
|
|
|
var dpmsOffCmd = &cobra.Command{
|
|
Use: "off [output]",
|
|
Short: "Turn display(s) off",
|
|
Args: cobra.MaximumNArgs(1),
|
|
ValidArgsFunction: func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
|
|
if len(args) != 0 {
|
|
return nil, cobra.ShellCompDirectiveNoFileComp
|
|
}
|
|
return getDPMSOutputs(), cobra.ShellCompDirectiveNoFileComp
|
|
},
|
|
Run: runDPMSOff,
|
|
}
|
|
|
|
var dpmsListCmd = &cobra.Command{
|
|
Use: "list",
|
|
Short: "List outputs",
|
|
Args: cobra.NoArgs,
|
|
Run: runDPMSList,
|
|
}
|
|
|
|
func init() {
|
|
dpmsCmd.AddCommand(dpmsOnCmd, dpmsOffCmd, dpmsListCmd)
|
|
}
|
|
|
|
func runDPMSOn(cmd *cobra.Command, args []string) {
|
|
outputName := ""
|
|
if len(args) > 0 {
|
|
outputName = args[0]
|
|
}
|
|
|
|
client, err := newDPMSClient()
|
|
if err != nil {
|
|
log.Fatalf("%v", err)
|
|
}
|
|
defer client.Close()
|
|
|
|
if err := client.SetDPMS(outputName, true); err != nil {
|
|
log.Fatalf("%v", err)
|
|
}
|
|
}
|
|
|
|
func runDPMSOff(cmd *cobra.Command, args []string) {
|
|
outputName := ""
|
|
if len(args) > 0 {
|
|
outputName = args[0]
|
|
}
|
|
|
|
client, err := newDPMSClient()
|
|
if err != nil {
|
|
log.Fatalf("%v", err)
|
|
}
|
|
defer client.Close()
|
|
|
|
if err := client.SetDPMS(outputName, false); err != nil {
|
|
log.Fatalf("%v", err)
|
|
}
|
|
}
|
|
|
|
func getDPMSOutputs() []string {
|
|
client, err := newDPMSClient()
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
defer client.Close()
|
|
return client.ListOutputs()
|
|
}
|
|
|
|
func runDPMSList(cmd *cobra.Command, args []string) {
|
|
client, err := newDPMSClient()
|
|
if err != nil {
|
|
log.Fatalf("%v", err)
|
|
}
|
|
defer client.Close()
|
|
|
|
for _, output := range client.ListOutputs() {
|
|
fmt.Println(output)
|
|
}
|
|
}
|