flokati/modules/sc.go

209 lines
6.3 KiB
Go

// vi:ts=4:sts=4:sw=4:noet:tw=72
package modules
import (
"encoding/json"
"fmt"
"html"
"io/ioutil"
"net/http"
"net/url"
"regexp"
"strings"
"time"
"code.dnix.de/an/xlog"
"code.dnix.de/an/flokatilib/util"
"github.com/sorcix/irc"
)
const (
API_URL = "https://robertsspaceindustries.com/api"
API_METHOD = "/stats/getCrowdfundStats"
QUERY_CIT_URL = "https://robertsspaceindustries.com/citizens/"
QUERY_ORG_URL = "https://robertsspaceindustries.com/orgs/"
RE_CIT_NAME = `<div class="info">\s*<p class="entry">\s*<strong class="value">(.*)</strong>\s*</p>\s*<p class="entry">`
RE_CIT_RECORD = `<p class="entry citizen-record">\s*<span class="label">UEE Citizen Record</span>\s*<strong class="value">#(.*)</strong>`
RE_CIT_HANDLE = `<p class="entry">\s*<span class="label">Handle name</span>\s*<strong class="value">(.*)</strong>\s*</p>`
RE_CIT_ORG = `<a href=.*orgs.*class=.*style=.*background-position.*px center.*>(.*)</a>`
RE_CIT_SID = `Spectrum Identification \(SID\)</span>\s*<strong class="value data.*">([A-Z0-9]*)</strong>`
RE_CIT_RANK = `Organization rank</span>\s*<strong class="value data.*">(.*)</strong>`
RE_ORG_NAME = `<h1>(.*) ?/ ?<span class="symbol">`
RE_ORG_COUNT = `<span class="count">(.*) member`
RE_ORG_MODEL = `<li class="model">(.*)</li>`
RE_ORG_FOCUS = `<ul class="focus clearfix">\s*<li class="primary tooltip-wrap">\s*<img src=".*" alt=".*" />\s*<div class="rsi-tooltip">\s*<div class="content">(.*)</div>\s*<span class="bottom"></span>\s*</div>\s*</li>\s*<li class="secondary tooltip-wrap">\s*<img src=".*" alt=".*" />\s*<div class="rsi-tooltip">\s*<div class="content">(.*)</div>\s*<span class="bottom"></span>\s*</div>\s*</li>\s*</ul>`
RE_ORG_COMM = `<li class="commitment">(.*)</li>`
FANS_INT = 1000
FLEET_INT = 1000
FUNDS_INT = 50000
)
var (
fans = 0
fleet = 0
funds = 0
)
func init() {
MsgFuncs["sc"] = scHandleMessage
RunFuncs["sc"] = scScrapeLoop
}
func scHandleMessage(m *irc.Message) {
tok := strings.Split(m.Trailing, " ")
if len(tok) < 1 {
return
}
switch tok[0] {
case "!scstats":
showScStats()
case "!sccit":
if len(tok) > 1 {
showCitizen(tok[1])
}
case "!scorg":
if len(tok) > 1 {
showOrganization(tok[1])
}
default:
}
}
func scScrapeLoop() {
for {
scScraper()
time.Sleep(1 * time.Minute)
}
}
func scScraper() {
var data interface{}
xlog.Info("Scraping SC stats")
resp, err := http.PostForm(API_URL+API_METHOD,
url.Values{"fans": {"true"}, "fleet": {"true"}, "funds": {"true"}})
if err != nil {
xlog.Info("Error: %v", err)
return
}
defer resp.Body.Close()
body, err := ioutil.ReadAll(resp.Body)
if err != nil {
xlog.Info("Error: %v", err)
return
}
err = json.Unmarshal(body, &data)
if err != nil {
xlog.Info("Error: %v", err)
return
}
stats := data.(map[string]interface{})["data"].(map[string]interface{})
//fmt.Println(stats["fans"].(type))
curFans := util.ToInt(stats["fans"])
curFleet := util.ToInt(stats["fleet"])
curFunds := util.ToInt(stats["funds"]) / 100
nextFans := ((fans / FANS_INT) * FANS_INT) + FANS_INT
nextFleet := ((fleet / FLEET_INT) * FLEET_INT) + FLEET_INT
nextFunds := ((funds / FUNDS_INT) * FUNDS_INT) + FUNDS_INT
if curFans >= nextFans {
SayCh <- "*\n[SC] Star Citizens: " + util.NumberToString(curFans, '.')
}
if curFleet >= nextFleet {
SayCh <- "*\n[SC] The UEE Fleet: " + util.NumberToString(curFleet, '.')
}
if curFunds >= nextFunds {
SayCh <- "*\n[SC] Funds raised: " + util.NumberToString(curFunds, '.')
}
fans = curFans
fleet = curFleet
funds = curFunds
}
func showScStats() {
SayCh <- "*\n**SC User and Funding Stats**"
SayCh <- fmt.Sprintf("*\nFans: %s", util.NumberToString(fans, '.'))
SayCh <- fmt.Sprintf("*\nFleet: %s", util.NumberToString(fleet, '.'))
SayCh <- fmt.Sprintf("*\nFunds: $ %s", util.NumberToString(funds, '.'))
}
func showCitizen(handle string) {
resp, err := http.Get(QUERY_CIT_URL + handle)
if err != nil {
xlog.Info("Error: %v", err)
return
}
defer resp.Body.Close()
body, err := ioutil.ReadAll(resp.Body)
if err != nil {
xlog.Info("Error: %v", err)
return
}
reName := regexp.MustCompile(RE_CIT_NAME)
reRecord := regexp.MustCompile(RE_CIT_RECORD)
reHandle := regexp.MustCompile(RE_CIT_HANDLE)
reOrg := regexp.MustCompile(RE_CIT_ORG)
reSid := regexp.MustCompile(RE_CIT_SID)
reRank := regexp.MustCompile(RE_CIT_RANK)
name := reName.FindStringSubmatch(string(body))
record := reRecord.FindStringSubmatch(string(body))
handle_ := reHandle.FindStringSubmatch(string(body))
org := reOrg.FindStringSubmatch(string(body))
sid := reSid.FindStringSubmatch(string(body))
rank := reRank.FindStringSubmatch(string(body))
if len(name) > 1 {
SayCh <- "*\n**Citizen Info**"
SayCh <- "*\n" + "Name: " + html.UnescapeString(string(name[1])) + " [" + string(handle_[1]) + "]"
SayCh <- "*\n" + "URL: " + QUERY_CIT_URL + string(handle_[1])
SayCh <- "*\n" + "UEE #: " + string(record[1])
if len(org) > 1 {
SayCh <- "*\n" + "Organization: " + html.UnescapeString(string(org[1])) + " [" + string(sid[1]) + "]"
SayCh <- "*\n" + "Rank: " + html.UnescapeString(string(rank[1]))
} else {
SayCh <- "*\n" + "Organization: <none>"
}
} else {
SayCh <- "*\n" + "***No Such Citizen***"
}
}
func showOrganization(handle string) {
resp, err := http.Get(QUERY_ORG_URL + handle)
if err != nil {
xlog.Info("Error: %v", err)
return
}
defer resp.Body.Close()
body, err := ioutil.ReadAll(resp.Body)
if err != nil {
xlog.Info("Error: %v", err)
return
}
reName := regexp.MustCompile(RE_ORG_NAME)
reCount := regexp.MustCompile(RE_ORG_COUNT)
reModel := regexp.MustCompile(RE_ORG_MODEL)
reFocus := regexp.MustCompile(RE_ORG_FOCUS)
reComm := regexp.MustCompile(RE_ORG_COMM)
name := reName.FindStringSubmatch(string(body))
count := reCount.FindStringSubmatch(string(body))
model := reModel.FindStringSubmatch(string(body))
focus := reFocus.FindStringSubmatch(string(body))
comm := reComm.FindStringSubmatch(string(body))
if len(name) > 1 {
SayCh <- "*\n**Organization Info**"
SayCh <- "*\nName: " + string(name[1]) + " [" + strings.ToUpper(handle) + "]"
SayCh <- "*\nURL: " + QUERY_ORG_URL + strings.ToUpper(handle)
SayCh <- "*\nMembers: " + string(count[1])
SayCh <- "*\nModel: " + string(model[1])
SayCh <- "*\nCommitment: " + string(comm[1])
SayCh <- "*\nFocus: " + string(focus[1]) + ", " + string(focus[2])
} else {
SayCh <- "*\n***No Such Organization***"
}
}