Compare commits

..

1 Commits

Author SHA1 Message Date
lamp 25de4eb7d8 commit 2024-09-16 00:16:41 -07:00
5 changed files with 113 additions and 163 deletions
-2
View File
@@ -1,2 +0,0 @@
credentials.json
past_bookmarks.json
-4
View File
@@ -1,4 +0,0 @@
{
"deno.enable": true,
"deno.lint": false
}
+84 -104
View File
@@ -1,133 +1,113 @@
import ky from 'https://esm.sh/ky';
import credentials from "./credentials.json" with { type: "json" }; import credentials from "./credentials.json" with { type: "json" };
var {client_id, client_secret, username, password, pixiv_cookie} = credentials; var {matrix_token, pixiv_cookie, pixiv_user_id} = credentials;
var access_token; var ke = ky.create({
hooks: {
await (async function login() { beforeRequest: [request => console.log(`${request.method} ${request.url}`)],
var form = new FormData(); afterResponse: [(_request, _options, response) => console.log(response.status)]
form.append("client_id", client_id); },
form.append("client_secret", client_secret); retry: {
form.append("grant_type", "password"); limit: 100,
form.append("username", username); delay: attemptCount => (2 ** (attemptCount - 1)) * 1000
form.append("password", password); },
var data = await fetch("https://pleroma.lamp.wtf/oauth/token", { timeout: false
method: "POST",
body: form
}).then(res => res.json());
console.log ("logged in", data);
access_token = data.access_token;
})();
async function uploadFile({data, name}) {
var form = new FormData();
form.append("file", data, name);
var res = await fetch("https://pleroma.lamp.wtf/api/v1/media", {
method: "POST",
body: form,
headers: {"Authorization": `Bearer ${access_token}`}
}); });
var json = await res.json();
console.log("uploaded file", res.status, json.url);
return json; var matrix = {
ky: ke.extend({headers: {"Authorization": `Bearer ${credentials.matrix_token}`}}),
rooms: {
default: "!mxhaBcBiEmnGxrffll:matrix.org",
pixiv_sfw: "!zStOGJAGxEaQgwxGYI:matrix.org",
pixiv_nsfw: "!HtXDHeAhEKBFGfAJTi:matrix.org"
},
async sendMessage({room = this.rooms.default, type = "m.text", body, filename, info, url}) {
return await this.ky.put(`https://matrix-client.matrix.org/_matrix/client/v3/rooms/${encodeURIComponent(room)}/send/m.room.message/${Math.random().toString()}`, {
json: {msgtype: type, body, filename, info, url}
}).json();
},
async uploadFile(filename, data) {
var {content_uri} = await this.ky.post(`https://matrix-client.matrix.org/_matrix/media/v3/upload?filename=${encodeURIComponent(filename)}`, {body: data}).json();
return content_uri;
},
async sendImage({room = this.rooms.default, body, filename, data, info}) {
var content_uri = await this.uploadFile(filename, data);
return await this.sendMessage({room, type: "m.image", body: body || filename, filename, url: content_uri, info});
}
} }
async function postStatus({status, visibility = "unlisted", content_type = "text/plain", media_ids = [], sensitive, files}) {
if (files) {
media_ids = (await Promise.all(files.map(file => uploadFile(file)))).map(d => d.id);
}
var form = new FormData();
form.append("status", status);
form.append("visibility", visibility);
form.append("source", "bot");
form.append("content_type", content_type);
for (let media_id of media_ids) {
form.append("media_ids[]", media_id);
}
if (sensitive) form.append("sensitive", "true");
var res = await fetch("https://pleroma.lamp.wtf/api/v1/statuses", {
method: "POST",
body: form,
headers: {"Authorization": `Bearer ${access_token}`}
});
var json = await res.json();
console.log("posted", res.status, json.uri || json);
return json;
}
async function downloadPixivIllust(illust_id) { async function downloadPixivIllust(illust_id) {
var url = `https://www.pixiv.net/en/artworks/${illust_id}`; var html = await ke.get(`https://www.pixiv.net/en/artworks/${illust_id}`, {headers: {"Cookie": credentials.pixiv_cookie}}).text();
console.log("fetch", url);
var html = await fetch(url, {headers: {"Cookie": pixiv_cookie}}).then(res => res.text());
var illust = Object.values(JSON.parse(html.match(/<meta name="preload-data" id="meta-preload-data" content='(.*)'>/)[1]).illust)[0]; var illust = Object.values(JSON.parse(html.match(/<meta name="preload-data" id="meta-preload-data" content='(.*)'>/)[1]).illust)[0];
var images = []; var images = [];
for (let i = 0; i < illust.pageCount; i++) { for (let i = 0; i < illust.pageCount; i++) {
let url = illust.urls.original.replace('p0', 'p'+i); let url = illust.urls.original.replace('p0', 'p'+i);
console.log("fetch", url); let data = await ke.get(url, {headers: {"Referer": "https://www.pixiv.net/"}}).then(res => res.blob());
let data = await fetch(url, {headers: {"Referer": "https://www.pixiv.net/"}}).then(res => res.blob());
images.push({url, name: url.split('/').pop(), data}); images.push({url, name: url.split('/').pop(), data});
}; };
return {illust, images}; return {illust, images};
} }
export async function pixivToPleroma(illust_id) { export async function pixivToMatrix(illust_id) {
try { try {
var {illust, images} = await downloadPixivIllust(illust_id); var {illust, images} = await downloadPixivIllust(illust_id);
await postStatus({ var room = illust.xRestrict ? matrix.rooms.pixiv_nsfw : matrix.rooms.pixiv_sfw;
status: `https://www.pixiv.net/en/artworks/${illust_id}`, await matrix.sendMessage({room, body: `https://www.pixiv.net/en/artworks/${illust_id}`});
files: images, for (var image of images) {
visibility: "unlisted", try {
sensitive: Boolean(illust.xRestrict) await matrix.sendImage({
room,
filename: image.name,
data: image.data,
info: {
mimetype: image.data.type,
size: image.data.size
}
}); });
} catch (error) { } catch (error) {
onError(error);
}
}
} catch (error) {
onError(error);
}
function onError(error) {
console.error(error.stack);
matrix.sendMessage({room: room || matrix.rooms.pixiv_sfw, body: error.stack}).catch(error => {
console.error(error.stack); console.error(error.stack);
postStatus({
status: `https://www.pixiv.net/en/artworks/${illust_id}` +
`\n#error:\n${error.stack}`
}); });
} }
} }
if (import.meta.main) Deno.serve({
port: 18024,
hostname: "127.0.0.1"
}, async (req, info) => {
var {pathname} = new URL(req.url);
console.log(info.remoteAddr.hostname, pathname);
switch (pathname) {
case "/ajax/illusts/bookmarks/add": var new_ids = [];
var payload = await req.clone().json(); top: for (var offset = 0;;) {
console.log(payload); let url = `https://www.pixiv.net/ajax/user/${pixiv_user_id}/illusts/bookmarks?tag=&offset=${offset}&limit=100&rest=show&lang=en&version=5dc84ab282403a049abea4e2f2214b6a69d31da6`;
var {comment, illust_id, restrict, tags} = payload; console.log("get", url);
pixivToPleroma(illust_id); let data = await fetch(url, {headers: {Cookie: pixiv_cookie}}).then(res => res.json());
break; let ids = data.body.works.map(x => x.id);
case "/ajax/illusts/bookmarks/delete": if (!ids.length) break;
var payload = await req.clone().formData(); for (let id of ids) {
console.log(payload); if (["119320087", "119534122"].includes(id)) break top;
// todo new_ids.push(id);
break;
case "/touch/ajax_api/ajax_api.php":
var payload = await req.clone().formData();
console.log(payload);
var mode = payload.get("mode");
if (mode == "add_bookmark_illust") {
var restrict = payload.get("restrict");
var tag = payload.get("tag");
var id = payload.get("id");
var comment = payload.get("comment");
pixivToPleroma(id);
} else if (mode == "delete_bookmark_illust") {
var id = payload.get("id");
// todo
} }
break; offset += ids.length;
}
console.log("new ids", new_ids);
for (let id of new_ids.reverse()) {
await pixivToMatrix(id);
} }
return fetch("https://www.pixiv.net"+pathname, {
method: req.method,
headers: req.headers,
body: req.body,
redirect: "manual"
});
});
-18
View File
@@ -1,18 +0,0 @@
import credentials from "./credentials.json" with { type: "json" };
var {pixiv_cookie, pixiv_user_id} = credentials;
var all_ids = [];
var offset = 0;
while (true) {
let url = `https://www.pixiv.net/ajax/user/${pixiv_user_id}/illusts/bookmarks?tag=&offset=${offset}&limit=100&rest=show&lang=en&version=5dc84ab282403a049abea4e2f2214b6a69d31da6`;
console.log("get", url);
let data = await fetch(url, {headers: {Cookie: pixiv_cookie}}).then(res => res.json());
let ids = data.body.works.map(x => x.id);
if (!ids.length) break;
for (let id of ids) {
all_ids.push(id);
}
offset += ids.length;
}
Deno.writeTextFileSync("past_bookmarks.json", JSON.stringify(all_ids));
-6
View File
@@ -1,6 +0,0 @@
import past_bookmarks from "./past_bookmarks.json" with { type: "json" };
import { pixivToPleroma } from "./app.deno.js";
for (let id of past_bookmarks.reverse()) {
await pixivToPleroma(id);
}