generated from VLADIMIR/template
43 lines
1.1 KiB
Go
43 lines
1.1 KiB
Go
package data_parser
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"net/http"
|
||
"regexp"
|
||
|
||
"github.com/gocarina/gocsv"
|
||
)
|
||
|
||
type parser struct{}
|
||
|
||
func NewGoogleTableScheduleParser() IDataParser {
|
||
return &parser{}
|
||
}
|
||
|
||
func (p *parser) Parse(_ context.Context, url string, v interface{}) error {
|
||
re := regexp.MustCompile(`/d/([a-zA-Z0-9-_]+)`)
|
||
matches := re.FindStringSubmatch(url)
|
||
if len(matches) < 2 {
|
||
return fmt.Errorf("Не удалось найти ID таблицы в ссылке")
|
||
}
|
||
sheetID := matches[1]
|
||
|
||
csvURL := fmt.Sprintf("https://docs.google.com/spreadsheets/d/%s/export?format=csv", sheetID)
|
||
|
||
resp, err := http.Get(csvURL)
|
||
if err != nil {
|
||
return fmt.Errorf("Ошибка при скачивании таблицы: %v", err)
|
||
}
|
||
defer resp.Body.Close()
|
||
|
||
if resp.StatusCode != http.StatusOK {
|
||
return fmt.Errorf("Ошибка: статус код %d (убедитесь, что таблица публичная)", resp.StatusCode)
|
||
}
|
||
|
||
if err := gocsv.Unmarshal(resp.Body, v); err != nil {
|
||
return fmt.Errorf("Ошибка при парсинге CSV в структуру: %v", err)
|
||
}
|
||
return nil
|
||
}
|