構成ファイルのサポートを有効にし、組み込みの静的レンダラーの問題を修正しました
Signed-off-by: Izuru Yakumo <yakumo.izuru@chaotic.ninja> git-svn-id: file:///srv/svn/repo/mai/trunk@53 e410bdd4-646f-c54f-a7ce-fffcc4f439ae
This commit is contained in:
yakumo.izuru
257
cmd/simplytranslate/main.go
Normal file
257
cmd/simplytranslate/main.go
Normal file
@@ -0,0 +1,257 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"time"
|
||||
"flag"
|
||||
|
||||
"git.chaotic.ninja/yakumo.izuru/simplytranslate/engines"
|
||||
"github.com/gofiber/fiber/v2"
|
||||
"github.com/gofiber/template/html/v2"
|
||||
"gopkg.in/ini.v1"
|
||||
)
|
||||
|
||||
var conf struct {
|
||||
listen string
|
||||
staticpath string
|
||||
tmplpath string
|
||||
}
|
||||
func readConf(file string) error {
|
||||
cfg, err := ini.Load(file)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
conf.listen = cfg.Section("").Key("listen").String()
|
||||
conf.staticpath = cfg.Section("").Key("rootdir").String()
|
||||
conf.tmplpath = cfg.Section("").Key("tmplpath").String()
|
||||
|
||||
return nil
|
||||
}
|
||||
func main() {
|
||||
var configfile string
|
||||
flag.StringVar(&configfile, "f", "", "Configuration file")
|
||||
flag.Parse()
|
||||
|
||||
if configfile != "" {
|
||||
readConf(configfile)
|
||||
}
|
||||
|
||||
conf.listen = "127.0.0.1:5000"
|
||||
conf.staticpath = "./static"
|
||||
conf.tmplpath = "./views"
|
||||
|
||||
engine := html.New(conf.tmplpath, ".html")
|
||||
engine.AddFunc("inc", func(i int) int { return i + 1 })
|
||||
|
||||
app := fiber.New(fiber.Config{
|
||||
Views: engine,
|
||||
})
|
||||
|
||||
app.All("/", func(c *fiber.Ctx) error {
|
||||
engine := c.Cookies("engine")
|
||||
if c.Query("engine") != "" {
|
||||
engine = c.Query("engine")
|
||||
}
|
||||
if _, ok := engines.Engines[engine]; !ok {
|
||||
engine = "google"
|
||||
}
|
||||
targetLanguages, err := engines.Engines[engine].TargetLanguages()
|
||||
if err != nil {
|
||||
return c.SendStatus(500)
|
||||
}
|
||||
sourceLanguages, err := engines.Engines[engine].SourceLanguages()
|
||||
if err != nil {
|
||||
return c.SendStatus(500)
|
||||
}
|
||||
originalText := ""
|
||||
translatedText := ""
|
||||
from := ""
|
||||
to := ""
|
||||
ttsFrom := ""
|
||||
ttsTo := ""
|
||||
sourceLanguage := ""
|
||||
|
||||
var translation engines.TranslationResult
|
||||
if c.Method() == "POST" {
|
||||
from = c.FormValue("from")
|
||||
to = c.FormValue("to")
|
||||
originalText = c.FormValue("text")
|
||||
if result, err := engines.Engines[engine].Translate(originalText, from, to); err != nil {
|
||||
return c.SendStatus(500)
|
||||
} else {
|
||||
translatedText = result.TranslatedText
|
||||
translation = result
|
||||
sourceLanguage = result.SourceLanguage
|
||||
}
|
||||
|
||||
ttsFromURL, _ := url.Parse("api/tts")
|
||||
fromQuery := url.Values{}
|
||||
fromQuery.Add("lang", from)
|
||||
fromQuery.Add("text", originalText)
|
||||
ttsFromURL.RawQuery = fromQuery.Encode()
|
||||
ttsFrom = ttsFromURL.String()
|
||||
|
||||
ttsToURL, _ := url.Parse("api/tts")
|
||||
toQuery := url.Values{}
|
||||
toQuery.Add("lang", to)
|
||||
toQuery.Add("text", translatedText)
|
||||
ttsToURL.RawQuery = toQuery.Encode()
|
||||
ttsTo = ttsToURL.String()
|
||||
|
||||
fromCookie := new(fiber.Cookie)
|
||||
fromCookie.Name = "from"
|
||||
fromCookie.Value = from
|
||||
fromCookie.Expires = time.Now().Add(time.Hour * 24 * 365)
|
||||
c.Cookie(fromCookie)
|
||||
|
||||
toCookie := new(fiber.Cookie)
|
||||
toCookie.Name = "to"
|
||||
toCookie.Value = to
|
||||
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")
|
||||
to = c.Cookies("to")
|
||||
} else {
|
||||
return c.SendStatus(400)
|
||||
}
|
||||
|
||||
if from == "" {
|
||||
from = "auto"
|
||||
}
|
||||
|
||||
enginesNames := map[string]string{}
|
||||
for k, v := range engines.Engines {
|
||||
enginesNames[k] = v.DisplayName()
|
||||
}
|
||||
|
||||
return c.Render("index", fiber.Map{
|
||||
"Engine": engine,
|
||||
"enginesNames": enginesNames,
|
||||
"SourceLanguages": sourceLanguages,
|
||||
"TargetLanguages": targetLanguages,
|
||||
"OriginalText": originalText,
|
||||
"Translation": translation,
|
||||
"From": from,
|
||||
"To": to,
|
||||
"TtsFrom": ttsFrom,
|
||||
"TtsTo": ttsTo,
|
||||
"SourceLanguage": sourceLanguage,
|
||||
})
|
||||
})
|
||||
|
||||
app.All("/api/translate", func(c *fiber.Ctx) error {
|
||||
from := ""
|
||||
to := ""
|
||||
engine := ""
|
||||
text := ""
|
||||
if c.Method() == "GET" {
|
||||
engine = c.Query("engine")
|
||||
text = c.Query("text")
|
||||
from = c.Query("from")
|
||||
to = c.Query("to")
|
||||
} else if c.Method() == "POST" {
|
||||
engine = c.FormValue("engine")
|
||||
text = c.FormValue("text")
|
||||
from = c.FormValue("from")
|
||||
to = c.FormValue("to")
|
||||
} else {
|
||||
return c.SendStatus(400)
|
||||
}
|
||||
if _, ok := engines.Engines[engine]; !ok || engine == "" {
|
||||
engine = "google"
|
||||
}
|
||||
if to == "" {
|
||||
return c.SendStatus(400)
|
||||
}
|
||||
if result, err := engines.Engines[engine].Translate(text, from, to); err != nil {
|
||||
return c.SendStatus(500)
|
||||
} else {
|
||||
return c.JSON(result)
|
||||
}
|
||||
})
|
||||
|
||||
app.Get("/api/source_languages", func(c *fiber.Ctx) error {
|
||||
engine := c.Query("engine")
|
||||
if _, ok := engines.Engines[engine]; !ok || engine == "" {
|
||||
engine = "google"
|
||||
}
|
||||
if result, err := engines.Engines[engine].SourceLanguages(); err != nil {
|
||||
return c.SendStatus(500)
|
||||
} else {
|
||||
return c.JSON(result)
|
||||
}
|
||||
})
|
||||
|
||||
app.Get("/api/target_languages", func(c *fiber.Ctx) error {
|
||||
engine := c.Query("engine")
|
||||
if _, ok := engines.Engines[engine]; !ok || engine == "" {
|
||||
engine = "google"
|
||||
}
|
||||
if result, err := engines.Engines[engine].TargetLanguages(); err != nil {
|
||||
return c.SendStatus(500)
|
||||
} else {
|
||||
return c.JSON(result)
|
||||
}
|
||||
})
|
||||
|
||||
app.Get("/api/tts", func(c *fiber.Ctx) error {
|
||||
engine := c.Query("engine")
|
||||
if _, ok := engines.Engines[engine]; !ok || engine == "" {
|
||||
engine = "google"
|
||||
}
|
||||
|
||||
text := c.Query("text")
|
||||
if text == "" {
|
||||
return c.SendStatus(400)
|
||||
}
|
||||
|
||||
lang := c.Query("lang")
|
||||
|
||||
if url, err := engines.Engines[engine].Tts(text, lang); err != nil {
|
||||
return c.SendStatus(500)
|
||||
} else {
|
||||
if response, err := http.Get(url); err != nil {
|
||||
return c.SendStatus(500)
|
||||
} else {
|
||||
defer response.Body.Close()
|
||||
var buf bytes.Buffer
|
||||
response.Write(&buf)
|
||||
c.Context().SetContentType("audio/mpeg")
|
||||
return c.Send(buf.Bytes())
|
||||
}
|
||||
}
|
||||
})
|
||||
|
||||
app.Post("/switchlanguages", func(c *fiber.Ctx) error {
|
||||
if c.Cookies("from") != "" {
|
||||
fromCookie := new(fiber.Cookie)
|
||||
fromCookie.Name = "from"
|
||||
fromCookie.Value = c.Cookies("to")
|
||||
fromCookie.Expires = time.Now().Add(24 * time.Hour * 365)
|
||||
|
||||
toCookie := new(fiber.Cookie)
|
||||
toCookie.Name = "to"
|
||||
toCookie.Value = c.Cookies("from")
|
||||
toCookie.Expires = time.Now().Add(24 * time.Hour * 365)
|
||||
|
||||
c.Cookie(fromCookie)
|
||||
c.Cookie(toCookie)
|
||||
}
|
||||
return c.Redirect("/")
|
||||
})
|
||||
app.Static("/static", conf.staticpath, fiber.Static{
|
||||
Compress: true,
|
||||
ByteRange: true,
|
||||
Browse: true,
|
||||
})
|
||||
app.Listen(conf.listen)
|
||||
}
|
||||
Reference in New Issue
Block a user