banek/main.go
2023-10-16 05:21:10 +06:00

265 lines
6.7 KiB
Go
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

//
// анекдоты категории /b
//
// Руслан <im@fruw.org>, 2023
//
package main
import (
"encoding/json"
"flag"
"fmt"
"io/ioutil"
"math/rand"
"net/http"
"strconv"
"strings"
"time"
)
type Message struct {
ID int `json:"id"`
Text interface{} `json:"text"`
}
type Channel struct {
Name string `json:"name"`
ID int `json:"id"`
Messages []Message `json:"messages"`
}
var (
showVersion bool
port int
content string
html string
jsonParse string
messages Channel
maxID int
)
const (
version = "1.0.3"
cyan = "\033[1;36m"
reset = "\033[0m"
)
func init() {
flag.BoolVar(&showVersion, "v", false, "")
flag.BoolVar(&showVersion, "version", false, "")
flag.IntVar(&port, "p", 9999, "")
flag.IntVar(&port, "port", 9999, "")
flag.StringVar(&content, "c", "/etc/banek/content.json", "")
flag.StringVar(&content, "content", "/etc/banek/content.json", "")
flag.StringVar(&html, "h", "/etc/banek/template.html", "")
flag.StringVar(&html, "html", "/etc/banek/template.html", "")
flag.StringVar(&jsonParse, "j", "", "")
flag.StringVar(&jsonParse, "json-parse", "", "")
flag.Usage = func() {
fmt.Printf("banek %s%s%s - анекдоты категории /b\n\n", cyan, version, reset)
fmt.Printf("%s-h%s, %s--help%s Показать справку banek и завершить работу\n", cyan, reset, cyan, reset)
fmt.Printf("%s-v%s, %s--version%s Показать версию banek и завершить работу\n", cyan, reset, cyan, reset)
fmt.Printf("%s-p%s, %s--port%s Порт, на котором будет запущен сервис (по-умолчанию 9999)\n", cyan, reset, cyan, reset)
fmt.Printf("%s-c%s, %s--content%s Путь до json-файла с анеками (по-умолчанию /etc/banek/content.json)\n", cyan, reset, cyan, reset)
fmt.Printf("%s-h%s, %s--html%s Путь до html-шаблона (по-умолчанию /etc/banek/template.html)\n", cyan, reset, cyan, reset)
fmt.Printf("%s-j%s, %s--json-parse%s Спарсить выгрузку из группы в json\n\n", cyan, reset, cyan, reset)
fmt.Println("Руслан, <im@fruw.org>, 2023")
}
flag.Parse()
}
func main() {
if showVersion {
fmt.Printf("banek %s%s%s\n", cyan, version, reset)
return
}
if jsonParse != "" {
parseJSON()
return
}
rand.Seed(time.Now().UnixNano())
err := loadMessages(content)
if err != nil {
fmt.Println("Ошибка чтения JSON:", err)
return
}
_, err = ioutil.ReadFile(html)
if err != nil {
fmt.Println("Ошибка чтения HTML:", err)
return
}
http.HandleFunc("/", handleRequest)
if err != nil {
fmt.Println("Ошибка сервиса:", err)
return
}
fmt.Printf("Сервис запущен на порту :%d\n", port)
http.ListenAndServe(fmt.Sprintf(":%d", port), nil)
}
func parseJSON() {
jsonData, err := ioutil.ReadFile(jsonParse)
if err != nil {
fmt.Println("Ошибка чтения %d: %d", content, err)
return
}
var channel Channel
if err := json.Unmarshal(jsonData, &channel); err != nil {
fmt.Println("Ошибка парсинга JSON:", err)
return
}
messageNumber := 1
for i, message := range channel.Messages {
if message.Text != nil {
switch text := message.Text.(type) {
case string:
if text != "" {
channel.Messages[i].ID = messageNumber
channel.Messages[i].Text = fmt.Sprintf("%s", text)
messageNumber++
}
case []interface{}:
var newText []string
for _, item := range text {
if str, ok := item.(string); ok && str != "" {
channel.Messages[i].ID = messageNumber
newText = append(newText, str)
messageNumber++
}
}
if len(newText) > 0 {
channel.Messages[i].Text = strings.Join(newText, " ")
}
}
}
}
cleanedMessages := make([]Message, 0)
for _, message := range channel.Messages {
if message.Text != "" {
cleanedMessages = append(cleanedMessages, message)
}
}
channel.Messages = cleanedMessages
formattedJSON, err := json.MarshalIndent(channel, "", " ")
if err != nil {
fmt.Println("Ошибка форматирования JSON:", err)
return
}
if err := ioutil.WriteFile(content, formattedJSON, 0644); err != nil {
fmt.Printf("Ошибка записи JSON файла %d: %d", content, err)
return
}
fmt.Println("JSON файл успешно записан по пути", content)
return
}
func loadMessages(filename string) error {
data, err := ioutil.ReadFile(filename)
if err != nil {
return err
}
if err := json.Unmarshal(data, &messages); err != nil {
return err
}
maxID = determineMaxID()
return nil
}
func determineMaxID() int {
max := 0
for _, message := range messages.Messages {
if message.ID > max {
max = message.ID
}
}
return max
}
func handleRequest(w http.ResponseWriter, r *http.Request) {
idStr := r.URL.Path[1:]
id, err := strconv.Atoi(idStr)
if err != nil || id < 1 || id > maxID {
id = rand.Intn(maxID)
}
message := findMessageByID(id)
userAgent := r.UserAgent()
if strings.HasPrefix(userAgent, "curl") || idStr == "plain" {
if text, ok := message.Text.(string); ok {
writePlainTextResponse(w, text, id)
} else {
fmt.Println("Ошибка: неверный Анек")
}
} else {
prevID, nextID := getPrevNextIDs(id)
htmlTemplate, _ := ioutil.ReadFile(html)
if text, ok := message.Text.(string); ok {
htmlContent := generateHTMLContent(w, htmlTemplate, id, text, prevID, nextID)
writeHTMLResponse(w, htmlContent)
} else {
fmt.Println("Ошибка: неверный Анек")
}
}
}
func findMessageByID(id int) *Message {
for _, message := range messages.Messages {
if message.ID == id {
return &message
}
}
return nil
}
func getPrevNextIDs(id int) (int, int) {
prevID := id - 1
if id <= 1 {
prevID = maxID
}
nextID := id + 1
if id >= maxID {
nextID = 1
}
return prevID, nextID
}
func generateHTMLContent(w http.ResponseWriter, htmlTemplate []byte, id int, text string, prevID int, nextID int) string {
htmlContent := string(htmlTemplate)
htmlContent = strings.ReplaceAll(htmlContent, "CURRENT_ANEK", strconv.Itoa(id))
htmlContent = strings.ReplaceAll(htmlContent, "TEMPLATE_TEXT", text)
htmlContent = strings.ReplaceAll(htmlContent, "PREV_ANEK", strconv.Itoa(prevID))
htmlContent = strings.ReplaceAll(htmlContent, "NEXT_ANEK", strconv.Itoa(nextID))
return htmlContent
}
func writePlainTextResponse(w http.ResponseWriter, text string, id int) {
w.Header().Set("Content-Type", "text/plain; charset=utf-8")
w.Write([]byte("Анек №" + strconv.Itoa(id) + "\n\n" + text + "\n"))
}
func writeHTMLResponse(w http.ResponseWriter, htmlContent string) {
w.Header().Set("Content-Type", "text/html; charset=utf-8")
w.Write([]byte(htmlContent))
}