warlof · warlof · Nov 15, 2020
diff --git a/src/Config/seat-connector.config.php b/src/Config/seat-connector.config.php
@@ -43,5 +43,10 @@
             'label' => 'seat-connector-teamspeak::seat.api_key',
             'type'  => 'text',
         ],
+        [
+            'name'  => 'registration_group_name',
+            'label' => 'seat-connector-teamspeak::seat.registration_group',
+            'type'  => 'text',
+        ],
     ],
 ];
diff --git a/src/Driver/TeamspeakClient.php b/src/Driver/TeamspeakClient.php
@@ -24,11 +24,13 @@
 use GuzzleHttp\Exception\RequestException;
 use Illuminate\Support\Arr;
 use Seat\Services\Exceptions\SettingException;
+use Seat\Web\Models\User;
 use Warlof\Seat\Connector\Drivers\IClient;
 use Warlof\Seat\Connector\Drivers\ISet;
 use Warlof\Seat\Connector\Drivers\IUser;
 use Warlof\Seat\Connector\Drivers\Teamspeak\Exceptions\CommandException;
 use Warlof\Seat\Connector\Drivers\Teamspeak\Exceptions\ConnexionException;
+use Warlof\Seat\Connector\Drivers\Teamspeak\Exceptions\EmptyResultException;
 use Warlof\Seat\Connector\Drivers\Teamspeak\Exceptions\LoginException;
 use Warlof\Seat\Connector\Drivers\Teamspeak\Exceptions\ServerException;
 use Warlof\Seat\Connector\Drivers\Teamspeak\Exceptions\TeamspeakException;
@@ -145,6 +147,18 @@ public static function getInstance(): IClient
         return self::$instance;
     }
 
+    /**
+     * @param int $instance_id
+     *
+     * @return $this
+     */
+    public function setInstance(int $instance_id): self
+    {
+        $this->instance_id = $instance_id;
+
+        return $this;
+    }
+
     /**
      * @return \Warlof\Seat\Connector\Drivers\IUser[]
      * @throws \Warlof\Seat\Connector\Exceptions\DriverException
@@ -282,6 +296,78 @@ public function findUserByName(string $nickname)
         return $speaker;
     }
 
+    /**
+     * @param \Seat\Web\Models\User $user
+     *
+     * @return \Warlof\Seat\Connector\Drivers\IUser[]
+     * @throws \Warlof\Seat\Connector\Drivers\Teamspeak\Exceptions\CommandException
+     * @throws \Warlof\Seat\Connector\Drivers\Teamspeak\Exceptions\LoginException
+     * @throws \Warlof\Seat\Connector\Exceptions\DriverException
+     * @throws \Warlof\Seat\Connector\Exceptions\InvalidDriverIdentityException
+     */
+    public function customSearch(User $user): array
+    {
+        try {
+            $response = $this->sendCall('POST', '/{instance_id}/customsearch', [
+                'ident' => 'seat_user_id',
+                'pattern' => $user->getAuthIdentifier(),
+                'instance_id' => $this->instance_id,
+            ]);
+        } catch (EmptyResultException $e) {
+            logger()->debug('Teamspeak return empty results', [
+                'command' => 'customsearch',
+                'ident' => 'seat_user_id',
+                'pattern' => $user->getAuthIdentifier(),
+                'instance_id' => $this->instance_id,
+            ]);
+
+            return [];
+        }
+
+        $speakers = [];
+
+        foreach ($response as $speaker) {
+            $speakers[] = $this->getUser($speaker->cldbid);
+        }
+
+        return $speakers;
+    }
+
+    /**
+     * @param string $uid
+     *
+     * @return int[]
+     * @throws \Warlof\Seat\Connector\Drivers\Teamspeak\Exceptions\CommandException
+     * @throws \Warlof\Seat\Connector\Drivers\Teamspeak\Exceptions\LoginException
+     * @throws \Warlof\Seat\Connector\Exceptions\DriverException
+     * @throws \Warlof\Seat\Connector\Exceptions\InvalidDriverIdentityException
+     */
+    public function clientGetIds(string $uid): array
+    {
+        try {
+            $response = $this->sendCall('GET', '/{instance}/clientgetids', [
+                'cluid' => $uid,
+                'instance' => $this->instance_id,
+            ]);
+        } catch (EmptyResultException $e) {
+            logger()->debug('Teamspeak return empty results', [
+                'command' => 'clientgetids',
+                'cluid' => $uid,
+                'instance_id' => $this->instance_id,
+            ]);
+
+            return [];
+        }
+
+        $speakers = [];
+
+        foreach ($response as $speaker) {
+            $speakers[] = $speaker->clid;
+        }
+
+        return $speakers;
+    }
+
     /**
      * @param string $id
      * @return \Warlof\Seat\Connector\Drivers\ISet|null
@@ -385,12 +471,7 @@ public function getServerGroupMembers(ISet $server_group): array
      */
     public function getSpeakerServerGroups(IUser $speaker): array
     {
-        // scope: manage_scope
-        $response = $this->sendCall('GET', '/{instance}/serverinfo', [
-            'instance' => $this->instance_id,
-        ]);
-
-        $server_info = Arr::first($response);
+        $server_info = $server_info = $this->serverInfo();
 
         // scope: manage_scope
         $response = $this->sendCall('GET', '/{instance}/servergroupsbyclientid', [
@@ -415,6 +496,74 @@ public function getSpeakerServerGroups(IUser $speaker): array
         return $server_group;
     }
 
+    /**
+     * @param int[] $clients
+     *
+     * @throws \Warlof\Seat\Connector\Drivers\Teamspeak\Exceptions\CommandException
+     * @throws \Warlof\Seat\Connector\Drivers\Teamspeak\Exceptions\LoginException
+     */
+    public function kickClientFromServerInstance(array $clients)
+    {
+        $this->sendCall('POST', '/{instance}/clientkick', [
+            'clid' => $clients,
+            'reasonid' => 5,
+            'reasonmsg' => 'Duplicate account is not allowed !',
+            'instance' => $this->instance_id,
+        ]);
+    }
+
+    /**
+     * @param \Warlof\Seat\Connector\Drivers\IUser $speaker
+     *
+     * @throws \Warlof\Seat\Connector\Drivers\Teamspeak\Exceptions\CommandException
+     * @throws \Warlof\Seat\Connector\Drivers\Teamspeak\Exceptions\LoginException
+     */
+    public function deleteClientFromServerInstance(IUser $speaker)
+    {
+        $this->sendCall('POST', '/{instance}/clientdbdelete', [
+            'cldbid' => $speaker->getClientId(),
+            'instance' => $this->instance_id,
+        ]);
+    }
+
+    /**
+     * @param \Seat\Web\Models\User $user
+     * @param int $server_group
+     *
+     * @return string
+     * @throws \Warlof\Seat\Connector\Drivers\Teamspeak\Exceptions\CommandException
+     * @throws \Warlof\Seat\Connector\Drivers\Teamspeak\Exceptions\LoginException
+     */
+    public function tokenAdd(User $user, int $server_group): string
+    {
+        // scope: manage_scope
+        $response = $this->sendCall('POST', '/{instance}/tokenadd', [
+            'tokentype' => 0,
+            'tokenid1' => $server_group,
+            'tokenid2' => 0,
+            'tokendescription' => sprintf('Automatically created token from SeAT for %s', $user->name),
+            'tokencustomset' => sprintf('ident=seat_user_id value=%d', $user->id),
+            'instance' => $this->instance_id,
+        ]);
+
+        return Arr::first($response)->token;
+    }
+
+    /**
+     * @return mixed
+     * @throws \Warlof\Seat\Connector\Drivers\Teamspeak\Exceptions\CommandException
+     * @throws \Warlof\Seat\Connector\Drivers\Teamspeak\Exceptions\LoginException
+     */
+    private function serverInfo()
+    {
+        // scope: manage_scope
+        $response = $this->sendCall('GET', '/{instance}/serverinfo', [
+            'instance' => $this->instance_id,
+        ]);
+
+        return Arr::first($response);
+    }
+
     /**
      * @param string $method
      * @param string $endpoint
@@ -462,7 +611,7 @@ private function sendCall(string $method, string $endpoint, array $arguments = [
                     'response' => [
                         'body' => $response->getBody()->getContents(),
                     ],
-                ],
+                ]
             );
         } catch (ConnectException $e) {
             throw new ConnexionException($e->getMessage(), $e->getCode(), $e);
@@ -476,6 +625,9 @@ private function sendCall(string $method, string $endpoint, array $arguments = [
             if (in_array($result->status->code, [5122, 5124]))
                 throw new LoginException($result->status->message, $result->status->code);
 
+            if ($result->status->code == 1281)
+                throw new EmptyResultException($result->status->message, $result->status->code);
+
             throw new CommandException($result->status->message, $result->status->code);
         }
 
@@ -490,7 +642,7 @@ private function sendCall(string $method, string $endpoint, array $arguments = [
      */
     private function seedSpeakers()
     {
-        $from        = 0;
+        $from = 0;
 
         while (true) {
             try {
@@ -527,12 +679,7 @@ private function seedSpeakers()
      */
     private function seedServerGroups()
     {
-        // scope: manage_scope
-        $response = $this->sendCall('GET', '/{instance}/serverinfo', [
-            'instance' => $this->instance_id,
-        ]);
-
-        $server_info = Arr::first($response);
+        $server_info = $this->serverInfo();
 
         // scope: manage_scope
         $response = $this->sendCall('GET', '/{instance}/servergrouplist', [

diff --git a/src/Exceptions/EmptyResultException.php b/src/Exceptions/EmptyResultException.php
@@ -0,0 +1,31 @@
+<?php
+/*
+ * This file is part of SeAT Teamspeak Connector.
+ *
+ * Copyright (C) 2020  Warlof Tutsimo <[email protected]>
+ *
+ * SeAT Teamspeak Connector  is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * any later version.
+ *
+ * SeAT Teamspeak Connector is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <https://www.gnu.org/licenses/>.
+ */
+
+namespace Warlof\Seat\Connector\Drivers\Teamspeak\Exceptions;
+
+/**
+ * Class EmptyResultException.
+ *
+ * @package Warlof\Seat\Connector\Drivers\Teamspeak\Exceptions
+ */
+class EmptyResultException extends TeamspeakException
+{
+
+}
diff --git a/src/Exceptions/InvalidServerGroupException.php b/src/Exceptions/InvalidServerGroupException.php
@@ -0,0 +1,41 @@
+<?php
+/*
+ * This file is part of SeAT Teamspeak Connector.
+ *
+ * Copyright (C) 2020  Warlof Tutsimo <[email protected]>
+ *
+ * SeAT Teamspeak Connector  is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * any later version.
+ *
+ * SeAT Teamspeak Connector is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <https://www.gnu.org/licenses/>.
+ */
+
+namespace Warlof\Seat\Connector\Drivers\Teamspeak\Exceptions;
+
+use Throwable;
+
+/**
+ * Class InvalidServerGroupException.
+ *
+ * @package Warlof\Seat\Connector\Drivers\Teamspeak\Exceptions
+ */
+class InvalidServerGroupException extends TeamspeakException
+{
+    /**
+     * InvalidServerGroupException constructor.
+     *
+     * @param string $server_group
+     */
+    public function __construct(string $server_group)
+    {
+        parent::__construct(sprintf('Server Group %s is not found.', $server_group));
+    }
+}