feat(tags): enable tags to be set in the stasharr settings (#52)

Allows users to set which tags will be associated with studios, scenes, and performers when they are
added to Whisparr using Stasharr.

fix #49

src/builder/PerformerPayloadBuilder.ts

@@ -13,7 +13,7 @@ export class PerformerPayloadBuilder {
     };
   }
 
-  setTags(tags: string[]) {
+  setTags(tags: number[]) {
     this.payload.tags = tags;
     return this;
   }

src/builder/ScenePayloadBuilder.ts

@@ -11,9 +11,15 @@ export class ScenePayloadBuilder {
       monitored: true,
       addOptions: { searchForMovie: true },
       qualityProfileId: 0,
+      tags: [],
     };
   }
 
+  setTags(tags: number[]) {
+    this.payload.tags = tags;
+    return this;
+  }
+
   setTitle(title: string) {
     this.payload.title = title;
     return this;

src/builder/StudioPayloadBuilder.ts

@@ -13,7 +13,7 @@ export class StudioPayloadBuilder {
     };
   }
 
-  setTags(tags: string[]) {
+  setTags(tags: number[]) {
     this.payload.tags = tags;
     return this;
   }

src/components/scene/header/Details.tsx

@@ -21,7 +21,7 @@ const Details = (props: { config: Config; stashId: string }) => {
   const [qualityProfiles] = createResource(
     props,
     async (p: { config: Config }) => {
-      return WhisparrService.getQualityProfiles(p.config);
+      return WhisparrService.qualityProfiles(p.config);
     },
   );
 

src/components/DomainInput.tsx → src/components/settings/DomainInput.tsx

@@ -1,5 +1,5 @@
-import { Stasharr } from '../enums/Stasharr';
-import { useSettings } from '../contexts/useSettings';
+import { Stasharr } from '../../enums/Stasharr';
+import { useSettings } from '../../contexts/useSettings';
 
 const DomainInput = () => {
   const { store, setStore } = useSettings();

src/components/NavbarLink.tsx → src/components/settings/NavbarLink.tsx

@@ -1,4 +1,4 @@
-import { Stasharr } from '../enums/Stasharr';
+import { Stasharr } from '../../enums/Stasharr';
 
 const NavbarLink = () => (
   <a

src/components/ProtocolSwitch.tsx → src/components/settings/ProtocolSwitch.tsx

@@ -1,5 +1,5 @@
-import { Stasharr } from '../enums/Stasharr';
-import { useSettings } from '../contexts/useSettings';
+import { Stasharr } from '../../enums/Stasharr';
+import { useSettings } from '../../contexts/useSettings';
 
 const ProtocolSwitch = () => {
   const { store, setStore } = useSettings();

src/components/QualityProfile.tsx → src/components/settings/QualityProfile.tsx

@@ -1,8 +1,8 @@
 import { For } from 'solid-js';
-import { useSettings } from '../contexts/useSettings';
+import { useSettings } from '../../contexts/useSettings';
 import { Form } from 'solid-bootstrap';
-import { Whisparr } from '../types/whisparr';
-import { Stasharr } from '../enums/Stasharr';
+import { Whisparr } from '../../types/whisparr';
+import { Stasharr } from '../../enums/Stasharr';
 
 const QualityProfileSelect = (props: {
   qualityProfiles: Whisparr.QualityProfile[] | undefined;

src/components/RootFolderPath.tsx → src/components/settings/RootFolderPath.tsx

@@ -1,8 +1,8 @@
 import { For } from 'solid-js';
-import { useSettings } from '../contexts/useSettings';
+import { useSettings } from '../../contexts/useSettings';
 import { Form } from 'solid-bootstrap';
-import { Whisparr } from '../types/whisparr';
-import { Stasharr } from '../enums/Stasharr';
+import { Whisparr } from '../../types/whisparr';
+import { Stasharr } from '../../enums/Stasharr';
 
 const RootFolderPathSelect = (props: {
   rootFolderPaths: Whisparr.RootFolder[] | undefined;

src/components/SearchOnAdd.tsx → src/components/settings/SearchOnAdd.tsx

@@ -1,7 +1,7 @@
 import { For } from 'solid-js';
-import { useSettings } from '../contexts/useSettings';
+import { useSettings } from '../../contexts/useSettings';
 import { Form } from 'solid-bootstrap';
-import { Stasharr } from '../enums/Stasharr';
+import { Stasharr } from '../../enums/Stasharr';
 
 const SearchOnAddSelect = () => {
   const { store, setStore } = useSettings();

src/components/SettingsModal.tsx → src/components/settings/SettingsModal.tsx

@@ -8,16 +8,17 @@ import {
 import { Modal, Button, Alert } from 'solid-bootstrap';
 import ProtocolSwitch from './ProtocolSwitch';
 import DomainInput from './DomainInput';
-import WhisparrApiKeyInput from './ApiKeyInput';
+import WhisparrApiKeyInput from '../ApiKeyInput';
 import { createStore } from 'solid-js/store';
-import { Config } from '../models/Config';
-import WhisparrService from '../service/WhisparrService';
-import { SettingsContext } from '../contexts/useSettings';
+import { Config } from '../../models/Config';
+import WhisparrService from '../../service/WhisparrService';
+import { SettingsContext } from '../../contexts/useSettings';
 import QualityProfileSelect from './QualityProfile';
 import { parseInt } from 'lodash';
 import RootFolderPathSelect from './RootFolderPath';
 import SearchOnAddSelect from './SearchOnAdd';
-import { Stasharr } from '../enums/Stasharr';
+import { Stasharr } from '../../enums/Stasharr';
+import Tags from './Tags';
 
 function SettingsModal(props: { config: Config }) {
   const [show, setShow] = createSignal(false);
@@ -93,6 +94,7 @@ function SettingsModal(props: { config: Config }) {
             <QualityProfileSelect qualityProfiles={qualityProfiles()} />
             <RootFolderPathSelect rootFolderPaths={rootFolderPaths()} />
             <SearchOnAddSelect />
+            <Tags />
           </Show>
         </Modal.Body>
         <Modal.Footer>

src/components/StashInstance.tsx → src/components/settings/StashInstance.tsx

@@ -1,5 +1,5 @@
-import { useSettings } from '../contexts/useSettings';
-import { Stasharr } from '../enums/Stasharr';
+import { useSettings } from '../../contexts/useSettings';
+import { Stasharr } from '../../enums/Stasharr';
 
 const StashInstance = () => {
   const { store, setStore } = useSettings();

src/components/settings/Tags.tsx

@@ -0,0 +1,60 @@
+import { Form } from 'solid-bootstrap';
+import { useSettings } from '../../contexts/useSettings';
+import { Stasharr } from '../../enums/Stasharr';
+import { createMemo, createResource, For } from 'solid-js';
+import { Config } from '../../models/Config';
+import WhisparrService from '../../service/WhisparrService';
+import { parseInt } from 'lodash';
+
+const Tags = () => {
+  const { store, setStore } = useSettings();
+
+  const reactiveStore = createMemo(
+    () => new Config(store.protocol, store.domain, store.whisparrApiKey),
+  );
+
+  const [tags] = createResource(reactiveStore, async (s) => {
+    if (!s.domain || !s.whisparrApiKey) return [];
+    return WhisparrService.tags(s) || [];
+  });
+
+  const handleTagsChange = (
+    // eslint-disable-next-line no-undef
+    selectedOptions: HTMLCollectionOf<HTMLOptionElement>,
+  ) => {
+    if (selectedOptions.length === 0) setStore('tags', []);
+    let tags: number[] = [];
+    for (let i = 0; i < selectedOptions.length; i++) {
+      const option = selectedOptions.item(i);
+      if (option) tags.push(parseInt(option.value, 10));
+    }
+    setStore('tags', tags);
+  };
+  return (
+    <div class="input-group mb-3">
+      <label class="input-group-text">Tags</label>
+      <Form.Select
+        aria-label="Tags select"
+        multiple={true}
+        onChange={(e) => handleTagsChange(e.target.selectedOptions)}
+        // value={store.tags.map((tag) => tag + '')}
+        id={Stasharr.ID.Modal.Tags}
+        data-bs-toggle="tooltip"
+        data-bs-title="Enter tags to be associated with the scenes, studios, and performers added via Stasharr."
+      >
+        <For each={tags()}>
+          {(tag) => (
+            <option
+              value={tag.id}
+              selected={store.tags.filter((t) => t === tag.id).length > 0}
+            >
+              {tag.label}
+            </option>
+          )}
+        </For>
+      </Form.Select>
+    </div>
+  );
+};
+
+export default Tags;

src/controller/NavbarController.ts

@@ -1,5 +1,5 @@
 import { render } from 'solid-js/web';
-import SettingsModal from '../components/SettingsModal';
+import SettingsModal from '../components/settings/SettingsModal';
 import { Config } from '../models/Config';
 import { BaseController } from './BaseController';
 import { StashDB } from '../enums/StashDB';

src/enums/Stasharr.ts

@@ -20,6 +20,7 @@ const qualityProfile = `${stasharrPrefix}-qualityProfile`;
 const rootFolderPath = `${stasharrPrefix}-rootFolderPath`;
 const searchOnAdd = `${stasharrPrefix}-searchOnAdd`;
 const stashDomain = `${stasharrPrefix}-stashDomain`;
+const tags = `${stasharrPrefix}-tags`;
 const headerDetails = `${stasharrPrefix}-headerDetails`;
 const whisparrCardButton = `whisparr-card-button`;
 
@@ -45,6 +46,7 @@ export const Stasharr = {
       RootFolderPath: rootFolderPath,
       SearchOnAdd: searchOnAdd,
       StashDomain: stashDomain,
+      Tags: tags,
     },
   },
   DOMSelector: {

src/models/Config.ts

@@ -8,6 +8,7 @@ export class Config {
   rootFolderPath: string = '';
   searchForNewMovie: boolean = true;
   stashDomain: string = 'http://localhost:9999';
+  tags: number[] = [];
 
   constructor(protocol?: boolean, domain?: string, whisparrApiKey?: string) {
     if (protocol) this.protocol = protocol;

src/service/PerformerService.ts

@@ -74,6 +74,7 @@ export default class PerformerService extends ServiceBase {
       .setQualityProfileId(config.qualityProfile)
       .setRootFolderPath(config.rootFolderPath)
       .setSearchOnAdd(config.searchForNewMovie)
+      .setTags(config.tags)
       .build();
     const response = await ServiceBase.request(
       config,

src/service/SceneService.ts

@@ -163,6 +163,7 @@ export default class SceneService extends ServiceBase {
       .setQualityProfileId(config.qualityProfile)
       .setRootFolderPath(config.rootFolderPath)
       .setSearchForMovie(config.searchForNewMovie)
+      .setTags(config.tags)
       .build();
     let response;
     try {

src/service/StudioService.ts

@@ -69,6 +69,7 @@ export default class StudioService extends ServiceBase {
       .setQualityProfileId(config.qualityProfile)
       .setRootFolderPath(config.rootFolderPath)
       .setSearchOnAdd(config.searchForNewMovie)
+      .setTags(config.tags)
       .build();
     const response = await ServiceBase.request(
       config,

src/service/WhisparrService.ts

@@ -61,26 +61,6 @@ export default class WhisparrService extends ServiceBase {
     }
   }
 
-  static async getQualityProfiles(
-    config: Config,
-  ): Promise<Whisparr.QualityProfile[]> {
-    const endpoint = 'qualityProfile';
-    const response = await ServiceBase.request(
-      config,
-      endpoint,
-      'GET',
-      undefined,
-      {
-        'Content-Type': 'application/json',
-      },
-    )
-      .then((response) => response.response)
-      .then((json) => {
-        return json as Whisparr.QualityProfile[];
-      });
-    return response;
-  }
-
   static async qualityProfiles(
     config: Config,
   ): Promise<null | Whisparr.QualityProfile[]> {
@@ -95,60 +75,6 @@ export default class WhisparrService extends ServiceBase {
     return response.response as Whisparr.QualityProfile[];
   }
 
-  static async getQualityProfilesForSelectMenu(
-    config: Config,
-  ): Promise<{ id: number; name: string }[]> {
-    let options: { id: number; name: string }[] = [];
-    await WhisparrService.getQualityProfiles(config).then(
-      (response: Whisparr.QualityProfile[]) => {
-        response.forEach((qualityProfile) => {
-          options.push({
-            id: qualityProfile.id,
-            name: qualityProfile.name,
-          });
-        });
-      },
-    );
-    return options;
-  }
-
-  static async getRootFolderPathsForSelectMenu(
-    config: Config,
-  ): Promise<{ id: number; name: string }[]> {
-    let options: { id: number; name: string }[] = [];
-
-    await WhisparrService.getRootFolders(config).then(
-      (response: Whisparr.RootFolder[]) => {
-        response.forEach((rootFolder) => {
-          options.push({
-            id: rootFolder.id,
-            name: rootFolder.path,
-          });
-        });
-      },
-    );
-
-    return options;
-  }
-
-  static async getRootFolders(config: Config): Promise<Whisparr.RootFolder[]> {
-    const endpoint = 'rootFolder';
-    const response = await ServiceBase.request(
-      config,
-      endpoint,
-      'GET',
-      undefined,
-      {
-        'Content-Type': 'application/json',
-      },
-    )
-      .then((response) => response.response)
-      .then((json) => {
-        return json as Whisparr.RootFolder[];
-      });
-    return response;
-  }
-
   static async rootFolderPaths(
     config: Config,
   ): Promise<null | Whisparr.RootFolder[]> {
@@ -162,4 +88,16 @@ export default class WhisparrService extends ServiceBase {
     }
     return response.response as Whisparr.RootFolder[];
   }
+
+  static async tags(config: Config): Promise<Whisparr.Tag[] | null> {
+    const endpoint = 'tag';
+    let response;
+    try {
+      response = await ServiceBase.request(config, endpoint, 'GET', undefined);
+    } catch (e) {
+      console.log('Error getting Tags: ', e);
+      return null;
+    }
+    return response.response as Whisparr.Tag[];
+  }
 }