-
-
Notifications
You must be signed in to change notification settings - Fork 32
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Moved apps handlers into their own package. (#9)
* Rename an error * make a new package with apps * remove sync * slide a few more pices around
- Loading branch information
1 parent
6d13c98
commit 4c01de5
Showing
14 changed files
with
353 additions
and
323 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,185 @@ | ||
| package apps | ||
|
|
||
| import ( | ||
| "context" | ||
| "encoding/json" | ||
| "fmt" | ||
| "log" | ||
| "net/http" | ||
| "path" | ||
| "strconv" | ||
| "time" | ||
|
|
||
| "github.com/gorilla/mux" | ||
| "golift.io/starr/lidarr" | ||
| "golift.io/starr/radarr" | ||
| "golift.io/starr/readarr" | ||
| "golift.io/starr/sonarr" | ||
| ) | ||
|
|
||
| /* This file contains: | ||
| *** The middleware procedure that stores the app interface in a request context. | ||
| *** Procedures to save and fetch an app interface into/from a request content. | ||
| */ | ||
|
|
||
| // Apps is the input configuration to relay requests to Starr apps. | ||
| type Apps struct { | ||
| APIKey string `json:"api_key" toml:"api_key" xml:"api_key" yaml:"api_key"` | ||
| URLBase string `json:"urlbase" toml:"urlbase" xml:"urlbase" yaml:"urlbase"` | ||
| Sonarr []*SonarrConfig `json:"sonarr,omitempty" toml:"sonarr" xml:"sonarr" yaml:"sonarr,omitempty"` | ||
| Radarr []*RadarrConfig `json:"radarr,omitempty" toml:"radarr" xml:"radarr" yaml:"radarr,omitempty"` | ||
| Lidarr []*LidarrConfig `json:"lidarr,omitempty" toml:"lidarr" xml:"lidarr" yaml:"lidarr,omitempty"` | ||
| Readarr []*ReadarrConfig `json:"readarr,omitempty" toml:"readarr" xml:"readarr" yaml:"readarr,omitempty"` | ||
| Router *mux.Router `json:"-" toml:"-" xml:"-" yaml:"-"` | ||
| ErrorLog *log.Logger `json:"-" toml:"-" xml:"-" yaml:"-"` | ||
| } | ||
|
|
||
| // Responder converts all our data to a JSON response. | ||
|
|
||
| // App allows safely storing context values. | ||
| type App string | ||
|
|
||
| // Constant for each app to unique identify itself. | ||
| // These strings are also used as a suffix to the /api/ web path. | ||
| const ( | ||
| Sonarr App = "sonarr" | ||
| Readarr App = "readarr" | ||
| Radarr App = "radarr" | ||
| Lidarr App = "lidarr" | ||
| ) | ||
|
|
||
| // Errors sent to client web requests. | ||
| var ( | ||
| ErrNoTMDB = fmt.Errorf("TMDB ID must not be empty") | ||
| ErrNoGRID = fmt.Errorf("GRID ID must not be empty") | ||
| ErrNoTVDB = fmt.Errorf("TVDB ID must not be empty") | ||
| ErrNoMBID = fmt.Errorf("MBID ID must not be empty") | ||
| ErrNoRadarr = fmt.Errorf("configured radarr ID not found") | ||
| ErrNoSonarr = fmt.Errorf("configured sonarr ID not found") | ||
| ErrNoLidarr = fmt.Errorf("configured lidarr ID not found") | ||
| ErrNoReadarr = fmt.Errorf("configured readarr ID not found") | ||
| ErrExists = fmt.Errorf("the requested item already exists") | ||
| ) | ||
|
|
||
| // APIHandler is our custom handler function for APIs. | ||
| type APIHandler func(r *http.Request) (int, interface{}) | ||
|
|
||
| // HandleAPIpath makes adding API paths a little cleaner. | ||
| // This grabs the app struct and saves it in a context before calling the handler. | ||
| func (a *Apps) HandleAPIpath(app App, api string, next APIHandler, method ...string) { | ||
| if len(method) == 0 { | ||
| method = []string{"GET"} | ||
| } | ||
|
|
||
| id := "{id:[0-9]+}" | ||
| if app == "" { | ||
| id = "" | ||
| } | ||
|
|
||
| // disccordnotifier uses 1-indexes. | ||
| a.Router.Handle(path.Join("/", a.URLBase, "api", string(app), id, api), | ||
| a.checkAPIKey(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { | ||
| start := time.Now() // Capture the starr app request time in a response header. | ||
| switch id, _ := strconv.Atoi(mux.Vars(r)["id"]); { | ||
| default: // unknown app, just run the handler. | ||
| i, m := next(r) | ||
| a.Respond(w, i, m, start) | ||
| case app == Radarr && (id > len(a.Radarr) || id < 1): | ||
| a.Respond(w, http.StatusUnprocessableEntity, fmt.Errorf("%v: %w", id, ErrNoRadarr), start) | ||
| case app == Lidarr && (id > len(a.Lidarr) || id < 1): | ||
| a.Respond(w, http.StatusUnprocessableEntity, fmt.Errorf("%v: %w", id, ErrNoLidarr), start) | ||
| case app == Sonarr && (id > len(a.Sonarr) || id < 1): | ||
| a.Respond(w, http.StatusUnprocessableEntity, fmt.Errorf("%v: %w", id, ErrNoSonarr), start) | ||
| case app == Readarr && (id > len(a.Readarr) || id < 1): | ||
| a.Respond(w, http.StatusUnprocessableEntity, fmt.Errorf("%v: %w", id, ErrNoReadarr), start) | ||
|
|
||
| // These store the application configuration (starr) in a context then pass that into the next method. | ||
| // They retrieve the return code and output, then send a response (a.respond). | ||
| case app == Radarr: | ||
| i, m := next(r.WithContext(context.WithValue(r.Context(), Radarr, a.Radarr[id-1]))) | ||
| a.Respond(w, i, m, start) | ||
| case app == Lidarr: | ||
| i, m := next(r.WithContext(context.WithValue(r.Context(), Lidarr, a.Lidarr[id-1]))) | ||
| a.Respond(w, i, m, start) | ||
| case app == Sonarr: | ||
| i, m := next(r.WithContext(context.WithValue(r.Context(), Sonarr, a.Sonarr[id-1]))) | ||
| a.Respond(w, i, m, start) | ||
| case app == Readarr: | ||
| i, m := next(r.WithContext(context.WithValue(r.Context(), Readarr, a.Readarr[id-1]))) | ||
| a.Respond(w, i, m, start) | ||
| } | ||
| }))).Methods(method...) | ||
| } | ||
|
|
||
| // checkAPIKey drops a 403 if the API key doesn't match. | ||
| func (a *Apps) checkAPIKey(next http.Handler) http.Handler { | ||
| return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { | ||
| if r.Header.Get("X-API-Key") != a.APIKey { | ||
| w.WriteHeader(http.StatusUnauthorized) | ||
| } else { | ||
| next.ServeHTTP(w, r) | ||
| } | ||
| }) | ||
| } | ||
|
|
||
| // InitHandlers activates all our handlers. | ||
| func (a *Apps) InitHandlers() { | ||
| a.radarrHandlers() | ||
| a.readarrHandlers() | ||
| a.lidarrHandlers() | ||
| a.sonarrHandlers() | ||
| } | ||
|
|
||
| // Setup creates request interfaces and sets the timeout for each server. | ||
| func (a *Apps) Setup(timeout time.Duration) { | ||
| for i := range a.Radarr { | ||
| a.Radarr[i].fix(timeout) | ||
| } | ||
|
|
||
| for i := range a.Readarr { | ||
| a.Readarr[i].fix(timeout) | ||
| } | ||
|
|
||
| for i := range a.Sonarr { | ||
| a.Sonarr[i].fix(timeout) | ||
| } | ||
|
|
||
| for i := range a.Lidarr { | ||
| a.Lidarr[i].fix(timeout) | ||
| } | ||
| } | ||
|
|
||
| func (a *Apps) Respond(w http.ResponseWriter, stat int, msg interface{}, start time.Time) { | ||
| w.Header().Set("X-Request-Time", time.Since(start).Round(time.Microsecond).String()) | ||
| w.Header().Set("Content-Type", "application/json") | ||
| w.WriteHeader(stat) | ||
|
|
||
| statusTxt := strconv.Itoa(stat) + ": " + http.StatusText(stat) | ||
|
|
||
| if m, ok := msg.(error); ok { | ||
| a.ErrorLog.Printf("Status: %s, Message: %v", statusTxt, m) | ||
| msg = m.Error() | ||
| } | ||
|
|
||
| b, _ := json.Marshal(map[string]interface{}{"status": statusTxt, "message": msg}) | ||
| _, _ = w.Write(b) | ||
| _, _ = w.Write([]byte("\n")) // curl likes new lines. | ||
| } | ||
|
|
||
| /* Every API call runs one of these methods to find the interface for the respective app. */ | ||
|
|
||
| func getLidarr(r *http.Request) *lidarr.Lidarr { | ||
| return r.Context().Value(Lidarr).(*LidarrConfig).lidarr | ||
| } | ||
|
|
||
| func getRadarr(r *http.Request) *radarr.Radarr { | ||
| return r.Context().Value(Radarr).(*RadarrConfig).radarr | ||
| } | ||
|
|
||
| func getReadarr(r *http.Request) *readarr.Readarr { | ||
| return r.Context().Value(Readarr).(*ReadarrConfig).readarr | ||
| } | ||
|
|
||
| func getSonarr(r *http.Request) *sonarr.Sonarr { | ||
| return r.Context().Value(Sonarr).(*SonarrConfig).sonarr | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.