feat(lists): steam wishlist (#85)

Author: s0up4200

.gitignore

@@ -3,6 +3,7 @@
 dist
 .dev
 .DS_Store
+config/
 
 # Binaries for programs and plugins
 *.exe

README.md

@@ -112,6 +112,12 @@ lists:
     url: https://api.autobrr.com/lists/metacritic/upcoming-albums
     filters:
       - 20 # Change me
+
+  - name: Steam Wishlist
+    type: steam
+    url: https://store.steampowered.com/wishlist/id/USERNAME/wishlistdata
+    filters:
+      - 20 # Change me
 ```
 
 If you're trying to reach radarr or sonarr hosted on swizzin from some other location, you need to do it like this with basic auth:

internal/domain/config.go

@@ -38,6 +38,7 @@ var (
 	ListTypeMdblist    ListType = "mdblist"
 	ListTypeMetacritic ListType = "metacritic"
 	ListTypePlaintext  ListType = "plaintext"
+	ListTypeSteam      ListType = "steam"
 )
 
 type ArrConfig struct {
@@ -319,4 +320,10 @@ lists:
   #  url: https://api.autobrr.com/lists/metacritic/upcoming-albums
   #  filters:
   #    - 20 # Change me
+
+  #- name: Steam Wishlist
+  #  type: steam
+  #  url: https://store.steampowered.com/wishlist/id/USERNAME/wishlistdata
+  #  filters:
+  #    - 20 # Change me
 `

internal/processor/service.go

@@ -127,6 +127,13 @@ func (s Service) ProcessLists(ctx context.Context, dryRun bool) []string {
 					log.Error().Err(err).Str("type", "plaintext").Str("client", listsClient.Name).Msg("error while processing Plaintext list, continuing with other lists")
 					processingErrors = append(processingErrors, fmt.Sprintf("Plaintext - %s: %v", listsClient.Name, err))
 				}
+
+			case domain.ListTypeSteam:
+				if err := s.steam(ctx, listsClient, dryRun, a); err != nil {
+					log.Error().Err(err).Str("type", "steam").Str("client", listsClient.Name).Msg("error while processing Steam wishlist, continuing with other lists")
+					processingErrors = append(processingErrors, fmt.Sprintf("Steam - %s: %v", listsClient.Name, err))
+				}
+
 			}
 		}
 	}

internal/processor/steam.go

@@ -0,0 +1,92 @@
+package processor
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"net/http"
+	"strings"
+
+	"github.com/autobrr/omegabrr/internal/domain"
+	"github.com/autobrr/omegabrr/pkg/autobrr"
+	"github.com/rs/zerolog/log"
+)
+
+func (s Service) steam(ctx context.Context, cfg *domain.ListConfig, dryRun bool, brr *autobrr.Client) error {
+	l := log.With().Str("type", "steam").Str("client", cfg.Name).Logger()
+
+	if cfg.URL == "" {
+		errMsg := "no URL provided for Steam wishlist"
+		l.Error().Msg(errMsg)
+		return fmt.Errorf(errMsg)
+	}
+
+	l.Debug().Msgf("fetching titles from %s", cfg.URL)
+
+	req, err := http.NewRequestWithContext(ctx, http.MethodGet, cfg.URL, nil)
+	if err != nil {
+		l.Error().Err(err).Msg("could not create new request")
+		return err
+	}
+
+	for k, v := range cfg.Headers {
+		req.Header.Set(k, v)
+	}
+
+	setUserAgent(req)
+
+	resp, err := s.httpClient.Do(req)
+	if err != nil {
+		l.Error().Err(err).Msg("failed to fetch titles")
+		return err
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode != http.StatusOK {
+		l.Error().Msg("failed to fetch titles, non-OK HTTP status received")
+		return fmt.Errorf("failed to fetch titles, non-OK HTTP status received")
+	}
+
+	var data map[string]struct {
+		Name string `json:"name"`
+	}
+
+	if err := json.NewDecoder(resp.Body).Decode(&data); err != nil {
+		l.Error().Err(err).Msg("failed to decode JSON data")
+		return err
+	}
+
+	var titles []string
+	for _, item := range data {
+		titles = append(titles, item.Name)
+	}
+
+	for _, filterID := range cfg.Filters {
+		filterTitles := []string{}
+		for _, title := range titles {
+			filterTitles = append(filterTitles, processTitle(title, cfg.MatchRelease)...)
+		}
+
+		joinedTitles := strings.Join(filterTitles, ",")
+
+		l.Trace().Msgf("%s", joinedTitles)
+
+		if len(joinedTitles) == 0 {
+			l.Debug().Msgf("no titles found for filter: %v", filterID)
+			continue
+		}
+
+		f := autobrr.UpdateFilter{MatchReleases: joinedTitles}
+
+		if !dryRun {
+			if err := brr.UpdateFilterByID(ctx, filterID, f); err != nil {
+				l.Error().Err(err).Msgf("error updating filter: %v", filterID)
+				return err
+			}
+		}
+
+		l.Debug().Msgf("successfully updated filter: %v", filterID)
+	}
+
+	return nil
+}

internal/processor/title.go

@@ -21,7 +21,7 @@ func processTitle(title string, matchRelease bool) []string {
 	// Regex patterns
 	// https://www.regular-expressions.info/unicode.html#category
 	// https://www.ncbi.nlm.nih.gov/staff/beck/charents/hex.html
-	replaceRegexp := regexp.MustCompile(`[\p{P}\p{Z}\x{00C0}-\x{017E}]`)
+	replaceRegexp := regexp.MustCompile(`[\p{P}\p{Z}\x{00C0}-\x{017E}\x{00AE}]`)
 	questionmarkRegexp := regexp.MustCompile(`[?]{2,}`)
 	regionCodeRegexp := regexp.MustCompile(`\(.+\)$`)
 	parenthesesEndRegexp := regexp.MustCompile(`\)$`)

internal/processor/title_test.go

@@ -216,6 +216,16 @@ func Test_processTitle(t *testing.T) {
 			},
 			want: []string{"Sh?gun*2024?", "Sh?gun*2024", "Sh?gun"},
 		},
+		{
+			name: "test_26",
+			args: args{
+				title:        "Pinball FX3 - Bethesda® Pinball",
+				matchRelease: false,
+			},
+			want: []string{
+				"Pinball?FX3*Bethesda*Pinball",
+			},
+		},
 	}
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {