mirror of
https://github.com/Ellpeck/WoodpeckerPlugins.git
synced 2024-11-24 19:28:34 +01:00
246 lines
10 KiB
JavaScript
246 lines
10 KiB
JavaScript
const Upload = require("nextcloud-chunk-file-upload");
|
|
const glob = require("glob");
|
|
const axios = require("axios");
|
|
const xml = require('xml2json');
|
|
|
|
// parse arguments
|
|
const serverEnv = process.env.PLUGIN_SERVER;
|
|
if (!serverEnv)
|
|
throw "Missing required arg server";
|
|
const userEnv = process.env.PLUGIN_USER;
|
|
if (!userEnv)
|
|
throw "Missing required arg user";
|
|
const tokenEnv = process.env.PLUGIN_TOKEN;
|
|
if (!tokenEnv)
|
|
throw "Missing required arg token";
|
|
const filesEnv = process.env.PLUGIN_FILES;
|
|
if (!filesEnv)
|
|
throw "Missing required arg files";
|
|
const destEnv = process.env.PLUGIN_DEST;
|
|
if (!destEnv)
|
|
throw "Missing required arg dest";
|
|
|
|
const baseDir = process.env.PLUGIN_BASEDIR || ".";
|
|
const chunkSizeEnv = process.env.PLUGIN_CHUNKSIZE || 10 * 1024 * 1024;
|
|
const quiet = process.env.PLUGIN_QUIET == "true";
|
|
const tags = process.env.PLUGIN_TAGS || "";
|
|
const flatten = process.env.PLUGIN_FLATTEN == "true";
|
|
const retentionBase = process.env.PLUGIN_RETENTIONBASE || "";
|
|
const retentionAmount = process.env.PLUGIN_RETENTIONAMOUNT || 0;
|
|
const retentionSkipTrash = process.env.PLUGIN_RETENTIONSKIPTRASH == "true";
|
|
|
|
upload();
|
|
|
|
async function upload() {
|
|
let basePath = `${serverEnv}/remote.php/dav`;
|
|
|
|
// delete files or directories that exceed the retention amount
|
|
if (retentionBase && retentionAmount) {
|
|
try {
|
|
let retentionPath = `${basePath}/files/${userEnv}/${retentionBase}`;
|
|
let response = await axios.request({
|
|
method: "propfind",
|
|
url: retentionPath,
|
|
auth: {
|
|
username: userEnv,
|
|
password: tokenEnv
|
|
},
|
|
// 404 means the directory doesn't exist, which is fine
|
|
validateStatus: s => s == 207 || s == 404
|
|
});
|
|
if (response.status != 404) {
|
|
let data = JSON.parse(xml.toJson(response.data));
|
|
let dirs = data["d:multistatus"]["d:response"].slice(1);
|
|
// sort directories by last modified
|
|
dirs.sort((a, b) => new Date(a["d:propstat"]["d:prop"]["d:getlastmodified"]) - new Date(b["d:propstat"]["d:prop"]["d:getlastmodified"]));
|
|
while (dirs.length >= parseInt(retentionAmount)) {
|
|
let dir = serverEnv + dirs[0]["d:href"];
|
|
let dirName = dir.substring(retentionPath.length - retentionBase.length).replace(/\/$/, "");
|
|
await axios.request({
|
|
method: "delete",
|
|
url: dir,
|
|
auth: {
|
|
username: userEnv,
|
|
password: tokenEnv
|
|
}
|
|
});
|
|
|
|
// if we skip the trash, we actually also have to delete the item from the trash
|
|
if (retentionSkipTrash) {
|
|
let trashResponse = await axios.request({
|
|
method: "propfind",
|
|
url: `${basePath}/trashbin/${userEnv}/trash`,
|
|
auth: {
|
|
username: userEnv,
|
|
password: tokenEnv
|
|
},
|
|
data: `<?xml version="1.0"?>
|
|
<d:propfind xmlns:d="DAV:" xmlns:oc="http://owncloud.org/ns" xmlns:nc="http://nextcloud.org/ns">
|
|
<d:prop>
|
|
<nc:trashbin-original-location/>
|
|
</d:prop>
|
|
</d:propfind>`
|
|
});
|
|
let trashContent = JSON.parse(xml.toJson(trashResponse.data))["d:multistatus"]["d:response"];
|
|
let trashItem = trashContent.find(e => e["d:propstat"]["d:prop"]["nc:trashbin-original-location"] == dirName);
|
|
await axios.request({
|
|
method: "delete",
|
|
url: serverEnv + trashItem["d:href"],
|
|
auth: {
|
|
username: userEnv,
|
|
password: tokenEnv
|
|
}
|
|
});
|
|
}
|
|
|
|
console.log(`Deleted directory ${dirName} because retention amount of ${retentionAmount} was reached${retentionSkipTrash ? " (skipped trash)" : ""}`);
|
|
dirs.splice(0, 1);
|
|
}
|
|
}
|
|
} catch (e) {
|
|
console.log(`Failed to delete old directories (${e})`);
|
|
process.exit(1);
|
|
}
|
|
}
|
|
|
|
// find ids for the tags we want to assign later
|
|
let tagIds = new Map();
|
|
if (tags) {
|
|
if (!quiet)
|
|
console.log(`Retrieving tag ids`);
|
|
// list tags and find the ones we want: https://doc.owncloud.com/server/next/developer_manual/webdav_api/tags.html#list-tags
|
|
let response = await axios.request({
|
|
method: "propfind",
|
|
url: `${basePath}/systemtags`,
|
|
auth: {
|
|
username: userEnv,
|
|
password: tokenEnv
|
|
},
|
|
data: `<?xml version="1.0"?>
|
|
<d:propfind xmlns:d="DAV:" xmlns:oc="http://owncloud.org/ns" xmlns:nc="http://nextcloud.org/ns">
|
|
<d:prop>
|
|
<oc:display-name/>
|
|
<oc:id/>
|
|
</d:prop>
|
|
</d:propfind>`
|
|
});
|
|
let data = JSON.parse(xml.toJson(response.data))["d:multistatus"]["d:response"].map(e => e["d:propstat"]["d:prop"]);
|
|
for (let tag of tags.split(",")) {
|
|
let entry = data.find(e => e["oc:display-name"] == tag);
|
|
if (!entry) {
|
|
console.log(`Couldn't find tag with name ${tag}`);
|
|
process.exit(1);
|
|
}
|
|
let tagId = entry["oc:id"];
|
|
tagIds.set(tag, tagId);
|
|
if (!quiet)
|
|
console.log(`Tag id of tag ${tag} is ${tagId}`);
|
|
}
|
|
}
|
|
|
|
// collect and upload files
|
|
if (!quiet)
|
|
console.log(`Uploading files`);
|
|
let upload = new Upload(basePath, userEnv, userEnv, tokenEnv);
|
|
for (let pattern of filesEnv.split(",")) {
|
|
let files = await glob.glob(pattern, { cwd: baseDir });
|
|
if (!files.length)
|
|
console.log(`No files found for pattern ${pattern}`);
|
|
for (let file of files) {
|
|
file = file.replaceAll("\\", "/");
|
|
let fileName = file.split("/").pop();
|
|
let dest = `${destEnv}/${flatten ? fileName : file}`;
|
|
|
|
// we have to explicitly create any directories that don't exist yet
|
|
// (https://github.com/shiftpi/nextcloud-chunk-file-upload/issues/22)
|
|
let currDir = "";
|
|
for (let dir of dest.split("/").slice(0, -1)) {
|
|
currDir += `${dir}/`;
|
|
try {
|
|
let response = await axios.request({
|
|
method: 'mkcol',
|
|
url: `${basePath}/files/${userEnv}/${currDir}`,
|
|
auth: {
|
|
username: userEnv,
|
|
password: tokenEnv
|
|
},
|
|
// 405 means the directory already exists
|
|
validateStatus: s => s == 201 || s == 405
|
|
});
|
|
if (response.status != 405) {
|
|
if (!quiet)
|
|
console.log(`Created directory ${dir}`);
|
|
|
|
// add tags to directory too
|
|
if (tagIds)
|
|
await addTags(basePath, tagIds, dir, currDir);
|
|
}
|
|
|
|
} catch (error) {
|
|
console.log(`Failed to create directory ${dir} (${error})`);
|
|
process.exit(1);
|
|
}
|
|
|
|
}
|
|
|
|
// use lib to upload file
|
|
if (!quiet)
|
|
console.log(`Uploading ${fileName} to ${dest}`);
|
|
await upload.uploadFile(`${baseDir}/${file}`, dest, parseInt(chunkSizeEnv)).then(e => {
|
|
if (!quiet)
|
|
console.log(`Uploaded ${fileName} to ${dest}`);
|
|
}).catch(e => {
|
|
console.log(`Failed to upload ${fileName} to ${dest} (${e})`);
|
|
process.exit(1);
|
|
});
|
|
|
|
// add tags
|
|
if (tagIds.size)
|
|
await addTags(basePath, tagIds, fileName, dest);
|
|
}
|
|
}
|
|
}
|
|
|
|
async function addTags(basePath, tagIds, fileName, location) {
|
|
try {
|
|
// get file id: https://docs.nextcloud.com/server/latest/developer_manual/client_apis/WebDAV/basic.html#requesting-properties
|
|
let response = await axios.request({
|
|
method: "propfind",
|
|
url: `${basePath}/files/${userEnv}/${location}`,
|
|
auth: {
|
|
username: userEnv,
|
|
password: tokenEnv
|
|
},
|
|
data: `<?xml version="1.0"?>
|
|
<d:propfind xmlns:d="DAV:" xmlns:oc="http://owncloud.org/ns" xmlns:nc="http://nextcloud.org/ns">
|
|
<d:prop>
|
|
<oc:fileid/>
|
|
</d:prop>
|
|
</d:propfind>`
|
|
});
|
|
let data = JSON.parse(xml.toJson(response.data));
|
|
let fileId = data["d:multistatus"]["d:response"]["d:propstat"]["d:prop"]["oc:fileid"];
|
|
if (!quiet)
|
|
console.log(`File id of ${fileName} is ${fileId}`);
|
|
|
|
// add tags: https://doc.owncloud.com/server/next/developer_manual/webdav_api/tags.html#assign-a-tag-to-a-file
|
|
for (let [tag, tagId] of tagIds.entries()) {
|
|
await axios.request({
|
|
method: "put",
|
|
url: `${basePath}/systemtags-relations/files/${fileId}/${tagId}`,
|
|
auth: {
|
|
username: userEnv,
|
|
password: tokenEnv
|
|
},
|
|
// 409 conflicted means the tag is already applied
|
|
validateStatus: s => s == 201 || s == 409
|
|
});
|
|
if (!quiet)
|
|
console.log(`Added tag ${tag} to ${fileName}`);
|
|
}
|
|
|
|
} catch (error) {
|
|
console.log(`Failed to assign tags ${tags} to ${fileName} (${error})`);
|
|
process.exit(1);
|
|
}
|
|
}
|