kng-feed-api/cron-job/cronJobs.go

package cronjob

import (
	"encoding/xml"
	"fmt"
	"kng_feed_api/helper"
	"kng_feed_api/model"
	"os"
	"path/filepath"
	"strconv"
	"strings"
	"time"
)

func CreateXMLFeed() {

	type Category struct {
		Text     string `xml:",chardata"`
		Id       string `xml:"id,attr"`
		ParentId string `xml:"parentId,attr"`
		URL      string `xml:"url,attr"`
	}

	type Param struct {
		Text string `xml:",chardata"`
		Name string `xml:"name,attr"`
	}

	type Offer struct {
		Id         string  `xml:"id,attr"`
		Avalieble  string  `xml:"available,attr"`
		Name       string  `xml:"name"`
		URL        string  `xml:"url"`
		CategoryId string  `xml:"categoryId"`
		Vendor     string  `xml:"vendor"`
		Param      []Param `xml:"param"`
	}

	type Shop struct {
		Name     string     `xml:"name"`
		URL      string     `xml:"url"`
		Category []Category `xml:"categories>category"`
		Offer    []Offer    `xml:"offers>offer"`
	}

	type YamlCatalog struct {
		XMLName xml.Name `xml:"yml_catalog"`
		Date    string   `xml:"date,attr"`
		Shop    Shop     `xml:"shop"`
	}

	fmt.Println("CRON JOB WORK - FROM EXTERNAL FILE ", time.Now())

	categoryRows, err := model.GetAllCategories()
	if err != nil {
		return
	}

	rows, err := model.GetAllProducts()
	if err != nil {
		return
	}

	var yml_catalog = &YamlCatalog{}

	time_now := time.Now().UTC()
	yml_catalog.Date = time_now.Format("2006-01-02 15:04")

	var shop = &Shop{}
	shop.Name = "ООО \"Княгиня\""
	shop.URL = "https://b2bn.kngnn.ru"

	for _, value := range categoryRows {
		var category = Category{
			Text:     value.Name,
			ParentId: value.ParentId,
			Id:       value.Id,
			URL:      value.Id,
		}

		shop.Category = append(shop.Category, category)
	}

	for _, value := range rows {
		var offer = Offer{
			Id:         value.CodeUT10,
			Avalieble:  "true",
			Name:       value.Name,
			URL:        "https://b2bn.kngnn.ru/?item-code=" + value.CodeUT10,
			CategoryId: value.CategoryId,
			Vendor:     value.Manufacturer,
		}
		offer.Param = append(offer.Param, Param{
			Text: value.CodeUT10,
			Name: "code",
		})
		offer.Param = append(offer.Param, Param{
			Text: value.CodeUT10,
			Name: "code_ut10",
		})
		offer.Param = append(offer.Param, Param{
			Text: value.Manufacturer,
			Name: "manufacturer",
		})
		offer.Param = append(offer.Param, Param{
			Text: value.ArticleNumber,
			Name: "article_number",
		})
		offer.Param = append(offer.Param, Param{
			Text: value.Brand,
			Name: "brand",
		})
		offer.Param = append(offer.Param, Param{
			Text: value.Unit,
			Name: "unit",
		})
		offer.Param = append(offer.Param, Param{
			Text: value.GroupLimit,
			Name: "group_limit",
		})
		offer.Param = append(offer.Param, Param{
			Text: value.GroupPrice,
			Name: "group_price",
		})
		offer.Param = append(offer.Param, Param{
			Text: value.NumberCatalog,
			Name: "number_catalog",
		})
		offer.Param = append(offer.Param, Param{
			Text: value.NumberDrawing,
			Name: "number_drawing",
		})
		offer.Param = append(offer.Param, Param{
			Text: value.NumberBrand,
			Name: "number_brand",
		})
		offer.Param = append(offer.Param, Param{
			Text: value.NumberArticle,
			Name: "number_article",
		})
		offer.Param = append(offer.Param, Param{
			Text: value.NumberSuffix,
			Name: "number_suffix",
		})

		if value.IdNumbers != nil {
			numbers := strings.Split(*value.IdNumbers, ";")
			for idx, num := range numbers {
				offer.Param = append(offer.Param, Param{
					Text: num,
					Name: "article_number" + strconv.Itoa(idx+1),
				})
			}
		}

		shop.Offer = append(shop.Offer, offer)

	}

	yml_catalog.Shop = *shop

	byteXmlText, err := xml.MarshalIndent(yml_catalog, " ", " ")

	if err != nil {
		fmt.Printf("error: %v\n", err)
	} else {
		var Header = `<?xml version="1.0" encoding="UTF-8"?>` + "\n"
		var byteXmlText_withHeasder = []byte(Header + string(byteXmlText))

		os.WriteFile(filepath.Join(helper.RootDir(), "feed.xml"), byteXmlText_withHeasder, 0666)
	}
}