494 lines
13 KiB
JavaScript
494 lines
13 KiB
JavaScript
const api = require('@libs/api-action')
|
|
const DB = require('@libs/database')
|
|
const axios = require('axios')
|
|
|
|
async function run (fn = null, txt, source) {
|
|
if (!fn || typeof fn !== 'function') return null
|
|
let res = null
|
|
let db = null
|
|
try {
|
|
db = await DB.connect()
|
|
res = await fn(txt, source, db)
|
|
} catch (err) {
|
|
console.log(err)
|
|
}
|
|
if (db !== null && 'release' in db) db.release()
|
|
return res
|
|
}
|
|
/**
|
|
* add group to database
|
|
* @param {string} txt command body format => groupName notifyEnable(0,1)
|
|
* @param {object} source
|
|
* @param {object} db
|
|
*/
|
|
const addGroup = async (txt, source = {}, db) => {
|
|
if (!db) return null
|
|
if (!('type' in source) || !('groupId' in source) || !('userId' in source)) return null
|
|
let {groupId, userId} = source
|
|
let arr = txt.split(' ')
|
|
if (arr.length < 2 || !isFinite(arr[1])) return null
|
|
let name = arr[0]
|
|
let notify = arr[1]
|
|
if (typeof notify === 'string') notify = parseInt(notify)
|
|
if (notify !== 0 && notify !== 1) return null
|
|
let text = `select "id" from "public"."line_group" where "id" = $1`
|
|
let values = [groupId]
|
|
let result = await db.query({
|
|
text,
|
|
values
|
|
})
|
|
if (result.rowCount > 0) {
|
|
let reply = 'group exists'
|
|
return { reply }
|
|
}
|
|
text = `insert into "public"."line_group" ("id", "name", "notify", "owner", "ctime", "mtime") values ($1, $2, $3, $4, now(), now())`
|
|
values = [groupId, name, (notify === 1), userId]
|
|
await db.query({
|
|
text,
|
|
values
|
|
})
|
|
let reply = 'add success'
|
|
return { reply }
|
|
}
|
|
|
|
/**
|
|
* add facebook page notify to group
|
|
* @param {string} txt command body format => pageid tmpl
|
|
* @param {object} source
|
|
* @param {object} db
|
|
*/
|
|
const addPage = async (txt, source = {}, db) => {
|
|
if (!db) return null
|
|
if (!('type' in source) || !('groupId' in source) || !('userId' in source)) return null
|
|
let {groupId, userId} = source
|
|
let arr = txt.split(' ')
|
|
if (arr.length < 2) return null
|
|
let page = arr[0]
|
|
let tmpl = arr.slice(1).join(' ')
|
|
let text = `select "id","owner" from "public"."line_group" where "id" = $1`
|
|
let values = [groupId]
|
|
let result = await db.query({
|
|
text,
|
|
values
|
|
})
|
|
if (result.rowCount === 0) {
|
|
let reply = 'group not register'
|
|
return { reply }
|
|
}
|
|
if (result.rows[0].owner !== userId) {
|
|
let reply = 'not owner'
|
|
return { reply }
|
|
}
|
|
// check page exists
|
|
text = `select "id" from "public"."facebook_page" where "id"=$1`
|
|
values = [page]
|
|
let fb = await db.query({
|
|
text,
|
|
values
|
|
})
|
|
// if no page data insert page data and insert rt
|
|
if (fb.rowCount === 0) {
|
|
text = `insert into "public"."facebook_page" ("id", "lastpost", "ctime", "mtime") values
|
|
($1, '', now(), now())`
|
|
values = [page]
|
|
await db.query({
|
|
text,
|
|
values
|
|
})
|
|
text = `insert into "public"."line_fb_rt" ("line", "facebook", "tmpl") values ($1, $2, $3)`
|
|
values = [groupId, page, tmpl]
|
|
await db.query({
|
|
text,
|
|
values
|
|
})
|
|
} else {
|
|
// check rt exists
|
|
text = `select rt.* from "public"."line_fb_rt" rt
|
|
left join "public"."line_group" line
|
|
on line."id" = rt."line"
|
|
left join "public"."facebook_page" fb
|
|
on fb."id" = rt."facebook"
|
|
where
|
|
fb."id" = $1
|
|
and line."id" = $2`
|
|
values = [page, groupId]
|
|
let rt = await db.query({
|
|
text,
|
|
values
|
|
})
|
|
if (rt.rowCount === 0) {
|
|
text = `insert into "public"."line_fb_rt" ("line", "facebook", "tmpl") values ($1, $2, $3)`
|
|
values = [groupId, page, tmpl]
|
|
await db.query({
|
|
text,
|
|
values
|
|
})
|
|
}
|
|
}
|
|
|
|
let reply = 'add page success'
|
|
return { reply }
|
|
}
|
|
|
|
/**
|
|
* del facebook page notify from group
|
|
* @param {string} txt command body format => pageid tmpl
|
|
* @param {object} source
|
|
* @param {object} db
|
|
*/
|
|
const delPage = async (txt = '', source = {}, db) => {
|
|
if (!db) return null
|
|
if (!('type' in source) || !('groupId' in source) || !('userId' in source)) return null
|
|
let {groupId, userId} = source
|
|
let arr = txt.split(' ')
|
|
if (arr.length < 1) return null
|
|
let page = arr[0]
|
|
let text = `select "id","owner" from "public"."line_group" where "id" = $1`
|
|
let values = [groupId]
|
|
let result = await db.query({
|
|
text,
|
|
values
|
|
})
|
|
if (result.rowCount === 0) {
|
|
let reply = 'group not register'
|
|
return { reply }
|
|
}
|
|
if (result.rows[0].owner !== userId) {
|
|
let reply = 'not owner'
|
|
return { reply }
|
|
}
|
|
|
|
text = `select count(rt.*) as c from "public"."line_fb_rt" rt
|
|
left join "public"."facebook_page" fb
|
|
on fb."id" = rt."facebook"
|
|
where
|
|
fb."id" = $1`
|
|
values = [page]
|
|
let count = await db.query({
|
|
text,
|
|
values
|
|
})
|
|
if (count.rowCount === 0 || count.rows[0].c === 0) return null
|
|
|
|
text = `select rt.* from "public"."line_fb_rt" rt
|
|
left join "public"."facebook_page" fb
|
|
on fb."id" = rt."facebook"
|
|
where
|
|
fb."id" = $1
|
|
and rt."line" = $2`
|
|
values = [page, groupId]
|
|
let rt = await db.query({
|
|
text,
|
|
values
|
|
})
|
|
if (rt.rowCount === 0) return null
|
|
if (rt.rowCount == 1 && count.rows[0].c == 1) { // eslint-disable-line
|
|
let text = `delete from "public"."facebook_page" where "id" = $1`
|
|
let values = [page]
|
|
await db.query({
|
|
text,
|
|
values
|
|
})
|
|
} else {
|
|
let text = `delete from "public"."line_fb_rt" where "line" = $1 and "facebook" = $2`
|
|
let values = [groupId, page]
|
|
await db.query({
|
|
text,
|
|
values
|
|
})
|
|
}
|
|
|
|
return {
|
|
reply: 'delete page notify success'
|
|
}
|
|
}
|
|
|
|
/**
|
|
* add twitch channel notify to group
|
|
* @param {string} txt command body format => twitch_user type tmpl
|
|
* @param {object} source
|
|
* @param {object} db
|
|
*/
|
|
const addTwitch = async (txt = '', source = {}, db) => {
|
|
if (!db) return null
|
|
if (!('type' in source) || !('groupId' in source) || !('userId' in source)) return null
|
|
let {groupId, userId} = source
|
|
let arr = txt.split(' ')
|
|
if (arr.length < 3) return null
|
|
let name = arr[0]
|
|
let type = arr[1]
|
|
let tmpl = arr.slice(2).join(' ')
|
|
|
|
if (['live'].indexOf(type) < 0) return null
|
|
|
|
let text = `select "id","owner" from "public"."line_group" where "id" = $1`
|
|
let values = [groupId]
|
|
let result = await db.query({
|
|
text,
|
|
values
|
|
})
|
|
if (result.rowCount === 0) {
|
|
let reply = 'group not register'
|
|
return { reply }
|
|
}
|
|
if (result.rows[0].owner !== userId) {
|
|
let reply = 'not owner'
|
|
return { reply }
|
|
}
|
|
|
|
let twitchUser = await api.twitchPublic.getUserIDByName(name)
|
|
if (twitchUser === null) return null
|
|
|
|
// check channel count
|
|
text = `select "id" from "public"."twitch_channel" where "id" = $1`
|
|
values = [twitchUser.id]
|
|
let twch = await db.query({
|
|
text,
|
|
values
|
|
})
|
|
|
|
if (twch.rowCount === 0) {
|
|
let text = `insert into "public"."twitch_channel" ("id", "name", "ctime", "mtime") values
|
|
($1, $2, now(), now())`
|
|
let values = [twitchUser.id, twitchUser.login]
|
|
await db.query({
|
|
text,
|
|
values
|
|
})
|
|
text = `insert into "public"."line_twitch_rt" ("line", "twitch", "tmpl", "type") values
|
|
($1, $2, $3, $4)`
|
|
values = [groupId, twitchUser.id, tmpl, type]
|
|
await db.query({
|
|
text,
|
|
values
|
|
})
|
|
} else {
|
|
let text = `select rt.* from "public"."line_twitch_rt" rt
|
|
left join "public"."twitch_channel" twitch
|
|
on twitch."id" = rt."twitch"
|
|
left join "public"."line_group" line
|
|
on line."id" = rt."line"
|
|
where
|
|
line."id" = $1
|
|
and twitch."id" = $2
|
|
and rt."type" = $3`
|
|
let values = [groupId, twitchUser.id, type]
|
|
let rt = await db.query({
|
|
text,
|
|
values
|
|
})
|
|
if (rt.rowCount === 0) {
|
|
let text = `insert into "public"."line_twitch_rt" ("line", "twitch", "tmpl", "type") values
|
|
($1, $2, $3, $4)`
|
|
let values = [groupId, twitchUser.id, tmpl, type]
|
|
await db.query({
|
|
text,
|
|
values
|
|
})
|
|
}
|
|
}
|
|
|
|
// api.twitch.registerWebHook(twitchUser.id, type)
|
|
return {
|
|
reply: 'add channel success'
|
|
}
|
|
}
|
|
|
|
/**
|
|
* del twitch channel notify from group
|
|
* @param {string} txt command body format => twitch_user type
|
|
* @param {object} source
|
|
* @param {object} db
|
|
*/
|
|
const delTwitch = async (txt = '', source = {}, db) => {
|
|
if (!db) return null
|
|
if (!('type' in source) || !('groupId' in source) || !('userId' in source)) return null
|
|
let {groupId, userId} = source
|
|
let arr = txt.split(' ')
|
|
if (arr.length < 2) return null
|
|
let twitch = arr[0]
|
|
let type = arr[1]
|
|
let text = `select "id","owner" from "public"."line_group" where "id" = $1`
|
|
let values = [groupId]
|
|
let result = await db.query({
|
|
text,
|
|
values
|
|
})
|
|
if (result.rowCount === 0) {
|
|
let reply = 'group not register'
|
|
return { reply }
|
|
}
|
|
if (result.rows[0].owner !== userId) {
|
|
let reply = 'not owner'
|
|
return { reply }
|
|
}
|
|
|
|
text = `select count(rt.*) as c from "public"."line_twitch_rt" rt
|
|
left join "public"."twitch_channel" twitch
|
|
on twitch."id" = rt."twitch"
|
|
where
|
|
twitch."name" = $1`
|
|
values = [twitch]
|
|
let count = await db.query({
|
|
text,
|
|
values
|
|
})
|
|
if (count.rowCount === 0 || count.rows[0].c === 0) return null
|
|
|
|
text = `select rt.* from "public"."line_twitch_rt" rt
|
|
left join "public"."twitch_channel" twitch
|
|
on twitch."id" = rt."twitch"
|
|
where
|
|
twitch."name" = $1
|
|
and rt."line" = $2`
|
|
values = [twitch, groupId]
|
|
let rt = await db.query({
|
|
text,
|
|
values
|
|
})
|
|
if (rt.rowCount === 0) return null
|
|
if (rt.rowCount == 1 && count.rows[0].c == 1) { // eslint-disable-line
|
|
let text = `delete from "public"."twitch_channel" where "name" = $1`
|
|
let values = [twitch]
|
|
await db.query({
|
|
text,
|
|
values
|
|
})
|
|
} else {
|
|
let text = `select "id" from "public"."twitch_channel" where "name" = $1`
|
|
let values = [twitch]
|
|
let tc = await db.query({
|
|
text,
|
|
values
|
|
})
|
|
if (tc.rowCount > 0) {
|
|
let text = `delete from "public"."line_twitch_rt" where "line" = $1 and "type" = $2 and "twitch" = $3`
|
|
let values = [groupId, type, tc.rows[0].id]
|
|
await db.query({
|
|
text,
|
|
values
|
|
})
|
|
}
|
|
}
|
|
|
|
return {
|
|
reply: 'delete twitch notify success'
|
|
}
|
|
}
|
|
|
|
/**
|
|
* add youtube channel notify to group
|
|
* @param {string} txt command body format => youtube tmpl
|
|
* @param {object} source
|
|
* @param {object} db
|
|
*/
|
|
const addYoutube = async (txt = '', source = {}, db) => {
|
|
if (!db) return null
|
|
if (!('type' in source) || !('groupId' in source) || !('userId' in source)) return null
|
|
let {groupId, userId} = source
|
|
let arr = txt.split(' ')
|
|
if (arr.length < 2) return null
|
|
let id = arr[0]
|
|
let tmpl = arr.slice(1).join(' ')
|
|
let text = `select "id","owner" from "public"."line_group" where "id" = $1`
|
|
let values = [groupId]
|
|
let result = await db.query({
|
|
text,
|
|
values
|
|
})
|
|
if (result.rowCount === 0) {
|
|
let reply = 'group not register'
|
|
return { reply }
|
|
}
|
|
if (result.rows[0].owner !== userId) {
|
|
let reply = 'not owner'
|
|
return { reply }
|
|
}
|
|
|
|
console.log('youtube id ::: ', id)
|
|
let youtubeName = await api.google.queryYoutubeName(id)
|
|
console.log('youtubr name :::: ', youtubeName)
|
|
if (youtubeName === null) return null
|
|
|
|
text = `select "id" from "public"."youtube_channel" where "id" = $1`
|
|
values = [id]
|
|
let ytch = await db.query({
|
|
text,
|
|
values
|
|
})
|
|
|
|
if (ytch.rowCount === 0) {
|
|
let text = `insert into "public"."youtube_channel" ("id", "name", "lastvideo", "expire", "ctime", "mtime") values
|
|
($1, $2, '', -1, now(), now())`
|
|
let values = [id, youtubeName]
|
|
await db.query({
|
|
text, values
|
|
})
|
|
text = `insert into "public"."line_youtube_rt" ("line", "youtube", "tmpl") values ($1, $2, $3)`
|
|
values = [groupId, id, tmpl]
|
|
await db.query({
|
|
text, values
|
|
})
|
|
} else {
|
|
let text = `insert into "public"."line_youtube_rt" ("line", "youtube", "tmpl") values ($1, $2, $3)`
|
|
let values = [groupId, id, tmpl]
|
|
await db.query({
|
|
text, values
|
|
})
|
|
}
|
|
|
|
try {
|
|
await api.google.subYoutube(id)
|
|
} catch (err) {
|
|
console.log(err)
|
|
}
|
|
|
|
return {
|
|
reply: 'add youtube channel success'
|
|
}
|
|
}
|
|
|
|
const image = async (txt, source, db) => {
|
|
if (typeof txt !== 'string' || txt.trim().length === 0) return null
|
|
let imgs = txt.split(';')
|
|
if (imgs.length !== 2) return null
|
|
|
|
try {
|
|
await axios({
|
|
url: imgs[0],
|
|
method: 'head'
|
|
})
|
|
await axios({
|
|
url: imgs[1],
|
|
method: 'head'
|
|
})
|
|
} catch (err) {
|
|
console.log(err)
|
|
return null
|
|
}
|
|
return {
|
|
reply: `$image$${txt}`
|
|
}
|
|
}
|
|
|
|
const hello = async () => {
|
|
return {reply: 'Hello World'}
|
|
}
|
|
|
|
const roll = async () => {
|
|
let r = Math.floor(Math.random() * 10000)
|
|
return {reply: r}
|
|
}
|
|
|
|
module.exports = {
|
|
addgroup: run.bind(null, addGroup),
|
|
addpage: run.bind(null, addPage),
|
|
delpage: run.bind(null, delPage),
|
|
addtwitch: run.bind(null, addTwitch),
|
|
deltwitch: run.bind(null, delTwitch),
|
|
addyoutube: run.bind(null, addYoutube),
|
|
hello: run.bind(null, hello),
|
|
image: run.bind(null, image),
|
|
roll: run.bind(null, roll)
|
|
}
|