hpz937/zurg
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Report traffic for all download tokens

Browse Source
This commit is contained in:
Ben Adrian Sarmiento
2024-07-04 04:14:39 +02:00
parent 49432dd810
commit 38a1a9e096
5 changed files with 59 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
internal/handlers/home.go
View File

@@ -43,7 +43,7 @@ type RootResponse struct {
DownloadTokens []string `json:"download_tokens"`
IDsToDelete []string `json:"ids_to_delete"`
Hosts []string `json:"hosts"`
TrafficServedPerAPI uint64 `json:"traffic_served_per_api"`
TrafficServedPerAPI map[string]uint64 `json:"traffic_served_per_api"`
}
func (zr *Handlers) generateResponse(resp http.ResponseWriter, req *http.Request) (*RootResponse, error) {
@@ -77,15 +77,16 @@ func (zr *Handlers) generateResponse(resp http.ResponseWriter, req *http.Request
sortedIDs := zr.torMgr.OnceDoneBin.ToSlice()
sort.Strings(sortedIDs)
trafficDetails, err := zr.rd.GetTrafficDetails()
if err != nil {
http.Error(resp, err.Error(), http.StatusInternalServerError)
return nil, err
}
var trafficFromAPI int64
trafficFromAPI = 0
if _, ok := trafficDetails["real-debrid.com"]; ok {
trafficFromAPI = trafficDetails["real-debrid.com"]
trafficFromAPI := make(map[string]uint64)
for _, token := range zr.rd.TokenManager.GetAllTokens() {
trafficDetails, err := zr.rd.GetTrafficDetails(token)
if err != nil {
trafficDetails = make(map[string]uint64)
}
trafficFromAPI[token] = 0
if _, ok := trafficDetails["real-debrid.com"]; ok {
trafficFromAPI[token] = trafficDetails["real-debrid.com"]
}
}
userInfo.Premium = userInfo.Premium / 86400
@@ -105,7 +106,7 @@ func (zr *Handlers) generateResponse(resp http.ResponseWriter, req *http.Request
Infuse: fmt.Sprintf("//%s/infuse/", req.Host),
Logs: fmt.Sprintf("//%s/logs/", req.Host),
UserInfo: userInfo,
TrafficServedPerAPI: uint64(trafficFromAPI),
TrafficServedPerAPI: trafficFromAPI,
LibrarySize: allTorrents.Count(),
TorrentsToRepair: repairQueueStr,
MemAlloc: bToMb(mem.Alloc),
@@ -220,10 +221,6 @@ func (zr *Handlers) handleHome(resp http.ResponseWriter, req *http.Request) {
<td>Process ID</td>
<td colspan="2">%d</td>
</tr>
<tr>
<td>Traffic Served (main token)</td>
<td colspan="2">%d MB (%d MB since startup)</td>
</tr>
<tr>
<td>Traffic Served (zurg)</td>
<td colspan="2">%d MB</td>
@@ -234,11 +231,22 @@ func (zr *Handlers) handleHome(resp http.ResponseWriter, req *http.Request) {
response.Sys,
response.NumGC,
response.PID,
bToMb(response.TrafficServedPerAPI), // traffic served *api*
bToMb(response.TrafficServedPerAPI-zr.downloader.TrafficOnStartup.Load()), // traffic served *api* since startup
bToMb(zr.downloader.TrafficServed.Load()), // traffic served *zurg*
bToMb(zr.downloader.TrafficServed.Load()), // traffic served *zurg*
)
for token, traffic := range response.TrafficServedPerAPI {
trafficOnStartup, _ := zr.downloader.TrafficOnStartup.Get(token)
out += fmt.Sprintf(`
<tr>
<td>Traffic Served (%s)</td>
<td colspan="2">%d MB (%d MB since startup)</td>
</tr>`,
utils.MaskToken(token)[40:],
bToMb(traffic), // traffic served *api*
bToMb(traffic-trafficOnStartup.Load()), // traffic served *api* since startup
)
}
out += fmt.Sprintf(`
<tr>
<td rowspan="3">Sponsor Zurg</td>