deranjer/plex-prometheus-exporter
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

updating plex json api parts

Browse Source
This commit is contained in:
deranjer
2023-09-26 15:46:00 -04:00
parent 3c9149f7c0
commit 147ef10c01
2 changed files with 82 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

57
main.go
View File

@@ -1,6 +1,7 @@
package main
import (
"encoding/json"
"fmt"
"log"
"net/http"
@@ -14,6 +15,7 @@ import (
type serverSettings struct {
port int
jsonPort int
plexAddr string
plexToken string
plexTimeout int
@@ -33,6 +35,10 @@ func setupServerSettings() serverSettings {
if err != nil {
log.Fatal("Unable to parse port string to int, failing....")
}
jsonPort, err := strconv.Atoi(envGetter("JSONPORT", "9546"))
if err != nil {
log.Fatal("Unable to parse jsonapi port string to int, failing....")
}
plexToken, value := os.LookupEnv("PLEX_TOKEN")
if !value {
log.Fatal("No plex token provided, failing....")
@@ -48,6 +54,7 @@ func setupServerSettings() serverSettings {
return serverSettings{
port: port,
jsonPort: jsonPort,
plexAddr: envGetter("PLEX_ADDR", "http://localhost:32400"),
plexToken: plexToken,
plexTimeout: plexTimeout,
@@ -56,9 +63,42 @@ func setupServerSettings() serverSettings {
}
}
func setupJSONAPI(plexStatsChannel chan *plexStats, settings serverSettings) {
// Create a new router
mux := http.NewServeMux()
// Create a handler for the `/plex-stats` endpoint
mux.HandleFunc("/plex-stats", func(w http.ResponseWriter, r *http.Request) {
// Create a new plexStats struct
plexStats := <-plexStatsChannel
// Marshal the plexStats struct to JSON
jsonBody, err := json.Marshal(plexStats)
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
// Set the content type to JSON
w.Header().Set("Content-Type", "application/json")
// Write the JSON body to the response
w.Write(jsonBody)
})
// Start the server
fmt.Println("Starting JSON API Server.....")
err := http.ListenAndServe(fmt.Sprintf(":%d", settings.jsonPort), mux)
if err != nil {
log.Fatal("Unable to start JSON Server on port: ", settings.jsonPort, " with err: ", err)
}
}
func main() {
os.Setenv("PORT", "9545")
os.Setenv("JSONPORT", "9546")
os.Setenv("PLEX_ADDR", "https://plex.derajnet.duckdns.org:32400")
os.Setenv("PLEX_TOKEN", "5ezJu5cjnhoAbPJRKngs")
os.Setenv("PLEX_TIMEOUT", "10")
@@ -75,16 +115,21 @@ func main() {
prometheus.MustRegister(plexTranscodeSessions)
prometheus.MustRegister(plexAllSessions)
plexStatsChannel := make(chan *plexStats)
go setupJSONAPI(plexStatsChannel, settings)
go func() {
for {
time.Sleep(time.Second * 60)
time.Sleep(time.Second * 5)
fmt.Println("Collecting info...")
stats := pc.gatherAllStats()
plexActiveSessions.Set(stats.currentSessions)
plexNumMovies.Set(stats.numMovies)
plexNumTVShows.Set(stats.numTV)
plexTranscodeSessions.Set(stats.numTranscodes)
plexAllSessions.Set(stats.numAllSessions)
plexStatsChannel <- stats
plexActiveSessions.Set(stats.CurrentSessions)
plexNumMovies.Set(stats.NumMovies)
plexNumTVShows.Set(stats.NumTV)
plexTranscodeSessions.Set(stats.NumTranscodes)
plexAllSessions.Set(stats.NumAllSessions)
}
}()

43
plex_client.go
View File

@@ -9,12 +9,22 @@ import (
"net/http"
)
type CurrentlyPlayingDetails struct {
CurrentPlayingTitle string `json:"currentPlayingTitle"`
IpAddress string `json:"ipAddress"`
Platform string `json:"platform"`
Product string `json:"product"`
Title string `json:"title"`
State string `json:"state"`
}
type plexStats struct {
currentSessions float64
numMovies float64
numTV float64
numTranscodes float64
numAllSessions float64
CurrentSessions float64 `json:"currentSessions"`
NumMovies float64 `json:"numMovies"`
NumTV float64 `json:"numTV"`
NumTranscodes float64 `json:"numTranscodes"`
NumAllSessions float64 `json:"numAllSessions"`
CurrentlyPlayingDetails CurrentlyPlayingDetails `json:"currentlyPlayingDetails"`
}
type PlexClient struct {
@@ -73,9 +83,9 @@ func (pc *PlexClient) getActiveSessions() *ActiveSessions {
return &currentActiveSessions
}
func (pc *PlexClient) getNumMovies() *ActiveSessions {
func (pc *PlexClient) getNumMovies() *RawPlexModel {
result := pc.sendRequest("/library/sections/3/all", nil) //Hard code library ID for now TODO: Grab from library/sections call
libraryAll := ActiveSessions{}
libraryAll := RawPlexModel{}
err := json.Unmarshal(result, &libraryAll)
if err != nil {
fmt.Println("Error unmarshalling num movies: ", err)
@@ -123,10 +133,19 @@ func (pc *PlexClient) gatherAllStats() *plexStats {
transcodeDetails := pc.getNumTranscodes()
allSessionsData := pc.getServerHistory()
allStats.currentSessions = float64(sessionsData.MediaContainer.Size)
allStats.numMovies = float64(libraryDetails.MediaContainer.Size)
allStats.numTV = float64(tvshowDetails.MediaContainer.Size)
allStats.numTranscodes = float64(transcodeDetails.MediaContainer.Size)
allStats.numAllSessions = float64(allSessionsData.MediaContainer.Size)
allStats.CurrentSessions = float64(sessionsData.MediaContainer.Size)
allStats.NumMovies = float64(libraryDetails.MediaContainer.Size)
allStats.NumTV = float64(tvshowDetails.MediaContainer.Size)
allStats.NumTranscodes = float64(transcodeDetails.MediaContainer.Size)
allStats.NumAllSessions = float64(allSessionsData.MediaContainer.Size)
if len(sessionsData.MediaContainer.Metadata) > 0 {
allStats.CurrentlyPlayingDetails.CurrentPlayingTitle = sessionsData.MediaContainer.Metadata[0].Title
allStats.CurrentlyPlayingDetails.IpAddress = sessionsData.MediaContainer.Metadata[0].Player.Address
allStats.CurrentlyPlayingDetails.Platform = sessionsData.MediaContainer.Metadata[0].Player.Platform
allStats.CurrentlyPlayingDetails.Product = sessionsData.MediaContainer.Metadata[0].Player.Product
allStats.CurrentlyPlayingDetails.State = sessionsData.MediaContainer.Metadata[0].Player.State
allStats.CurrentlyPlayingDetails.Title = sessionsData.MediaContainer.Metadata[0].Player.Title
}
return &allStats
}