albert/ncore-stats
1
0
Fork 0
mirror of https://github.com/skidoodle/ncore-stats.git synced 2025-02-15 05:09:14 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactor: profiles are now stored in json, added displaynames

Browse source
This commit is contained in:
hasitotabla 2024-10-10 16:01:08 +02:00
parent 8a994c3bb8
commit 6ffa3de9bb
4 changed files with 42 additions and 27 deletions
Download patch file Download diff file Expand all files Collapse all files

63
main.go
View file

@ -7,45 +7,56 @@ import (
"log"
"net/http"
"os"
"strings"
"time"
"github.com/PuerkitoBio/goquery"
"github.com/joho/godotenv"
)
type ProfileData struct {
Owner string `json:"owner"`
Timestamp time.Time `json:"timestamp"`
Rank string `json:"rank"`
Upload string `json:"upload"`
CurrentUpload string `json:"current_upload"`
CurrentDownload string `json:"current_download"`
Points string `json:"points"`
Owner string `json:"owner"`
Timestamp time.Time `json:"timestamp"`
Rank string `json:"rank"`
Upload string `json:"upload"`
CurrentUpload string `json:"current_upload"`
CurrentDownload string `json:"current_download"`
Points string `json:"points"`
}
var (
profiles = map[string]string{}
jsonFile = "data.json"
baseUrl = "https://ncore.pro/profile.php?id="
nick string
pass string
client *http.Client
)
func init() {
godotenv.Load()
nick = os.Getenv("NICK")
pass = os.Getenv("PASS")
for i := 1; i <= 10; i++ {
profileKey := fmt.Sprintf("PROFILE_%d", i)
if profileURL := os.Getenv(profileKey); profileURL != "" {
profiles[profileURL] = profileURL
}
file, err := os.Open("profiles.json")
if err != nil {
log.Fatal(err)
}
defer file.Close()
jsonBytes, _ := io.ReadAll(file)
var tempProfiles map[string]string = make(map[string]string)
json.Unmarshal(jsonBytes, &tempProfiles)
for k, v := range tempProfiles {
profiles[k] = baseUrl + v
}
client = &http.Client{}
}
func fetchProfile(url string) (*ProfileData, error) {
func fetchProfile(url string, displayName string) (*ProfileData, error) {
req, err := http.NewRequest("GET", url, nil)
if err != nil {
return nil, err
@ -57,10 +68,11 @@ func fetchProfile(url string) (*ProfileData, error) {
if err != nil {
return nil, err
}
defer resp.Body.Close()
if resp.StatusCode != http.StatusOK {
return nil, fmt.Errorf("failed to fetch profile: %s", url)
return nil, fmt.Errorf("failed to fetch profile: %s", displayName)
}
doc, err := goquery.NewDocumentFromReader(resp.Body)
@ -68,11 +80,8 @@ func fetchProfile(url string) (*ProfileData, error) {
return nil, err
}
owner := strings.TrimSpace(doc.Find(".fobox_fej").Text())
owner = strings.Replace(owner, " profilja", "", 1)
profile := &ProfileData{
Owner: owner,
Owner: displayName,
Timestamp: time.Now(),
}
@ -108,10 +117,10 @@ func readExistingProfiles() ([]ProfileData, error) {
}
defer file.Close()
var profiles []ProfileData
var existingProfiles []ProfileData
byteValue, _ := io.ReadAll(file)
err = json.Unmarshal(byteValue, &profiles)
return profiles, err
err = json.Unmarshal(byteValue, &existingProfiles)
return existingProfiles, err
}
func logToJSON(profile *ProfileData) error {
@ -134,14 +143,14 @@ func logToJSON(profile *ProfileData) error {
}
func dataHandler(w http.ResponseWriter, r *http.Request) {
profiles, err := readExistingProfiles()
existingProfiles, err := readExistingProfiles()
if err != nil {
http.Error(w, "Could not read data", http.StatusInternalServerError)
return
}
w.Header().Set("Content-Type", "application/json")
json.NewEncoder(w).Encode(profiles)
json.NewEncoder(w).Encode(existingProfiles)
}
func serveHTML(w http.ResponseWriter, r *http.Request) {
@ -149,13 +158,13 @@ func serveHTML(w http.ResponseWriter, r *http.Request) {
}
func main() {
ticker := time.NewTicker(24 * time.Hour)
ticker := time.NewTicker(time.Minute * 30)
defer ticker.Stop()
go func() {
for {
for _, url := range profiles {
profile, err := fetchProfile(url)
for displayName, url := range profiles {
profile, err := fetchProfile(url, displayName)
if err != nil {
log.Println(err)
continue