refactor: вынести конфиг в config.go, добавить секции pachca/email
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
48
config.go
Normal file
48
config.go
Normal file
@@ -0,0 +1,48 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os"
|
||||||
|
|
||||||
|
"gopkg.in/yaml.v3"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Config — структура для чтения YAML-файла
|
||||||
|
type Config struct {
|
||||||
|
ListenAddresses []string `yaml:"listen_addresses"`
|
||||||
|
LogDir string `yaml:"log_dir"`
|
||||||
|
Telegram *ConfigTelegram `yaml:"telegram"`
|
||||||
|
Pachca *ConfigPachca `yaml:"pachca"`
|
||||||
|
Email *ConfigEmail `yaml:"email"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type ConfigTelegram struct {
|
||||||
|
Token string `yaml:"token"`
|
||||||
|
GroupID int64 `yaml:"group_id"`
|
||||||
|
DisableIPV6 *bool `yaml:"disable_ipv6"` // nil → true
|
||||||
|
}
|
||||||
|
|
||||||
|
type ConfigPachca struct {
|
||||||
|
WebhookURL string `yaml:"webhook_url"` // входящий вебхук Pachca
|
||||||
|
}
|
||||||
|
|
||||||
|
type ConfigEmail struct {
|
||||||
|
SMTPAddr string `yaml:"smtp_addr"`
|
||||||
|
Username string `yaml:"username"`
|
||||||
|
Password string `yaml:"password"`
|
||||||
|
From string `yaml:"from"`
|
||||||
|
To []string `yaml:"to"`
|
||||||
|
Subject string `yaml:"subject"` // "" → "HTTP Logger"
|
||||||
|
}
|
||||||
|
|
||||||
|
// loadConfig читает YAML-конфиг из файла
|
||||||
|
func loadConfig(path string) (*Config, error) {
|
||||||
|
data, err := os.ReadFile(path)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
var cfg Config
|
||||||
|
if err := yaml.Unmarshal(data, &cfg); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return &cfg, nil
|
||||||
|
}
|
||||||
28
main.go
28
main.go
@@ -6,27 +6,12 @@ import (
|
|||||||
"log"
|
"log"
|
||||||
"net"
|
"net"
|
||||||
"net/http"
|
"net/http"
|
||||||
"os"
|
|
||||||
"strings"
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"gopkg.in/yaml.v3"
|
|
||||||
|
|
||||||
"gitea.mediatoday.ru/mt/notify"
|
"gitea.mediatoday.ru/mt/notify"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Config — структура для чтения YAML-файла
|
|
||||||
type Config struct {
|
|
||||||
ListenAddresses []string `yaml:"listen_addresses"`
|
|
||||||
LogDir string `yaml:"log_dir"`
|
|
||||||
Telegram *ConfigTelegraam `yaml:"telegram"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type ConfigTelegraam struct {
|
|
||||||
Token string `yaml:"token"`
|
|
||||||
GroupID int64 `yaml:"group_id"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func GetRemoteAddr(r *http.Request) string {
|
func GetRemoteAddr(r *http.Request) string {
|
||||||
// Сначала смотрим X-Forwarded-For
|
// Сначала смотрим X-Forwarded-For
|
||||||
if xff := r.Header.Get("X-Forwarded-For"); xff != "" {
|
if xff := r.Header.Get("X-Forwarded-For"); xff != "" {
|
||||||
@@ -138,16 +123,3 @@ func main() {
|
|||||||
// Блокируем main, чтобы программа не завершилась
|
// Блокируем main, чтобы программа не завершилась
|
||||||
select {}
|
select {}
|
||||||
}
|
}
|
||||||
|
|
||||||
// loadConfig читает YAML-конфиг из файла
|
|
||||||
func loadConfig(path string) (*Config, error) {
|
|
||||||
data, err := os.ReadFile(path)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
var cfg Config
|
|
||||||
if err := yaml.Unmarshal(data, &cfg); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
return &cfg, nil
|
|
||||||
}
|
|
||||||
|
|||||||
Reference in New Issue
Block a user