From 457549b7cc7a0d52cb2f9f052e5599b41399f323 Mon Sep 17 00:00:00 2001 From: eeeXun Date: Fri, 19 May 2023 19:24:11 +0800 Subject: [PATCH] refactor: set api key in for-loop --- config.go | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/config.go b/config.go index 120bb80..55fa6dd 100644 --- a/config.go +++ b/config.go @@ -113,11 +113,10 @@ func configInit() { uiStyle.SetSrcBorderColor(config.GetString("source.border_color")). SetDstBorderColor(config.GetString("destination.border_color")) // Set API Keys - if config.Get("api_key.chatgpt") != nil { - translators["ChatGPT"].SetAPIKey(config.GetString("api_key.chatgpt")) - } - if config.Get("api_key.deepl") != nil { - translators["DeepL"].SetAPIKey(config.GetString("api_key.deepl")) + for _, name := range []string{"ChatGPT", "DeepL"} { + if config.Get(fmt.Sprintf("api_key.%s", name)) != nil { + translators[name].SetAPIKey(config.GetString(fmt.Sprintf("api_key.%s", name))) + } } // Set argument language if len(*srcLangArg) > 0 {