Skip to content

Instantly share code, notes, and snippets.

@aNNiMON

aNNiMON/.gitignore

Last active Oct 29, 2020
Embed
What would you like to do?
Reddit Images to Telegram
own-modules
main_*.own
redditimages.db
st = conn.createStatement()
st.executeUpdate(
"CREATE TABLE IF NOT EXISTS posts (
id INTEGER PRIMARY KEY AUTOINCREMENT,
post_id STRING NOT NULL,
subreddit STRING NOT NULL,
url STRING NOT NULL,
created_at INTEGER NOT NULL,
sent_at INTEGER NOT NULL
)")
st.executeUpdate(
"CREATE UNIQUE INDEX IF NOT EXISTS url_idx ON posts (url)")
st.close()
stIsPostExists = conn.prepareStatement(
"SELECT COUNT(*) FROM posts
WHERE url = ?")
stAddPost = conn.prepareStatement(
"INSERT INTO posts(post_id, subreddit, url, created_at, sent_at)
VALUES(?, ?, ?, ?, ?)")
def isPostUnique(post) {
stIsPostExists.setString(1, post.url)
rs = stIsPostExists.executeQuery()
return rs.getInt(1) == 0
}
def addPost(post) {
stAddPost.setString(1, post.id)
stAddPost.setString(2, post.sub)
stAddPost.setString(3, post.url)
stAddPost.setLong(4, post.time)
stAddPost.setLong(5, time() / 1000)
stAddPost.executeUpdate()
}
config = {
"token": "1234567890:AABBCCDDEE",
"peer": 1234, // chat_id
"items-in-top": 3,
"subreddits": [
"wholesomememes", "aww"
]
}
include "redditimages.own"
[
{
"id": "18f1894447dfa72000a83011511a817c",
"name": "telegram-bot"
}
]
use ["std"]
if (ARGS.length >= 2 && ARGS[0] == "owm") {
use ["files", "json", "java"]
File = newClass("java.io.File")
runtime = newClass("java.lang.Runtime").getRuntime()
def loadModulesJson(path = "modules.json") {
f = fopen(path, "r")
modules = jsondecode(readText(f))
fclose(f)
return modules
}
def exec(cmd, dir = ".") = runtime.exec(cmd, null, new File(dir))
match (ARGS[1]) {
case "install": {
modulesDir = "own-modules"
if (!exists(modulesDir)) {
mkdir(modulesDir)
}
for module : loadModulesJson() {
print module.name
moduleDir = modulesDir + "/" + module.name
if (!exists(moduleDir)) {
mkdir(moduleDir)
cmd = "git clone https://gist.github.com/" + module.id + ".git " + module.name
exec(cmd, modulesDir)
println " installed"
} else {
exec("git pull origin master", moduleDir)
println " updated"
}
}
}
}
}
class Reddit {
def fetchSubreddit(subreddit, maxItems = 5) {
url = "https://www.reddit.com/r/" + subreddit + ".json"
data = sync(def(ret) = http(url, combine(::jsondecode, ret))).data ?? []
if (!length(data)) return []
return stream(data.children)
.map(def(child) = child.data)
.limit(min(maxItems, data.dist ?? 0))
.map(def(post) = {
"id": post.id,
"sub": subreddit,
"url": post.url,
"time": post.created_utc,
"title": post.title,
"permalink": post.permalink,
"flair_text": arrayKeyExists("link_flair_text", post) ? post.link_flair_text : ""
})
.toArray()
}
def fetchSubreddits(subreddits, maxItems = 5) =
stream(subreddits)
.flatMap(def(r) = this.fetchSubreddit(r, maxItems))
.toArray()
}
use ["std", "math", "http", "json", "functional", "files", "jdbc"]
conn = getConnection("jdbc:sqlite:redditimages.db")
include "own-modules/telegram-bot/TelegramBot.own"
include "Reddit.own"
include "database.own"
bot = new TelegramBot(config.token)
reddit = new Reddit()
subreddits = reddit.fetchSubreddits(config.subreddits, config["items-in-top"])
media = stream(subreddits)
.filter(def(p) = reduce([".jpg", ".png"], false, def(acc, ext) = acc || indexOf(p.url, ext) > 0))
.filter(::isPostUnique)
//.peek(def(p) = bot.sendPhoto(config.peer, p.url))
.peek(::addPost)
.map(def(p) = {
"type": "photo",
"media": p.url,
"caption": getCaption(p),
"parse_mode": "html"
})
.limit(10)
.toArray()
debug(jsonencode(media))
if (length(media) > 0) {
bot.sendMediaGroup(config.peer, media)
}
stIsPostExists.close()
stAddPost.close()
conn.close()
// Helpers
def debug(r) {
// echo(r)
}
def strToHashtag(str) =
str.toLowerCase()
.replaceAll("[^a-z_0-9\s]", "")
.replaceAll("\s+", "_")
def safe(str) = str.replace("&", "&")
.replace("<", "&lt;").replace(">", "&gt;")
def getCaption(post) {
tag = ""
if (length(post.flair_text) > 0) {
tag = " #" + strToHashtag(post.sub + "_" + post.flair_text)
}
return sprintf(
"<a href=\"%s\">%s</a>\n" +
"<a href=\"https://reddit.com%s\">🗨 Comments</a>%s\n" +
"🔎 <a href=\"https://saucenao.com/search.php?url=%s\">SauceNAO</a>, " +
"<a href=\"https://yandex.ua/images/search?rpt=imageview&url=%s\">Yandex</a>",
safe(post.url), safe(post.title),
safe(post.permalink), tag,
urlencode(post.url),
urlencode(post.url)
)
}
sleep(4000)
use "java"
System = newClass("java.lang.System")
System.exit(0)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment