From 6e4fe9d5885efd75891102be86b0e264852ff92d Mon Sep 17 00:00:00 2001
From: Hexah <hexer504@gmail.com>
Date: Sat, 6 Jun 2020 11:28:36 +0200
Subject: [PATCH] Add SearchClient, Closes #32

---
 lib/src/channels/channel_client.dart |  4 +--
 lib/src/search/search_client.dart    | 49 ++++++++++++++++++++++++++++
 lib/src/youtube_explode_base.dart    |  6 +++-
 lib/youtube_explode_dart.dart        |  1 +
 test/search_test.dart                | 23 ++++++++++++-
 5 files changed, 79 insertions(+), 4 deletions(-)
 create mode 100644 lib/src/search/search_client.dart

diff --git a/lib/src/channels/channel_client.dart b/lib/src/channels/channel_client.dart
index 8364c87..743187e 100644
--- a/lib/src/channels/channel_client.dart
+++ b/lib/src/channels/channel_client.dart
@@ -42,9 +42,9 @@ class ChannelClient {
     videoId = VideoId.fromString(videoId);
     var videoInfoResponse =
         await VideoInfoResponse.get(_httpClient, videoId.value);
-    var playerReponse = videoInfoResponse.playerResponse;
+    var playerResponse = videoInfoResponse.playerResponse;
 
-    var channelId = playerReponse.videoChannelId;
+    var channelId = playerResponse.videoChannelId;
     return await get(ChannelId(channelId));
   }
 
diff --git a/lib/src/search/search_client.dart b/lib/src/search/search_client.dart
new file mode 100644
index 0000000..8ece938
--- /dev/null
+++ b/lib/src/search/search_client.dart
@@ -0,0 +1,49 @@
+import '../common/common.dart';
+import '../reverse_engineering/responses/playerlist_response.dart';
+import '../reverse_engineering/youtube_http_client.dart';
+import '../videos/video.dart';
+import '../videos/video_id.dart';
+
+/// YouTube search queries.
+class SearchClient {
+  final YoutubeHttpClient _httpClient;
+
+  /// Initializes an instance of [SearchClient]
+  SearchClient(this._httpClient);
+
+  /// Enumerates videos returned by the specified search query.
+  Stream<Video> getVideosAsync(String searchQuery) async* {
+    var encounteredVideoIds = <String>{};
+
+    for (var page = 0; page < double.maxFinite; page++) {
+      var response =
+          await PlaylistResponse.searchResults(_httpClient, searchQuery);
+
+      var countDelta = 0;
+      for (var video in response.videos) {
+        var videoId = video.id;
+
+        if (!encounteredVideoIds.add(videoId)) {
+          continue;
+        }
+
+        yield Video(
+            VideoId(videoId),
+            video.title,
+            video.author,
+            video.uploadDate,
+            video.description,
+            video.duration,
+            ThumbnailSet(videoId),
+            video.keywords,
+            Engagement(video.viewCount, video.likes, video.dislikes));
+        countDelta++;
+      }
+
+      // Videos loop around, so break when we stop seeing new videos
+      if (countDelta <= 0) {
+        break;
+      }
+    }
+  }
+}
diff --git a/lib/src/youtube_explode_base.dart b/lib/src/youtube_explode_base.dart
index 7de3ca2..d711671 100644
--- a/lib/src/youtube_explode_base.dart
+++ b/lib/src/youtube_explode_base.dart
@@ -3,6 +3,7 @@ library youtube_explode.base;
 import 'channels/channels.dart';
 import 'playlists/playlist_client.dart';
 import 'reverse_engineering/youtube_http_client.dart';
+import 'search/search_client.dart';
 import 'videos/video_client.dart';
 
 /// Library entry point.
@@ -19,17 +20,20 @@ class YoutubeExplode {
   ChannelClient get channels => _channels;
 
   /// YouTube search queries.
-//TODO: Add SearchClient
+  SearchClient get search => _search;
 
+  /// Initializes an instance of [YoutubeClient].
   YoutubeExplode() : _httpClient = YoutubeHttpClient() {
     _videos = VideoClient(_httpClient);
     _playlists = PlaylistClient(_httpClient);
     _channels = ChannelClient(_httpClient);
+    _search = SearchClient(_httpClient);
   }
 
   VideoClient _videos;
   PlaylistClient _playlists;
   ChannelClient _channels;
+  SearchClient _search;
 
   /// Closes the HttpClient assigned to this [YoutubeHttpClient].
   /// Should be called after this is not used anymore.
diff --git a/lib/youtube_explode_dart.dart b/lib/youtube_explode_dart.dart
index 6e355ff..bb1671a 100644
--- a/lib/youtube_explode_dart.dart
+++ b/lib/youtube_explode_dart.dart
@@ -4,5 +4,6 @@ export 'src/channels/channels.dart';
 export 'src/common/common.dart';
 export 'src/exceptions/exceptions.dart';
 export 'src/playlists/playlists.dart';
+export 'src/search/search_client.dart';
 export 'src/videos/videos.dart';
 export 'src/youtube_explode_base.dart';
diff --git a/test/search_test.dart b/test/search_test.dart
index c532bde..8a33daf 100644
--- a/test/search_test.dart
+++ b/test/search_test.dart
@@ -1 +1,22 @@
-//TODO: Implement this
+import 'package:test/test.dart';
+import 'package:youtube_explode_dart/youtube_explode_dart.dart';
+
+void main() {
+  group('Search', () {
+    YoutubeExplode yt;
+    setUp(() {
+      yt = YoutubeExplode();
+    });
+
+    tearDown(() {
+      yt.close();
+    });
+
+    test('SearchYouTubeVideos', () async {
+      var videos = await yt.search
+          .getVideosAsync('undead corporation megalomania')
+          .toList();
+      expect(videos, isNotEmpty);
+    });
+  });
+}