add a cryptocurrency feature

which is able to fetch cryptocurrency price

Author: Hann

discord_bot.go

@@ -23,6 +23,7 @@ func main() {
 
 	store := refs.Store{
 		Asset: &refs.AssetStore{},
+		Ticker: &refs.TickerStore{},
 	}
 
 	// Add handler

handlers/cryptocurrency/filter.go

@@ -0,0 +1,26 @@
+package cryptocurrency
+
+import (
+	coreCrypto "github.com/QOLPlus/core/commands/cryptocurrency"
+)
+
+func arrowOfTicker(ticker coreCrypto.Ticker) string {
+	switch ticker.Change {
+	case "EVEN":
+		return "-"
+	case "RISE":
+		return "▲"
+	case "FALL":
+		return "▼"
+	default:
+		return ""
+	}
+}
+func pmChangeRate(ticker coreCrypto.Ticker) float64 {
+	switch ticker.Change {
+	case "FALL":
+		return -1 * ticker.ChangeRate
+	default:
+		return ticker.ChangeRate
+	}
+}

handlers/cryptocurrency/process.go

@@ -0,0 +1,94 @@
+package cryptocurrency
+
+import (
+	"fmt"
+	coreCrypto "github.com/QOLPlus/core/commands/cryptocurrency"
+	"github.com/QOLPlus/discord-bot/refs"
+	"github.com/bwmarrin/discordgo"
+	"github.com/dustin/go-humanize"
+	"math"
+	"regexp"
+	"strings"
+)
+
+
+var commands = []string{"코인", "크립토", "가상자산", "가상화폐", "CC", "cc", "c"}
+
+var Registry = &refs.HandlerRegistry{
+	Commands: commands,
+	Proc: Process,
+}
+
+var notFoundMessage string = fmt.Sprintf(
+	"가상자산 정보를 찾을 수 없습니다. `[(%s) 검색어]`",
+	strings.Join(commands, "|"),
+)
+
+var KRW string = "KRW"
+
+func Process(store *refs.Store, s *discordgo.Session, m *discordgo.MessageCreate, data []string) {
+	tickerStore := store.Ticker
+	tickerStore.Reload(3600)
+
+	var selectedMarkets []coreCrypto.MarketMaster
+	for _, phrase := range data {
+		if len(phrase) == 0 || KRW == phrase {
+			continue
+		}
+
+		for _, master := range  *tickerStore.Data {
+			if similar(master, phrase) && strings.HasPrefix(master.Market, KRW){
+				selectedMarkets = append(selectedMarkets, master)
+			}
+		}
+	}
+
+	if len(selectedMarkets) == 0 {
+		_, _ = s.ChannelMessageSend(m.ChannelID, notFoundMessage)
+		return
+	}
+
+	for _, market := range selectedMarkets {
+		tickers, err := market.FetchTicker()
+
+		if err != nil {
+			continue
+		}
+
+		for _, ticker := range *tickers {
+			_, _ = s.ChannelMessageSend(m.ChannelID, buildSimpleMessage(market, ticker))
+		}
+	}
+}
+
+func similar(market coreCrypto.MarketMaster, keyword string) bool {
+	match, _ := regexp.MatchString(
+		keyword,
+		strings.Join([]string{
+			market.KoreanName,
+			market.EnglishName,
+			market.Market,
+		}, "|"),
+	)
+	return match
+}
+
+func buildSimpleMessage(market coreCrypto.MarketMaster, ticker coreCrypto.Ticker) string {
+	return fmt.Sprintf(
+		"**%s**: %s %s",
+		market.KoreanName,
+		strings.Join([]string{
+			humanize.Commaf(ticker.TradePrice),
+			fmt.Sprintf("%s%s", arrowOfTicker(ticker), humanize.CommafWithDigits(math.Abs(ticker.ChangePrice), 2)),
+			"(" + fmt.Sprintf("%.2f", pmChangeRate(ticker) * 100) + "%)",
+		}, " "),
+		fmt.Sprintf(
+			"%s / 52주 신고가: %s(%s) / 52주 신저가: %s(%s)",
+			ticker.Market,
+			humanize.CommafWithDigits(ticker.Highest52WeekPrice, 2),
+			ticker.Highest52WeekDate,
+			humanize.CommafWithDigits(ticker.Lowest52WeekPrice, 2),
+			ticker.Lowest52WeekDate,
+		),
+	)
+}

handlers/handler.go

@@ -1,6 +1,7 @@
 package handlers
 
 import (
+	"github.com/QOLPlus/discord-bot/handlers/cryptocurrency"
 	"github.com/QOLPlus/discord-bot/handlers/stock_trend"
 	"github.com/QOLPlus/discord-bot/handlers/weather"
 	"github.com/bwmarrin/discordgo"
@@ -16,6 +17,7 @@ func HandlerFactory(store *refs.Store) interface{} {
 	onHandlers.Register(stock_trend.Registry)
 	onHandlers.Register(weather.Registry)
 	onHandlers.Register(asset.Registry)
+	onHandlers.Register(cryptocurrency.Registry)
 	onCommands := onHandlers.GetKeys()
 
 	return func(s *discordgo.Session, m *discordgo.MessageCreate) {

refs/store.go

@@ -2,18 +2,25 @@ package refs
 
 import (
 	coreAsset "github.com/QOLPlus/core/commands/asset"
+	coreCrypto "github.com/QOLPlus/core/commands/cryptocurrency"
 	"time"
 )
 
 type Store struct {
 	Asset *AssetStore
+	Ticker *TickerStore
 }
 
 type AssetStore struct {
 	Timestamp int64
 	Data      *[]AssetData
 }
 
+type TickerStore struct {
+	Timestamp int64
+	Data      *[]coreCrypto.MarketMaster
+}
+
 type AssetData struct {
 	Master *coreAsset.StockMaster
 	Assets *[]coreAsset.Asset
@@ -47,3 +54,19 @@ func (as *AssetStore) Reload(offset int64) {
 		as.Timestamp = now
 	}
 }
+func (ts *TickerStore) Reload(offset int64) {
+	now := time.Now().Unix()
+	if (now - ts.Timestamp) < offset {
+		return
+	}
+
+	masters, err := coreCrypto.FetchMarketMasters()
+	if err != nil {
+		return
+	}
+
+	if len(*masters) > 0 {
+		ts.Data = masters
+		ts.Timestamp = now
+	}
+}