@muhammedaksam/easiarr 0.1.10 → 0.3.0

package/src/data/trash-profiles.ts

@@ -0,0 +1,252 @@
+/**
+ * TRaSH Guide Quality Profile Presets
+ * Pre-configured quality profiles based on TRaSH Guides recommendations
+ */
+
+export interface TRaSHProfilePreset {
+  id: string
+  name: string
+  description: string
+  app: "radarr" | "sonarr" | "both"
+  cutoffQuality: string
+  allowedQualities: string[]
+  cfScores: Record<string, number>
+}
+
+// Quality names for different resolutions
+export const RADARR_QUALITIES = {
+  // HD
+  "Bluray-1080p": "Bluray-1080p",
+  "WEB-1080p": "WEBDL-1080p",
+  "HDTV-1080p": "HDTV-1080p",
+  "Bluray-720p": "Bluray-720p",
+  "WEB-720p": "WEBDL-720p",
+  // UHD
+  "Bluray-2160p": "Bluray-2160p",
+  "WEB-2160p": "WEBDL-2160p",
+  // Remux
+  "Remux-1080p": "Remux-1080p",
+  "Remux-2160p": "Remux-2160p",
+}
+
+export const SONARR_QUALITIES = {
+  "WEB-1080p": "WEBDL-1080p",
+  "WEB-720p": "WEBDL-720p",
+  "WEB-2160p": "WEBDL-2160p",
+  "HDTV-1080p": "HDTV-1080p",
+  "HDTV-720p": "HDTV-720p",
+  "Bluray-1080p": "Bluray-1080p",
+  "Bluray-720p": "Bluray-720p",
+  "Bluray-2160p": "Bluray-2160p",
+  "Remux-1080p": "Remux-1080p",
+  "Remux-2160p": "Remux-2160p",
+}
+
+// TRaSH recommended Custom Format scores
+export const CF_SCORES = {
+  // Unwanted (use negative scores)
+  "BR-DISK": -10000,
+  LQ: -10000,
+  "LQ (Release Title)": -10000,
+  "3D": -10000,
+  x265: -10000, // Only for HD, not UHD
+  Extras: -10000,
+
+  // Preferred (positive scores)
+  "Repack/Proper": 5,
+  Repack2: 6,
+
+  // HDR Formats
+  "DV HDR10Plus": 1600,
+  "DV HDR10": 1500,
+  DV: 1400,
+  "DV HLG": 1300,
+  "DV SDR": 1200,
+  HDR10Plus: 700,
+  HDR10: 600,
+  HDR: 500,
+  "HDR (undefined)": 400,
+  PQ: 300,
+  HLG: 200,
+
+  // Audio Formats
+  "TrueHD Atmos": 5000,
+  "DTS X": 4500,
+  TrueHD: 4000,
+  "DTS-HD MA": 3500,
+  FLAC: 3000,
+  PCM: 2500,
+  "DTS-HD HRA": 2000,
+  "DD+ Atmos": 1500,
+  "DD+": 1000,
+  "DTS-ES": 800,
+  DTS: 600,
+  AAC: 400,
+  DD: 300,
+
+  // Streaming Services
+  AMZN: 0,
+  ATVP: 100,
+  DSNP: 100,
+  HBO: 0,
+  HMAX: 0,
+  Hulu: 0,
+  MA: 0,
+  NF: 0,
+  PCOK: 0,
+  PMTP: 0,
+
+  // Movie Versions
+  "IMAX Enhanced": 800,
+  IMAX: 700,
+  Hybrid: 100,
+  "Criterion Collection": 100,
+  "Special Edition": 50,
+  "Theatrical Cut": 0,
+
+  // HQ Release Groups
+  "HQ-Remux": 1750,
+  "HQ-WEBDL": 1700,
+  HQ: 1600,
+}
+
+// Radarr Profile Presets
+export const RADARR_PRESETS: TRaSHProfilePreset[] = [
+  {
+    id: "hd-bluray-web",
+    name: "HD Bluray + WEB",
+    description: "High-Quality HD Encodes (Bluray-720p/1080p). Size: 6-15 GB",
+    app: "radarr",
+    cutoffQuality: "Bluray-1080p",
+    allowedQualities: ["Bluray-1080p", "Bluray-720p", "WEBDL-1080p", "WEBDL-720p", "WEBRip-1080p", "WEBRip-720p"],
+    cfScores: {
+      "BR-DISK": -10000,
+      LQ: -10000,
+      "LQ (Release Title)": -10000,
+      "3D": -10000,
+      "x265 (HD)": -10000,
+      "Repack/Proper": 5,
+      "HQ-WEBDL": 1700,
+      HQ: 1600,
+    },
+  },
+  {
+    id: "uhd-bluray-web",
+    name: "UHD Bluray + WEB",
+    description: "High-Quality UHD Encodes (Bluray-2160p). Size: 20-60 GB",
+    app: "radarr",
+    cutoffQuality: "Bluray-2160p",
+    allowedQualities: ["Bluray-2160p", "WEBDL-2160p", "WEBRip-2160p"],
+    cfScores: {
+      "BR-DISK": -10000,
+      LQ: -10000,
+      "LQ (Release Title)": -10000,
+      "DV HDR10Plus": 1600,
+      "DV HDR10": 1500,
+      DV: 1400,
+      HDR10Plus: 700,
+      HDR10: 600,
+      "Repack/Proper": 5,
+      "TrueHD Atmos": 5000,
+      "DTS X": 4500,
+      "HQ-WEBDL": 1700,
+    },
+  },
+  {
+    id: "remux-web-1080p",
+    name: "Remux + WEB 1080p",
+    description: "1080p Remuxes. Size: 20-40 GB",
+    app: "radarr",
+    cutoffQuality: "Remux-1080p",
+    allowedQualities: ["Remux-1080p", "WEBDL-1080p", "WEBRip-1080p"],
+    cfScores: {
+      "BR-DISK": -10000,
+      LQ: -10000,
+      "x265 (HD)": -10000,
+      "HQ-Remux": 1750,
+      "Repack/Proper": 5,
+      "TrueHD Atmos": 5000,
+      "DTS X": 4500,
+      TrueHD: 4000,
+      "DTS-HD MA": 3500,
+    },
+  },
+  {
+    id: "remux-web-2160p",
+    name: "Remux + WEB 2160p",
+    description: "2160p Remuxes. Size: 40-100 GB",
+    app: "radarr",
+    cutoffQuality: "Remux-2160p",
+    allowedQualities: ["Remux-2160p", "WEBDL-2160p", "WEBRip-2160p"],
+    cfScores: {
+      "BR-DISK": -10000,
+      LQ: -10000,
+      "DV HDR10Plus": 1600,
+      "DV HDR10": 1500,
+      DV: 1400,
+      HDR10Plus: 700,
+      HDR10: 600,
+      "HQ-Remux": 1750,
+      "Repack/Proper": 5,
+      "TrueHD Atmos": 5000,
+      "DTS X": 4500,
+    },
+  },
+]
+
+// Sonarr Profile Presets
+export const SONARR_PRESETS: TRaSHProfilePreset[] = [
+  {
+    id: "web-1080p",
+    name: "WEB-1080p",
+    description: "720p/1080p WEBDL. Balanced quality and size",
+    app: "sonarr",
+    cutoffQuality: "WEBDL-1080p",
+    allowedQualities: ["WEBDL-1080p", "WEBRip-1080p", "WEBDL-720p", "WEBRip-720p"],
+    cfScores: {
+      "BR-DISK": -10000,
+      LQ: -10000,
+      "x265 (HD)": -10000,
+      "Repack/Proper": 5,
+      "HQ-WEBDL": 1700,
+      AMZN: 100,
+      ATVP: 100,
+      DSNP: 100,
+      NF: 100,
+    },
+  },
+  {
+    id: "web-2160p",
+    name: "WEB-2160p",
+    description: "2160p WEBDL with HDR. Premium quality",
+    app: "sonarr",
+    cutoffQuality: "WEBDL-2160p",
+    allowedQualities: ["WEBDL-2160p", "WEBRip-2160p"],
+    cfScores: {
+      "BR-DISK": -10000,
+      LQ: -10000,
+      "DV HDR10Plus": 1600,
+      "DV HDR10": 1500,
+      DV: 1400,
+      HDR10Plus: 700,
+      HDR10: 600,
+      "Repack/Proper": 5,
+      "HQ-WEBDL": 1700,
+      AMZN: 100,
+      ATVP: 100,
+      NF: 100,
+    },
+  },
+]
+
+// Get all presets for an app
+export function getPresetsForApp(app: "radarr" | "sonarr"): TRaSHProfilePreset[] {
+  if (app === "radarr") return RADARR_PRESETS
+  if (app === "sonarr") return SONARR_PRESETS
+  return []
+}
+
+// Get a specific preset by ID
+export function getPresetById(id: string): TRaSHProfilePreset | undefined {
+  return [...RADARR_PRESETS, ...SONARR_PRESETS].find((p) => p.id === id)
+}

package/src/ui/screens/AdvancedSettings.ts

@@ -17,6 +17,7 @@ import { createPageLayout } from "../components/PageLayout"
 import { FileEditor } from "../components/FileEditor"
 import { readFile, writeFile } from "node:fs/promises"
 import { getConfigPath, getComposePath } from "../../config/manager"
+import { getEnvPath } from "../../utils/env"
 import { existsSync } from "node:fs"
 
 export class AdvancedSettings {
@@ -100,7 +101,7 @@ export class AdvancedSettings {
           })
           break
         case 1: {
-          const envPath = getComposePath().replace("docker-compose.yml", ".env")
+          const envPath = getEnvPath()
           await this.editFile(".env", envPath, async (content) => {
             await writeFile(envPath, content, "utf-8")
             this.renderMenu()

package/src/ui/screens/ApiKeyViewer.ts

@@ -1,13 +1,12 @@
 import { BoxRenderable, CliRenderer, TextRenderable, KeyEvent } from "@opentui/core"
 import { existsSync, readFileSync, writeFileSync } from "node:fs"
-import { writeFile, readFile } from "node:fs/promises"
 import { join } from "node:path"
 import { randomBytes } from "node:crypto"
 import { parse as parseYaml } from "yaml"
 import { createPageLayout } from "../components/PageLayout"
 import { EasiarrConfig, AppDefinition } from "../../config/schema"
 import { getApp } from "../../apps/registry"
-import { getComposePath } from "../../config/manager"
+import { updateEnv } from "../../utils/env"
 
 /** Generate a random 32-character hex API key */
 function generateApiKey(): string {
@@ -337,30 +336,13 @@ export class ApiKeyViewer extends BoxRenderable {
     if (foundKeys.length === 0) return
 
     try {
-      const envPath = getComposePath().replace("docker-compose.yml", ".env")
-
-      // Read existing .env if present
-      const currentEnv: Record<string, string> = {}
-      if (existsSync(envPath)) {
-        const content = await readFile(envPath, "utf-8")
-        content.split("\n").forEach((line) => {
-          const [key, ...val] = line.split("=")
-          if (key && val.length > 0) currentEnv[key.trim()] = val.join("=").trim()
-        })
-      }
-
-      // Add API keys with format API_KEY_SONARR, API_KEY_RADARR, etc.
+      // Build updates object with API keys
+      const updates: Record<string, string> = {}
       for (const k of foundKeys) {
-        const envKey = `API_KEY_${k.appId.toUpperCase()}`
-        currentEnv[envKey] = k.key
+        updates[`API_KEY_${k.appId.toUpperCase()}`] = k.key
       }
 
-      // Reconstruct .env content
-      const envContent = Object.entries(currentEnv)
-        .map(([k, v]) => `${k}=${v}`)
-        .join("\n")
-
-      await writeFile(envPath, envContent, "utf-8")
+      await updateEnv(updates)
 
       // Update status
       if (this.statusText) {

package/src/ui/screens/AppConfigurator.ts

@@ -11,13 +11,13 @@ import {
   InputRenderableEvents,
   KeyEvent,
 } from "@opentui/core"
-import { existsSync, readFileSync } from "node:fs"
-import { writeFile, readFile } from "node:fs/promises"
 import { createPageLayout } from "../components/PageLayout"
 import { EasiarrConfig, AppId } from "../../config/schema"
 import { getApp } from "../../apps/registry"
 import { ArrApiClient, createQBittorrentConfig, createSABnzbdConfig } from "../../api/arr-api"
-import { getComposePath } from "../../config/manager"
+import { QBittorrentClient } from "../../api/qbittorrent-api"
+import { getCategoriesForApps } from "../../utils/categories"
+import { readEnvSync, updateEnv } from "../../utils/env"
 
 interface ConfigResult {
   appId: AppId
@@ -82,29 +82,11 @@ export class AppConfigurator extends BoxRenderable {
   }
 
   private loadSavedCredentials() {
-    try {
-      const envPath = getComposePath().replace("docker-compose.yml", ".env")
-      if (!existsSync(envPath)) return
-
-      const content = readFileSync(envPath, "utf-8")
-      content.split("\n").forEach((line) => {
-        const [key, ...val] = line.split("=")
-        if (key && val.length > 0) {
-          const value = val.join("=").trim()
-          if (key.trim() === "GLOBAL_USERNAME") {
-            this.globalUsername = value
-          } else if (key.trim() === "GLOBAL_PASSWORD") {
-            this.globalPassword = value
-          } else if (key.trim() === "QBITTORRENT_PASSWORD") {
-            this.qbPass = value
-          } else if (key.trim() === "SABNZBD_API_KEY") {
-            this.sabApiKey = value
-          }
-        }
-      })
-    } catch {
-      // Ignore errors
-    }
+    const env = readEnvSync()
+    if (env.GLOBAL_USERNAME) this.globalUsername = env.GLOBAL_USERNAME
+    if (env.GLOBAL_PASSWORD) this.globalPassword = env.GLOBAL_PASSWORD
+    if (env.QBITTORRENT_PASSWORD) this.qbPass = env.QBITTORRENT_PASSWORD
+    if (env.SABNZBD_API_KEY) this.sabApiKey = env.SABNZBD_API_KEY
   }
 
   private renderCredentialsPrompt() {
@@ -216,32 +198,10 @@ export class AppConfigurator extends BoxRenderable {
 
   private async saveGlobalCredentialsToEnv() {
     try {
-      const envPath = getComposePath().replace("docker-compose.yml", ".env")
-
-      // Read existing .env if present
-      const currentEnv: Record<string, string> = {}
-      if (existsSync(envPath)) {
-        const content = await readFile(envPath, "utf-8")
-        content.split("\n").forEach((line) => {
-          const [key, ...val] = line.split("=")
-          if (key && val.length > 0) currentEnv[key.trim()] = val.join("=").trim()
-        })
-      }
-
-      // Add global credentials
-      if (this.globalUsername) {
-        currentEnv["GLOBAL_USERNAME"] = this.globalUsername
-      }
-      if (this.globalPassword) {
-        currentEnv["GLOBAL_PASSWORD"] = this.globalPassword
-      }
-
-      // Reconstruct .env content
-      const envContent = Object.entries(currentEnv)
-        .map(([k, v]) => `${k}=${v}`)
-        .join("\n")
-
-      await writeFile(envPath, envContent, "utf-8")
+      const updates: Record<string, string> = {}
+      if (this.globalUsername) updates.GLOBAL_USERNAME = this.globalUsername
+      if (this.globalPassword) updates.GLOBAL_PASSWORD = this.globalPassword
+      await updateEnv(updates)
     } catch {
       // Ignore errors - not critical
     }
@@ -331,8 +291,18 @@ export class AppConfigurator extends BoxRenderable {
       throw new Error("Already configured")
     }
 
-    // Add root folder
-    await client.addRootFolder(appDef.rootFolder.path)
+    // Add root folder - Lidarr requires profile IDs
+    if (appId === "lidarr") {
+      const metadataProfiles = await client.getMetadataProfiles()
+      const qualityProfiles = await client.getQualityProfiles()
+      await client.addRootFolder({
+        path: appDef.rootFolder.path,
+        defaultMetadataProfileId: metadataProfiles[0]?.id || 1,
+        defaultQualityProfileId: qualityProfiles[0]?.id || 1,
+      })
+    } else {
+      await client.addRootFolder(appDef.rootFolder.path)
+    }
 
     // Set up authentication if credentials provided
     if (this.globalPassword) {
@@ -346,24 +316,16 @@ export class AppConfigurator extends BoxRenderable {
 
   private extractApiKey(appId: AppId): string | null {
     // Use API keys from .env file (format: API_KEY_APPNAME)
-    try {
-      const envPath = getComposePath().replace("docker-compose.yml", ".env")
-      if (!existsSync(envPath)) return null
-
-      const content = readFileSync(envPath, "utf-8")
-      const envKey = `API_KEY_${appId.toUpperCase()}`
-
-      for (const line of content.split("\n")) {
-        const [key, ...val] = line.split("=")
-        if (key?.trim() === envKey && val.length > 0) {
-          return val.join("=").trim()
-        }
-      }
+    const envKey = `API_KEY_${appId.toUpperCase()}`
+    return readEnvSync()[envKey] ?? null
+  }
 
-      return null
-    } catch {
-      return null
-    }
+  /**
+   * Get qBittorrent categories based on enabled *arr apps
+   */
+  private getEnabledCategories(): { name: string; savePath: string }[] {
+    const enabledAppIds = this.config.apps.filter((a) => a.enabled).map((a) => a.id)
+    return getCategoriesForApps(enabledAppIds)
   }
 
   private renderConfigProgress() {
@@ -543,6 +505,22 @@ export class AppConfigurator extends BoxRenderable {
   }
 
   private async addDownloadClients(type: "qbittorrent" | "sabnzbd") {
+    // Configure qBittorrent settings via its API first
+    if (type === "qbittorrent") {
+      try {
+        const qbClient = new QBittorrentClient(this.qbHost, this.qbPort, this.qbUser, this.qbPass)
+        const loggedIn = await qbClient.login()
+        if (loggedIn) {
+          // Generate categories from enabled *arr apps that use download clients
+          const categories = this.getEnabledCategories()
+          // Configure TRaSH-compliant settings: save_path, auto_tmm, categories
+          await qbClient.configureTRaSHCompliant(categories)
+        }
+      } catch {
+        // Ignore qBittorrent config errors - may not be ready or have different auth
+      }
+    }
+
     // Add download client to all *arr apps
     const servarrApps = this.config.apps.filter((a) => {
       const def = getApp(a.id)
@@ -587,31 +565,13 @@ export class AppConfigurator extends BoxRenderable {
 
   private async saveCredentialsToEnv(type: "qbittorrent" | "sabnzbd") {
     try {
-      const envPath = getComposePath().replace("docker-compose.yml", ".env")
-
-      // Read existing .env if present
-      const currentEnv: Record<string, string> = {}
-      if (existsSync(envPath)) {
-        const content = await readFile(envPath, "utf-8")
-        content.split("\n").forEach((line) => {
-          const [key, ...val] = line.split("=")
-          if (key && val.length > 0) currentEnv[key.trim()] = val.join("=").trim()
-        })
-      }
-
-      // Add credentials
+      const updates: Record<string, string> = {}
       if (type === "qbittorrent" && this.qbPass) {
-        currentEnv["QBITTORRENT_PASSWORD"] = this.qbPass
+        updates.QBITTORRENT_PASSWORD = this.qbPass
       } else if (type === "sabnzbd" && this.sabApiKey) {
-        currentEnv["SABNZBD_API_KEY"] = this.sabApiKey
+        updates.SABNZBD_API_KEY = this.sabApiKey
       }
-
-      // Reconstruct .env content
-      const envContent = Object.entries(currentEnv)
-        .map(([k, v]) => `${k}=${v}`)
-        .join("\n")
-
-      await writeFile(envPath, envContent, "utf-8")
+      await updateEnv(updates)
     } catch {
       // Ignore errors - not critical
     }

package/src/ui/screens/MainMenu.ts

@@ -11,6 +11,8 @@ import { createPageLayout } from "../components/PageLayout"
 import { saveCompose } from "../../compose"
 import { ApiKeyViewer } from "./ApiKeyViewer"
 import { AppConfigurator } from "./AppConfigurator"
+import { TRaSHProfileSetup } from "./TRaSHProfileSetup"
+import { ProwlarrSetup } from "./ProwlarrSetup"
 
 export class MainMenu {
   private renderer: RenderContext
@@ -100,10 +102,18 @@ export class MainMenu {
           name: "⚙️  Configure Apps",
           description: "Set root folders and download clients via API",
         },
+        {
+          name: "🎯 TRaSH Guide Setup",
+          description: "Apply TRaSH quality profiles and custom formats",
+        },
         {
           name: "🔄 Regenerate Compose",
           description: "Rebuild docker-compose.yml",
         },
+        {
+          name: "🔗 Prowlarr Setup",
+          description: "Sync indexers to *arr apps, FlareSolverr",
+        },
         { name: "❌ Exit", description: "Close easiarr" },
       ],
     })
@@ -143,11 +153,33 @@ export class MainMenu {
           break
         }
         case 5: {
+          // TRaSH Profile Setup
+          this.menu.blur()
+          this.page.visible = false
+          const trashSetup = new TRaSHProfileSetup(this.renderer as CliRenderer, this.config, () => {
+            this.page.visible = true
+            this.menu.focus()
+          })
+          this.container.add(trashSetup)
+          break
+        }
+        case 6: {
           // Regenerate compose
           await saveCompose(this.config)
           break
         }
-        case 6:
+        case 7: {
+          // Prowlarr Setup
+          this.menu.blur()
+          this.page.visible = false
+          const prowlarrSetup = new ProwlarrSetup(this.renderer as CliRenderer, this.config, () => {
+            this.page.visible = true
+            this.menu.focus()
+          })
+          this.container.add(prowlarrSetup)
+          break
+        }
+        case 8:
           process.exit(0)
           break
       }