From 2d3c12f7d0e58a42056365f783c45c3290dfaba9 Mon Sep 17 00:00:00 2001
From: office-pc nix root <paul@fauchon.dev>
Date: Sat, 14 Sep 2024 12:39:52 +0800
Subject: [PATCH] fix login with cookie stop displaying the encoded auth

---
 lib/service.dart                              | 67 ++++++++++---------
 lib/settings/states/server_settings.dart      |  7 +-
 .../states/server_settings.freezed.dart       |  8 ++-
 .../views/screens/manage_single_server.dart   | 29 +++++---
 4 files changed, 63 insertions(+), 48 deletions(-)

diff --git a/lib/service.dart b/lib/service.dart
index c79442a8..88d888ab 100644
--- a/lib/service.dart
+++ b/lib/service.dart
@@ -115,7 +115,7 @@ class Service {
     return db.getSettings(useProxySettingName)?.value == 'true';
   }
 
-  Future<ServerRequest> buildUrl(
+  Future<ServerRequest> buildRequest(
     String baseUrl, {
     Map<String, String>? pathParams,
     Map<String, String?>? query,
@@ -163,7 +163,7 @@ class Service {
       if (forceJson) {
         headers['Content-Type'] = 'application/json';
       }
-
+      print(headers);
       return ServerRequest(uri: uri, headers: headers);
     } catch (err) {
       log.severe('Couldn\'t build url', err);
@@ -174,10 +174,9 @@ class Service {
   handleErrors(Response response) {}
 
   Future<Video> getVideo(String videoId) async {
-    var req = await buildUrl(urlGetVideo, pathParams: {':id': videoId});
-    req.headers['Content-Type'] = 'application/json; charset=utf-16';
-    final response = await http.get(req.uri,
-        headers: {'Content-Type': 'application/json; charset=utf-16'});
+    var req = await buildRequest(urlGetVideo,
+        pathParams: {':id': videoId}, utf16: true);
+    final response = await http.get(req.uri, headers: req.headers);
 
     var video = Video.fromJson(handleResponse(response));
     await DeArrow.processVideos([video]);
@@ -187,12 +186,14 @@ class Service {
   }
 
   Future<String> loginWithCookies(
-      String serverUrl, String username, String password) async {
+      String serverUrl, String username, String password,
+      {Map<String, String>? headers}) async {
     try {
       String url = '$serverUrl/login?type=invidious';
       var map = {'email': username, 'password': password};
 
-      final response = await http.post(Uri.parse(url), body: map);
+      final response =
+          await http.post(Uri.parse(url), body: map, headers: headers);
       if (response.statusCode == 302 &&
           response.headers.containsKey('set-cookie')) {
         // we have a cookie to parse
@@ -258,7 +259,7 @@ class Service {
       query.putIfAbsent('type', () => type);
     }
 
-    var req = await buildUrl(urlGetTrending, query: query);
+    var req = await buildRequest(urlGetTrending, query: query);
     final response = await http.get(req.uri, headers: req.headers);
 
     Iterable i = handleResponse(response);
@@ -268,7 +269,7 @@ class Service {
   }
 
   Future<List<VideoInList>> getPopular() async {
-    var req = await buildUrl(urlGetPopular);
+    var req = await buildRequest(urlGetPopular);
     final response = await http.get(req.uri, headers: req.headers);
     Iterable i = handleResponse(response);
     var list = List<VideoInList>.from(i.map((e) => VideoInList.fromJson(e)));
@@ -283,7 +284,7 @@ class Service {
       SearchDate date = SearchDate.any,
       SearchDuration duration = SearchDuration.any}) async {
     String countryCode = db.getSettings(browsingCountry)?.value ?? 'US';
-    final req = await buildUrl(urlSearch, query: {
+    final req = await buildRequest(urlSearch, query: {
       'q': Uri.encodeQueryComponent(query),
       'type': type?.name,
       'page': page?.toString() ?? '1',
@@ -322,7 +323,7 @@ class Service {
   Future<UserFeed> getUserFeed(
       {int? maxResults, int? page, bool saveLastSeen = true}) async {
     // for background service to be able to use
-    final req = await buildUrl(urlGetUserFeed,
+    final req = await buildRequest(urlGetUserFeed,
         query: {
           'max_results': maxResults?.toString(),
           'page': page?.toString()
@@ -396,7 +397,7 @@ class Service {
 
   Future<SearchSuggestion> getSearchSuggestion(String query) async {
     if (query.isEmpty) return SearchSuggestion(query, []);
-    var request = await buildUrl(urlSearchSuggestions,
+    var request = await buildRequest(urlSearchSuggestions,
         query: {"q": Uri.encodeQueryComponent(query)});
     final response = await http.get(request.uri, headers: request.headers);
     SearchSuggestion search =
@@ -443,7 +444,7 @@ class Service {
 
   Future<bool> subscribe(String channelId) async {
     if (!await isLoggedIn()) return false;
-    final req = await buildUrl(urlAddDeleteSubscriptions,
+    final req = await buildRequest(urlAddDeleteSubscriptions,
         pathParams: {":ucid": channelId}, authenticated: true);
 
     final response = await http.post(req.uri, headers: req.headers);
@@ -459,7 +460,7 @@ class Service {
   Future<bool> unSubscribe(String channelId) async {
     if (!await isLoggedIn()) return false;
 
-    final req = await buildUrl(urlAddDeleteSubscriptions,
+    final req = await buildRequest(urlAddDeleteSubscriptions,
         pathParams: {":ucid": channelId}, authenticated: true);
 
     final response = await http.delete(req.uri, headers: req.headers);
@@ -483,7 +484,7 @@ class Service {
   Future<List<Subscription>> getSubscriptions() async {
     if (!await isLoggedIn()) return [];
 
-    var req = await buildUrl(urlGetSubscriptions, authenticated: true);
+    var req = await buildRequest(urlGetSubscriptions, authenticated: true);
 
     final response = await http.get(req.uri, headers: req.headers);
     Iterable i = handleResponse(response);
@@ -500,7 +501,7 @@ class Service {
     if (sortBy != null) queryStr.putIfAbsent('sort_by', () => sortBy);
     if (source != null) queryStr.putIfAbsent('source', () => source);
 
-    var req = await buildUrl(urlGetComments,
+    var req = await buildRequest(urlGetComments,
         pathParams: {':id': videoId}, query: queryStr);
     final response = await http.get(req.uri, headers: req.headers);
     return VideoComments.fromJson(handleResponse(response));
@@ -509,7 +510,7 @@ class Service {
   Future<Channel> getChannel(String channelId) async {
     // sometimes the api gives the channel with /channel/<channelid> format
     channelId = channelId.replaceAll("/channel/", '');
-    var req = await buildUrl(urlGetChannel,
+    var req = await buildRequest(urlGetChannel,
         pathParams: {':id': channelId}, utf16: true);
 
     final response = await http.get(req.uri, headers: req.headers);
@@ -529,7 +530,7 @@ class Service {
       String channelId, String? continuation,
       {bool saveLastSeen = true,
       ChannelSortBy sortBy = ChannelSortBy.newest}) async {
-    final req = await buildUrl(urlGetChannelVideos,
+    final req = await buildRequest(urlGetChannelVideos,
         pathParams: {':id': channelId},
         query: {
           'continuation': continuation,
@@ -552,7 +553,7 @@ class Service {
 
   Future<VideosWithContinuation> getChannelStreams(
       String channelId, String? continuation) async {
-    final req = await buildUrl(urlGetChannelStreams,
+    final req = await buildRequest(urlGetChannelStreams,
         pathParams: {':id': channelId},
         query: {'continuation': continuation},
         utf16: true);
@@ -567,7 +568,7 @@ class Service {
 
   Future<VideosWithContinuation> getChannelShorts(
       String channelId, String? continuation) async {
-    final req = await buildUrl(urlGetChannelShorts,
+    final req = await buildRequest(urlGetChannelShorts,
         pathParams: {':id': channelId},
         query: {'continuation': continuation},
         utf16: true);
@@ -582,7 +583,7 @@ class Service {
 
   Future<List<Playlist>> getUserPlaylists({bool postProcessing = true}) async {
     try {
-      var req = await buildUrl(urlGetUserPlaylists, authenticated: true);
+      var req = await buildRequest(urlGetUserPlaylists, authenticated: true);
 
       final response = await http.get(req.uri, headers: req.headers);
       Iterable i = handleResponse(response);
@@ -600,7 +601,7 @@ class Service {
 
   Future<ChannelPlaylists> getChannelPlaylists(String channelId,
       {String? continuation}) async {
-    final req = await buildUrl(urlGetChannelPlaylists,
+    final req = await buildRequest(urlGetChannelPlaylists,
         pathParams: {':id': channelId}, query: {'continuation': continuation});
 
     final response = await http.get(req.uri, headers: req.headers);
@@ -612,7 +613,7 @@ class Service {
   }
 
   Future<String?> createPlayList(String name, String type) async {
-    var req = await buildUrl(urlPostUserPlaylists,
+    var req = await buildRequest(urlPostUserPlaylists,
         authenticated: true, forceJson: true);
 
     Map<String, String> body = {
@@ -629,7 +630,7 @@ class Service {
   }
 
   Future<void> addVideoToPlaylist(String playListId, String videoId) async {
-    var req = await buildUrl(urlPostUserPlaylistVideo,
+    var req = await buildRequest(urlPostUserPlaylistVideo,
         pathParams: {":id": playListId}, authenticated: true, forceJson: true);
 
     Map<String, String> body = {
@@ -642,7 +643,7 @@ class Service {
   }
 
   Future<void> deleteUserPlaylist(String playListId) async {
-    var req = await buildUrl(urlDeleteUserPlaylist,
+    var req = await buildRequest(urlDeleteUserPlaylist,
         pathParams: {":id": playListId}, authenticated: true, forceJson: true);
 
     final response = await http.delete(req.uri, headers: req.headers);
@@ -651,7 +652,7 @@ class Service {
 
   Future<void> deleteUserPlaylistVideo(
       String playListId, String indexId) async {
-    final req = await buildUrl(urlDeleteUserPlaylistVideo,
+    final req = await buildRequest(urlDeleteUserPlaylistVideo,
         pathParams: {':id': playListId, ':index': indexId},
         authenticated: true,
         forceJson: true);
@@ -661,7 +662,7 @@ class Service {
   }
 
   Future<List<String>> getUserHistory(int page, int maxResults) async {
-    final req = await buildUrl(urlGetClearHistory,
+    final req = await buildRequest(urlGetClearHistory,
         query: {'page': page.toString(), 'max_results': maxResults.toString()},
         authenticated: true,
         forceJson: true);
@@ -688,7 +689,7 @@ class Service {
   }
 
   Future<void> clearUserHistory() async {
-    var req = await buildUrl(urlGetClearHistory,
+    var req = await buildRequest(urlGetClearHistory,
         authenticated: true, forceJson: true);
 
     final response = await http.delete(req.uri, headers: req.headers);
@@ -696,14 +697,14 @@ class Service {
   }
 
   Future<void> deleteFromUserHistory(String videoId) async {
-    var req = await buildUrl(urlAddDeleteHistory,
+    var req = await buildRequest(urlAddDeleteHistory,
         pathParams: {':id': videoId}, authenticated: true, forceJson: true);
     final response = await http.delete(req.uri, headers: req.headers);
     handleResponse(response);
   }
 
   Future<void> addToUserHistory(String videoId) async {
-    var req = await buildUrl(urlAddDeleteHistory,
+    var req = await buildRequest(urlAddDeleteHistory,
         pathParams: {':id': videoId}, authenticated: true, forceJson: true);
 
     final response = await http.post(req.uri, headers: req.headers);
@@ -747,7 +748,7 @@ class Service {
 
   Future<Playlist> getPublicPlaylists(String playlistId,
       {int? page, bool saveLastSeen = true}) async {
-    final req = await buildUrl(urlGetPublicPlaylist,
+    final req = await buildRequest(urlGetPublicPlaylist,
         pathParams: {':id': playlistId}, query: {'page': page?.toString()});
 
     final response = await http.get(req.uri, headers: req.headers);
@@ -765,7 +766,7 @@ class Service {
   }
 
   Future<Playlist> getUserPlaylist(String playlistId) async {
-    final req = await buildUrl(urlGetUserPlaylist,
+    final req = await buildRequest(urlGetUserPlaylist,
         pathParams: {':id': playlistId}, authenticated: true);
 
     final response = await http.get(req.uri, headers: req.headers);
diff --git a/lib/settings/states/server_settings.dart b/lib/settings/states/server_settings.dart
index b624cab9..81ae8607 100644
--- a/lib/settings/states/server_settings.dart
+++ b/lib/settings/states/server_settings.dart
@@ -64,7 +64,8 @@ class ServerSettingsCubit extends Cubit<ServerSettingsState> {
 
   Future<void> logInWithCookie(String username, String password) async {
     var s = state.server.copyWith();
-    String cookie = await service.loginWithCookies(s.url, username, password);
+    String cookie = await service.loginWithCookies(s.url, username, password,
+        headers: s.customHeaders);
     s = s.copyWith(sidCookie: cookie);
     await db.upsertServer(s);
     if (s.inUse) {
@@ -106,4 +107,8 @@ class ServerSettingsState with _$ServerSettingsState {
   const factory ServerSettingsState(
       {required Server server,
       @Default(false) bool canDelete}) = _ServerSettingsState;
+
+  const ServerSettingsState._();
+
+  bool get hasBasicAuth => server.customHeaders.containsKey("Authorization");
 }
diff --git a/lib/settings/states/server_settings.freezed.dart b/lib/settings/states/server_settings.freezed.dart
index 78d3c1c8..1484903f 100644
--- a/lib/settings/states/server_settings.freezed.dart
+++ b/lib/settings/states/server_settings.freezed.dart
@@ -115,9 +115,10 @@ class __$$ServerSettingsStateImplCopyWithImpl<$Res>
 
 /// @nodoc
 
-class _$ServerSettingsStateImpl implements _ServerSettingsState {
+class _$ServerSettingsStateImpl extends _ServerSettingsState {
   const _$ServerSettingsStateImpl(
-      {required this.server, this.canDelete = false});
+      {required this.server, this.canDelete = false})
+      : super._();
 
   @override
   final Server server;
@@ -151,10 +152,11 @@ class _$ServerSettingsStateImpl implements _ServerSettingsState {
           this, _$identity);
 }
 
-abstract class _ServerSettingsState implements ServerSettingsState {
+abstract class _ServerSettingsState extends ServerSettingsState {
   const factory _ServerSettingsState(
       {required final Server server,
       final bool canDelete}) = _$ServerSettingsStateImpl;
+  const _ServerSettingsState._() : super._();
 
   @override
   Server get server;
diff --git a/lib/settings/views/screens/manage_single_server.dart b/lib/settings/views/screens/manage_single_server.dart
index 0e25c5c0..4663bcdb 100644
--- a/lib/settings/views/screens/manage_single_server.dart
+++ b/lib/settings/views/screens/manage_single_server.dart
@@ -181,7 +181,7 @@ class ManageSingleServerScreen extends StatelessWidget {
                               leading: server.authToken?.isNotEmpty ?? false
                                   ? const Icon(Icons.check)
                                   : const Icon(Icons.token),
-                              enabled: !isLoggedIn,
+                              enabled: !isLoggedIn && !state.hasBasicAuth,
                               title: Text(locals.tokenLogin),
                               value: Text(server.authToken?.isNotEmpty ?? false
                                   ? locals.loggedIn
@@ -222,16 +222,23 @@ class ManageSingleServerScreen extends StatelessWidget {
                             ],
                           ),
                           tiles: [
-                            ...state.server.customHeaders.keys
-                                .map((k) => SettingsTile(
-                                      title: Text(k),
-                                      description: Text(
-                                          state.server.customHeaders[k] ?? ''),
-                                      trailing: IconButton(
-                                        icon: const Icon(Icons.delete),
-                                        onPressed: () => cubit.removeHeader(k),
-                                      ),
-                                    )),
+                            ...state.server.customHeaders.keys.map((k) {
+                              String value =
+                                  state.server.customHeaders[k] ?? '';
+                              if (k == 'Authorization') {
+                                value = value.replaceAll(
+                                    value.replaceAll("Basic ", ""), "········");
+                              }
+
+                              return SettingsTile(
+                                title: Text(k),
+                                description: Text(value),
+                                trailing: IconButton(
+                                  icon: const Icon(Icons.delete),
+                                  onPressed: () => cubit.removeHeader(k),
+                                ),
+                              );
+                            }),
                             SettingsTile(
                               title: Text(locals.addBasicAuth),
                               leading: const Icon(Icons.key),