import 'dart:async'; import 'dart:io'; import 'package:dio/io.dart'; import 'package:drift/drift.dart'; import 'package:get/get.dart' hide Value; import 'package:dio/dio.dart'; import 'package:rhythm_box/providers/database.dart'; import 'package:rhythm_box/services/database/database.dart'; extension ExpirationAuthenticationTableData on AuthenticationTableData { bool get isExpired => DateTime.now().isAfter(spotifyExpiration); String? getCookie(String key) => spotifyCookie.value .split('; ') .firstWhereOrNull((c) => c.trim().startsWith('$key=')) ?.trim() .split('=') .last .replaceAll(';', ''); String? getNeteaseCookie(String key) => neteaseCookie?.value .split(';') .firstWhereOrNull((c) => c.trim().startsWith('$key=')) ?.trim() .split('=') .last .replaceAll(';', ''); } class AuthenticationProvider extends GetxController { static final Dio dio = () { final dio = Dio(); (dio.httpClientAdapter as IOHttpClientAdapter) .createHttpClient = () => HttpClient() ..badCertificateCallback = (X509Certificate cert, String host, int port) { return host.endsWith('spotify.com') && port == 443; }; return dio; }(); var auth = Rxn(); Timer? refreshTimer; @override void onInit() { super.onInit(); loadAuthenticationData(); } Future loadAuthenticationData() async { final database = Get.find().database; final data = await (database.select(database.authenticationTable) ..where((s) => s.id.equals(0))) .getSingleOrNull(); auth.value = data; _setRefreshTimer(); } void _setRefreshTimer() { refreshTimer?.cancel(); if (auth.value != null && auth.value!.isExpired) { refreshSpotifyCredentials(); } refreshTimer = Timer( auth.value!.spotifyExpiration.difference(DateTime.now()), () => refreshSpotifyCredentials(), ); } Future refreshSpotifyCredentials() async { final database = Get.find().database; final refreshedCredentials = await credentialsFromCookie(auth.value!.spotifyCookie.value); await database .update(database.authenticationTable) .replace(refreshedCredentials); loadAuthenticationData(); // Reload data after refreshing } Future login(String cookie) async { final database = Get.find().database; final refreshedCredentials = await credentialsFromCookie(cookie); await database .into(database.authenticationTable) .insert(refreshedCredentials, mode: InsertMode.replace); loadAuthenticationData(); // Reload data after login } Future credentialsFromCookie( String cookie) async { try { final spDc = cookie .split('; ') .firstWhereOrNull((c) => c.trim().startsWith('sp_dc=')) ?.trim(); final res = await dio.getUri( Uri.parse( 'https://open.spotify.com/get_access_token?reason=transport&productType=web_player'), options: Options( headers: { 'Cookie': spDc ?? '', 'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/124.0.0.0 Safari/537.36', }, validateStatus: (status) => true, ), ); final body = res.data; if ((res.statusCode ?? 500) >= 400) { throw Exception( "Failed to get access token: ${body['error'] ?? res.statusMessage}"); } return AuthenticationTableCompanion.insert( id: const Value(0), spotifyCookie: DecryptedText("${res.headers["set-cookie"]?.join(";")}; $spDc"), spotifyAccessToken: DecryptedText(body['accessToken']), spotifyExpiration: DateTime.fromMillisecondsSinceEpoch( body['accessTokenExpirationTimestampMs']), ); } catch (e) { // Handle error rethrow; } } Future setNeteaseCredentials(String cookie) async { final database = Get.find().database; await database.update(database.authenticationTable).replace( AuthenticationTableCompanion.insert( id: const Value(0), spotifyCookie: auth.value!.spotifyCookie, spotifyAccessToken: auth.value!.spotifyAccessToken, spotifyExpiration: auth.value!.spotifyExpiration, neteaseCookie: Value(DecryptedText(cookie)), neteaseExpiration: const Value(null), ), ); await loadAuthenticationData(); } Future logout() async { auth.value = null; final database = Get.find().database; await (database.delete(database.authenticationTable) ..where((s) => s.id.equals(0))) .go(); // Additional cleanup if necessary } Future logoutNetease() async { final database = Get.find().database; await database.update(database.authenticationTable).replace( AuthenticationTableCompanion.insert( id: const Value(0), spotifyCookie: auth.value!.spotifyCookie, spotifyAccessToken: auth.value!.spotifyAccessToken, spotifyExpiration: auth.value!.spotifyExpiration, neteaseCookie: const Value(null), neteaseExpiration: const Value(null), ), ); await loadAuthenticationData(); } @override void onClose() { refreshTimer?.cancel(); super.onClose(); } }