Changeset 23 in code for trunk/web/main.go

Timestamp:

Sep 2, 2023, 12:15:30 PM (21 months ago)

Author:

manerakai

Message:

Improved code

File:

• trunk/web/main.go (modified) (5 diffs)

trunk/web/main.go

@@ -21 +21 @@
 
         app.All("/", func(c *fiber.Ctx) error {
-                engine := c.Query("engine")
-                if _, ok := engines.Engines[engine]; !ok || engine == "" {
+                engine := c.Cookies("engine")
+                if c.Query("engine") != "" {
+                        engine = c.Query("engine")
+                }
+                if _, ok := engines.Engines[engine]; !ok {
                         engine = "google"
                 }
@@ -50 +53 @@
                                 translatedText = result.TranslatedText
                                 translation = result
+                                from = result.SourceLanguage
                         }
 
@@ -69 +73 @@
                         fromCookie.Name = "from"
                         fromCookie.Value = from
-                        fromCookie.Expires = time.Now().Add(24 * time.Hour * 365)
+                        fromCookie.Expires = time.Now().Add(time.Hour * 24 * 365)
                         c.Cookie(fromCookie)
 
@@ -75 +79 @@
                         toCookie.Name = "to"
                         toCookie.Value = to
-                        toCookie.Expires = time.Now().Add(24 * time.Hour * 365)
+                        toCookie.Expires = time.Now().Add(time.Hour * 24 * 365)
                         c.Cookie(toCookie)
 
+                        engineCookie := new(fiber.Cookie)
+                        engineCookie.Name = "engine"
+                        engineCookie.Value = engine
+                        engineCookie.Expires = time.Now().Add(time.Hour * 24 * 365)
+                        c.Cookie(engineCookie)
                 } else if c.Method() == "GET" {
                         from = c.Cookies("from")
@@ -185 +194 @@
                         }
                 }
-
         })