initial commit
All checks were successful
continuous-integration/drone/push Build is passing

This commit is contained in:
shibao 2022-12-15 19:31:12 -05:00
commit 3078d7cf6c
10 changed files with 115 additions and 0 deletions

6
.dockerignore Normal file
View File

@ -0,0 +1,6 @@
.drone.yml
.gitignore
.tool-versions
crontab
crontab.example
docker-compose.yml

44
.drone.yml Normal file
View File

@ -0,0 +1,44 @@
kind: pipeline
type: docker
name: misskey-szuru-bot
steps:
- name: build and publish stable
image: thegeeklab/drone-docker-buildx
privileged: true
settings:
repo: shibaobun/misskey-szuru-bot
purge: true
compress: true
platforms: linux/amd64,linux/arm/v7
username:
from_secret: docker_username
password:
from_secret: docker_password
tags: latest
when:
branch:
- stable
- name: build and publish tagged version
image: thegeeklab/drone-docker-buildx
privileged: true
settings:
repo: shibaobun/misskey-szuru-bot
purge: true
compress: true
platforms: linux/amd64,linux/arm/v7
username:
from_secret: docker_username
password:
from_secret: docker_password
tags:
- ${DRONE_TAG}
when:
event:
- tag
volumes:
- name: docker_sock
host:
path: /var/run/docker.sock

1
.gitignore vendored Normal file
View File

@ -0,0 +1 @@
crontab

1
.tool-versions Normal file
View File

@ -0,0 +1 @@
python 3.11.1

9
Dockerfile Normal file
View File

@ -0,0 +1,9 @@
FROM python:3.11.1-alpine3.17
WORKDIR /usr/src/app
COPY . .
RUN pip install -r requirements.txt
RUN chmod +x entrypoint.sh
ENTRYPOINT ["./entrypoint.sh"]

1
crontab.example Normal file
View File

@ -0,0 +1 @@
0 * * * * python misskey-szuru-bot.py --instance misskey.io --token token_here --booru szurubooru.com --username my-username --apiKey szuru-login-token --query "sort:random"

9
docker-compose.yml Normal file
View File

@ -0,0 +1,9 @@
version: "3"
services:
misskey-szuru-bot:
image: shibaobun/misskey-szuru-bot
container_name: misskey-szuru-bot
restart: always
volumes:
- ./crontab:/usr/src/app/crontab

4
entrypoint.sh Normal file
View File

@ -0,0 +1,4 @@
#!/usr/bin/env bash
set -ou pipefail
crontab ./crontab && crond -f

37
misskey-szuru-bot.py Normal file
View File

@ -0,0 +1,37 @@
#!/usr/bin/python3
import argparse
import pyszuru
import requests
from misskey import Misskey
# arguments
parser = argparse.ArgumentParser(prog='misskey-szuru-bot',
description='Bot that posts a link to a random szurubooru image')
parser.add_argument('-i', '--instance', metavar='instance', type=str,
help='Domain of misskey instance i.e: misskey.io', required=True)
parser.add_argument('-t', '--token', metavar='token', type=str,
help='Token used for posting to misskey instance', required=True)
parser.add_argument('-b', '--booru', metavar='booru', type=str,
help='Domain of szurubooru instance to query i.e: szurubooru.com', required=True)
parser.add_argument('-u', '--username', metavar='username', type=str,
help='Username for szurubooru account', required=True)
parser.add_argument('-a', '--apiKey', metavar='apiKey', type=str,
help='API key for szurubooru account', required=True)
parser.add_argument('-q', '--query', metavar='query', type=str,
help='Query to search szurubooru', default="sort:random")
args = parser.parse_args()
# get post
booru = pyszuru.API(f"https://{args.booru}", username=args.username, token=args.apiKey)
post = next(booru.search_post(args.query))
# compose note
text = f"link: {post.content}"
if post.safety == "unsafe":
text += " (nsfw)"
text += f"\nsource: https://{args.booru}/post/{post.id_}"
# post a note :D
Misskey(args.instance, i=args.token).notes_create(text=text)

3
requirements.txt Normal file
View File

@ -0,0 +1,3 @@
Misskey.py
requests
pyszuru