initial commit

1 files changed, 520 insertions, 0 deletions

diff --git a/ponymix.cc b/ponymix.cc
new file mode 100644
index 0000000..1bd256f
--- /dev/null
+++ b/ponymix.cc
@@ -0,0 +1,520 @@
+#include "pulse.h"
+
+#include <err.h>
+#include <getopt.h>
+
+#include <algorithm>
+#include <map>
+#include <stdexcept>
+
+#define _unused_ __attribute__((unused))
+
+enum Action {
+  ACTION_DEFAULTS,
+  ACTION_LIST,
+  ACTION_LISTCARDS,
+  ACTION_LISTPROFILES,
+  ACTION_GETVOL,
+  ACTION_SETVOL,
+  ACTION_GETBAL,
+  ACTION_SETBAL,
+  ACTION_ADJBAL,
+  ACTION_INCREASE,
+  ACTION_DECREASE,
+  ACTION_MUTE,
+  ACTION_UNMUTE,
+  ACTION_TOGGLE,
+  ACTION_ISMUTED,
+  ACTION_SETDEFAULT,
+  ACTION_GETPROFILE,
+  ACTION_SETPROFILE,
+  ACTION_MOVE,
+  ACTION_KILL,
+  ACTION_INVALID,
+};
+
+static enum DeviceType opt_devtype;
+static enum Action opt_action;
+static const char* opt_device;
+static const char* opt_card;
+
+static const int kMinVolume = 0;
+static const int kMaxVolume = 150;
+static const int kMinBalance = -100;
+static const int kMaxBalance = 100;
+
+static const char* type_to_string(enum DeviceType t) {
+  switch (t) {
+  case DEVTYPE_SINK:
+    return "sink";
+  case DEVTYPE_SOURCE:
+    return "source";
+  case DEVTYPE_SINK_INPUT:
+    return "sink input";
+  case DEVTYPE_SOURCE_OUTPUT:
+    return "source output";
+  }
+
+  /* impossibiru! */
+  return NULL;
+}
+
+static enum Action string_to_action(const char* str) {
+  static std::map<string, enum Action> actionmap = {
+    { "defaults",       ACTION_DEFAULTS     },
+    { "list",           ACTION_LIST         },
+    { "list-cards",     ACTION_LISTCARDS    },
+    { "list-profiles",  ACTION_LISTPROFILES },
+    { "get-volume",     ACTION_GETVOL       },
+    { "set-volume",     ACTION_SETVOL       },
+    { "get-balance",    ACTION_GETBAL       },
+    { "set-balance",    ACTION_SETBAL       },
+    { "adj-balance",    ACTION_ADJBAL       },
+    { "increase",       ACTION_INCREASE     },
+    { "decrease",       ACTION_DECREASE     },
+    { "mute",           ACTION_MUTE         },
+    { "unmute",         ACTION_UNMUTE       },
+    { "toggle",         ACTION_TOGGLE       },
+    { "is-muted",       ACTION_ISMUTED      },
+    { "set-default",    ACTION_SETDEFAULT   },
+    { "get-profile",    ACTION_GETPROFILE   },
+    { "set-profile",    ACTION_SETPROFILE   },
+    { "move",           ACTION_MOVE         },
+    { "kill",           ACTION_KILL         }
+  };
+
+  try {
+    return actionmap.at(str);
+  } catch(std::out_of_range) {
+    errx(1, "error: Invalid option specified: %s", str);
+  }
+}
+
+static enum DeviceType string_to_devtype_or_die(const char* str) {
+  static std::map<string, enum DeviceType> typemap = {
+    { "sink",           DEVTYPE_SINK          },
+    { "source",         DEVTYPE_SOURCE        },
+    { "sink-input",     DEVTYPE_SINK_INPUT    },
+    { "source-output",  DEVTYPE_SOURCE_OUTPUT },
+  };
+  try {
+    return typemap.at(str);
+  } catch(std::out_of_range) {
+    errx(1, "error: Invalid device type specified: %s", str);
+  }
+}
+
+static Device* string_to_device(Pulse& ponymix, string arg, enum DeviceType type) {
+  switch (type) {
+  case DEVTYPE_SINK:
+    return ponymix.GetSink(arg);
+  case DEVTYPE_SOURCE:
+    return ponymix.GetSource(arg);
+  case DEVTYPE_SOURCE_OUTPUT:
+    return ponymix.GetSourceOutput(arg);
+  case DEVTYPE_SINK_INPUT:
+    return ponymix.GetSinkInput(arg);
+  default:
+  return nullptr;
+  }
+}
+
+static Device* string_to_device_or_die(Pulse& ponymix,
+                                       string arg,
+                                       enum DeviceType type) {
+  Device* device = string_to_device(ponymix, arg, type);
+  if (device == nullptr) errx(1, "no match found for device: %s", arg.c_str());
+  return device;
+}
+
+static void Print(const Device& device) {
+  printf("%s %d: %s\n"
+         "  %s\n"
+         "  Avg. Volume: %d%%\n",
+         type_to_string(device.Type()),
+         device.Index(),
+         device.Name().c_str(),
+         device.Desc().c_str(),
+         device.Volume());
+}
+
+static void Print(const Card& card) {
+  printf("%s\n", card.Name().c_str());
+}
+
+static void Print(const Profile& profile, bool active) {
+  printf("%s: %s%s\n",
+         profile.name.c_str(), profile.desc.c_str(), active ? " [active]" : "");
+}
+
+static int ShowDefaults(Pulse& ponymix,
+                        int argc _unused_,
+                        char* argv[] _unused_) {
+  const auto& info = ponymix.GetDefaults();
+  Print(*ponymix.GetSink(info.sink));
+  Print(*ponymix.GetSource(info.source));
+  return 0;
+}
+
+static int List(Pulse& ponymix, int argc _unused_, char* argv[] _unused_) {
+  if (argc != 0) errx(1, "error: list requires 0 arguments");
+
+  const auto& sinks = ponymix.GetSinks();
+  for (const auto& s : sinks) Print(s);
+
+  const auto& sources = ponymix.GetSources();
+  for (const auto& s : sources) Print(s);
+
+  const auto& sinkinputs = ponymix.GetSinkInputs();
+  for (const auto& s : sinkinputs) Print(s);
+
+  const auto& sourceoutputs = ponymix.GetSourceOutputs();
+  for (const auto& s : sourceoutputs) Print(s);
+
+  return 0;
+}
+
+static int ListCards(Pulse& ponymix, int argc _unused_, char* argv[] _unused_) {
+  if (argc != 0) errx(1, "error: list-cards requires 0 arguments");
+
+  const auto& cards = ponymix.GetCards();
+  for (const auto& c : cards) Print(c);
+
+  return 0;
+}
+
+static int ListProfiles(Pulse& ponymix, int argc _unused_, char* argv[] _unused_) {
+  if (argc != 0) errx(1, "error: list-profiles requires 0 arguments");
+
+  // TODO: figure out how to get a list of cards?
+  auto card = ponymix.GetCard(opt_card);
+  if (card == nullptr) errx(1, "error: no match found for card: %s", opt_card);
+
+  const auto& profiles = card->Profiles();
+  for (const auto& p : profiles) Print(p, p.name == card->ActiveProfile().name);
+
+  return 0;
+}
+
+static int GetVolume(Pulse& ponymix, int argc, char* argv[] _unused_) {
+  if (argc != 0) errx(1, "error: get-volume requires 0 arguments");
+
+  auto device = string_to_device_or_die(ponymix, opt_device, opt_devtype);
+  printf("%d\n", device->Volume());
+  return 0;
+}
+
+static int SetVolume(Pulse& ponymix, int argc, char* argv[]) {
+  if (argc != 1) errx(1, "error: set-volume requires exactly 1 argument");
+
+  auto device = string_to_device_or_die(ponymix, opt_device, opt_devtype);
+
+  long volume;
+  try {
+    volume = std::stol(argv[0]);
+  } catch (std::invalid_argument _) {
+    errx(1, "error: failed to convert string to integer: %s", argv[0]);
+  }
+
+  if (!ponymix.SetVolume(*device, volume)) return 1;
+
+  printf("%d\n", device->Volume());
+
+  return 0;
+}
+
+static int GetBalance(Pulse& ponymix, int argc, char* argv[] _unused_) {
+  if (argc != 0) errx(1, "error: get-balance requires 0 arguments");
+
+  auto device = string_to_device_or_die(ponymix, opt_device, opt_devtype);
+  printf("%d\n", device->Balance());
+  return 0;
+}
+
+static int SetBalance(Pulse& ponymix, int argc, char* argv[]) {
+  if (argc != 1) errx(1, "error: set-balance requires exactly 1 argument");
+
+  auto device = string_to_device_or_die(ponymix, opt_device, opt_devtype);
+
+  long balance;
+  try {
+    balance = std::stol(argv[0]);
+  } catch (std::invalid_argument _) {
+    errx(1, "error: failed to convert string to integer: %s", argv[0]);
+  }
+
+  if (!ponymix.SetBalance(*device, balance)) return 1;
+
+  printf("%d\n", device->Balance());
+
+  return 0;
+}
+
+static int AdjBalance(Pulse& ponymix, int argc, char* argv[]) {
+  if (argc != 1) errx(1, "error: adj-balance requires exactly 1 argument");
+
+  auto device = string_to_device_or_die(ponymix, opt_device, opt_devtype);
+
+  long balance;
+  try {
+    balance = std::stol(argv[0]);
+  } catch (std::invalid_argument _) {
+    errx(1, "error: failed to convert string to integer: %s", argv[0]);
+  }
+
+  if (!ponymix.SetBalance(*device, device->Balance() + balance)) return 1;
+
+  printf("%d\n", device->Balance());
+
+  return 0;
+}
+
+static int IncreaseVolume(Pulse& ponymix, int argc, char* argv[]) {
+  if (argc != 1) errx(1, "error: increase requires exactly 1 argument");
+
+  auto device = string_to_device_or_die(ponymix, opt_device, opt_devtype);
+
+  long delta;
+  try {
+    delta = std::stol(argv[0]);
+  } catch (std::invalid_argument _) {
+    errx(1, "error: failed to convert string to integer: %s", argv[0]);
+  }
+
+  if (!ponymix.SetVolume(*device, device->Volume() + delta)) return 1;
+
+  printf("%d\n", device->Volume());
+
+  return 0;
+}
+
+static int DecreaseVolume(Pulse& ponymix, int argc, char* argv[]) {
+  if (argc != 1) errx(1, "error: decrease requires exactly 1 argument");
+
+  auto device = string_to_device_or_die(ponymix, opt_device, opt_devtype);
+
+  long delta;
+  try {
+    delta = std::stol(argv[0]);
+  } catch (std::invalid_argument _) {
+    errx(1, "error: failed to convert string to integer: %s", argv[0]);
+  }
+
+  if (!ponymix.SetVolume(*device, device->Volume() - delta)) return 1;
+
+  printf("%d\n", device->Volume());
+
+  return 0;
+}
+
+static int Mute(Pulse& ponymix, int argc, char* argv[] _unused_) {
+  if (argc != 0) errx(1, "error: mute requires 0 arguments");
+
+  auto device = string_to_device_or_die(ponymix, opt_device, opt_devtype);
+
+  if (!ponymix.SetMute(*device, true)) return 1;
+
+  printf("%d\n", device->Volume());
+
+  return 0;
+}
+
+static int Unmute(Pulse& ponymix, int argc, char* argv[] _unused_) {
+  if (argc != 0) errx(1, "error: unmute requires 0 arguments");
+
+  auto device = string_to_device_or_die(ponymix, opt_device, opt_devtype);
+
+  if (!ponymix.SetMute(*device, false)) return 1;
+
+  printf("%d\n", device->Volume());
+
+  return 0;
+}
+
+static int ToggleMute(Pulse& ponymix, int argc, char* argv[] _unused_) {
+  if (argc != 0) errx(1, "error: toggle requires 0 arguments");
+
+  auto device = string_to_device_or_die(ponymix, opt_device, opt_devtype);
+
+  if (!ponymix.SetMute(*device, !ponymix.IsMuted(*device))) return 1;
+
+  printf("%d\n", device->Volume());
+
+  return 0;
+}
+
+static int IsMuted(Pulse& ponymix, int argc, char* argv[] _unused_) {
+  if (argc != 0) errx(1, "error: is-muted requires 0 arguments"); 
+
+  auto device = string_to_device_or_die(ponymix, opt_device, opt_devtype);
+  return !ponymix.IsMuted(*device);
+}
+
+static int SetDefault(Pulse& ponymix, int argc, char* argv[] _unused_) {
+  if (argc != 0) errx(1, "error: set-default requires 0 arguments"); 
+
+  auto device = string_to_device_or_die(ponymix, opt_device, opt_devtype);
+
+  return !ponymix.SetDefault(*device);
+}
+
+static int GetProfile(Pulse& ponymix, int argc, char* argv[] _unused_) {
+  if (argc != 0) errx(1, "error: get-profile requires 0 arguments");
+
+  auto card = ponymix.GetCard(opt_card);
+  if (card == nullptr) errx(1, "error: no match found for card: %s", opt_card);
+
+  printf("%s\n", card->ActiveProfile().name.c_str());
+
+  return true;
+}
+
+static int SetProfile(Pulse& ponymix, int argc, char* argv[] _unused_) {
+  if (argc != 1) errx(1, "error: set-profile requires 1 argument");
+
+  auto card = ponymix.GetCard(opt_card);
+  if (card == nullptr) errx(1, "error: no match found for card: %s", opt_card);
+
+  return !ponymix.SetProfile(*card, argv[0]);
+}
+
+static int Move(Pulse& ponymix, int argc, char* argv[]) {
+  if (argc != 1) errx(1, "error: move requires 1 argument");
+
+  // Does this even work?
+  auto source = string_to_device_or_die(ponymix, opt_device, opt_devtype);
+  auto target = string_to_device_or_die(ponymix, argv[0], opt_devtype);
+
+  return !ponymix.Move(*source, *target);
+}
+
+static int Kill(Pulse& ponymix, int argc, char* argv[] _unused_) {
+  if (argc != 0) errx(1, "error: set-default requires 0 arguments"); 
+
+  auto device = string_to_device_or_die(ponymix, opt_device, opt_devtype);
+
+  return !ponymix.Kill(*device);
+}
+
+static int (*fn[])(Pulse& ponymix, int argc, char* argv[]) = {
+  [ACTION_DEFAULTS] = ShowDefaults,
+  [ACTION_LIST] = List,
+  [ACTION_LISTCARDS] = ListCards,
+  [ACTION_LISTPROFILES] = ListProfiles,
+  [ACTION_GETVOL] = GetVolume,
+  [ACTION_SETVOL] = SetVolume,
+  [ACTION_GETBAL] = GetBalance,
+  [ACTION_SETBAL] = SetBalance,
+  [ACTION_ADJBAL] = AdjBalance,
+  [ACTION_INCREASE] = IncreaseVolume,
+  [ACTION_DECREASE] = DecreaseVolume,
+  [ACTION_MUTE] = Mute,
+  [ACTION_UNMUTE] = Unmute,
+  [ACTION_TOGGLE] = ToggleMute,
+  [ACTION_ISMUTED] = IsMuted,
+  [ACTION_SETDEFAULT] = SetDefault,
+  [ACTION_GETPROFILE] = GetProfile,
+  [ACTION_SETPROFILE] = SetProfile,
+  [ACTION_MOVE] = Move,
+  [ACTION_KILL] = Kill,
+};
+
+void usage() {
+  printf("usage: %s [options] <command>...\n", program_invocation_short_name);
+  fputs("\nOptions:\n"
+        " -h, --help              display this help and exit\n\n"
+
+        " -c, --card CARD         target card (index or name)\n"
+        " -d, --device DEVICE     target device (index or name)\n"
+        " -t, --devtype TYPE      device type\n", stdout);
+
+  fputs("\nCommon Commands:\n"
+        "  list                   list available devices\n"
+        "  list-cards             list available cards\n"
+        "  get-volume             get volume for device\n"
+        "  set-volume VALUE       set volume for device\n"
+        "  get-balance            get balance for device\n"
+        "  set-balance VALUE      set balance for device\n"
+        "  adj-balance VALUE      increase or decrease balance for device\n"
+        "  increase VALUE         increase volume\n", stdout);
+  fputs("  decrease VALUE         decrease volume\n"
+        "  mute                   mute device\n"
+        "  unmute                 unmute device\n"
+        "  toggle                 toggle mute\n"
+        "  is-muted               check if muted\n", stdout);
+
+  fputs("\nCard Commands:\n"
+        "  list-profiles          list available profiles for a card\n"
+        "  get-profile            get active profile for card\n"
+        "  set-profile PROFILE    set profile for a card\n"
+
+        "\nDevice Commands:\n"
+        "  defaults               list default devices (default command)\n"
+        "  set-default DEVICE     set default device by ID\n"
+
+        "\nApplication Commands:\n"
+        "  move DEVICE            move target device to DEVICE\n"
+        "  kill DEVICE            kill target DEVICE\n", stdout);
+
+  exit(EXIT_SUCCESS);
+}
+
+bool parse_options(int argc, char** argv) {
+  static const struct option opts[] = {
+    { "card",   required_argument, 0, 'c' },
+    { "device", required_argument, 0, 'd' },
+    { "help",   no_argument,       0, 'h' },
+    { "type",   required_argument, 0, 't' },
+    { 0, 0, 0, 0 },
+  };
+
+  for (;;) {
+    int opt = getopt_long(argc, argv, "c:d:ht:", opts, NULL);
+    if (opt == -1)
+      break;
+
+    switch (opt) {
+    case 'c':
+      opt_card = optarg;
+      break;
+    case 'd':
+      opt_device = optarg;
+      break;
+    case 'h':
+      usage();
+      break;
+    case 't':
+      opt_devtype = string_to_devtype_or_die(optarg);
+      break;
+    default:
+      return false;
+    }
+  }
+
+  return true;
+}
+
+int main(int argc, char* argv[]) {
+  Pulse ponymix("ponymix");
+  ponymix.Populate();
+
+  // defaults
+  opt_action = ACTION_DEFAULTS;
+  opt_devtype = DEVTYPE_SINK;
+  opt_device = ponymix.GetDefaults().sink.c_str();
+  opt_card = ponymix.GetCards()[0].Name().c_str();
+
+  if (!parse_options(argc, argv)) return 1;
+  argc -= optind;
+  argv += optind;
+
+  if (argc > 0) {
+    opt_action = string_to_action(argv[0]);
+    argc--;
+    argv++;
+  }
+
+  return fn[opt_action](ponymix, argc, argv);
+}
+
+// vim: set et ts=2 sw=2: