Compare commits

..

4 Commits

Author SHA1 Message Date
Kingkor Roy Tirtho
2b9c5730c9 feat: add yt engine plugin support 2025-12-07 19:58:05 +06:00
Kingkor Roy Tirtho
4129a61d85 feat: add timezone and encrypt cookies endpoint 2025-12-07 19:17:59 +06:00
Kingkor Roy Tirtho
949519aa61 feat: add localStorage API support 2025-12-07 18:36:42 +06:00
Kingkor Roy Tirtho
6da7fb7ac3 feat: add form api 2025-12-07 17:00:27 +06:00
20 changed files with 672 additions and 255 deletions

View File

@ -45,8 +45,6 @@ import 'package:spotube/services/kv_store/encrypted_kv_store.dart';
import 'package:spotube/services/kv_store/kv_store.dart';
import 'package:spotube/services/logger/logger.dart';
import 'package:spotube/services/wm_tools/wm_tools.dart';
import 'package:spotube/src/rust/api/plugin/models/core.dart';
import 'package:spotube/src/rust/api/plugin/plugin.dart';
import 'package:spotube/src/rust/frb_generated.dart';
import 'package:spotube/utils/migrations/sandbox.dart';
import 'package:spotube/utils/platform.dart';
@ -178,58 +176,6 @@ class Spotube extends HookConsumerWidget {
HomeWidget.registerInteractivityCallback(glanceBackgroundCallback);
}
start() async {
final server = await ref.read(serverProvider.future);
final plugin = SpotubePlugin();
const pluginConfiguration = PluginConfiguration(
name: "Spotube Plugin",
description: "Spotube Plugin",
version: "1.0.0",
author: "Spotube",
entryPoint: "Plugin",
pluginApiVersion: "2.0.0",
apis: [PluginApi.localstorage, PluginApi.webview],
abilities: [PluginAbility.metadata],
);
final pluginContext = plugin.createContext(
serverEndpointUrl:
"http://${server.server.address.host}:${server.port}",
serverSecret: ref.read(serverRandomSecretProvider),
pluginScript: """
class AuthEndpoint {
}
class CoreEndpoint {
async checkUpdate() {
console.log(globalThis);
const webview = await WebView.create("https://spotube.krtirtho.dev");
webview.onUrlChange((url) => {
console.log("url_request: ", url);
if (url.includes("/about")) {
webview.close();
}
});
await webview.open();
await new Promise((resolve) => setTimeout(resolve, 5000));
}
}
class Plugin {
constructor() {
this.auth = new AuthEndpoint();
this.core = new CoreEndpoint();
}
}
""",
pluginConfig: pluginConfiguration,
);
await plugin.core.checkUpdate(
mpscTx: pluginContext, pluginConfig: pluginConfiguration);
}
start();
return () {
/// For enabling hot reload for audio player
if (!kDebugMode) return;

View File

@ -3,7 +3,10 @@ import 'package:shelf/shelf.dart';
import 'package:shelf_router/shelf_router.dart';
import 'package:spotube/provider/server/routes/connect.dart';
import 'package:spotube/provider/server/routes/playback.dart';
import 'package:spotube/provider/server/routes/plugin_apis/form.dart';
import 'package:spotube/provider/server/routes/plugin_apis/path_provider.dart';
import 'package:spotube/provider/server/routes/plugin_apis/webview.dart';
import 'package:spotube/provider/server/routes/plugin_apis/yt_engine.dart';
Handler pluginApiAuthMiddleware(Handler handler) {
return (Request request) {
@ -20,6 +23,8 @@ final serverRouterProvider = Provider((ref) {
final playbackRoutes = ref.watch(serverPlaybackRoutesProvider);
final connectRoutes = ref.watch(serverConnectRoutesProvider);
final webviewRoutes = ref.watch(serverWebviewRoutesProvider);
final formRoutes = ref.watch(serverFormRoutesProvider);
final ytEngineRoutes = ref.watch(serverYTEngineRoutesProvider);
final router = Router();
@ -52,6 +57,27 @@ final serverRouterProvider = Provider((ref) {
"/plugin-api/webview/cookies",
pluginApiAuthMiddleware(webviewRoutes.postGetWebviewCookies),
);
router.post(
"/plugin-api/form/show",
pluginApiAuthMiddleware(formRoutes.showForm),
);
router.get(
"/plugin/localstorage/directories",
pluginApiAuthMiddleware(ServerPathProviderRoutes.getDirectories),
);
router.get(
"/plugin-api/yt-engine/search",
pluginApiAuthMiddleware(ytEngineRoutes.search),
);
router.get(
"/plugin-api/yt-engine/video",
pluginApiAuthMiddleware(ytEngineRoutes.getVideo),
);
router.get(
"/plugin-api/yt-engine/stream-manifest",
pluginApiAuthMiddleware(ytEngineRoutes.streamManifest),
);
router.all("/ws", connectRoutes.websocket);

View File

@ -0,0 +1,30 @@
import 'dart:convert';
import 'package:auto_route/auto_route.dart';
import 'package:riverpod/riverpod.dart';
import 'package:shelf/shelf.dart';
import 'package:spotube/collections/routes.dart';
import 'package:spotube/collections/routes.gr.dart';
import 'package:spotube/models/metadata/metadata.dart';
class ServerFormRoutes {
Future<Response> showForm(Request request) async {
final body = jsonDecode(await request.readAsString());
final res = await rootNavigatorKey.currentContext!.router
.push<List<Map<String, dynamic>>?>(
SettingsMetadataProviderFormRoute(
title: body["title"],
fields: (body["fields"] as List)
.map((e) => MetadataFormFieldObject.fromJson(e))
.toList(),
),
);
return Response.ok(
jsonEncode(res),
headers: {'Content-Type': 'application/json'},
);
}
}
final serverFormRoutesProvider = Provider((ref) => ServerFormRoutes());

View File

@ -0,0 +1,29 @@
import 'dart:convert';
import 'dart:io';
import 'package:path_provider/path_provider.dart' as pp;
import 'package:shelf/shelf.dart';
class ServerPathProviderRoutes {
static Future<Response> getDirectories(Request request) async {
final directories = <String, Directory?>{
'temporary': await Future<Directory?>.value(pp.getTemporaryDirectory())
.catchError((e) => null),
'applicationDocuments':
await Future<Directory?>.value(pp.getApplicationDocumentsDirectory())
.catchError((e) => null),
'applicationSupport':
await Future<Directory?>.value(pp.getApplicationSupportDirectory())
.catchError((e) => null),
'library': await Future<Directory?>.value(pp.getLibraryDirectory())
.catchError((e) => null),
'externalStorage':
await pp.getExternalStorageDirectory().catchError((e) => null),
'downloads': await pp.getDownloadsDirectory().catchError((e) => null),
}.map((key, value) => MapEntry(key, value?.path));
return Response.ok(
jsonEncode(directories),
headers: {'Content-Type': 'application/json'},
);
}
}

View File

@ -1,16 +1,38 @@
import 'dart:async';
import 'dart:convert';
import 'dart:math';
import 'dart:typed_data';
import 'package:hooks_riverpod/hooks_riverpod.dart';
import 'package:shadcn_flutter/shadcn_flutter.dart';
import 'package:shelf/shelf.dart';
import 'package:shelf_router/shelf_router.dart';
import 'package:spotube/provider/server/server.dart';
import 'package:spotube/src/plugin_api/webview/webview.dart';
import 'package:async/async.dart';
import 'package:encrypt/encrypt.dart' as encrypt;
class ServerWebviewRoutes {
final Ref ref;
ServerWebviewRoutes({required this.ref});
final Map<String, Webview> _webviews = {};
String _encryptCookies(dynamic cookies, String secret) {
final keyBytes = base64.decode(secret);
final key = encrypt.Key(keyBytes);
final ivBytes = List<int>.generate(16, (_) => Random.secure().nextInt(256));
final iv = encrypt.IV(Uint8List.fromList(ivBytes));
final encrypter = encrypt.Encrypter(
encrypt.AES(key, mode: encrypt.AESMode.cbc, padding: 'PKCS7'),
);
final encrypted = encrypter.encrypt(jsonEncode(cookies), iv: iv);
final combined = iv.bytes + encrypted.bytes;
return base64.encode(combined);
}
Future<Response> postCreateWebview(Request request) async {
final payload = jsonDecode(await request.readAsString());
final uri = Uri.parse(payload['url'] as String);
@ -105,6 +127,8 @@ class ServerWebviewRoutes {
}
Future<Response> postGetWebviewCookies(Request request) async {
final secret = ref.read(serverRandomSecretProvider);
final body = jsonDecode(await request.readAsString());
final uid = body['uid'] as String;
final url = body['url'] as String;
@ -114,8 +138,9 @@ class ServerWebviewRoutes {
return Response.notFound('Webview with uid $uid not found');
}
final cookies = await webview.getCookies(url);
final encryptedCookies = _encryptCookies(cookies, secret);
return Response.ok(
jsonEncode(cookies),
jsonEncode({'data': encryptedCookies}),
encoding: utf8,
headers: {
'Content-Type': 'application/json',
@ -131,4 +156,5 @@ class ServerWebviewRoutes {
}
}
final serverWebviewRoutesProvider = Provider((ref) => ServerWebviewRoutes());
final serverWebviewRoutesProvider =
Provider((ref) => ServerWebviewRoutes(ref: ref));

View File

@ -0,0 +1,114 @@
import 'dart:convert';
import 'package:hooks_riverpod/hooks_riverpod.dart';
import 'package:shelf/shelf.dart';
import 'package:spotube/provider/youtube_engine/youtube_engine.dart';
import 'package:spotube/services/youtube_engine/youtube_engine.dart';
class ServerYTEngineRoutes {
final Ref ref;
ServerYTEngineRoutes({required this.ref});
YouTubeEngine get youtubeEngine => ref.read(youtubeEngineProvider);
Future<Response> search(Request request) async {
final query = request.url.queryParameters["query"];
if (query == null || query.isEmpty) {
return Response.badRequest(
body: 'Query parameter "query" is required',
);
}
final result = await youtubeEngine.searchVideos(query);
final mappedResult = result
.map((video) => {
'id': video.id.value,
'title': video.title,
'author': video.author,
'duration': video.duration?.inSeconds,
'description': video.description,
'uploadDate': video.uploadDate?.toIso8601String(),
'viewCount': video.engagement.viewCount,
'likeCount': video.engagement.likeCount,
'isLive': video.isLive,
})
.toList();
return Response.ok(
jsonEncode(mappedResult),
encoding: utf8,
headers: {
'Content-Type': 'application/json',
},
);
}
Future<Response> getVideo(Request request) async {
final videoId = request.url.queryParameters["videoId"];
if (videoId == null || videoId.isEmpty) {
return Response.badRequest(
body: 'Query parameter "videoId" is required',
);
}
final video = await youtubeEngine.getVideo(videoId);
final res = {
'id': video.id.value,
'title': video.title,
'author': video.author,
'duration': video.duration?.inSeconds,
'description': video.description,
'uploadDate': video.uploadDate?.toIso8601String(),
'viewCount': video.engagement.viewCount,
'likeCount': video.engagement.likeCount,
'isLive': video.isLive,
};
return Response.ok(
jsonEncode(res),
encoding: utf8,
headers: {
'Content-Type': 'application/json',
},
);
}
Future<Response> streamManifest(Request request) async {
final videoId = request.url.queryParameters["videoId"];
if (videoId == null || videoId.isEmpty) {
return Response.badRequest(
body: 'Query parameter "videoId" is required',
);
}
final streamManifest =
await youtubeEngine.getStreamManifest(videoId).then((manifest) {
final streams = manifest.audioOnly
.map(
(stream) => {
'url': stream.url.toString(),
'quality': stream.qualityLabel,
'bitrate': stream.bitrate.bitsPerSecond,
'container': stream.container.name,
'videoId': stream.videoId,
},
)
.toList();
return streams;
});
return Response.ok(
jsonEncode(streamManifest),
encoding: utf8,
headers: {
'Content-Type': 'application/json',
},
);
}
}
final serverYTEngineRoutesProvider = Provider<ServerYTEngineRoutes>(
(ref) => ServerYTEngineRoutes(ref: ref),
);

View File

@ -15,7 +15,7 @@ final serverRandomSecretProvider = Provider<String>(
(ref) {
final random = Random.secure();
final values = List<int>.generate(16, (i) => random.nextInt(256));
return base64Url.encode(values);
return base64.encode(values);
},
);
final serverProvider = FutureProvider(

View File

@ -2,10 +2,11 @@ import 'package:flutter_riverpod/flutter_riverpod.dart';
import 'package:spotube/models/database/database.dart';
import 'package:spotube/provider/user_preferences/user_preferences_provider.dart';
import 'package:spotube/services/youtube_engine/newpipe_engine.dart';
import 'package:spotube/services/youtube_engine/youtube_engine.dart';
import 'package:spotube/services/youtube_engine/youtube_explode_engine.dart';
import 'package:spotube/services/youtube_engine/yt_dlp_engine.dart';
final youtubeEngineProvider = Provider((ref) {
final youtubeEngineProvider = Provider<YouTubeEngine>((ref) {
final engineMode = ref.watch(
userPreferencesProvider.select((value) => value.youtubeClientEngine),
);

View File

@ -82,7 +82,7 @@ packages:
source: hosted
version: "1.6.5"
async:
dependency: transitive
dependency: "direct main"
description:
name: async
sha256: "758e6d74e971c3e5aceb4110bfd6698efc7f501675bcfe0c775459a8140750eb"

View File

@ -162,6 +162,7 @@ dependencies:
flutter_rust_bridge: 2.11.1
json_annotation: ^4.9.0
random_user_agents: ^1.0.18
async: ^2.13.0
dev_dependencies:
build_runner: ^2.4.13

128
rust/Cargo.lock generated
View File

@ -364,6 +364,16 @@ dependencies = [
"windows-link 0.2.1",
]
[[package]]
name = "chrono-tz"
version = "0.10.4"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "a6139a8597ed92cf816dfb33f5dd6cf0bb93a6adc938f11039f371bc5bcd26c3"
dependencies = [
"chrono",
"phf 0.12.1",
]
[[package]]
name = "cipher"
version = "0.5.0-rc.2"
@ -404,6 +414,19 @@ dependencies = [
"crossbeam-utils",
]
[[package]]
name = "confy"
version = "2.0.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "8807c397789cbe02bbdb1a27ea5f345584132808697b2a3f957c829829ee4814"
dependencies = [
"etcetera",
"lazy_static",
"serde",
"thiserror",
"toml",
]
[[package]]
name = "console_error_panic_hook"
version = "0.1.7"
@ -721,6 +744,17 @@ dependencies = [
"windows-sys 0.61.2",
]
[[package]]
name = "etcetera"
version = "0.10.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "26c7b13d0780cb82722fd59f6f57f925e143427e4a75313a6c77243bf5326ae6"
dependencies = [
"cfg-if",
"home",
"windows-sys 0.59.0",
]
[[package]]
name = "event-listener"
version = "5.4.0"
@ -1558,9 +1592,9 @@ dependencies = [
[[package]]
name = "lazy_static"
version = "1.4.0"
version = "1.5.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "e2abad23fbc42b3700f2f279844dc832adb2b2eb069b2df918f455c4e18cc646"
checksum = "bbd2bcb4c963f2ddae06a2efc7e9f3591312473c50c6685e1f298068316e66fe"
[[package]]
name = "libc"
@ -1788,7 +1822,7 @@ dependencies = [
"hex-simd",
"llrt_build",
"memchr",
"phf",
"phf 0.13.1",
]
[[package]]
@ -2432,6 +2466,15 @@ version = "2.3.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "9b4f627cb1b25917193a259e49bdad08f671f8d9708acfd5fe0a8c1455d87220"
[[package]]
name = "phf"
version = "0.12.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "913273894cec178f401a31ec4b656318d95473527be05c0752cc41cdc32be8b7"
dependencies = [
"phf_shared 0.12.1",
]
[[package]]
name = "phf"
version = "0.13.1"
@ -2439,7 +2482,7 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "c1562dc717473dbaa4c1f85a36410e03c047b2e7df7f45ee938fbef64ae7fadf"
dependencies = [
"phf_macros",
"phf_shared",
"phf_shared 0.13.1",
]
[[package]]
@ -2449,7 +2492,7 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "135ace3a761e564ec88c03a77317a7c6b80bb7f7135ef2544dbe054243b89737"
dependencies = [
"fastrand",
"phf_shared",
"phf_shared 0.13.1",
]
[[package]]
@ -2459,12 +2502,21 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "812f032b54b1e759ccd5f8b6677695d5268c588701effba24601f6932f8269ef"
dependencies = [
"phf_generator",
"phf_shared",
"phf_shared 0.13.1",
"proc-macro2",
"quote",
"syn",
]
[[package]]
name = "phf_shared"
version = "0.12.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "06005508882fb681fd97892ecff4b7fd0fee13ef1aa569f8695dae7ab9099981"
dependencies = [
"siphasher",
]
[[package]]
name = "phf_shared"
version = "0.13.1"
@ -2898,10 +2950,15 @@ name = "rust_lib_spotube"
version = "0.1.0"
dependencies = [
"anyhow",
"base64 0.22.1",
"chrono-tz",
"confy",
"eventsource-client",
"flutter_rust_bridge",
"heck",
"iana-time-zone",
"llrt_modules",
"openssl",
"reqwest",
"rquickjs",
"serde",
@ -3163,6 +3220,15 @@ dependencies = [
"serde_core",
]
[[package]]
name = "serde_spanned"
version = "1.0.3"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "e24345aa0fe688594e73770a5f6d1b216508b4f93484c0026d521acd30134392"
dependencies = [
"serde_core",
]
[[package]]
name = "serde_urlencoded"
version = "0.7.1"
@ -3373,6 +3439,26 @@ dependencies = [
"windows-sys 0.61.2",
]
[[package]]
name = "thiserror"
version = "2.0.17"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "f63587ca0f12b72a0600bcba1d40081f830876000bb46dd2337a3051618f4fc8"
dependencies = [
"thiserror-impl",
]
[[package]]
name = "thiserror-impl"
version = "2.0.17"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "3ff15c8ecd7de3849db632e14d18d2571fa09dfc5ed93479bc4485c7a517c913"
dependencies = [
"proc-macro2",
"quote",
"syn",
]
[[package]]
name = "threadpool"
version = "1.8.1"
@ -3474,6 +3560,21 @@ dependencies = [
"tokio",
]
[[package]]
name = "toml"
version = "0.9.8"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "f0dc8b1fb61449e27716ec0e1bdf0f6b8f3e8f6b05391e8497b8b6d7804ea6d8"
dependencies = [
"indexmap",
"serde_core",
"serde_spanned",
"toml_datetime",
"toml_parser",
"toml_writer",
"winnow",
]
[[package]]
name = "toml_datetime"
version = "0.7.3"
@ -3504,6 +3605,12 @@ dependencies = [
"winnow",
]
[[package]]
name = "toml_writer"
version = "1.0.4"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "df8b2b54733674ad286d16267dcfc7a71ed5c776e4ac7aa3c3e2561f7c637bf2"
[[package]]
name = "tower"
version = "0.5.2"
@ -3884,6 +3991,15 @@ dependencies = [
"windows-targets 0.52.6",
]
[[package]]
name = "windows-sys"
version = "0.59.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "1e38bc4d79ed67fd075bcc251a1c39b32a1776bbe92e5bef1f0bf1f8c531853b"
dependencies = [
"windows-targets 0.52.6",
]
[[package]]
name = "windows-sys"
version = "0.60.2"

View File

@ -21,6 +21,11 @@ heck = "0.5.0"
llrt_modules = { git = "https://github.com/awslabs/llrt.git", rev = "7d749dd18cf26a2e51119094c3b945975ae57bd4", features = ["abort", "buffer", "console", "crypto", "events", "exceptions", "fetch", "navigator", "url", "timers"] }
eventsource-client = "0.15.1"
reqwest = { version = "0.12", features = ["json"] }
confy = "2.0.0"
chrono-tz = "0.10"
iana-time-zone = "0.1"
base64 = "0.22.1"
openssl = "0.10.75"
[patch."https://github.com/DelSkayn/rquickjs"]
rquickjs = "0.10.0"

View File

@ -11,7 +11,8 @@ use crate::api::plugin::senders::{
PluginTrackSender, PluginUserSender,
};
use crate::frb_generated::StreamSink;
use crate::internal::apis::webview;
use crate::internal::apis;
use crate::internal::apis::{form, get_platform_directories, timezone, webview, yt_engine};
use anyhow::anyhow;
use flutter_rust_bridge::{frb, Rust2DartSendError};
use llrt_modules::module_builder::ModuleBuilder;
@ -35,6 +36,7 @@ pub struct OpaqueSender {
async fn create_context(
server_endpoint_url: String,
server_secret: String,
plugin_slug: String,
) -> anyhow::Result<(AsyncContext, AsyncRuntime)> {
let runtime = AsyncRuntime::new().expect("Unable to create async runtime");
@ -51,7 +53,11 @@ async fn create_context(
.with_global(navigator::init)
.with_global(url::init)
.with_global(timers::init)
.with_global(util::init);
.with_global(util::init)
.with_global(form::init)
.with_global(webview::init)
.with_global(yt_engine::init)
.with_global(timezone::init);
let (module_resolver, module_loader, global_attachment) = module_builder.build();
runtime
@ -62,13 +68,19 @@ async fn create_context(
.await
.expect("Unable to create async context");
let directories =
get_platform_directories(server_endpoint_url.clone(), server_secret.clone()).await?;
let local_storage_dir = directories
.application_support
.ok_or_else(|| anyhow!("Application support directory not found"))?;
async_with!(context => |ctx| {
apis::init(&ctx, server_endpoint_url, server_secret).catch(&ctx).map_err(|e| anyhow!("Failed to initialize APIs: {}", e))?;
apis::local_storage::init(&ctx, plugin_slug, local_storage_dir).catch(&ctx).map_err(|e| anyhow!("Failed to initialize LocalStorage API: {}", e))?;
global_attachment.attach(&ctx).catch(&ctx).map_err(|e| anyhow!("Failed to attach global modules: {}", e))?;
webview::init(&ctx, server_endpoint_url, server_secret).catch(&ctx).map_err(|e| anyhow!("Failed to initialize WebView API: {}", e))?;
anyhow::Ok(())
})
.await
.map_err(|e| anyhow!("Failed to register globals: {}", e))?;
.await?;
Ok((context, runtime))
}
@ -155,7 +167,7 @@ impl SpotubePlugin {
Ok(())
}
// #[frb(sync)]
#[frb(sync)]
pub fn create_context(
&self,
plugin_script: String,
@ -175,6 +187,7 @@ impl SpotubePlugin {
let (ctx, _) = create_context(
server_endpoint_url,
server_secret,
plugin_config.slug(),
).await?;
let injection = format!(

View File

@ -1,167 +0,0 @@
use eventsource_client::{ClientBuilder, Client, SSE};
use flutter_rust_bridge::for_generated::futures::StreamExt;
use rquickjs::function::Func;
use rquickjs::{CatchResultExt, Ctx, Error as JsError, Function, Object, Value};
use tokio::sync::mpsc;
fn connect_sse<'js>(ctx: Ctx<'js>, config: Object<'js>) -> rquickjs::Result<Object<'js>> {
let url: String = config.get("url")?;
let on_connecting: Function = config.get("onConnecting")?;
let on_open: Function = config.get("onOpen")?;
let on_message: Function = config.get("onMessage")?;
let on_error: Function = config.get("onError")?;
let (close_tx, mut close_rx) = mpsc::unbounded_channel::<()>();
if let Err(e) = on_connecting.call::<_, ()>(()).catch(&ctx) {
eprintln!("Error in onConnecting callback: {}", e);
}
// Spawn the SSE background task using Ctx::spawn
let _ = ctx.clone().spawn(async move {
let client = ClientBuilder::for_url(&url);
if let Err(err) = client {
eprintln!("Error in ClientBuilder::for_url: {}", err);
return;
}
let client = client.unwrap().build();
// Notify "open"
if let Err(e) = on_open.call::<(), ()>(()) {
eprintln!("Error in onOpen callback: {}", e);
}
// Now listen to SSE events OR close signal
let mut stream = Box::pin(client.stream());
loop {
tokio::select! {
// Check for close signal first
_ = close_rx.recv() => {
// Close requested — drop stream and exit
drop(stream);
break;
}
event = stream.next() => {
match event {
Some(Ok(SSE::Event(msg))) => {
let data = msg.data.clone();
if let Err(e) = on_message.call::<_, ()>((data,)) {
eprintln!("Error in onMessage callback: {}", e);
}
}
Some(Ok(SSE::Connected(details))) => {
println!("SSE Connected: {:?}", details);
}
Some(Ok(SSE::Comment(comment))) => {
println!("SSE Comment: {}", comment);
}
Some(Err(err)) => {
if let Err(e) = on_error.call::<_, ()>((err.to_string(),)) {
eprintln!("Error in onError callback: {}", e);
}
break;
}
None => {
println!("SSE Stream ended gracefully");
break;
}
}
}
}
}
});
// Create the close function that sends signal via channel
let close_fn = Function::new(ctx.clone(), move |_ctx: Ctx<'_>| {
// Send close signal — ignore errors if receiver is gone
let _ = close_tx.send(());
Ok::<(), JsError>(())
})?;
// Return { close: () => void }
let result = Object::new(ctx)?;
result.set("close", close_fn)?;
Ok(result)
}
pub fn init(ctx: &Ctx) -> rquickjs::Result<()> {
let globals = ctx.globals();
globals.set("__connectSSE", Func::new(connect_sse))?;
ctx.eval::<Value, _>(
r#"
globalThis.EventSource = class EventSource {
#listeners = {};
constructor(url, options) {
this.url = url;
this.options = options;
this.close = __connectSSE({
url: this.url,
onConnecting: this.#onConnecting.bind(this),
onOpen: this.#onOpen.bind(this),
onMessage: this.#onMessage.bind(this),
onError: this.#onError.bind(this),
}).close;
}
#onMessage(data) {
console.log("Received message:", data);
if (this.onmessage) {
this.onmessage(data);
}
const eventLines = data.split('\n');
if(eventLines.length === 0) return;
const eventNameChunks = eventLines[0].split("event:");
if(eventNameChunks.length === 0) return;
const eventName = eventNameChunks[1].trim();
if (!this.#listeners[eventName]) return;
const eventDataChunks = eventLines[1].split("data:");
if(eventDataChunks.length === 0) return;
const eventData = eventDataChunks[1].trim();
if (!eventData) return;
this.#listeners[eventName](eventData);
}
#onConnecting() {
this.readyState = 0;
}
#onOpen() {
this.readyState = 1;
if (this.onopen) {
this.onopen();
}
}
#onError(error) {
this.readyState = 2;
if (this.onerror) {
this.onerror(error);
}
}
addEventListener(event, callback) {
this.#listeners[event] ??= [];
this.#listeners[event].push(callback);
}
}
"#,
)?;
Ok(())
}

View File

@ -0,0 +1,25 @@
use rquickjs::{Ctx, Value};
pub fn init(ctx: &Ctx) -> rquickjs::Result<()> {
ctx.eval::<Value, _>(
r#"
globalThis.SpotubeForm = class SpotubeForm {
static async show(title, fields) {
return await fetch(
`${__serverUrl}/plugin-api/form/show`,
{
method: 'POST',
headers: {
'Content-Type': 'application/json',
'X-Plugin-Secret': __serverSecret,
},
body: JSON.stringify({ title, fields }),
}
).then(res=>res.json());
}
}
"#,
)?;
Ok(())
}

View File

@ -0,0 +1,115 @@
use rquickjs::class::Trace;
use rquickjs::{Class, Ctx, JsLifetime, Value};
use serde::{Deserialize, Serialize};
use std::collections::HashMap;
use std::path::Path;
use std::sync::{Arc, Mutex};
/// All values stored as strings; we convert at the edges.
#[derive(Debug, Default, Serialize, Deserialize, Clone)]
struct LocalStorageConfig {
map: HashMap<String, String>,
}
/// LocalStorage backed by `confy`.
#[derive(Clone, JsLifetime, Trace)]
#[rquickjs::class]
pub struct LocalStorage {
#[qjs(skip_trace)]
prefix: String,
#[qjs(skip_trace)]
path: String,
#[qjs(skip_trace)]
state: Arc<Mutex<LocalStorageConfig>>,
}
fn merge_prefix(prefix: String, key: String) -> String {
format!("{}->{}", prefix, key)
}
#[rquickjs::methods]
impl LocalStorage {
#[qjs(constructor)]
pub fn new(prefix: String, directory: String) -> rquickjs::Result<Self> {
let path = Path::new(&directory).join("plugin_configs.toml");
let cfg: LocalStorageConfig = confy::load_path(path.clone()).map_err(|e| {
rquickjs::Error::new_from_js_message(
"local_storage",
"PersistenceError",
&e.to_string(),
)
})?;
Ok(Self {
prefix,
path: path.to_string_lossy().to_string(),
state: Arc::new(Mutex::new(cfg)),
})
}
/// Persist current state to disk.
fn persist(&self) -> rquickjs::Result<()> {
let cfg = self.state.lock().unwrap().clone();
confy::store_path(self.path.clone(), cfg).map_err(|e| {
rquickjs::Error::new_from_js_message(
"local_storage",
"PersistenceError",
&e.to_string(),
)
})?;
Ok(())
}
#[qjs(rename = "setItem")]
pub fn set_item(&self, key: String, value: String) -> rquickjs::Result<()> {
{
let mut state = self.state.lock().unwrap();
state
.map
.insert(merge_prefix(self.prefix.clone(), key), value);
}
self.persist()
}
#[qjs(rename = "getItem")]
pub fn get_item(&self, key: String) -> rquickjs::Result<Option<String>> {
let state = self.state.lock().map_err(|e| {
rquickjs::Error::new_from_js_message("local_storage", "LockError", &e.to_string())
})?;
let key = merge_prefix(self.prefix.clone(), key.clone());
Ok(state.map.get(key.as_str()).cloned())
}
#[qjs(rename = "removeItem")]
pub fn remove_item(&self, key: String) -> rquickjs::Result<()> {
{
let mut state = self.state.lock().map_err(|e| {
rquickjs::Error::new_from_js_message("local_storage", "LockError", &e.to_string())
})?;
state
.map
.remove(merge_prefix(self.prefix.clone(), key).as_str());
}
self.persist()
}
pub fn clear(&self) -> rquickjs::Result<()> {
{
let mut state = self.state.lock().map_err(|e| {
rquickjs::Error::new_from_js_message("local_storage", "LockError", &e.to_string())
})?;
state.map.clear();
}
self.persist()
}
}
pub fn init(ctx: &Ctx, prefix: String, directory: String) -> rquickjs::Result<()> {
let global = ctx.globals();
Class::<LocalStorage>::define(&global)?;
ctx.eval::<Value, _>(format!(
"globalThis.localStorage = new LocalStorage('{}', '{}');",
prefix, directory
))?;
Ok(())
}

View File

@ -1,2 +1,40 @@
pub mod event_source;
use rquickjs::Ctx;
use serde::{Deserialize, Serialize};
pub mod form;
pub mod local_storage;
pub mod webview;
pub mod timezone;
pub mod yt_engine;
pub fn init(ctx: &Ctx, endpoint_url: String, secret: String) -> rquickjs::Result<()> {
ctx.globals().set("__serverUrl", endpoint_url)?;
ctx.globals().set("__serverSecret", secret)?;
Ok(())
}
#[derive(Debug, Serialize, Deserialize, Clone)]
#[serde(rename_all = "camelCase")]
pub struct DirectoriesResponse {
pub temporary: Option<String>,
pub application_documents: Option<String>,
pub application_support: Option<String>,
pub library: Option<String>,
pub external_storage: Option<String>,
pub downloads: Option<String>,
}
pub async fn get_platform_directories(
server_url: String,
server_secret: String,
) -> anyhow::Result<DirectoriesResponse> {
let client = reqwest::Client::new();
Ok(client
.get(format!("{}/plugin/localstorage/directories", server_url).as_str())
.header("X-Plugin-Secret", server_secret.as_str())
.send()
.await?
.json::<DirectoriesResponse>()
.await?)
}

View File

@ -0,0 +1,27 @@
use rquickjs::prelude::Func;
use rquickjs::{Class, Ctx, Object};
pub fn get_local_timezone() -> rquickjs::Result<String> {
let timezone = iana_time_zone::get_timezone()
.map_err(|e| rquickjs::Error::new_from_js_message("Timezone", "Error", &e.to_string()))?;
Ok(timezone)
}
pub fn get_available_timezones() -> rquickjs::Result<Vec<String>> {
let timezones: Vec<String> = chrono_tz::TZ_VARIANTS
.iter()
.map(|tz| tz.name().to_string())
.collect();
Ok(timezones)
}
pub fn init(ctx: &Ctx) -> rquickjs::Result<()> {
let globals = ctx.globals();
let timezone_obj = Object::new(ctx.clone())?;
timezone_obj.set("getLocalTimezone", Func::new(get_local_timezone))?;
timezone_obj.set("getAvailableTimezones", Func::new(get_available_timezones))?;
globals.set("Timezone", timezone_obj)?;
Ok(())
}

View File

@ -1,6 +1,8 @@
use base64::{engine::general_purpose::STANDARD, Engine as _};
use eventsource_client::{Client as EventSourceClient, ClientBuilder};
use flutter_rust_bridge::for_generated::futures::StreamExt;
use rquickjs::{class::Trace, Class, Ctx, Function, JsLifetime};
use openssl::symm::{decrypt, Cipher};
use rquickjs::{class::Trace, CatchResultExt, Class, Ctx, Function, JsLifetime, Value};
use serde::{Deserialize, Serialize};
use std::collections::HashMap;
@ -52,8 +54,8 @@ impl<'js> WebView<'js> {
#[qjs(static)]
pub async fn create(ctx: Ctx<'js>, url: String) -> rquickjs::Result<Class<'js, WebView<'js>>> {
let endpoint_url: String = ctx.globals().get("__webviewUrl").unwrap_or_default();
let secret: String = ctx.globals().get("__webviewSecret").unwrap_or_default();
let endpoint_url: String = ctx.globals().get("__serverUrl").unwrap_or_default();
let secret: String = ctx.globals().get("__serverSecret").unwrap_or_default();
let client = reqwest::Client::new();
let endpoint = format!("{}/plugin-api/webview/create", endpoint_url.clone());
@ -78,7 +80,7 @@ impl<'js> WebView<'js> {
Class::instance(ctx, webview)
}
pub async fn open(&self) -> rquickjs::Result<()> {
pub async fn open(&self, ctx: Ctx<'js>) -> rquickjs::Result<()> {
let client = reqwest::Client::new();
let endpoint = format!("{}/plugin-api/webview/open", self.endpoint_url);
@ -95,12 +97,13 @@ impl<'js> WebView<'js> {
rquickjs::Error::new_from_js_message("reqwest", "Error", &e.to_string())
})?;
self.url_change_task().await;
self.url_change_task(ctx).await;
Ok(())
}
pub async fn cookies(&self, ctx: Ctx<'js>) -> rquickjs::Result<rquickjs::Value<'js>> {
pub async fn cookies(&self, ctx: Ctx<'js>) -> rquickjs::Result<Value<'js>> {
let secret: String = ctx.globals().get("__serverSecret").unwrap_or_default();
let client = reqwest::Client::new();
let endpoint = format!("{}/plugin-api/webview/cookies", self.endpoint_url);
@ -122,7 +125,36 @@ impl<'js> WebView<'js> {
rquickjs::Error::new_from_js_message("reqwest", "Error", &e.to_string())
})?;
let value = ctx.json_parse(data.to_string())?;
let enc = data.get("data").and_then(|v| v.as_str()).ok_or_else(|| {
rquickjs::Error::new_from_js_message("cookies", "Error", "missing encrypted data")
})?;
let combined = STANDARD.decode(enc.trim()).map_err(|e| {
rquickjs::Error::new_from_js_message("cookies", "Error", &format!("b64 decode: {}", e))
})?;
if combined.len() < 16 {
return Err(rquickjs::Error::new_from_js_message(
"cookies",
"Error",
"invalid payload (too short)",
));
}
let (iv, cipher_bytes) = combined.split_at(16);
let key = STANDARD.decode(secret.trim()).map_err(|e| {
rquickjs::Error::new_from_js_message("cookies", "Error", &format!("key decode: {}", e))
})?;
let plain = decrypt(Cipher::aes_128_cbc(), &key, Some(iv), cipher_bytes).map_err(|e| {
rquickjs::Error::new_from_js_message("cookies", "Error", &format!("decrypt: {}", e))
})?;
let cookies_json: serde_json::Value = serde_json::from_slice(&plain).map_err(|e| {
rquickjs::Error::new_from_js_message("cookies", "Error", &format!("json decode: {}", e))
})?;
let value = ctx.json_parse(cookies_json.to_string())?;
Ok(value)
}
@ -152,7 +184,7 @@ impl<'js> WebView<'js> {
Ok(())
}
async fn url_change_task(&self) {
async fn url_change_task(&self, ctx: Ctx<'js>) {
let endpoint = format!(
"{}/plugin-api/webview/{}/on-url-request",
self.endpoint_url, self.uid
@ -181,10 +213,19 @@ impl<'js> WebView<'js> {
{
let url = data.get("url").cloned().unwrap_or_default();
for callback in self.callbacks.iter() {
if let Err(e) = callback.call::<_, ()>((url.clone(),)) {
match callback.call::<_, Value>((url.clone(),)) {
Ok(res) => {
if let Some(promise) = res.into_promise() {
if let Err(e) = promise.into_future::<()>().await.catch(&ctx) {
eprintln!("Error in onUrlChange promise: {}", e);
}
}
}
Err(e) => {
eprintln!("Error in onUrlChange callback: {}", e);
}
}
}
} else {
eprintln!("Failed to parse event data: {}", msg.data);
}
@ -203,12 +244,7 @@ impl<'js> WebView<'js> {
}
}
pub fn init(ctx: &Ctx, endpoint_url: String, secret: String) -> rquickjs::Result<()> {
// Store config in globals for access in static methods
ctx.globals().set("__webviewUrl", endpoint_url)?;
ctx.globals().set("__webviewSecret", secret)?;
// Register the WebView class
pub fn init(ctx: &Ctx) -> rquickjs::Result<()> {
Class::<WebView>::define(&ctx.globals())?;
Ok(())

View File

@ -0,0 +1,36 @@
use rquickjs::{Ctx, Value};
pub fn init(ctx: &Ctx) -> rquickjs::Result<()> {
ctx.eval::<Value, _>(
r#"
globalThis.YouTubeEngine = class YouTubeEngine {
async request(endpoint, qName, qValue) {
return await fetch(
`${__serverUrl}/plugin-api/yt-engine/${endpoint}?${qName}=${encodeURIComponent(qValue)}`,
{
method: 'GET',
headers: {
'Content-Type': 'application/json',
'X-Plugin-Secret': __serverSecret,
},
}
).then(res=>res.json())
}
async search(query) {
return await this.request('search', 'query', query);
}
async video(videoId) {
return await this.request('video', 'videoId', videoId);
}
async streamManifest(videoId) {
return await this.request('stream-manifest', 'videoId', videoId);
}
}
"#,
)?;
Ok(())
}