This commit is contained in:
parent
8d030f4db3
commit
b69323fc00
1 changed files with 0 additions and 207 deletions
207
modules/sc.go
207
modules/sc.go
|
|
@ -1,207 +0,0 @@
|
|||
// vi:ts=4:sts=4:sw=4:noet:tw=72
|
||||
|
||||
package modules
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"flag"
|
||||
"fmt"
|
||||
"html"
|
||||
"io/ioutil"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"regexp"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"git.dnix.de/an/flokati/util"
|
||||
"git.dnix.de/an/xlog"
|
||||
)
|
||||
|
||||
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
|
||||
targetChannel = flag.String("sc_target_channel", "", "Default channel for stats output")
|
||||
)
|
||||
|
||||
func init() {
|
||||
MsgFuncs["sc"] = scHandleMessage
|
||||
RunFuncs["sc"] = scScrapeLoop
|
||||
}
|
||||
|
||||
func scHandleMessage(m *Message) {
|
||||
tok := strings.Split(m.Text, " ")
|
||||
if len(tok) < 1 {
|
||||
return
|
||||
}
|
||||
switch tok[0] {
|
||||
case "!scstats":
|
||||
showScStats(m.Channel)
|
||||
case "!sccit":
|
||||
if len(tok) > 1 {
|
||||
showCitizen(tok[1], m.Channel)
|
||||
}
|
||||
case "!scorg":
|
||||
if len(tok) > 1 {
|
||||
showOrganization(tok[1], m.Channel)
|
||||
}
|
||||
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 <- *targetChannel + "\n[SC] Star Citizens: " + util.NumberToString(curFans, '.')
|
||||
}
|
||||
if curFleet >= nextFleet {
|
||||
SayCh <- *targetChannel + "\n[SC] The UEE Fleet: " + util.NumberToString(curFleet, '.')
|
||||
}
|
||||
if curFunds >= nextFunds {
|
||||
SayCh <- *targetChannel + "\n[SC] Funds raised: " + util.NumberToString(curFunds, '.')
|
||||
}
|
||||
|
||||
fans = curFans
|
||||
fleet = curFleet
|
||||
funds = curFunds
|
||||
}
|
||||
|
||||
func showScStats(channel string) {
|
||||
SayCh <- channel + "\n**SC User and Funding Stats**"
|
||||
SayCh <- fmt.Sprintf("%s\nFans: %s", channel, util.NumberToString(fans, '.'))
|
||||
SayCh <- fmt.Sprintf("%s\nFleet: %s", channel, util.NumberToString(fleet, '.'))
|
||||
SayCh <- fmt.Sprintf("%s*\nFunds: $ %s", channel, util.NumberToString(funds, '.'))
|
||||
}
|
||||
|
||||
func showCitizen(handle, channel 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 <- channel + "\n**Citizen Info**"
|
||||
SayCh <- channel + "\nName: " + html.UnescapeString(string(name[1])) + " [" + string(handle_[1]) + "]"
|
||||
SayCh <- channel + "\nURL: " + QUERY_CIT_URL + string(handle_[1])
|
||||
SayCh <- channel + "\nUEE #: " + string(record[1])
|
||||
if len(org) > 1 {
|
||||
SayCh <- channel + "\nOrganization: " + html.UnescapeString(string(org[1])) + " [" + string(sid[1]) + "]"
|
||||
SayCh <- channel + "\nRank: " + html.UnescapeString(string(rank[1]))
|
||||
} else {
|
||||
SayCh <- channel + "\nOrganization: <none>"
|
||||
}
|
||||
} else {
|
||||
SayCh <- channel + "\n***No Such Citizen***"
|
||||
}
|
||||
}
|
||||
|
||||
func showOrganization(handle, channel 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 <- channel + "\n**Organization Info**"
|
||||
SayCh <- channel + "\nName: " + string(name[1]) + " [" + strings.ToUpper(handle) + "]"
|
||||
SayCh <- channel + "\nURL: " + QUERY_ORG_URL + strings.ToUpper(handle)
|
||||
SayCh <- channel + "\nMembers: " + string(count[1])
|
||||
SayCh <- channel + "\nModel: " + string(model[1])
|
||||
SayCh <- channel + "\nCommitment: " + string(comm[1])
|
||||
SayCh <- channel + "\nFocus: " + string(focus[1]) + ", " + string(focus[2])
|
||||
} else {
|
||||
SayCh <- channel + "\n***No Such Organization***"
|
||||
}
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue