From 4b5d1347a4eba415292b53de71b85697ed03561a Mon Sep 17 00:00:00 2001
From: Andrei <b1naryth1ef@gmail.com>
Date: Mon, 17 Oct 2016 17:30:07 -0500
Subject: [PATCH] more voice

---
 disco/gateway/events.py |   2 +
 disco/voice/client.py   | 120 +++++---------------------------------
 disco/voice/opus.py     | 125 ++++++++++++++++++++++++++++++++++++++++
 disco/voice/player.py   |  93 ++++++++++++++++++++++++++++++
 4 files changed, 234 insertions(+), 106 deletions(-)
 create mode 100644 disco/voice/opus.py
 create mode 100644 disco/voice/player.py

diff --git a/disco/gateway/events.py b/disco/gateway/events.py
index 5b2e8b5..6d55690 100644
--- a/disco/gateway/events.py
+++ b/disco/gateway/events.py
@@ -49,6 +49,8 @@ class GatewayEvent(six.with_metaclass(GatewayEventMeta, Model)):
         """
         Create this GatewayEvent class from data and the client.
         """
+        cls.raw_data = obj
+
         # If this event is wrapping a model, pull its fields
         if hasattr(cls, '_wraps_model'):
             alias, model = cls._wraps_model
diff --git a/disco/voice/client.py b/disco/voice/client.py
index b293ec3..2ca00ba 100644
--- a/disco/voice/client.py
+++ b/disco/voice/client.py
@@ -3,7 +3,6 @@ import socket
 import struct
 import time
 
-from six.moves import queue
 from holster.enum import Enum
 from holster.emitter import Emitter
 
@@ -23,11 +22,6 @@ VoiceState = Enum(
     VOICE_CONNECTED=6,
 )
 
-# TODO:
-#   - player implementation
-#   - encryption
-#   - cleanup
-
 
 class VoiceException(Exception):
     def __init__(self, msg, client):
@@ -45,18 +39,14 @@ class UDPVoiceClient(LoggingClass):
         self.run_task = None
         self.connected = False
 
-        self.seq = 0
-        self.ts = 0
-
-    def send_frame(self, frame):
-        self.seq += 1
-        data = '\x80\x78'
-        data += struct.pack('>H', self.seq)
-        data += struct.pack('>I', self.ts)
-        data += struct.pack('>I', self.vc.ssrc)
-        data += ''.join(frame)
-        self.send(data)
-        self.ts += 960
+    def send_frame(self, frame, sequence, timestamp):
+        data = bytearray(12)
+        data[0] = 0x80
+        data[1] = 0x78
+        struct.pack_into('>H', data, 2, sequence)
+        struct.pack_into('>I', data, 4, timestamp)
+        struct.pack_into('>i', data, 8, self.vc.ssrc)
+        self.send(data + ''.join(frame))
 
     def run(self):
         while True:
@@ -126,7 +116,6 @@ class VoiceClient(LoggingClass):
     def set_state(self, state):
         prev_state = self.state
         self.state = state
-        print 'State Change %s to %s' % (prev_state, state)
         self.state_emitter.emit(state, prev_state)
 
     def heartbeat(self, interval):
@@ -208,7 +197,6 @@ class VoiceClient(LoggingClass):
         self.log.warning('Voice websocket error: {}'.format(err))
 
     def on_open(self):
-        print 'open'
         self.send(VoiceOPCode.IDENTIFY, {
             'server_id': self.channel.guild_id,
             'user_id': self.client.state.me.id,
@@ -217,9 +205,12 @@ class VoiceClient(LoggingClass):
         })
 
     def on_close(self, code, error):
-        # TODO
         self.log.warning('Voice websocket disconnected (%s, %s)', code, error)
 
+        if self.state == VoiceState.CONNECTED:
+            self.log.info('Attempting voice reconnection')
+            self.connect()
+
     def connect(self, timeout=5, mute=False, deaf=False):
         self.set_state(VoiceState.AWAITING_ENDPOINT)
 
@@ -255,88 +246,5 @@ class VoiceClient(LoggingClass):
             'channel_id': None,
         })
 
-    def send_frame(self, frame):
-        self.udp.send_frame(frame)
-
-
-class OpusItem(object):
-    __slots__ = ('frames', 'idx')
-
-    def __init__(self):
-        self.frames = []
-        self.idx = 0
-
-    @classmethod
-    def from_raw_file(cls, path):
-        inst = cls()
-        obj = open(path, 'r')
-
-        while True:
-            buff = obj.read(2)
-            if not buff:
-                return inst
-            size = struct.unpack('<h', buff)[0]
-            inst.frames.append(obj.read(size))
-
-    def have_frame(self):
-        return self.idx + 1 < len(self.frames)
-
-    def next_frame(self):
-        self.idx += 1
-        return self.frames[self.idx]
-
-
-class Player(object):
-    def __init__(self, client):
-        self.client = client
-        self.queue = queue.Queue()
-        self.playing = True
-        self.run_task = gevent.spawn(self.run)
-        self.paused = None
-        self.complete = gevent.event.Event()
-
-    def disconnect(self):
-        self.client.disconnect()
-
-    def pause(self):
-        if self.paused:
-            return
-        self.paused = gevent.event.Event()
-
-    def resume(self):
-        self.paused.set()
-        self.paused = None
-
-    def play(self, item):
-        start = time.time()
-        loops = 0
-
-        while True:
-            loops += 1
-            if self.paused:
-                self.paused.wait()
-
-            if self.client.state == VoiceState.DISCONNECTED:
-                return
-
-            if self.client.state != VoiceState.CONNECTED:
-                self.client.state_emitter.wait(VoiceState.CONNECTED)
-
-            if not item.have_frame():
-                return
-
-            self.client.send_frame(item.next_frame())
-            next_time = start + 0.02 * loops
-            delay = max(0, 0.02 + (next_time - time.time()))
-            gevent.sleep(delay)
-
-    def run(self):
-        self.client.set_speaking(True)
-        while self.playing:
-            self.play(self.queue.get())
-
-            if self.client.state == VoiceState.DISCONNECTED:
-                self.playing = False
-                self.complete.set()
-                return
-        self.client.set_speaking(False)
+    def send_frame(self, *args, **kwargs):
+        self.udp.send_frame(*args, **kwargs)
diff --git a/disco/voice/opus.py b/disco/voice/opus.py
new file mode 100644
index 0000000..a1ad20e
--- /dev/null
+++ b/disco/voice/opus.py
@@ -0,0 +1,125 @@
+import sys
+import array
+import ctypes
+import ctypes.util
+
+from holster.enum import Enum
+
+from disco.util.logging import LoggingClass
+
+
+c_int_ptr = ctypes.POINTER(ctypes.c_int)
+c_int16_ptr = ctypes.POINTER(ctypes.c_int16)
+c_float_ptr = ctypes.POINTER(ctypes.c_float)
+
+
+class EncoderStruct(ctypes.Structure):
+    pass
+
+
+class DecoderStruct(ctypes.Structure):
+    pass
+
+
+EncoderStructPtr = ctypes.POINTER(EncoderStruct)
+DecoderStructPtr = ctypes.POINTER(DecoderStruct)
+
+
+class BaseOpus(LoggingClass):
+    BASE_EXPORTED = {
+        'opus_strerror': ([ctypes.c_int], ctypes.c_char_p),
+    }
+
+    EXPORTED = {}
+
+    def __init__(self, library_path=None):
+        self.path = library_path or self.find_library()
+        self.lib = ctypes.cdll.LoadLibrary(self.path)
+
+        methods = {}
+        methods.update(self.BASE_EXPORTED)
+        methods.update(self.EXPORTED)
+
+        for name, item in methods.items():
+            func = getattr(self.lib, name)
+
+            if item[1]:
+                func.argtypes = item[1]
+
+            func.restype = item[2]
+
+            setattr(self, name.replace('opus_', ''), func)
+
+    @staticmethod
+    def find_library():
+        if sys.platform == 'win32':
+            raise Exception('Cannot auto-load opus on Windows, please specify full library path')
+
+        return ctypes.util.find_library('opus')
+
+
+Application = Enum(
+    AUDIO=2049,
+    VOIP=2048,
+    LOWDELAY=2051
+)
+
+
+Control = Enum(
+    SET_BITRATE=4002,
+    SET_BANDWIDTH=4008,
+    SET_FEC=4012,
+    SET_PLP=4014,
+)
+
+
+class OpusEncoder(BaseOpus):
+    EXPORTED = {
+        'opus_encoder_get_size': ([ctypes.c_int], ctypes.c_int),
+        'opus_encoder_create': ([ctypes.c_int, ctypes.c_int, ctypes.c_int, c_int_ptr], EncoderStructPtr),
+        'opus_encode': ([EncoderStructPtr, c_int16_ptr, ctypes.c_int, ctypes.c_char_p, ctypes.c_int32], ctypes.c_int32),
+        'opus_encoder_ctl': (None, ctypes.c_int32),
+        'opus_encoder_destroy': ([EncoderStructPtr], None),
+    }
+
+    def __init__(self, sampling, channels, application=Application.AUDIO, library_path=None):
+        super(OpusDecoder, self).__init__(library_path)
+        self.sampling_rate = sampling
+        self.channels = channels
+        self.application = application
+
+        self.frame_length = 20
+        self.sample_size = 2 * self.channels
+        self.samples_per_frame = int(self.sampling_rate / 1000 * self.frame_length)
+        self.frame_size = self.samples_per_frame * self.sample_size
+
+        self.inst = self.create()
+
+    def create(self):
+        ret = ctypes.c_int()
+        result = self.encoder_create(self.sampling_rate, self.channels, self.application.value, ctypes.byref(ret))
+
+        if ret.value != 0:
+            raise Exception('Failed to create opus encoder: {}'.format(ret.value))
+
+        return result
+
+    def __del__(self):
+        if self.inst:
+            self.encoder_destroy(self.inst)
+            self.inst = None
+
+    def encode(self, pcm, frame_size):
+        max_data_bytes = len(pcm)
+        pcm = ctypes.cast(pcm, c_int16_ptr)
+        data = (ctypes.c_char * max_data_bytes)()
+
+        ret = self.encode(self.inst, pcm, frame_size, data, max_data_bytes)
+        if ret < 0:
+            raise Exception('Failed to encode: {}'.format(ret))
+
+        return array.array('b', data[:ret]).tobytes()
+
+
+class OpusDecoder(BaseOpus):
+    pass
diff --git a/disco/voice/player.py b/disco/voice/player.py
new file mode 100644
index 0000000..144a25b
--- /dev/null
+++ b/disco/voice/player.py
@@ -0,0 +1,93 @@
+import gevent
+import struct
+import time
+
+from six.moves import queue
+
+from disco.voice.client import VoiceState
+
+
+class OpusItem(object):
+    def __init__(self, frame_length=20, channels=2):
+        self.frames = []
+        self.idx = 0
+        self.frame_length = frame_length
+        self.channels = channels
+
+    @classmethod
+    def from_raw_file(cls, path):
+        inst = cls()
+        obj = open(path, 'r')
+
+        while True:
+            buff = obj.read(2)
+            if not buff:
+                return inst
+            size = struct.unpack('<h', buff)[0]
+            inst.frames.append(obj.read(size))
+
+    def have_frame(self):
+        return self.idx + 1 < len(self.frames)
+
+    def next_frame(self):
+        self.idx += 1
+        return self.frames[self.idx]
+
+
+class Player(object):
+    def __init__(self, client):
+        self.client = client
+        self.queue = queue.Queue()
+        self.playing = True
+        self.run_task = gevent.spawn(self.run)
+        self.paused = None
+        self.complete = gevent.event.Event()
+
+    def disconnect(self):
+        self.client.disconnect()
+
+    def pause(self):
+        if self.paused:
+            return
+        self.paused = gevent.event.Event()
+
+    def resume(self):
+        self.paused.set()
+        self.paused = None
+
+    def play(self, item):
+        start = time.time()
+        loops = 0
+
+        while True:
+            loops += 1
+            if self.paused:
+                self.client.set_speaking(False)
+                self.paused.wait()
+                gevent.sleep(2)
+                self.client.set_speaking(True)
+
+            if self.client.state == VoiceState.DISCONNECTED:
+                return
+
+            if self.client.state != VoiceState.CONNECTED:
+                self.client.state_emitter.wait(VoiceState.CONNECTED)
+
+            if not item.have_frame():
+                return
+
+            self.client.send_frame(item.next_frame())
+            next_time = start + 0.02 * loops
+            delay = max(0, 0.02 + (next_time - time.time()))
+            gevent.sleep(delay)
+
+    def run(self):
+        self.client.set_speaking(True)
+        while self.playing:
+            self.play(self.queue.get())
+
+            if self.client.state == VoiceState.DISCONNECTED:
+                self.playing = False
+                self.complete.set()
+                return
+        self.client.set_speaking(False)