Skip to content

Commit

Permalink
update with fear and greed index support
Browse files Browse the repository at this point in the history
  • Loading branch information
@hawooni
hawooni committed Jan 24, 2022
1 parent f99bfb1 commit c3ed412
Show file tree
Hide file tree
Showing 2 changed files with 66 additions and 14 deletions.
2 changes: 1 addition & 1 deletion config/message.js
View file

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

78 changes: 65 additions & 13 deletions src/telegram.js
View file
Original file line number Diff line number Diff line change
Expand Up @@ -9,15 +9,14 @@ const CB_ACTION_PRICE_SYMBOL = 'price-symbol'
const CB_ACTION_CHART_SYMBOL = 'chart-symbol'
const CB_ACTION_CHART_INTERVAL = 'chart-interval'
const CB_ACTION_CHART_STUDIES = 'chart-studies'
const CB_ACTION_FG_INDEX = 'fear-greed-index'

module.exports = (log, argv, version, setting) => {
const TelegramBot = require('@lib/telegram')

const chatLimit = new NodeCache({ stdTTL: 60 }) // rate limit ttl per minute
const teleBot = new TelegramBot(argv.telegramToken, { polling: true })

const BOT_NAME = process.env.BOT_NAME || setting.BOT_NAME

log.verbose('\nRunning Telegram Server ✓')

axios.defaults.headers.common['user-agent'] = `cryptoOpnBot/${version}`
Expand All @@ -35,6 +34,10 @@ module.exports = (log, argv, version, setting) => {
command: '/chart',
description: 'Advanced Chart.',
},
{
command: '/fear_greed_index',
description: 'Crypto Market Sentiment Index',
},
{
command: '/example',
description: 'Command examples.',
Expand All @@ -56,7 +59,7 @@ module.exports = (log, argv, version, setting) => {
if (text === '/start') {
return teleBot.sendMessage(
chat.id,
MESSAGE.TELEGRAM_START.replace('{{botName}}', BOT_NAME),
MESSAGE.TELEGRAM_START.replace('{{botName}}', setting.BOT_NAME),
{
parse_mode: 'HTML',
}
Expand All @@ -69,7 +72,6 @@ module.exports = (log, argv, version, setting) => {
.then((res) =>
teleBot.sendPhoto(chat.id, res.data, {
caption: getPriceCaption(eSymbol),
parse_mode: 'HTML',
reply_markup: {
inline_keyboard: getChunkInputObjs(
CB_ACTION_PRICE_SYMBOL,
Expand All @@ -88,7 +90,6 @@ module.exports = (log, argv, version, setting) => {
.then((res) =>
teleBot.sendPhoto(chat.id, res.data, {
caption: getChartCaption(eSymbol),
parse_mode: 'HTML',
reply_markup: {
inline_keyboard: getChunkInputObjs(
CB_ACTION_CHART_SYMBOL,
Expand All @@ -109,7 +110,6 @@ module.exports = (log, argv, version, setting) => {
.then((res) =>
teleBot.sendPhoto(chat.id, res.data, {
caption: getPriceCaption(eSymbol, interval),
parse_mode: 'HTML',
})
)
} else if (text?.startsWith('/chart')) {
Expand All @@ -123,7 +123,18 @@ module.exports = (log, argv, version, setting) => {
.then((res) =>
teleBot.sendPhoto(chat.id, res.data, {
caption: getChartCaption(eSymbol, interval, splitStudies),
parse_mode: 'HTML',
})
)
} else if (text?.startsWith('/fear_greed_index')) {
return axios
.get(setting.CRYPTO_FEAR_GREED_INDEX_URL, {
responseType: 'arraybuffer',
})
.then((res) =>
teleBot.sendPhoto(chat.id, res.data, {
reply_markup: {
inline_keyboard: getInputReload(CB_ACTION_FG_INDEX),
},
})
)
} else if (text?.startsWith('/example')) {
Expand Down Expand Up @@ -202,6 +213,13 @@ module.exports = (log, argv, version, setting) => {
)
)
}
} else if (cbKey === CB_ACTION_FG_INDEX) {
return reqReloadEditMsgPhoto(
CB_ACTION_FG_INDEX,
setting.CRYPTO_FEAR_GREED_INDEX_URL,
chat.id,
message_id
)
} else {
throw Error('Unable to get symbol from callback query.')
}
Expand Down Expand Up @@ -300,8 +318,7 @@ module.exports = (log, argv, version, setting) => {
* @returns {String} price image caption
*/
function getPriceCaption(eSymbol, interval = null) {
const url = getPriceImgApiUrl(eSymbol, interval) // external to use default size
return `<a href="${url}">${eSymbol.toUpperCase()} ${interval || setting.DEFAULT_PRICE_INTERVAL}</a>` // prettier-ignore
return `${eSymbol.toUpperCase()} ${interval || setting.DEFAULT_PRICE_INTERVAL}`
}

/**
Expand All @@ -311,11 +328,11 @@ module.exports = (log, argv, version, setting) => {
* @returns {String} chart image caption
*/
function getChartCaption(eSymbol, interval = null, studies = null) {
const url = getChartImgApiUrl(eSymbol, interval, studies) // external to use default size
const dInterval = interval || setting.DEFAULT_CHART_INTERVAL
const dStudies = studies || setting.DEFAULT_CHART_STUDIES
const studyIds = lodash.uniq(dStudies.map((dStudy) => dStudy.split(':')[0]))
return `<a href="${url}">${eSymbol.toUpperCase()} ${dInterval} ${studyIds.toString()}</a>`

return `${eSymbol.toUpperCase()} ${dInterval} ${studyIds.toString()}`
}

/**
Expand All @@ -338,7 +355,6 @@ module.exports = (log, argv, version, setting) => {
.then((res) => {
const opt = getEditMsgPhotoOpt(chatId, msgId, res.data, {
caption: getPriceCaption(...inputs),
parse_mode: 'HTML',
})
if (inputKeys) {
opt.qs.reply_markup = {
Expand All @@ -349,6 +365,28 @@ module.exports = (log, argv, version, setting) => {
})
}

/**
* @param {String} cbKey
* @param {String} url
* @param {Integer} chatId
* @param {Integer} msgId
* @returns {Promise}
*/
function reqReloadEditMsgPhoto(cbKey, url, chatId, msgId) {
return axios
.get(url, {
responseType: 'arraybuffer',
})
.then((res) => {
const opt = getEditMsgPhotoOpt(chatId, msgId, res.data)

opt.qs.reply_markup = {
inline_keyboard: getInputReload(cbKey),
}
return teleBot._request('editMessageMedia', opt)
})
}

/**
* @param {Integer} chatId
* @param {Integer} msgId
Expand All @@ -366,7 +404,6 @@ module.exports = (log, argv, version, setting) => {
.then((res) => {
const opt = getEditMsgPhotoOpt(chatId, msgId, res.data, {
caption: getChartCaption(...inputs),
parse_mode: 'HTML',
})
if (inputKeys) {
opt.qs.reply_markup = {
Expand Down Expand Up @@ -427,6 +464,21 @@ module.exports = (log, argv, version, setting) => {
]
}

/**
* @param {String} cbKey
* @returns {Array}
*/
function getInputReload(cbKey) {
return [
[
{
text: '↻ ' + new Date().toISOString(), // must be a unique message (400 Bad Request: message is not modified)
callback_data: cbKey,
},
],
]
}

/**
* generate inline keyboard inputs
*
Expand Down

0 comments on commit c3ed412

Please sign in to comment.