booru: save to 🌶️ folder if nsfw

This commit is contained in:
end-4 2024-08-19 08:51:15 +07:00
parent b5f98347a7
commit 7d39885ce3
2 changed files with 7 additions and 6 deletions

View file

@ -187,7 +187,6 @@ const BooruPage = (taglist, serviceName = 'Booru') => {
// Show
// const downloadCommand = `wget -O '${imagePath}' '${data.preview_url}'`;
const downloadCommand = `curl -L -o '${imagePath}' '${data.preview_url}'`;
// console.log(downloadCommand)
if (!force && fileExists(imagePath)) showImage();
else Utils.timeout(delay, () => Utils.execAsync(['bash', '-c', downloadCommand])
.then(showImage)
@ -222,13 +221,13 @@ const BooruPage = (taglist, serviceName = 'Booru') => {
name: 'Save image',
icon: 'save',
action: (self) => {
const currentTags = BooruService.queries.at(-1).realTagList.filter(tag => !tag.includes('rating:'));
const currentTags = BooruService.queries.at(-1).realTagList.filter(tag => !tag.includes('rating:'));
const tagDirectory = currentTags.join('_');
let fileExtension = data.file_ext || 'jpg';
const saveCommand = `mkdir -p $(xdg-user-dir PICTURES)/homework/${tagDirectory} && curl -L -o $(xdg-user-dir PICTURES)/homework/${tagDirectory}/${data.md5}.${fileExtension} '${data.file_url}'`;
let fileExtension = data.file_ext || 'jpg';
const saveCommand = `mkdir -p $(xdg-user-dir PICTURES)/homework/${data.is_nsfw ? '🌶️/' : ''}${tagDirectory} && curl -L -o $(xdg-user-dir PICTURES)/homework/${data.is_nsfw ? '🌶️/' : ''}${tagDirectory}/${data.md5}.${fileExtension} '${data.file_url}'`;
execAsync(['bash', '-c', saveCommand])
.then(() => self.label = 'done')
.catch(print);
.then(() => self.label = 'done')
.catch(print);
},
}),
]

View file

@ -129,6 +129,8 @@ class BooruService extends Service {
aspect_ratio: obj.width / obj.height,
id: obj.id,
tags: obj.tags,
rating: obj.rating,
is_nsfw: (obj.rating != 's'),
md5: obj.md5,
preview_url: obj.preview_url,
preview_width: obj.preview_width,