Work on #20 - Better API; Work on #16;

modules/api/components/WebSocketAPI.php

@@ -0,0 +1,102 @@
+<?php
+
+namespace app\modules\api\components;
+
+use app\models\User;
+use Ratchet\Client\Connector;
+use Ratchet\Client\WebSocket;
+use React\EventLoop\Factory;
+use Yii;
+use yii\helpers\Json;
+
+class WebSocketAPI
+{
+    /**
+     * @var User
+     */
+    protected $user;
+
+    /**
+     * @var string
+     */
+    public $localWSSUrl;
+
+    /**
+     * WebSocketAPI constructor.
+     * @param User $user
+     * @param string|null $localWSSUrl
+     */
+    public function __construct($user, $localWSSUrl = null)
+    {
+        $this->user = $user;
+        $this->localWSSUrl = $localWSSUrl;
+
+        if (is_null($this->localWSSUrl)) {
+            $this->localWSSUrl = Yii::$app->params['localWSURL'];
+        }
+    }
+
+    /**
+     * @return string
+     */
+    protected function getWSSUrl()
+    {
+        return $this->localWSSUrl . '/?type=user&id=' . $this->user->id . '&auth_key=' . $this->user->auth_key;
+    }
+
+    /**
+     * @param int $itemID
+     * @return bool
+     */
+    public function turnOn($itemID)
+    {
+        return $this->send([
+            'type' => 'turnON',
+            'item_id' => $itemID,
+        ]);
+    }
+
+    /**
+     * @param int $itemID
+     * @return bool
+     */
+    public function turnOff($itemID)
+    {
+        return $this->send([
+            'type' => 'turnOFF',
+            'item_id' => $itemID,
+        ]);
+    }
+
+    /**
+     * @param array $data
+     * @return bool
+     */
+    protected function send($data)
+    {
+        $loop = Factory::create();
+        $connector = new Connector($loop);
+
+        $afterConnect = function (WebSocket $conn) use ($data) {
+            // Send data
+            $conn->send(Json::encode($data));
+
+            // Job done. Close the connection
+            $conn->close();
+
+            return true;
+        };
+
+        $afterDisconnect = function(\Exception $e) use ($loop) {
+            echo "Could not connect: {$e->getMessage()}\n";
+            $loop->stop();
+        };
+
+        $connector($this->getWSSUrl(), [], ['Origin' => 'origin'])
+            ->then($afterConnect, $afterDisconnect);
+
+        $loop->run();
+
+        return ($afterConnect == true and $afterDisconnect == false);
+    }
+}