126 lines
3.5 KiB
TypeScript
126 lines
3.5 KiB
TypeScript
import { RqbitDesktopConfig } from "./configuration";
|
|
import {
|
|
AddTorrentResponse,
|
|
ListTorrentsResponse,
|
|
RqbitAPI,
|
|
TorrentDetails,
|
|
TorrentStats,
|
|
ErrorDetails,
|
|
} from "rqbit-webui/src/api-types";
|
|
|
|
import { InvokeArgs, invoke } from "@tauri-apps/api/tauri";
|
|
|
|
interface InvokeErrorResponse {
|
|
error_kind: string;
|
|
human_readable: string;
|
|
status: number;
|
|
status_text: string;
|
|
}
|
|
|
|
function errorToUIError(
|
|
path: string,
|
|
): (e: InvokeErrorResponse) => Promise<never> {
|
|
return (e: InvokeErrorResponse) => {
|
|
console.log(e);
|
|
let reason: ErrorDetails = {
|
|
method: "INVOKE",
|
|
path: path,
|
|
text: e.human_readable ?? e.toString(),
|
|
status: e.status,
|
|
statusText: e.status_text,
|
|
};
|
|
return Promise.reject(reason);
|
|
};
|
|
}
|
|
|
|
export async function invokeAPI<Response>(
|
|
name: string,
|
|
params?: InvokeArgs,
|
|
): Promise<Response> {
|
|
console.log("invoking", name, params);
|
|
const result = await invoke<Response>(name, params).catch(
|
|
errorToUIError(name),
|
|
);
|
|
console.log(result);
|
|
return result;
|
|
}
|
|
|
|
async function readFileAsBase64(file: File): Promise<string> {
|
|
return new Promise((resolve, reject) => {
|
|
const reader = new FileReader();
|
|
|
|
reader.onload = function (event) {
|
|
const base64String = (event?.target?.result as string)?.split(",")[1];
|
|
if (base64String) {
|
|
resolve(base64String);
|
|
} else {
|
|
reject(new Error("Failed to read file as base64."));
|
|
}
|
|
};
|
|
|
|
reader.onerror = function (error) {
|
|
console.log(error);
|
|
reject(error);
|
|
};
|
|
|
|
reader.readAsDataURL(file);
|
|
});
|
|
}
|
|
|
|
export const makeAPI = (configuration: RqbitDesktopConfig): RqbitAPI => {
|
|
return {
|
|
getStreamLogsUrl: () => {
|
|
if (!configuration.http_api.listen_addr) {
|
|
return null;
|
|
}
|
|
let port = configuration.http_api.listen_addr.split(":")[1];
|
|
if (!port) {
|
|
return null;
|
|
}
|
|
return `http://127.0.0.1:${port}/stream_logs`;
|
|
},
|
|
listTorrents: async function (): Promise<ListTorrentsResponse> {
|
|
return await invokeAPI<ListTorrentsResponse>("torrents_list");
|
|
},
|
|
getTorrentDetails: async function (id: number): Promise<TorrentDetails> {
|
|
return await invokeAPI<TorrentDetails>("torrent_details", { id });
|
|
},
|
|
getTorrentStats: async function (id: number): Promise<TorrentStats> {
|
|
return await invokeAPI<TorrentStats>("torrent_stats", { id });
|
|
},
|
|
uploadTorrent: async function (data, opts): Promise<AddTorrentResponse> {
|
|
if (data instanceof File) {
|
|
let contents = await readFileAsBase64(data);
|
|
return await invokeAPI<AddTorrentResponse>(
|
|
"torrent_create_from_base64_file",
|
|
{
|
|
contents,
|
|
opts: opts ?? {},
|
|
},
|
|
);
|
|
}
|
|
return await invokeAPI<AddTorrentResponse>("torrent_create_from_url", {
|
|
url: data,
|
|
opts: opts ?? {},
|
|
});
|
|
},
|
|
updateOnlyFiles: function (id, files): Promise<void> {
|
|
return invokeAPI<void>("torrent_action_configure", {
|
|
id: id,
|
|
onlyFiles: files,
|
|
});
|
|
},
|
|
pause: function (id: number): Promise<void> {
|
|
return invokeAPI<void>("torrent_action_pause", { id });
|
|
},
|
|
start: function (id: number): Promise<void> {
|
|
return invokeAPI<void>("torrent_action_start", { id });
|
|
},
|
|
forget: function (id: number): Promise<void> {
|
|
return invokeAPI<void>("torrent_action_forget", { id });
|
|
},
|
|
delete: function (id: number): Promise<void> {
|
|
return invokeAPI<void>("torrent_action_delete", { id });
|
|
},
|
|
};
|
|
};
|