about summary refs log tree commit homepage
path: root/lib/dtas/unix_client.rb
diff options
context:
space:
mode:
authorEric Wong <normalperson@yhbt.net>2013-08-24 09:54:45 +0000
committerEric Wong <normalperson@yhbt.net>2013-08-24 09:54:45 +0000
commit3e09ac0c10c95bb24a08af62393b4f761e2743d0 (patch)
tree778dffa2ba8798503fc047db0feef6d65426ea22 /lib/dtas/unix_client.rb
downloaddtas-3e09ac0c10c95bb24a08af62393b4f761e2743d0.tar.gz
Diffstat (limited to 'lib/dtas/unix_client.rb')
-rw-r--r--lib/dtas/unix_client.rb52
1 files changed, 52 insertions, 0 deletions
diff --git a/lib/dtas/unix_client.rb b/lib/dtas/unix_client.rb
new file mode 100644
index 0000000..f46eddf
--- /dev/null
+++ b/lib/dtas/unix_client.rb
@@ -0,0 +1,52 @@
+# -*- encoding: binary -*-
+# :stopdoc:
+# Copyright (C) 2013, Eric Wong <normalperson@yhbt.net>
+# License: GPLv3 or later (https://www.gnu.org/licenses/gpl-3.0.txt)
+require 'dtas'
+require 'socket'
+require 'io/wait'
+require 'shellwords'
+
+class DTAS::UNIXClient
+  attr_reader :to_io
+
+  def self.default_path
+    (ENV["DTAS_PLAYER_SOCK"] || File.expand_path("~/.dtas/player.sock")).b
+  end
+
+  def initialize(path = self.class.default_path)
+    @to_io = begin
+      raise if ENV["_DTAS_NOSEQPACKET"]
+      Socket.new(:AF_UNIX, :SOCK_SEQPACKET, 0)
+    rescue
+      warn("get your operating system developers to support " \
+           "SOCK_SEQPACKET for AF_UNIX sockets")
+      warn("falling back to SOCK_DGRAM, reliability possibly compromised")
+      Socket.new(:AF_UNIX, :SOCK_DGRAM, 0)
+    end
+    @to_io.connect(Socket.pack_sockaddr_un(path))
+  end
+
+  def req_start(args)
+    args = Shellwords.join(args) if Array === args
+    @to_io.send(args, Socket::MSG_EOR)
+  end
+
+  def req_ok(args, timeout = nil)
+    res = req(args, timeout)
+    res == "OK" or raise "Unexpected response: #{res}"
+    res
+  end
+
+  def req(args, timeout = nil)
+    req_start(args)
+    res_wait(timeout)
+  end
+
+  def res_wait(timeout = nil)
+    @to_io.wait(timeout)
+    nr = @to_io.nread
+    nr > 0 or raise EOFError, "unexpected EOF from server"
+    @to_io.recvmsg[0]
+  end
+end