From ba08c7b4e793c3c3de29ba797c2ed693248c1582 Mon Sep 17 00:00:00 2001
From: AnimeDL <multidownloadernx@gmail.com>
Date: Wed, 17 Apr 2024 11:59:47 -0700
Subject: [PATCH] [AO] [ADN] Improve API locale handling

---
 adn.ts                             | 2 +-
 ao.ts                              | 2 +-
 gui/server/services/adn.ts         | 3 ++-
 gui/server/services/animeonegai.ts | 3 ++-
 4 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/adn.ts b/adn.ts
index 7c393be..bea0556 100644
--- a/adn.ts
+++ b/adn.ts
@@ -78,7 +78,7 @@ export default class AnimationDigitalNetwork implements ServiceClass {
   public async cli() {
     console.info(`\n=== Multi Downloader NX ${packageJson.version} ===\n`);
     const argv = yargs.appArgv(this.cfg.cli);
-    if (argv.locale !== 'en-US')
+    if (['fr', 'de'].includes(argv.locale))
       this.locale = argv.locale;
     if (argv.debug)
       this.debug = true;
diff --git a/ao.ts b/ao.ts
index 21ac51c..4cbbccf 100644
--- a/ao.ts
+++ b/ao.ts
@@ -84,7 +84,7 @@ export default class AnimeOnegai implements ServiceClass {
   public async cli() {
     console.info(`\n=== Multi Downloader NX ${packageJson.version} ===\n`);
     const argv = yargs.appArgv(this.cfg.cli);
-    if (argv.locale !== 'en-US')
+    if (['pt', 'es'].includes(argv.locale))
       this.locale = argv.locale;
     if (argv.debug)
       this.debug = true;
diff --git a/gui/server/services/adn.ts b/gui/server/services/adn.ts
index 1b72f3d..88bbe2d 100644
--- a/gui/server/services/adn.ts
+++ b/gui/server/services/adn.ts
@@ -19,7 +19,8 @@ class ADNHandler extends Base implements MessageHandler {
 
   public getDefaults() {
     const _default = yargs.appArgv(this.adn.cfg.cli, true);
-    this.adn.locale = _default.locale;
+    if (['fr', 'de'].includes(_default.locale))
+      this.adn.locale = _default.locale;
   }
 
   public async auth(data: AuthData) {
diff --git a/gui/server/services/animeonegai.ts b/gui/server/services/animeonegai.ts
index 75bc47e..ef413bf 100644
--- a/gui/server/services/animeonegai.ts
+++ b/gui/server/services/animeonegai.ts
@@ -19,7 +19,8 @@ class AnimeOnegaiHandler extends Base implements MessageHandler {
 
   public getDefaults() {
     const _default = yargs.appArgv(this.ao.cfg.cli, true);
-    this.ao.locale = _default.locale;
+    if (['es', 'pt'].includes(_default.locale))
+      this.ao.locale = _default.locale;
   }
 
   public async auth(data: AuthData) {