1
#!/usr/bin/env bash
2
set -euo pipefail
3

4
LOCKDIR="/tmp/viraudio"
5
mkdir -p "$LOCKDIR"
6

7
command -v pactl >/dev/null || { echo "Missing pactl, please install it first." >&2; exit 1; }
8

9
usage() {
10
  cat <<EOF
11
Usage:
12
  $0 sink [NAME]
13
  $0 mic [NAME]
14
  $0 del <NAME>
15
  $0 rm <NAME>
16
  $0 rm --all
17
  $0 rm --all --sink
18
  $0 rm --all --mic
19

20
Source: https://gist.github.com/wyf9/ff12240ae023da0a068f2466968e3681
21
Help: https://wyf9.top/p/virtual-sink#usage
22
EOF
23
  exit 1
24
}
25

26
die() {
27
  echo "$*" >&2
28
  exit 1
29
}
30

31
lock_file() {
32
  local type="$1"
33
  local name="$2"
34
  printf '%s/viraudio-%s-%s-%s.lock' "$LOCKDIR" "$USER" "$name" "$type"
35
}
36

37
exists_name() {
38
  local name="$1"
39
  [[ -f "$(lock_file sink "$name")" || -f "$(lock_file mic "$name")" ]]
40
}
41

42
generate_name() {
43
  local type="$1"
44
  local prefix
45
  case "$type" in
46
    sink) prefix="Virtual Sink" ;;
47
    mic) prefix="Virtual Mic" ;;
48
  esac
49
  local i=1
50
  while exists_name "$prefix $i"; do
51
    ((i++))
52
  done
53
  echo "$prefix $i"
54
}
55

56
create_sink() {
57
  local name="${1:-}"
58

59
  if [[ -z "$name" ]]; then
60
    name=$(generate_name sink)
61
  fi
62

63
  if exists_name "$name"; then
64
    die "Virtual device '$name' already exists."
65
  fi
66

67
  local module_index
68
  module_index=$(
69
    pactl load-module module-null-sink \
70
      media.class=Audio/Sink \
71
      sink_name="$name" \
72
      sink_properties=device.description="$name"
73
  )
74

75
  printf '%s\n' "$module_index" > "$(lock_file sink "$name")"
76
  echo "Created virtual sink '$name' (module $module_index)."
77
}
78

79
create_mic() {
80
  local name="${1:-}"
81

82
  if [[ -z "$name" ]]; then
83
    name=$(generate_name mic)
84
  fi
85

86
  if exists_name "$name"; then
87
    die "Virtual device '$name' already exists."
88
  fi
89

90
  local module_index
91
  module_index=$(
92
    pactl load-module module-null-sink \
93
      media.class=Audio/Source/Virtual \
94
      sink_name="$name" \
95
      sink_properties=device.description="$name"
96
  )
97

98
  printf '%s\n' "$module_index" > "$(lock_file mic "$name")"
99
  echo "Created virtual mic '$name' (module $module_index)."
100
}
101

102
remove_name() {
103
  local name="$1"
104
  local removed=0
105

106
  for type in sink mic; do
107
    local lf
108
    lf="$(lock_file "$type" "$name")"
109

110
    if [[ -f "$lf" ]]; then
111
      local module_index
112
      module_index="$(<"$lf")"
113
      pactl unload-module "$module_index"
114
      rm -f "$lf"
115
      echo "Removed virtual $type '$name' (module $module_index)."
116
      removed=1
117
    fi
118
  done
119

120
  if [[ $removed -eq 0 ]]; then
121
    die "No virtual sink or mic named '$name' found."
122
  fi
123
}
124

125
remove_all() {
126
  local type_filter="${1:-}"
127
  local types=()
128
  local removed=0
129

130
  if [[ -z "$type_filter" ]]; then
131
    types=(sink mic)
132
  else
133
    types=("$type_filter")
134
  fi
135

136
  for type in "${types[@]}"; do
137
    for lf in "$LOCKDIR"/viraudio-*"-$type".lock; do
138
      if [[ -f "$lf" ]]; then
139
        local module_index
140
        module_index="$(<"$lf")"
141
        local name
142
        name=$(basename "$lf" "-$type.lock" | sed "s/^viraudio-[^-]*-//")
143
        pactl unload-module "$module_index"
144
        rm -f "$lf"
145
        echo "Removed virtual $type '$name' (module $module_index)."
146
        removed=$((removed + 1))
147
      fi
148
    done
149
  done
150

151
  if [[ $removed -eq 0 ]]; then
152
    die "No virtual devices found."
153
  fi
154
  echo "Total removed: $removed."
155
}
156

157
if [[ $# -lt 1 ]]; then
158
  usage
159
fi
160

161
cmd="$1"
162
shift
163
name="$*"
164

165
case "$cmd" in
166
  sink)
167
    create_sink "$name"
168
    ;;
169
  mic)
170
    create_mic "$name"
171
    ;;
172
  del|rm)
173
    if [[ "$name" == "--all" ]]; then
174
      remove_all
175
    elif [[ "$name" == "--all --sink" ]] || [[ "$name" == "--sink --all" ]]; then
176
      remove_all sink
177
    elif [[ "$name" == "--all --mic" ]] || [[ "$name" == "--mic --all" ]]; then
178
      remove_all mic
179
    elif [[ -z "$name" ]]; then
180
      die "NAME required for del/rm, or use --all."
181
    else
182
      remove_name "$name"
183
    fi
184
    ;;
185
  *)
186
    usage
187
    ;;
188
esac