65b4d9d9e7* Add Unsloth desktop deep links * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Address deep-link review feedback --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
188 lines
7.2 KiB
Python
188 lines
7.2 KiB
Python
# SPDX-License-Identifier: AGPL-3.0-only
|
|
# Copyright 2026-present the Unsloth AI Inc. team. All rights reserved. See /studio/LICENSE.AGPL-3.0
|
|
|
|
"""Desktop deep-link configuration, routing, and validation contracts."""
|
|
|
|
import json
|
|
from pathlib import Path
|
|
import shutil
|
|
import subprocess
|
|
import textwrap
|
|
|
|
import pytest
|
|
|
|
try:
|
|
import tomllib
|
|
except ModuleNotFoundError:
|
|
tomllib = pytest.importorskip("tomli")
|
|
|
|
|
|
REPO = Path(__file__).resolve().parents[2]
|
|
FRONTEND = REPO / "studio/frontend"
|
|
TAURI = REPO / "studio/src-tauri"
|
|
PARSER = FRONTEND / "src/features/deep-links/parse-deep-link.ts"
|
|
INTENT_GATE = FRONTEND / "src/features/deep-links/deep-link-intent.ts"
|
|
GGUF_FILENAME = FRONTEND / "src/features/hub/lib/gguf-filename.ts"
|
|
|
|
|
|
def test_unsloth_deep_link_parser_guardrails(tmp_path: Path) -> None:
|
|
if shutil.which("node") is None:
|
|
pytest.skip("node not available")
|
|
probe = subprocess.run(
|
|
["node", "--experimental-strip-types", "--version"],
|
|
capture_output = True,
|
|
text = True,
|
|
timeout = 5,
|
|
)
|
|
if probe.returncode != 0:
|
|
pytest.skip("node --experimental-strip-types not available")
|
|
|
|
(tmp_path / "parse-deep-link.ts").write_text(
|
|
PARSER.read_text(encoding = "utf-8"), encoding = "utf-8"
|
|
)
|
|
|
|
(tmp_path / "gguf-filename.ts").write_text(
|
|
GGUF_FILENAME.read_text(encoding = "utf-8"), encoding = "utf-8"
|
|
)
|
|
|
|
(tmp_path / "deep-link-intent.ts").write_text(
|
|
INTENT_GATE.read_text(encoding = "utf-8"), encoding = "utf-8"
|
|
)
|
|
script = textwrap.dedent("""
|
|
import assert from "node:assert/strict";
|
|
import { parseUnslothDeepLink } from "./parse-deep-link.ts";
|
|
|
|
import { createDeepLinkIntentGate } from "./deep-link-intent.ts";
|
|
import {
|
|
ggufFilenamesMatch,
|
|
ggufSelectionOverrideMatchesIntent,
|
|
} from "./gguf-filename.ts";
|
|
|
|
const valid = new Map([
|
|
[
|
|
"unsloth://open_from_hf?model=unsloth/Laguna-S-2.1-GGUF",
|
|
{ model: "unsloth/Laguna-S-2.1-GGUF" },
|
|
],
|
|
[
|
|
"unsloth://open_from_hf/?model=org/repo_name",
|
|
{ model: "org/repo_name" },
|
|
],
|
|
[
|
|
"unsloth://open_from_hf?model=org%2Frepo",
|
|
{ model: "org/repo" },
|
|
],
|
|
[
|
|
"unsloth://open_from_hf?model=unsloth/Laguna-S-2.1-GGUF&file=Laguna-S-2.1-UD-IQ3_XXS.gguf",
|
|
{
|
|
model: "unsloth/Laguna-S-2.1-GGUF",
|
|
file: "Laguna-S-2.1-UD-IQ3_XXS.gguf",
|
|
},
|
|
],
|
|
[
|
|
"unsloth://open_from_hf?file=weights%2Fmodel-Q4_K_M.gguf&model=org/repo",
|
|
{ model: "org/repo", file: "weights/model-Q4_K_M.gguf" },
|
|
],
|
|
[
|
|
`unsloth://open_from_hf?model=${"a".repeat(96)}/${"b".repeat(96)}`,
|
|
{ model: `${"a".repeat(96)}/${"b".repeat(96)}` },
|
|
],
|
|
]);
|
|
for (const [url, intent] of valid) {
|
|
assert.deepEqual(parseUnslothDeepLink(url), intent, url);
|
|
}
|
|
|
|
assert.equal(
|
|
ggufFilenamesMatch(
|
|
"weights/model-Q4_K_M-00002-of-00002.gguf",
|
|
"weights/model-Q4_K_M-00001-of-00002.gguf",
|
|
),
|
|
true,
|
|
);
|
|
assert.equal(
|
|
ggufFilenamesMatch("model-Q4_K_M.GGUF", "model-q4_k_m.gguf"),
|
|
true,
|
|
);
|
|
assert.equal(ggufFilenamesMatch("mmproj-F16.gguf", "model-F16.gguf"), false);
|
|
|
|
assert.equal(ggufSelectionOverrideMatchesIntent("a.gguf", 2, "a.gguf", 2), true);
|
|
assert.equal(ggufSelectionOverrideMatchesIntent("a.gguf", 2, "a.gguf", 1), false);
|
|
|
|
let now = 1_000;
|
|
const acceptIntent = createDeepLinkIntentGate(2_000, () => now);
|
|
assert.equal(acceptIntent("org/repo", "a.gguf"), 1);
|
|
assert.equal(acceptIntent("org/repo", "a.gguf"), null);
|
|
assert.equal(acceptIntent("org/repo", "b.gguf"), 2);
|
|
now = 3_000;
|
|
assert.equal(acceptIntent("org/repo", "b.gguf"), 3);
|
|
|
|
|
|
const invalid = [
|
|
"",
|
|
"https://open_from_hf?model=org/repo",
|
|
"UNSLOTH://open_from_hf?model=org/repo",
|
|
"unsloth://OPEN_FROM_HF?model=org/repo",
|
|
"unsloth://open_from_hf/path?model=org/repo",
|
|
"unsloth://open_from_hf/%2e%2e?model=org/repo",
|
|
"unsloth://user@open_from_hf?model=org/repo",
|
|
"unsloth://open_from_hf:42?model=org/repo",
|
|
"unsloth://open_from_hf?model=org/repo#fragment",
|
|
"unsloth://open_from_hf?model=org/repo&download=true",
|
|
|
|
"unsloth://open_from_hf?model=org/repo&file=model.gguf&file=other.gguf",
|
|
"unsloth://open_from_hf?model=org/repo&file=",
|
|
"unsloth://open_from_hf?model=org/repo&file=../model.gguf",
|
|
"unsloth://open_from_hf?model=org/repo&file=%2Fmodel.gguf",
|
|
"unsloth://open_from_hf?model=org/repo&file=model.safetensors",
|
|
"unsloth://open_from_hf?model=org/repo&model=other/repo",
|
|
"unsloth://open_from_hf?model=repo",
|
|
"unsloth://open_from_hf?model=org/repo/extra",
|
|
"unsloth://open_from_hf?model=-org/repo",
|
|
"unsloth://open_from_hf?model=org/repo.",
|
|
|
|
"unsloth://open_from_hf?model=org/repo.git",
|
|
"unsloth://open_from_hf?model=org/repo--name",
|
|
"unsloth://open_from_hf?model=org/repo..name",
|
|
];
|
|
for (const url of invalid) {
|
|
assert.equal(parseUnslothDeepLink(url), null, url);
|
|
}
|
|
""")
|
|
result = subprocess.run(
|
|
["node", "--experimental-strip-types", "--no-warnings", "--input-type=module"],
|
|
input = script,
|
|
cwd = tmp_path,
|
|
capture_output = True,
|
|
text = True,
|
|
timeout = 30,
|
|
)
|
|
assert result.returncode == 0, f"stderr: {result.stderr}\nstdout: {result.stdout}"
|
|
|
|
|
|
def test_tauri_registers_only_the_unsloth_scheme() -> None:
|
|
cargo = tomllib.loads((TAURI / "Cargo.toml").read_text(encoding = "utf-8"))
|
|
dependencies = cargo["dependencies"]
|
|
assert "tauri-plugin-deep-link" in dependencies
|
|
single_instance = dependencies["tauri-plugin-single-instance"]
|
|
assert isinstance(single_instance, dict)
|
|
assert "deep-link" in single_instance.get("features", [])
|
|
|
|
config = json.loads((TAURI / "tauri.conf.json").read_text(encoding = "utf-8"))
|
|
assert config["plugins"]["deep-link"]["desktop"]["schemes"] == ["unsloth"]
|
|
|
|
capabilities = json.loads((TAURI / "capabilities/default.json").read_text(encoding = "utf-8"))
|
|
assert "deep-link:default" in capabilities["permissions"]
|
|
assert "core:window:allow-unminimize" in capabilities["permissions"]
|
|
|
|
main = (TAURI / "src/main.rs").read_text(encoding = "utf-8")
|
|
assert main.index("tauri_plugin_single_instance::init") < main.index(
|
|
"tauri_plugin_deep_link::init()"
|
|
)
|
|
assert "DeepLinkExt" in main
|
|
assert "if let Err(error) = app.deep_link().register_all()" in main
|
|
assert 'warn!("Failed to register deep-link handlers: {error}")' in main
|
|
assert 'target_os = "linux"' in main
|
|
desktop_template = TAURI / "linux/unsloth.desktop"
|
|
assert config["bundle"]["linux"]["deb"]["desktopTemplate"] == "./linux/unsloth.desktop"
|
|
desktop = desktop_template.read_text(encoding = "utf-8")
|
|
assert "Exec={{exec}} %u" in desktop
|
|
assert "MimeType=x-scheme-handler/unsloth;" in desktop
|