Compare commits
16 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
e054f458bb
|
|||
|
6c3de4b2ef
|
|||
|
a99349e75d
|
|||
|
4f6f54b631
|
|||
|
895146b472
|
|||
|
522012d7d5
|
|||
|
a3c33d7ec1
|
|||
|
d7f28413b2
|
|||
|
d4260f85ec
|
|||
|
6ea3683350
|
|||
|
bc03abeb9d
|
|||
|
14c251f3e4
|
|||
|
317ebfdf5f
|
|||
|
68bb6199b9
|
|||
|
d6b133a8e0
|
|||
|
c9486c54b2
|
@@ -1,28 +0,0 @@
|
|||||||
name: release
|
|
||||||
|
|
||||||
on:
|
|
||||||
push:
|
|
||||||
tags:
|
|
||||||
- 'v*'
|
|
||||||
|
|
||||||
jobs:
|
|
||||||
release:
|
|
||||||
runs-on: ubuntu-latest
|
|
||||||
steps:
|
|
||||||
- uses: actions/checkout@v4
|
|
||||||
name: Checkout
|
|
||||||
with:
|
|
||||||
fetch-depth: 0
|
|
||||||
- name: Setup go
|
|
||||||
uses: https://github.com/actions/setup-go@v4
|
|
||||||
with:
|
|
||||||
go-version: '>=1.23.6'
|
|
||||||
- name: Compile
|
|
||||||
run: make release
|
|
||||||
- name: Make release
|
|
||||||
id: use-go-action
|
|
||||||
uses: https://gitea.com/actions/release-action@main
|
|
||||||
with:
|
|
||||||
files: |-
|
|
||||||
bin/*.zip
|
|
||||||
api_key: '${{secrets.RELEASE_TOKEN}}'
|
|
||||||
@@ -0,0 +1,65 @@
|
|||||||
|
# https://docs.gitea.com/usage/actions/overview
|
||||||
|
# https://docs.github.com/ru/actions/reference/workflows-and-actions/contexts
|
||||||
|
|
||||||
|
name: Release
|
||||||
|
|
||||||
|
on:
|
||||||
|
push:
|
||||||
|
tags:
|
||||||
|
- 'v*'
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
release:
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
steps:
|
||||||
|
- name: Checkout
|
||||||
|
uses: actions/checkout@v4
|
||||||
|
with: # https://github.com/actions/checkout
|
||||||
|
fetch-depth: 0
|
||||||
|
|
||||||
|
- name: Set up go
|
||||||
|
uses: actions/setup-go@v4
|
||||||
|
with: # https://github.com/actions/setup-go
|
||||||
|
go-version: '>=1.24.2'
|
||||||
|
|
||||||
|
- name: Build release files (amd64)
|
||||||
|
run: make release ARCH=amd64
|
||||||
|
|
||||||
|
- name: Build release files (arm64)
|
||||||
|
run: make release ARCH=arm64
|
||||||
|
|
||||||
|
- name: Create new release
|
||||||
|
id: use-go-action
|
||||||
|
uses: https://gitea.com/actions/gitea-release-action@main
|
||||||
|
with: # https://gitea.com/actions/gitea-release-action
|
||||||
|
server_url: https://git.axenov.dev
|
||||||
|
token: '${{secrets.RELEASE_TOKEN}}'
|
||||||
|
files: |-
|
||||||
|
bin/*.zip
|
||||||
|
|
||||||
|
- name: Set up Docker Buildx
|
||||||
|
uses: docker/setup-buildx-action@v3
|
||||||
|
with: # https://github.com/docker/setup-buildx-action
|
||||||
|
build-args: "IPTVC_VERSION=${{ github.ref_name }}"
|
||||||
|
buildkitd-config-inline: |
|
||||||
|
# https://github.com/moby/buildkit/blob/master/docs/buildkitd.toml.md
|
||||||
|
[ registry."docker.io" ]
|
||||||
|
mirrors = ["https://dockerhub.timeweb.cloud", "https://dh-mirror.gitverse.ru"]
|
||||||
|
http = true
|
||||||
|
insecure = true
|
||||||
|
|
||||||
|
- name: Log in to Gitea Container Registry
|
||||||
|
uses: docker/login-action@v3
|
||||||
|
with: # https://github.com/docker/login-action
|
||||||
|
registry: git.axenov.dev
|
||||||
|
username: ${{ secrets.USERNAME }}
|
||||||
|
password: ${{ secrets.RELEASE_TOKEN }}
|
||||||
|
|
||||||
|
- name: Build and push Docker images
|
||||||
|
uses: docker/build-push-action@v5
|
||||||
|
with: # https://github.com/docker/build-push-action
|
||||||
|
context: .
|
||||||
|
push: true
|
||||||
|
tags: |
|
||||||
|
git.axenov.dev/iptv/iptvc:${{ github.ref_name }}
|
||||||
|
git.axenov.dev/iptv/iptvc:latest
|
||||||
@@ -12,3 +12,4 @@ output/
|
|||||||
iptvc
|
iptvc
|
||||||
|
|
||||||
!/**/*.gitkeep
|
!/**/*.gitkeep
|
||||||
|
.DS_Store
|
||||||
|
|||||||
@@ -0,0 +1,187 @@
|
|||||||
|
# Контекст проекта: IPTV Checker (iptvc)
|
||||||
|
|
||||||
|
> **Важно:** Этот файл создан автоматически на основе анализа кодовой базы. Используй его как справочный контекст при внесении изменений.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## Обзор проекта
|
||||||
|
|
||||||
|
**IPTV Checker (iptvc)** — консольная утилита для проверки доступности IPTV-плейлистов и каналов в форматах `m3u` / `m3u8`.
|
||||||
|
|
||||||
|
- **Назначение:** Быстрая массовая проверка плейлистов из файлов, по URL или из INI-реестра с расчётом статистики (online/offline).
|
||||||
|
- **Экосистема:** Часть проекта [m3u.su](https://m3u.su).
|
||||||
|
- **Лицензия:** MIT (см. `LICENSE`).
|
||||||
|
- **Автор:** Антон Аксенов.
|
||||||
|
|
||||||
|
### Основные технологии
|
||||||
|
|
||||||
|
- **Язык:** Go 1.23.6+
|
||||||
|
- **CLI-фреймворк:** [spf13/cobra](https://github.com/spf13/cobra)
|
||||||
|
- **Кеширование:** [Redis / KeyDB](https://github.com/redis/go-redis) (опционально, через `go-redis/v9`)
|
||||||
|
- **Конфигурация окружения:** [godotenv](https://github.com/joho/godotenv) + переменные окружения
|
||||||
|
- **Парсинг INI:** [gopkg.in/ini.v1](https://gopkg.in/ini.v1)
|
||||||
|
|
||||||
|
### Архитектура
|
||||||
|
|
||||||
|
Приложение построено по классической для Go CLI схеме:
|
||||||
|
|
||||||
|
```
|
||||||
|
cmd/ — команды Cobra (root, check)
|
||||||
|
app/ — бизнес-логика и инфраструктура
|
||||||
|
app.go — инициализация конфигурации, логгера, кеша; глобальные аргументы
|
||||||
|
checker/ — ядро проверки: подготовка списков, HTTP-опрос каналов, семафоры
|
||||||
|
playlist/ — модели Playlist/Channel/Group, парсинг m3u/m3u8
|
||||||
|
inifile/ — чтение реестра плейлистов из INI
|
||||||
|
config/ — чтение конфигурации из env (.env)
|
||||||
|
cache/ — подключение к Redis/KeyDB
|
||||||
|
logger/ — настройка вывода логов
|
||||||
|
tagfile/ — работа с файлом тегов (channels.json)
|
||||||
|
utils/ — вспомогательные функции (MD5, Fetch, ExpandPath и др.)
|
||||||
|
main.go — точка входа: контекст с отменой по сигналу, делегация cmd.ExecuteContext
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## Сборка и запуск
|
||||||
|
|
||||||
|
### Локальная разработка
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# Быстрый запуск из исходников
|
||||||
|
go run .
|
||||||
|
|
||||||
|
# Сборка бинаря
|
||||||
|
go build -o iptvc .
|
||||||
|
|
||||||
|
# Показать справку
|
||||||
|
./iptvc help
|
||||||
|
|
||||||
|
# Проверить плейлист из файла
|
||||||
|
./iptvc check -f mypls.m3u
|
||||||
|
|
||||||
|
# Проверить плейлист по ссылке
|
||||||
|
./iptvc check -u http://m3u.su/XYZ
|
||||||
|
|
||||||
|
# Проверить плейлисты из INI-файла
|
||||||
|
./iptvc check -i playlists.ini
|
||||||
|
|
||||||
|
# Только JSON-результат, без логов
|
||||||
|
./iptvc check -f mypls.m3u -j -q
|
||||||
|
```
|
||||||
|
|
||||||
|
### Makefile
|
||||||
|
|
||||||
|
| Команда | Описание |
|
||||||
|
|---------|----------|
|
||||||
|
| `make help` | Справка по целям |
|
||||||
|
| `make fmt` | `go fmt ./...` |
|
||||||
|
| `make vet` | `go vet ./...` |
|
||||||
|
| `make tidy` | `go mod tidy && go mod verify` |
|
||||||
|
| `make lint` | `fmt` + `vet` + сборка в `/dev/null` |
|
||||||
|
| `make clean` | Удаление скомпилированных бинарей и `bin/` |
|
||||||
|
| `make linux` | Сборка для Linux (`GOARCH=amd64` по умолчанию) |
|
||||||
|
| `make win` | Сборка для Windows |
|
||||||
|
| `make darwin` | Сборка для macOS |
|
||||||
|
| `make release` | Полный релиз: все ОС × amd64 + arm64, упаковка в zip |
|
||||||
|
|
||||||
|
> **Примечание:** Переменная `GOARCH` переопределяется через окружение, например: `make linux GOARCH=arm64`.
|
||||||
|
|
||||||
|
### Docker
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# Ручная сборка образа
|
||||||
|
docker build -t iptvc:latest .
|
||||||
|
|
||||||
|
# Скрипт сборки и публикации (с версионированием через git-тег)
|
||||||
|
./build-docker-image.sh [TAG]
|
||||||
|
```
|
||||||
|
|
||||||
|
Dockerfile использует многоступенчатую сборку:
|
||||||
|
1. `golang:1.25-alpine` — компиляция бинаря (`CGO_ENABLED=0`, `trimpath`, `ldflags`)
|
||||||
|
2. `alpine:3.22.2` — финальный минимальный образ с непривилегированным пользователем.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## Конфигурация
|
||||||
|
|
||||||
|
Приложение читает переменные окружения (файл `.env` опционален).
|
||||||
|
|
||||||
|
| Переменная | По умолчанию | Описание |
|
||||||
|
|------------|--------------|----------|
|
||||||
|
| `CACHE_ENABLED` | `false` | Включить подключение к кешу |
|
||||||
|
| `CACHE_HOST` | `localhost` | Хост Redis/KeyDB |
|
||||||
|
| `CACHE_PORT` | `6379` | Порт |
|
||||||
|
| `CACHE_USERNAME` | — | Пользователь |
|
||||||
|
| `CACHE_PASSWORD` | — | Пароль |
|
||||||
|
| `CACHE_DB` | `0` | Номер БД |
|
||||||
|
| `CACHE_TTL` | `1800` | TTL записей в секундах |
|
||||||
|
|
||||||
|
> Файл `.env.example` содержит пример заполнения.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## Ключевые файлы и пакеты
|
||||||
|
|
||||||
|
| Путь | Назначение |
|
||||||
|
|------|------------|
|
||||||
|
| `main.go` | Точка входа: `context.WithCancel`, обработка `SIGINT`/`SIGTERM`, вызов `cmd.ExecuteContext` |
|
||||||
|
| `cmd/root.go` | Корневая команда Cobra, глобальный флаг `--verbose` |
|
||||||
|
| `cmd/check.go` | Команда `check`: флаги (`-f`, `-u`, `-i`, `-c`, `-t`, `-r`, `-j`, `-q`, `--repeat`, `--every`), цикл повторений, вызов `checker` |
|
||||||
|
| `app/app.go` | Глобальные объекты `Args`, `Cache`, `Config`; `Init()` и `Shutdown()` |
|
||||||
|
| `app/checker/checker.go` | **Ядро:** подготовка списков, скачивание плейлистов, HTTP-опрос каналов с семафором, динамический расчёт `timeout`/`routines`, кеширование результатов |
|
||||||
|
| `app/playlist/playlist.go` | Структуры `Playlist`, `Channel`, `Group`; парсинг m3u/m3u8 (атрибуты `#EXTINF`, `#EXTGRP`, заголовок `#EXTM3U`) |
|
||||||
|
| `app/config/config.go` | Чтение конфигурации из env с дефолтами |
|
||||||
|
| `app/inifile/inifile.go` | Чтение реестра плейлистов в формате INI |
|
||||||
|
| `app/cache/cache.go` | Инициализация клиента Redis/KeyDB |
|
||||||
|
| `app/tagfile/tagfile.go` | Загрузка и поиск тегов для каналов |
|
||||||
|
| `app/utils/utils.go` | Утилиты: `Fetch`, `ExpandPath`, `Md5str`, `ArrayUnique` |
|
||||||
|
| `app/utils/utils_test.go` | Тесты утилит |
|
||||||
|
| `app/config/config_test.go` | Тесты конфигурации |
|
||||||
|
| `app/playlist/playlist_test.go` | Тесты парсера плейлистов |
|
||||||
|
| `docs/json.md` | Документация по структуре JSON-вывода |
|
||||||
|
| `build-docker-image.sh` | Скрипт сборки и пуша Docker-образа с версией из git |
|
||||||
|
| `Dockerfile` | Многоступенчатая сборка контейнера |
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## Правила разработки
|
||||||
|
|
||||||
|
### Стиль кода
|
||||||
|
|
||||||
|
- Код оформлен в едином стиле с лидирующими комментариями-заголовками (автор, лицензия, ссылка на репозиторий).
|
||||||
|
- Используются экспортируемые имена с подробными комментариями на русском языке (доминирующий язык комментариев).
|
||||||
|
- Глобальные переменные приложения (`Args`, `Cache`, `Config`) расположены в `app/app.go`.
|
||||||
|
|
||||||
|
### Тестирование
|
||||||
|
|
||||||
|
- Модульные тесты присутствуют для пакетов `utils`, `config`, `playlist`.
|
||||||
|
- Для запуска: `go test ./...`.
|
||||||
|
|
||||||
|
### Линтинг
|
||||||
|
|
||||||
|
- В Makefile есть цель `lint`, объединяющая `go fmt`, `go vet` и проверочную сборку.
|
||||||
|
- Перед коммитом рекомендуется выполнять `make lint`.
|
||||||
|
|
||||||
|
### Работа с зависимостями
|
||||||
|
|
||||||
|
- Модули Go (`go.mod` / `go.sum`).
|
||||||
|
- Обновление/очистка: `make tidy`.
|
||||||
|
|
||||||
|
### Соглашения по логике
|
||||||
|
|
||||||
|
- **Динамические параметры проверки:** количество горутин и HTTP-таймаут рассчитываются в `calcParameters(count)` на основе числа каналов. Чем больше каналов — тем больше параллелизм, но таймаут сокращается.
|
||||||
|
- **Семафор:** количество одновременных проверок ограничено через буферизированный канал (`chSemaphores`).
|
||||||
|
- **User-Agent:** при проверке каналов используется фиксированный заголовок `Mozilla/5.0 WINK/1.31.1 (AndroidTV/9) HlsWinkPlayer`.
|
||||||
|
- **TLS:** проверка выполняется с `InsecureSkipVerify: true`.
|
||||||
|
- **Кеш:** результаты проверки плейлистов сериализуются в JSON и сохраняются в Redis с TTL.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## Коды возврата
|
||||||
|
|
||||||
|
| Код | Значение |
|
||||||
|
|-----|----------|
|
||||||
|
| `0` | Успех |
|
||||||
|
| `1` | Общая ошибка |
|
||||||
|
| `2` | Команда `check` запущена без параметров `-f`, `-u` и `-c` (и нет INI по умолчанию) |
|
||||||
|
| `130` | Прерывание по сигналу (`SIGINT` / `SIGTERM`) |
|
||||||
+24
@@ -0,0 +1,24 @@
|
|||||||
|
FROM golang:1.25-alpine AS iptv-img-builder
|
||||||
|
ARG GOOS
|
||||||
|
ARG GOARCH
|
||||||
|
ARG IPTVC_VERSION
|
||||||
|
ENV CGO_ENABLED=0
|
||||||
|
ENV GOOS=${GOOS:-linux}
|
||||||
|
ENV GOARCH=${GOARCH:-amd64}
|
||||||
|
COPY go.mod go.sum ./
|
||||||
|
RUN go mod download
|
||||||
|
COPY . .
|
||||||
|
RUN go build \
|
||||||
|
-trimpath \
|
||||||
|
-ldflags="-s -w -X main.version=${IPTVC_VERSION}" \
|
||||||
|
-o /app/iptvc \
|
||||||
|
.
|
||||||
|
|
||||||
|
FROM alpine:3.22.2 AS iptv-img-checker
|
||||||
|
LABEL org.opencontainers.image.authors="Anthony Axenov <anthonyaxenov@gmail.com>"
|
||||||
|
WORKDIR /app
|
||||||
|
RUN apk add --no-cache ca-certificates tzdata
|
||||||
|
RUN addgroup -S iptvc-user && adduser -S -G iptvc-user -H -s /sbin/nologin iptvc-user
|
||||||
|
COPY --from=iptv-img-builder --chown=iptvc-user:iptvc-user /app/iptvc /app/iptvc
|
||||||
|
USER iptvc-user
|
||||||
|
ENTRYPOINT ["/app/iptvc"]
|
||||||
@@ -1,46 +1,62 @@
|
|||||||
.DEFAULT_GOAL=help
|
.DEFAULT_GOAL=help
|
||||||
|
|
||||||
BINARY_NAME=iptvc
|
BINARY_NAME := iptvc
|
||||||
ARCH=amd64
|
GOARCH ?= amd64
|
||||||
|
|
||||||
LINUX_PATH="bin/linux_${ARCH}"
|
|
||||||
WINDOWS_PATH="bin/windows_${ARCH}"
|
|
||||||
DARWIN_PATH="bin/darwin_${ARCH}"
|
|
||||||
|
|
||||||
LINUX_FILE="${LINUX_PATH}/${BINARY_NAME}"
|
|
||||||
WINDOWS_FILE="${WINDOWS_PATH}/${BINARY_NAME}.exe"
|
|
||||||
DARWIN_FILE="${DARWIN_PATH}/${BINARY_NAME}"
|
|
||||||
|
|
||||||
## clean: Remove all compiled binaries
|
## clean: Remove all compiled binaries
|
||||||
clean:
|
clean:
|
||||||
@go clean
|
@go clean
|
||||||
@rm -rf bin/
|
@rm -rf bin/
|
||||||
|
|
||||||
## linux: Build new binaries for linux (x64)
|
## linux: Build new binaries for linux
|
||||||
linux:
|
linux:
|
||||||
@rm -rf ${LINUX_PATH}
|
@rm -rf bin/linux_$(GOARCH)
|
||||||
@GOARCH=${ARCH} GOOS=linux go build -o ${LINUX_FILE} . && echo "Compiled: ${LINUX_FILE}"
|
@GOARCH=$(GOARCH) GOOS=linux go build -o bin/linux_$(GOARCH)/$(BINARY_NAME) .
|
||||||
|
@zip -j bin/linux_$(GOARCH).zip bin/linux_$(GOARCH)/$(BINARY_NAME)
|
||||||
|
@echo "Compiled: bin/linux_$(GOARCH)/$(BINARY_NAME) ($(GOARCH))"
|
||||||
|
|
||||||
## win: Build new binaries for windows (x64)
|
## win: Build new binaries for windows
|
||||||
win:
|
win:
|
||||||
@rm -rf ${WINDOWS_PATH}
|
@rm -rf bin/windows_$(GOARCH)
|
||||||
@GOARCH=${ARCH} GOOS=windows go build -o ${WINDOWS_FILE} . && echo "Compiled: ${WINDOWS_FILE}"
|
@GOARCH=$(GOARCH) GOOS=windows go build -o bin/windows_$(GOARCH)/$(BINARY_NAME).exe .
|
||||||
|
@zip -j bin/windows_$(GOARCH).zip bin/windows_$(GOARCH)/$(BINARY_NAME).exe
|
||||||
|
@echo "Compiled: bin/windows_$(GOARCH)/$(BINARY_NAME).exe ($(GOARCH))"
|
||||||
|
|
||||||
## darwin: Build new binaries for darwin (x64)
|
## darwin: Build new binaries for darwin
|
||||||
darwin:
|
darwin:
|
||||||
@rm -rf ${DARWIN_PATH}
|
@rm -rf bin/darwin_$(GOARCH)
|
||||||
@GOARCH=${ARCH} GOOS=darwin go build -o ${DARWIN_FILE} . && echo "Compiled: ${DARWIN_FILE}"
|
@GOARCH=$(GOARCH) GOOS=darwin go build -o bin/darwin_$(GOARCH)/$(BINARY_NAME) .
|
||||||
|
@zip -j bin/darwin_$(GOARCH).zip bin/darwin_$(GOARCH)/$(BINARY_NAME)
|
||||||
## all: Build new binaries for linux, windows and darwin (x64)
|
@echo "Compiled: bin/darwin_$(GOARCH)/$(BINARY_NAME) ($(GOARCH))"
|
||||||
all: clean linux win darwin
|
|
||||||
|
|
||||||
## release: Build all binaries and zip them
|
## release: Build all binaries and zip them
|
||||||
release: clean darwin linux win
|
release: clean
|
||||||
@zip -j ${LINUX_PATH}.zip ${LINUX_FILE}
|
@make linux GOARCH=amd64
|
||||||
@zip -j ${DARWIN_PATH}.zip ${DARWIN_FILE}
|
@make linux GOARCH=arm64
|
||||||
@zip -j ${WINDOWS_PATH}.zip ${WINDOWS_FILE}
|
@make win GOARCH=amd64
|
||||||
|
@make win GOARCH=arm64
|
||||||
|
@make darwin GOARCH=amd64
|
||||||
|
@make darwin GOARCH=arm64
|
||||||
|
|
||||||
|
## fmt: Format Go source code
|
||||||
|
fmt:
|
||||||
|
@go fmt ./...
|
||||||
|
|
||||||
|
## vet: Run go vet
|
||||||
|
vet:
|
||||||
|
@go vet ./...
|
||||||
|
|
||||||
|
## tidy: Tidy and verify Go modules
|
||||||
|
tidy:
|
||||||
|
@go mod tidy
|
||||||
|
@go mod verify
|
||||||
|
|
||||||
|
## lint: Run fmt, vet and build as basic linting
|
||||||
|
lint: fmt vet
|
||||||
|
@go build -o /dev/null .
|
||||||
|
@echo "Linting complete"
|
||||||
|
|
||||||
## help: Show this message and exit
|
## help: Show this message and exit
|
||||||
help: Makefile
|
help: Makefile
|
||||||
@echo "Choose a command run:"
|
@echo "Available recipes:"
|
||||||
@sed -n 's/^##//p' $< | column -t -s ':' | sed -e 's/^/ /'
|
@sed -n 's/^## //p' $< | column -t -s ':'
|
||||||
|
|||||||
@@ -5,22 +5,21 @@
|
|||||||
Консольная программа для проверки IPTV-плейлистов в формате m3u или m3u8.
|
Консольная программа для проверки IPTV-плейлистов в формате m3u или m3u8.
|
||||||
|
|
||||||
> **Веб-сайт:** [m3u.su](https://m3u.su)
|
> **Веб-сайт:** [m3u.su](https://m3u.su)
|
||||||
> Исходный код: [git.axenov.dev/IPTV/iptvc](https://git.axenov.dev/IPTV/iptvc)
|
> **Документация:** [m3u.su/docs](https://m3u.su/docs)
|
||||||
|
> Исходный код: [git.axenov.dev/IPTV](https://git.axenov.dev/IPTV)
|
||||||
> Telegram-канал: [@iptv_aggregator](https://t.me/iptv_aggregator)
|
> Telegram-канал: [@iptv_aggregator](https://t.me/iptv_aggregator)
|
||||||
> Обсуждение: [@iptv_aggregator_chat](https://t.me/iptv_aggregator_chat)
|
> Обсуждение: [@iptv_aggregator_chat](https://t.me/iptv_aggregator_chat)
|
||||||
> Дополнительные сведения:
|
> Бот: [@iptv_aggregator_bot](https://t.me/iptv_aggregator_bot)
|
||||||
> * [./docs](./docs)
|
|
||||||
> * [git.axenov.dev/IPTV/.profile](https://git.axenov.dev/IPTV/.profile)
|
|
||||||
|
|
||||||
## Установка
|
## Установка
|
||||||
|
|
||||||
Достаточно скачать и распаковать архив с подходящим исполняемым файлом [со страницы последнего релиза](https://git.axenov.dev/IPTV/iptvc/releases/latest):
|
Достаточно скачать и распаковать архив с подходящим исполняемым файлом [со страницы последнего релиза](https://git.axenov.dev/IPTV/iptvc/releases/latest):
|
||||||
|
|
||||||
| ОС | Архив | Платформа |
|
| ОС | Скачать для `amd64` | Скачать для `arm64` |
|
||||||
|---------|----------------------|-----------|
|
| ------- | ------------------------------------------------------------------------------------------------- | ------------------------------------------------------------------------------------------------- |
|
||||||
| Linux | `linux_amd64.zip` | x64 |
|
| Linux | [linux_amd64.zip](https://git.axenov.dev/IPTV/iptvc/releases/download/latest/linux_amd64.zip) | [linux_arm64.zip](https://git.axenov.dev/IPTV/iptvc/releases/download/latest/linux_arm64.zip) |
|
||||||
| MacOS | `darwin_amd64.zip` | x64 |
|
| MacOS | [darwin_amd64.zip](https://git.axenov.dev/IPTV/iptvc/releases/download/latest/darwin_amd64.zip) | [darwin_arm64.zip](https://git.axenov.dev/IPTV/iptvc/releases/download/latest/darwin_arm64.zip) |
|
||||||
| Windows | `windows_amd64.zip` | x64 |
|
| Windows | [windows_amd64.zip](https://git.axenov.dev/IPTV/iptvc/releases/download/latest/windows_amd64.zip) | [windows_arm64.zip](https://git.axenov.dev/IPTV/iptvc/releases/download/latest/windows_arm64.zip) |
|
||||||
|
|
||||||
## Компиляция
|
## Компиляция
|
||||||
|
|
||||||
@@ -29,7 +28,7 @@
|
|||||||
|
|
||||||
1. Склонировать репозиторий
|
1. Склонировать репозиторий
|
||||||
2. Находясь в корне репозитория, следует выполнить `make` или `make help` для получения справки.
|
2. Находясь в корне репозитория, следует выполнить `make` или `make help` для получения справки.
|
||||||
3. Другой способ -- выполнить `go run .` для быстрого запуска.
|
3. Другой способ — выполнить `go run .` для быстрого запуска.
|
||||||
|
|
||||||
## Быстрый старт
|
## Быстрый старт
|
||||||
|
|
||||||
@@ -69,11 +68,11 @@
|
|||||||
|
|
||||||
### Другие возможности команды `check`
|
### Другие возможности команды `check`
|
||||||
|
|
||||||
* `--random|-r X` -- проверить X случайных плейлистов из ini-файла
|
* `--random|-r X` — проверить X случайных плейлистов из ini-файла
|
||||||
* `--json|-j` -- вывести результаты проверки в формате JSON
|
* `--json|-j` — вывести результаты проверки в формате JSON
|
||||||
* `--quiet|-q` -- полностью подавить вывод лога (включая отладочную информацию)
|
* `--quiet|-q` — полностью подавить вывод лога (включая отладочную информацию)
|
||||||
* `--verbose|-v` -- добавить в лог более подробную отладочную информацию (значительно увеличит количество строк!)
|
* `--verbose|-v` — добавить в лог более подробную отладочную информацию (значительно увеличит количество строк!)
|
||||||
* `--tags|-t` -- файл с перечислением тегов (подробности см. [здесь](https://git.axenov.dev/IPTV/playlists#файл-channelsjson))
|
* `--tags|-t` — файл с перечислением тегов (подробности см. [здесь](https://git.axenov.dev/IPTV/playlists#файл-channelsjson))
|
||||||
|
|
||||||
Например, можно получить только json с результатами, передать его в `jq` и, отфильтровав результат, вывести названия оффлайн каналов:
|
Например, можно получить только json с результатами, передать его в `jq` и, отфильтровав результат, вывести названия оффлайн каналов:
|
||||||
|
|
||||||
@@ -126,8 +125,8 @@ pls='https://example.com/list2.m3u'
|
|||||||
### Параметры проверки
|
### Параметры проверки
|
||||||
|
|
||||||
Выше в п.7 видно некоторые служебные данные:
|
Выше в п.7 видно некоторые служебные данные:
|
||||||
* `timeout` -- таймаут каждого запроса в секундах (макс. время ожидания ответа канала);
|
* `timeout` — таймаут каждого запроса в секундах (макс. время ожидания ответа канала);
|
||||||
* `routines` -- количество одновременных проверок.
|
* `routines` — количество одновременных проверок.
|
||||||
|
|
||||||
Эти параметры рассчитываются динамически для каждого плейлиста в отдельности, исходя из количества каналов в каждом (`count`).
|
Эти параметры рассчитываются динамически для каждого плейлиста в отдельности, исходя из количества каналов в каждом (`count`).
|
||||||
См. [app/checker/checker.go](app/checker/checker.go) для подробностей.
|
См. [app/checker/checker.go](app/checker/checker.go) для подробностей.
|
||||||
@@ -151,9 +150,9 @@ pls='https://example.com/list2.m3u'
|
|||||||
|
|
||||||
### Коды возврата
|
### Коды возврата
|
||||||
|
|
||||||
* 0 -- успех
|
* 0 — успех
|
||||||
* 1 -- общая ошибка, см. вывод
|
* 1 — общая ошибка, см. вывод
|
||||||
* 2 -- команде `check` не переданы параметры `--file`, `--url` и `--code`
|
* 2 — команде `check` не переданы параметры `--file`, `--url` и `--code`
|
||||||
|
|
||||||
## Лицензия
|
## Лицензия
|
||||||
|
|
||||||
|
|||||||
+25
-1
@@ -10,17 +10,20 @@ import (
|
|||||||
"axenov/iptv-checker/app/cache"
|
"axenov/iptv-checker/app/cache"
|
||||||
"axenov/iptv-checker/app/config"
|
"axenov/iptv-checker/app/config"
|
||||||
"axenov/iptv-checker/app/logger"
|
"axenov/iptv-checker/app/logger"
|
||||||
|
"log"
|
||||||
|
|
||||||
"github.com/redis/go-redis/v9"
|
"github.com/redis/go-redis/v9"
|
||||||
)
|
)
|
||||||
|
|
||||||
const VERSION = "1.0.5"
|
var version = "dev"
|
||||||
|
|
||||||
// Arguments описывает аргументы командной строки
|
// Arguments описывает аргументы командной строки
|
||||||
type Arguments struct {
|
type Arguments struct {
|
||||||
IniPath string
|
IniPath string
|
||||||
TagsPath string
|
TagsPath string
|
||||||
RandomCount uint
|
RandomCount uint
|
||||||
|
RepeatCount uint
|
||||||
|
RepeatEverySec uint
|
||||||
NeedJson bool
|
NeedJson bool
|
||||||
NeedQuiet bool
|
NeedQuiet bool
|
||||||
Verbose bool
|
Verbose bool
|
||||||
@@ -32,6 +35,18 @@ var (
|
|||||||
Config *config.Config
|
Config *config.Config
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// SetVersion устанавливает версию приложения
|
||||||
|
func SetVersion(v string) {
|
||||||
|
if v != "" {
|
||||||
|
version = v
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Version возвращает версию приложения
|
||||||
|
func Version() string {
|
||||||
|
return version
|
||||||
|
}
|
||||||
|
|
||||||
// Init инициализирует конфигурацию и подключение к keydb
|
// Init инициализирует конфигурацию и подключение к keydb
|
||||||
func Init() {
|
func Init() {
|
||||||
Config = config.Init()
|
Config = config.Init()
|
||||||
@@ -40,3 +55,12 @@ func Init() {
|
|||||||
Cache = cache.Init(&Config.Cache)
|
Cache = cache.Init(&Config.Cache)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Shutdown корректно завершает работу приложения
|
||||||
|
func Shutdown() {
|
||||||
|
if Cache != nil {
|
||||||
|
if err := Cache.Close(); err != nil {
|
||||||
|
log.Printf("Error closing cache connection: %s", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
+77
-28
@@ -28,13 +28,20 @@ import (
|
|||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
var (
|
// Checker выполняет проверку плейлистов и каналов.
|
||||||
|
type Checker struct {
|
||||||
tagBlocks []tagfile.TagBlock
|
tagBlocks []tagfile.TagBlock
|
||||||
ctx = context.Background()
|
}
|
||||||
)
|
|
||||||
|
// NewChecker создаёт новый экземпляр Checker.
|
||||||
|
func NewChecker(tagsPath string) *Checker {
|
||||||
|
return &Checker{
|
||||||
|
tagBlocks: tagfile.Init(tagsPath),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// PrepareListsToCheck готовит список плейлистов для проверки
|
// PrepareListsToCheck готовит список плейлистов для проверки
|
||||||
func PrepareListsToCheck(files []string, urls []string, codes []string) []playlist.Playlist {
|
func (c *Checker) PrepareListsToCheck(files []string, urls []string, codes []string) []playlist.Playlist {
|
||||||
var lists []playlist.Playlist
|
var lists []playlist.Playlist
|
||||||
|
|
||||||
if len(files) > 0 {
|
if len(files) > 0 {
|
||||||
@@ -51,7 +58,11 @@ func PrepareListsToCheck(files []string, urls []string, codes []string) []playli
|
|||||||
|
|
||||||
if len(urls) > 0 {
|
if len(urls) > 0 {
|
||||||
for _, url := range urls {
|
for _, url := range urls {
|
||||||
pls, _ := playlist.MakeFromUrl(url)
|
pls, err := playlist.MakeFromUrl(url)
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("Warning: %s, skipping\n", err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
lists = append(lists, pls)
|
lists = append(lists, pls)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -74,7 +85,7 @@ func PrepareListsToCheck(files []string, urls []string, codes []string) []playli
|
|||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
if app.Config.Cache.IsActive {
|
if app.Config.Cache.IsActive {
|
||||||
cachedLists := getCachedPlaylists()
|
cachedLists := c.getCachedPlaylists()
|
||||||
for key := range ini.Lists {
|
for key := range ini.Lists {
|
||||||
if _, ok := cachedLists[key]; ok {
|
if _, ok := cachedLists[key]; ok {
|
||||||
continue
|
continue
|
||||||
@@ -97,20 +108,30 @@ func PrepareListsToCheck(files []string, urls []string, codes []string) []playli
|
|||||||
}
|
}
|
||||||
|
|
||||||
// getCachedPlaylists возвращает из кеша проверенные ранее плейлисты
|
// getCachedPlaylists возвращает из кеша проверенные ранее плейлисты
|
||||||
func getCachedPlaylists() map[string]playlist.Playlist {
|
func (c *Checker) getCachedPlaylists() map[string]playlist.Playlist {
|
||||||
result := make(map[string]playlist.Playlist)
|
result := make(map[string]playlist.Playlist)
|
||||||
keys := app.Cache.Keys(ctx, "*")
|
ctx := context.Background()
|
||||||
for _, key := range keys.Val() {
|
iter := app.Cache.Scan(ctx, 0, "*", 100).Iterator()
|
||||||
value := app.Cache.Get(ctx, key).Val()
|
for iter.Next(ctx) {
|
||||||
|
key := iter.Val()
|
||||||
|
value, err := app.Cache.Get(ctx, key).Result()
|
||||||
|
if err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
var pls playlist.Playlist
|
var pls playlist.Playlist
|
||||||
_ = json.Unmarshal([]byte(value), &pls)
|
if err := json.Unmarshal([]byte(value), &pls); err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
result[pls.Code] = pls
|
result[pls.Code] = pls
|
||||||
}
|
}
|
||||||
|
if err := iter.Err(); err != nil {
|
||||||
|
log.Printf("Error scanning cache: %s", err)
|
||||||
|
}
|
||||||
return result
|
return result
|
||||||
}
|
}
|
||||||
|
|
||||||
// CheckPlaylists проверяет плейлисты и возвращает их же с результатами проверки
|
// CheckPlaylists проверяет плейлисты и возвращает их же с результатами проверки
|
||||||
func CheckPlaylists(lists []playlist.Playlist) (int, int) {
|
func (c *Checker) CheckPlaylists(ctx context.Context, lists []playlist.Playlist) (int, int) {
|
||||||
count := len(lists)
|
count := len(lists)
|
||||||
if count == 0 {
|
if count == 0 {
|
||||||
log.Println("There are no playlists to check")
|
log.Println("There are no playlists to check")
|
||||||
@@ -119,7 +140,6 @@ func CheckPlaylists(lists []playlist.Playlist) (int, int) {
|
|||||||
|
|
||||||
log.Printf("%d playlists will be checked\n", len(lists))
|
log.Printf("%d playlists will be checked\n", len(lists))
|
||||||
step, onlineCount, offlineCount := 0, 0, 0
|
step, onlineCount, offlineCount := 0, 0, 0
|
||||||
tagBlocks = tagfile.Init(app.Args.TagsPath)
|
|
||||||
|
|
||||||
for idx := range lists {
|
for idx := range lists {
|
||||||
pls := lists[idx]
|
pls := lists[idx]
|
||||||
@@ -146,7 +166,7 @@ func CheckPlaylists(lists []playlist.Playlist) (int, int) {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
log.Printf("Cannot read playlist [%s]: %s\n", pls.Url, err)
|
log.Printf("Cannot read playlist [%s]: %s\n", pls.Url, err)
|
||||||
offlineCount++
|
offlineCount++
|
||||||
cachePlaylist(pls)
|
c.cachePlaylist(pls)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -156,15 +176,15 @@ func CheckPlaylists(lists []playlist.Playlist) (int, int) {
|
|||||||
pls = pls.Parse()
|
pls = pls.Parse()
|
||||||
|
|
||||||
log.Printf("Parsed, checking channels (%d)...\n", len(pls.Channels))
|
log.Printf("Parsed, checking channels (%d)...\n", len(pls.Channels))
|
||||||
pls = CheckChannels(pls)
|
pls = c.CheckChannels(ctx, pls)
|
||||||
lists[idx] = pls
|
lists[idx] = pls
|
||||||
cachePlaylist(pls)
|
c.cachePlaylist(pls)
|
||||||
}
|
}
|
||||||
|
|
||||||
return onlineCount, offlineCount
|
return onlineCount, offlineCount
|
||||||
}
|
}
|
||||||
|
|
||||||
func cachePlaylist(pls playlist.Playlist) {
|
func (c *Checker) cachePlaylist(pls playlist.Playlist) {
|
||||||
if !app.Config.Cache.IsActive {
|
if !app.Config.Cache.IsActive {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -172,19 +192,27 @@ func cachePlaylist(pls playlist.Playlist) {
|
|||||||
jsonBytes, err := json.Marshal(pls)
|
jsonBytes, err := json.Marshal(pls)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Printf("Error while saving playlist to cache: %s", err)
|
log.Printf("Error while saving playlist to cache: %s", err)
|
||||||
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
key := pls.Code
|
||||||
|
if key == "" {
|
||||||
|
key = "raw:" + utils.Md5str(pls.Url)
|
||||||
|
}
|
||||||
|
|
||||||
|
ctx := context.Background()
|
||||||
ttl := time.Duration(app.Config.Cache.Ttl) * time.Second
|
ttl := time.Duration(app.Config.Cache.Ttl) * time.Second
|
||||||
written := app.Cache.Set(ctx, pls.Code, string(jsonBytes), ttl)
|
written := app.Cache.Set(ctx, key, string(jsonBytes), ttl)
|
||||||
if written.Err() != nil {
|
if written.Err() != nil {
|
||||||
log.Printf("Error while saving playlist to cache: %s", err)
|
log.Printf("Error while saving playlist to cache: %s", written.Err())
|
||||||
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
log.Println("Cached sucessfully")
|
log.Println("Cached sucessfully")
|
||||||
}
|
}
|
||||||
|
|
||||||
// CheckChannels проверяет каналы и возвращает их же с результатами проверки
|
// CheckChannels проверяет каналы и возвращает их же с результатами проверки
|
||||||
func CheckChannels(pls playlist.Playlist) playlist.Playlist {
|
func (c *Checker) CheckChannels(ctx context.Context, pls playlist.Playlist) playlist.Playlist {
|
||||||
type errorData struct {
|
type errorData struct {
|
||||||
tvChannel playlist.Channel
|
tvChannel playlist.Channel
|
||||||
err error
|
err error
|
||||||
@@ -196,8 +224,17 @@ func CheckChannels(pls playlist.Playlist) playlist.Playlist {
|
|||||||
return pls
|
return pls
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pls.OnlineCount = 0
|
||||||
|
pls.OfflineCount = 0
|
||||||
|
|
||||||
timeout, routines := calcParameters(count)
|
timeout, routines := calcParameters(count)
|
||||||
httpClient := http.Client{Timeout: timeout}
|
tr := &http.Transport{
|
||||||
|
TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
|
||||||
|
}
|
||||||
|
httpClient := http.Client{
|
||||||
|
Timeout: timeout,
|
||||||
|
Transport: tr,
|
||||||
|
}
|
||||||
chSemaphores := make(chan struct{}, routines)
|
chSemaphores := make(chan struct{}, routines)
|
||||||
chOnline := make(chan playlist.Channel, len(pls.Channels))
|
chOnline := make(chan playlist.Channel, len(pls.Channels))
|
||||||
chOffline := make(chan playlist.Channel, len(pls.Channels))
|
chOffline := make(chan playlist.Channel, len(pls.Channels))
|
||||||
@@ -208,16 +245,18 @@ func CheckChannels(pls playlist.Playlist) playlist.Playlist {
|
|||||||
for _, tvChannel := range pls.Channels {
|
for _, tvChannel := range pls.Channels {
|
||||||
wg.Add(1)
|
wg.Add(1)
|
||||||
go func(tvChannel playlist.Channel) {
|
go func(tvChannel playlist.Channel) {
|
||||||
chSemaphores <- struct{}{}
|
|
||||||
defer func() {
|
defer func() {
|
||||||
|
if r := recover(); r != nil {
|
||||||
|
log.Printf("Panic while checking channel '%s': %v", tvChannel.Title, r)
|
||||||
|
}
|
||||||
<-chSemaphores
|
<-chSemaphores
|
||||||
wg.Done()
|
wg.Done()
|
||||||
}()
|
}()
|
||||||
|
chSemaphores <- struct{}{}
|
||||||
|
|
||||||
tvChannel.Tags = getTagsForChannel(tvChannel)
|
tvChannel.Tags = c.getTagsForChannel(tvChannel)
|
||||||
|
|
||||||
http.DefaultTransport.(*http.Transport).TLSClientConfig = &tls.Config{InsecureSkipVerify: true}
|
req, err := http.NewRequestWithContext(ctx, "GET", tvChannel.URL, nil)
|
||||||
req, err := http.NewRequest("GET", tvChannel.URL, nil)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
data := errorData{tvChannel: tvChannel, err: err}
|
data := errorData{tvChannel: tvChannel, err: err}
|
||||||
chError <- data
|
chError <- data
|
||||||
@@ -238,7 +277,13 @@ func CheckChannels(pls playlist.Playlist) playlist.Playlist {
|
|||||||
tvChannel.IsOnline = tvChannel.Status < http.StatusBadRequest
|
tvChannel.IsOnline = tvChannel.Status < http.StatusBadRequest
|
||||||
tvChannel.ContentType = resp.Header.Get("Content-Type")
|
tvChannel.ContentType = resp.Header.Get("Content-Type")
|
||||||
chunk := io.LimitReader(resp.Body, 512) // just for sure
|
chunk := io.LimitReader(resp.Body, 512) // just for sure
|
||||||
bodyBytes, _ := io.ReadAll(chunk)
|
bodyBytes, err := io.ReadAll(chunk)
|
||||||
|
if err != nil {
|
||||||
|
_ = resp.Body.Close()
|
||||||
|
data := errorData{tvChannel: tvChannel, err: err}
|
||||||
|
chError <- data
|
||||||
|
return
|
||||||
|
}
|
||||||
bodyString := string(bodyBytes)
|
bodyString := string(bodyBytes)
|
||||||
_ = resp.Body.Close()
|
_ = resp.Body.Close()
|
||||||
contentType := http.DetectContentType(bodyBytes)
|
contentType := http.DetectContentType(bodyBytes)
|
||||||
@@ -322,6 +367,10 @@ func CheckChannels(pls playlist.Playlist) playlist.Playlist {
|
|||||||
|
|
||||||
// calcParameters вычисляет оптимальное количество горутин и таймаут запроса
|
// calcParameters вычисляет оптимальное количество горутин и таймаут запроса
|
||||||
func calcParameters(count int) (time.Duration, int) {
|
func calcParameters(count int) (time.Duration, int) {
|
||||||
|
if count <= 0 {
|
||||||
|
return 10 * time.Second, 1
|
||||||
|
}
|
||||||
|
|
||||||
routines := count
|
routines := count
|
||||||
if routines > 3000 {
|
if routines > 3000 {
|
||||||
routines = 3000
|
routines = 3000
|
||||||
@@ -357,10 +406,10 @@ func calcParameters(count int) (time.Duration, int) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// getTagsForChannel ищет и возвращает теги для канала
|
// getTagsForChannel ищет и возвращает теги для канала
|
||||||
func getTagsForChannel(tvChannel playlist.Channel) []string {
|
func (c *Checker) getTagsForChannel(tvChannel playlist.Channel) []string {
|
||||||
var foundTags []string
|
var foundTags []string
|
||||||
|
|
||||||
for _, block := range tagBlocks {
|
for _, block := range c.tagBlocks {
|
||||||
tags := block.GetTags(tvChannel)
|
tags := block.GetTags(tvChannel)
|
||||||
if tags != nil {
|
if tags != nil {
|
||||||
foundTags = append(foundTags, tags...)
|
foundTags = append(foundTags, tags...)
|
||||||
|
|||||||
@@ -7,14 +7,14 @@
|
|||||||
package config
|
package config
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/joho/godotenv"
|
|
||||||
"os"
|
"os"
|
||||||
"strconv"
|
"strconv"
|
||||||
|
|
||||||
|
"github.com/joho/godotenv"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Config описывает конфигурацию
|
// Config описывает конфигурацию
|
||||||
type Config struct {
|
type Config struct {
|
||||||
DebugMode bool
|
|
||||||
Cache CacheConfig
|
Cache CacheConfig
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -34,7 +34,6 @@ type CacheConfig struct {
|
|||||||
func Init() *Config {
|
func Init() *Config {
|
||||||
_ = godotenv.Load(".env")
|
_ = godotenv.Load(".env")
|
||||||
return &Config{
|
return &Config{
|
||||||
//DebugMode: readEnvBoolean("APP_DEBUG", false),
|
|
||||||
Cache: CacheConfig{
|
Cache: CacheConfig{
|
||||||
IsEnabled: readEnvBoolean("CACHE_ENABLED", false),
|
IsEnabled: readEnvBoolean("CACHE_ENABLED", false),
|
||||||
Host: readEnv("CACHE_HOST", "localhost"),
|
Host: readEnv("CACHE_HOST", "localhost"),
|
||||||
|
|||||||
@@ -0,0 +1,49 @@
|
|||||||
|
package config
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestReadEnv(t *testing.T) {
|
||||||
|
_ = os.Setenv("TEST_VAR_IPTVC", "value123")
|
||||||
|
defer os.Unsetenv("TEST_VAR_IPTVC")
|
||||||
|
|
||||||
|
got := readEnv("TEST_VAR_IPTVC", "default")
|
||||||
|
if got != "value123" {
|
||||||
|
t.Errorf("readEnv = %q, want %q", got, "value123")
|
||||||
|
}
|
||||||
|
|
||||||
|
gotDefault := readEnv("TEST_VAR_MISSING", "default")
|
||||||
|
if gotDefault != "default" {
|
||||||
|
t.Errorf("readEnv default = %q, want %q", gotDefault, "default")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestReadEnvBoolean(t *testing.T) {
|
||||||
|
_ = os.Setenv("TEST_BOOL", "true")
|
||||||
|
defer os.Unsetenv("TEST_BOOL")
|
||||||
|
|
||||||
|
if !readEnvBoolean("TEST_BOOL", false) {
|
||||||
|
t.Error("readEnvBoolean(true) returned false")
|
||||||
|
}
|
||||||
|
|
||||||
|
if readEnvBoolean("TEST_BOOL_MISSING", false) {
|
||||||
|
t.Error("readEnvBoolean(missing, false) returned true")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestReadEnvInteger(t *testing.T) {
|
||||||
|
_ = os.Setenv("TEST_INT", "42")
|
||||||
|
defer os.Unsetenv("TEST_INT")
|
||||||
|
|
||||||
|
got := readEnvInteger("TEST_INT", 0)
|
||||||
|
if got != 42 {
|
||||||
|
t.Errorf("readEnvInteger = %d, want 42", got)
|
||||||
|
}
|
||||||
|
|
||||||
|
gotDefault := readEnvInteger("TEST_INT_MISSING", 10)
|
||||||
|
if gotDefault != 10 {
|
||||||
|
t.Errorf("readEnvInteger default = %d, want 10", gotDefault)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -9,7 +9,6 @@ package logger
|
|||||||
import (
|
import (
|
||||||
"io"
|
"io"
|
||||||
"log"
|
"log"
|
||||||
"log/slog"
|
|
||||||
"os"
|
"os"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -20,22 +19,3 @@ func Init(quiet bool) {
|
|||||||
log.SetOutput(io.Discard)
|
log.SetOutput(io.Discard)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// InitSlog инициализирует продвинутый логгер
|
|
||||||
// TODO пока непонятно что с этим делать
|
|
||||||
func InitSlog(quiet bool, debug bool) {
|
|
||||||
writer := os.Stdout
|
|
||||||
if quiet {
|
|
||||||
writer = nil
|
|
||||||
}
|
|
||||||
|
|
||||||
level := slog.LevelInfo
|
|
||||||
if debug {
|
|
||||||
level = slog.LevelDebug
|
|
||||||
}
|
|
||||||
|
|
||||||
options := slog.HandlerOptions{Level: level, AddSource: false}
|
|
||||||
handler := slog.NewTextHandler(writer, &options)
|
|
||||||
logger := slog.New(handler)
|
|
||||||
slog.SetDefault(logger)
|
|
||||||
}
|
|
||||||
|
|||||||
+26
-14
@@ -54,8 +54,10 @@ type Playlist struct {
|
|||||||
CheckedAt int64 `json:"checkedAt"` // Время проверки в формате UNIX timestamp
|
CheckedAt int64 `json:"checkedAt"` // Время проверки в формате UNIX timestamp
|
||||||
}
|
}
|
||||||
|
|
||||||
// tmpChannel хранит временные данные о канале, который обрабатывается в Parse
|
var (
|
||||||
var tmpChannel = Channel{}
|
attrRegex = regexp.MustCompile(`(?U)([a-z-]+)="(.*)"`)
|
||||||
|
titleRegex = regexp.MustCompile(`['"]?\s*,\s*(.+)`)
|
||||||
|
)
|
||||||
|
|
||||||
// MakeFromFile создаёт экземпляр плейлиста из файла
|
// MakeFromFile создаёт экземпляр плейлиста из файла
|
||||||
func MakeFromFile(filepath string) (Playlist, error) {
|
func MakeFromFile(filepath string) (Playlist, error) {
|
||||||
@@ -98,25 +100,34 @@ func MakeFromUrl(url string) (Playlist, error) {
|
|||||||
// parseAttributes парсит атрибуты тегов #EXT*
|
// parseAttributes парсит атрибуты тегов #EXT*
|
||||||
func parseAttributes(line string) map[string]string {
|
func parseAttributes(line string) map[string]string {
|
||||||
result := make(map[string]string)
|
result := make(map[string]string)
|
||||||
regex := regexp.MustCompile(`(?U)([a-z-]+)="(.*)"`)
|
regexMatches := attrRegex.FindAllStringSubmatch(line, -1)
|
||||||
regexMatches := regex.FindAllStringSubmatch(line, -1)
|
|
||||||
for _, match := range regexMatches {
|
for _, match := range regexMatches {
|
||||||
result[match[1]] = match[2]
|
result[match[1]] = match[2]
|
||||||
}
|
}
|
||||||
return result
|
return result
|
||||||
}
|
}
|
||||||
|
|
||||||
// parseName парсит название канала из строки тега #EXTINF
|
// parseTitle парсит название канала из строки тега #EXTINF
|
||||||
func parseName(line string) string {
|
func parseTitle(line string) string {
|
||||||
//TODO https://git.axenov.dev/IPTV/iptvc/issues/7
|
// сначала пытаемся по-доброму: в строке есть тег, могут быть атрибуты,
|
||||||
parts := strings.Split(line, ",")
|
// есть запятая-разделитель, после неё -- название канала (с запятыми или без)
|
||||||
if len(parts) == 2 {
|
regexMatches := titleRegex.FindAllStringSubmatch(line, -1)
|
||||||
return strings.Trim(parts[1], " ")
|
if len(regexMatches) > 0 && len(regexMatches[0]) >= 2 {
|
||||||
|
return strings.TrimSpace(regexMatches[0][1])
|
||||||
}
|
}
|
||||||
|
|
||||||
regex := regexp.MustCompile(`['"]?\s*,\s*(.+)`)
|
// теперь пытаемся хоть как-то: в строке есть тег, могут быть атрибуты,
|
||||||
regexMatches := regex.FindAllStringSubmatch(line, -1)
|
// НЕТ запятой-разделителя и название канала (с запятыми или без)
|
||||||
return regexMatches[0][1]
|
lastQuotePos := strings.LastIndexAny(line, `,"'`)
|
||||||
|
if lastQuotePos != -1 && lastQuotePos < len(line)-1 {
|
||||||
|
afterLastQuote := line[lastQuotePos+1:]
|
||||||
|
name := strings.TrimSpace(afterLastQuote)
|
||||||
|
if name != "" {
|
||||||
|
return name
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return line // ну штош
|
||||||
}
|
}
|
||||||
|
|
||||||
// Download загружает плейлист по URL-адресу
|
// Download загружает плейлист по URL-адресу
|
||||||
@@ -148,6 +159,7 @@ func (pls *Playlist) ReadFromFs() error {
|
|||||||
// Parse разбирает плейлист
|
// Parse разбирает плейлист
|
||||||
func (pls *Playlist) Parse() Playlist {
|
func (pls *Playlist) Parse() Playlist {
|
||||||
isChannel := false
|
isChannel := false
|
||||||
|
tmpChannel := Channel{}
|
||||||
pls.Attributes = make(map[string]string)
|
pls.Attributes = make(map[string]string)
|
||||||
pls.Channels = make(map[string]Channel)
|
pls.Channels = make(map[string]Channel)
|
||||||
pls.Groups = make(map[string]Group)
|
pls.Groups = make(map[string]Group)
|
||||||
@@ -170,7 +182,7 @@ func (pls *Playlist) Parse() Playlist {
|
|||||||
if strings.HasPrefix(line, "#EXTINF") {
|
if strings.HasPrefix(line, "#EXTINF") {
|
||||||
isChannel = true
|
isChannel = true
|
||||||
tmpChannel.Attributes = parseAttributes(line)
|
tmpChannel.Attributes = parseAttributes(line)
|
||||||
tmpChannel.Title = parseName(line)
|
tmpChannel.Title = parseTitle(line)
|
||||||
|
|
||||||
if tmpChannel.Title == "" {
|
if tmpChannel.Title == "" {
|
||||||
if tvgid, ok := tmpChannel.Attributes["tvg-id"]; ok {
|
if tvgid, ok := tmpChannel.Attributes["tvg-id"]; ok {
|
||||||
|
|||||||
@@ -0,0 +1,63 @@
|
|||||||
|
package playlist
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestParseAttributes(t *testing.T) {
|
||||||
|
line := `#EXTINF:-1 tvg-id="test" group-title="News",Channel Name`
|
||||||
|
attrs := parseAttributes(line)
|
||||||
|
if attrs["tvg-id"] != "test" {
|
||||||
|
t.Errorf("tvg-id = %q, want %q", attrs["tvg-id"], "test")
|
||||||
|
}
|
||||||
|
if attrs["group-title"] != "News" {
|
||||||
|
t.Errorf("group-title = %q, want %q", attrs["group-title"], "News")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestParseTitle(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
input string
|
||||||
|
expected string
|
||||||
|
}{
|
||||||
|
{`#EXTINF:-1,Channel Name`, "Channel Name"},
|
||||||
|
{`#EXTINF:-1 tvg-id="x",Another Channel`, "Another Channel"},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.expected, func(t *testing.T) {
|
||||||
|
got := parseTitle(tt.input)
|
||||||
|
if got != tt.expected {
|
||||||
|
t.Errorf("parseTitle(%q) = %q, want %q", tt.input, got, tt.expected)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestPlaylistParse(t *testing.T) {
|
||||||
|
content := `#EXTM3U
|
||||||
|
#EXTINF:-1 tvg-id="1" group-title="News",First Channel
|
||||||
|
http://example.com/1
|
||||||
|
#EXTINF:-1 tvg-id="2",Second Channel
|
||||||
|
http://example.com/2
|
||||||
|
`
|
||||||
|
pls := Playlist{Content: content}
|
||||||
|
result := pls.Parse()
|
||||||
|
|
||||||
|
if len(result.Channels) != 2 {
|
||||||
|
t.Errorf("channels count = %d, want 2", len(result.Channels))
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(result.Groups) != 1 {
|
||||||
|
t.Errorf("groups count = %d, want 1", len(result.Groups))
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, ch := range result.Channels {
|
||||||
|
if ch.URL == "" {
|
||||||
|
t.Error("channel URL is empty")
|
||||||
|
}
|
||||||
|
if ch.Title == "" {
|
||||||
|
t.Error("channel Title is empty")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -75,8 +75,12 @@ func (block *TagBlock) GetTags(ch playlist.Channel) []string {
|
|||||||
|
|
||||||
// Init инициализирует объекты тегов из тегфайла
|
// Init инициализирует объекты тегов из тегфайла
|
||||||
func Init(path string) []TagBlock {
|
func Init(path string) []TagBlock {
|
||||||
pathNormalized, _ := utils.ExpandPath(path)
|
pathNormalized, err := utils.ExpandPath(path)
|
||||||
_, err := os.Stat(pathNormalized)
|
if err != nil {
|
||||||
|
log.Println("Warning: all channels will be untagged due to error:", err)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
_, err = os.Stat(pathNormalized)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Println("Warning: all channels will be untagged due to error:", err)
|
log.Println("Warning: all channels will be untagged due to error:", err)
|
||||||
return nil
|
return nil
|
||||||
|
|||||||
+7
-2
@@ -52,8 +52,13 @@ func Fetch(url string) ([]byte, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
req.Header.Set("User-Agent", "Mozilla/5.0 WINK/1.31.1 (AndroidTV/9) HlsWinkPlayer")
|
req.Header.Set("User-Agent", "Mozilla/5.0 WINK/1.31.1 (AndroidTV/9) HlsWinkPlayer")
|
||||||
http.DefaultTransport.(*http.Transport).TLSClientConfig = &tls.Config{InsecureSkipVerify: true}
|
transport := &http.Transport{
|
||||||
httpClient := http.Client{Timeout: 10 * time.Second}
|
TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
|
||||||
|
}
|
||||||
|
httpClient := http.Client{
|
||||||
|
Timeout: 10 * time.Second,
|
||||||
|
Transport: transport,
|
||||||
|
}
|
||||||
resp, err := httpClient.Do(req)
|
resp, err := httpClient.Do(req)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
|
|||||||
@@ -0,0 +1,51 @@
|
|||||||
|
package utils
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestMd5str(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
input string
|
||||||
|
expected string
|
||||||
|
}{
|
||||||
|
{"hello", "5d41402abc4b2a76b9719d911017c592"},
|
||||||
|
{"", "d41d8cd98f00b204e9800998ecf8427e"},
|
||||||
|
{"http://example.com/stream", "a1b2c3d4e5f6"}, // just length check
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.input, func(t *testing.T) {
|
||||||
|
got := Md5str(tt.input)
|
||||||
|
if len(got) != 32 {
|
||||||
|
t.Errorf("Md5str(%q) length = %d, want 32", tt.input, len(got))
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestArrayUnique(t *testing.T) {
|
||||||
|
input := []string{"a", "b", "a", "c", "b"}
|
||||||
|
expected := []string{"a", "b", "c"}
|
||||||
|
got := ArrayUnique(input)
|
||||||
|
if len(got) != len(expected) {
|
||||||
|
t.Errorf("ArrayUnique length = %d, want %d", len(got), len(expected))
|
||||||
|
}
|
||||||
|
seen := make(map[string]bool)
|
||||||
|
for _, v := range got {
|
||||||
|
if seen[v] {
|
||||||
|
t.Errorf("ArrayUnique returned duplicate: %s", v)
|
||||||
|
}
|
||||||
|
seen[v] = true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestExpandPath(t *testing.T) {
|
||||||
|
got, err := ExpandPath("/tmp/test.txt")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("ExpandPath error: %v", err)
|
||||||
|
}
|
||||||
|
if got == "" {
|
||||||
|
t.Error("ExpandPath returned empty string")
|
||||||
|
}
|
||||||
|
}
|
||||||
Executable
+17
@@ -0,0 +1,17 @@
|
|||||||
|
#!/usr/bin/env bash
|
||||||
|
|
||||||
|
[[ "$1" ]] && DOCKER_TAG="$1" || DOCKER_TAG="latest"
|
||||||
|
[[ "$1" ]] && GIT_TAG="$1" || GIT_TAG=$(git describe --tags --always)
|
||||||
|
GIT_HASH=$(git rev-parse --short HEAD)
|
||||||
|
IPTVC_VERSION="${GIT_TAG}-${GIT_HASH}"
|
||||||
|
|
||||||
|
git checkout "${GIT_TAG}" 2>/dev/null
|
||||||
|
docker build \
|
||||||
|
--build-arg IPTVC_VERSION="${IPTVC_VERSION}" \
|
||||||
|
--build-arg GOOS="${GOOS:-linux}" \
|
||||||
|
--build-arg GOARCH="${GOARCH:-amd64}" \
|
||||||
|
--tag iptvc:"${DOCKER_TAG}" \
|
||||||
|
--tag git.axenov.dev/iptv/iptvc:"${DOCKER_TAG}" \
|
||||||
|
.
|
||||||
|
|
||||||
|
docker push git.axenov.dev/iptv/iptvc:"${DOCKER_TAG}"
|
||||||
+42
-3
@@ -27,10 +27,30 @@ var checkCmd = &cobra.Command{
|
|||||||
files, _ := cmd.Flags().GetStringSlice("file")
|
files, _ := cmd.Flags().GetStringSlice("file")
|
||||||
urls, _ := cmd.Flags().GetStringSlice("url")
|
urls, _ := cmd.Flags().GetStringSlice("url")
|
||||||
codes, _ := cmd.Flags().GetStringSlice("code")
|
codes, _ := cmd.Flags().GetStringSlice("code")
|
||||||
lists := checker.PrepareListsToCheck(files, urls, codes)
|
|
||||||
|
|
||||||
|
waitSeconds := app.Args.RepeatEverySec
|
||||||
|
if waitSeconds <= 0 {
|
||||||
|
waitSeconds = 5
|
||||||
|
}
|
||||||
|
|
||||||
|
currentIteration := 1
|
||||||
|
for {
|
||||||
|
if app.Args.RepeatCount != 1 {
|
||||||
|
log.Printf(
|
||||||
|
"@ New iteration current=%d count=%d\n",
|
||||||
|
currentIteration,
|
||||||
|
app.Args.RepeatCount,
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
ch := checker.NewChecker(app.Args.TagsPath)
|
||||||
|
|
||||||
|
lists := ch.PrepareListsToCheck(files, urls, codes)
|
||||||
|
|
||||||
|
if len(lists) > 0 {
|
||||||
startTime := time.Now()
|
startTime := time.Now()
|
||||||
onlineCount, offlineCount := checker.CheckPlaylists(lists)
|
ctx := cmd.Context()
|
||||||
|
onlineCount, offlineCount := ch.CheckPlaylists(ctx, lists)
|
||||||
|
|
||||||
log.Printf(
|
log.Printf(
|
||||||
"Done! count=%d online=%d offline=%d elapsedTime=%.2fs\n",
|
"Done! count=%d online=%d offline=%d elapsedTime=%.2fs\n",
|
||||||
@@ -41,9 +61,26 @@ var checkCmd = &cobra.Command{
|
|||||||
)
|
)
|
||||||
|
|
||||||
if app.Args.NeedJson {
|
if app.Args.NeedJson {
|
||||||
marshal, _ := json.Marshal(lists)
|
marshal, err := json.Marshal(lists)
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("Error marshaling results: %s", err)
|
||||||
|
} else {
|
||||||
fmt.Println(string(marshal))
|
fmt.Println(string(marshal))
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
log.Println("There are no playlists to check")
|
||||||
|
}
|
||||||
|
|
||||||
|
if app.Args.RepeatCount != 0 {
|
||||||
|
if uint(currentIteration) == app.Args.RepeatCount {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
currentIteration++
|
||||||
|
}
|
||||||
|
log.Printf("Waiting for new iteration... seconds=%d\n", app.Args.RepeatEverySec)
|
||||||
|
time.Sleep(time.Duration(app.Args.RepeatEverySec) * time.Second)
|
||||||
|
}
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -51,6 +88,8 @@ func init() {
|
|||||||
checkCmd.Flags().StringVarP(&app.Args.TagsPath, "tags", "t", "./channels.json", "path to a local tagfile")
|
checkCmd.Flags().StringVarP(&app.Args.TagsPath, "tags", "t", "./channels.json", "path to a local tagfile")
|
||||||
checkCmd.Flags().StringVarP(&app.Args.IniPath, "ini", "i", "./playlists.ini", "path to a local ini-file")
|
checkCmd.Flags().StringVarP(&app.Args.IniPath, "ini", "i", "./playlists.ini", "path to a local ini-file")
|
||||||
checkCmd.Flags().UintVarP(&app.Args.RandomCount, "random", "r", 0, "take this count of random playlists to check from ini-file")
|
checkCmd.Flags().UintVarP(&app.Args.RandomCount, "random", "r", 0, "take this count of random playlists to check from ini-file")
|
||||||
|
checkCmd.Flags().UintVarP(&app.Args.RepeatCount, "repeat", "", 1, "repeat same check X times")
|
||||||
|
checkCmd.Flags().UintVarP(&app.Args.RepeatEverySec, "every", "", 5, "wait N seconds after every check")
|
||||||
checkCmd.Flags().BoolVarP(&app.Args.NeedJson, "json", "j", false, "print results in JSON format in the end")
|
checkCmd.Flags().BoolVarP(&app.Args.NeedJson, "json", "j", false, "print results in JSON format in the end")
|
||||||
checkCmd.Flags().BoolVarP(&app.Args.NeedQuiet, "quiet", "q", false, "suppress logs (does not affect on -j)")
|
checkCmd.Flags().BoolVarP(&app.Args.NeedQuiet, "quiet", "q", false, "suppress logs (does not affect on -j)")
|
||||||
checkCmd.Flags().StringSliceP("file", "f", []string{}, "path to a local playlist file (m3u/m3u8)")
|
checkCmd.Flags().StringSliceP("file", "f", []string{}, "path to a local playlist file (m3u/m3u8)")
|
||||||
|
|||||||
@@ -8,6 +8,7 @@ package cmd
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"axenov/iptv-checker/app"
|
"axenov/iptv-checker/app"
|
||||||
|
"context"
|
||||||
"os"
|
"os"
|
||||||
|
|
||||||
"github.com/spf13/cobra"
|
"github.com/spf13/cobra"
|
||||||
@@ -33,6 +34,11 @@ func Execute() {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// ExecuteContext runs the root command with the given context.
|
||||||
|
func ExecuteContext(ctx context.Context) error {
|
||||||
|
return rootCmd.ExecuteContext(ctx)
|
||||||
|
}
|
||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
rootCmd.PersistentFlags().BoolVarP(&app.Args.Verbose, "verbose", "v", false, "enable additional output")
|
rootCmd.PersistentFlags().BoolVarP(&app.Args.Verbose, "verbose", "v", false, "enable additional output")
|
||||||
}
|
}
|
||||||
|
|||||||
+1
-1
@@ -18,7 +18,7 @@ var versionCmd = &cobra.Command{
|
|||||||
Use: "version",
|
Use: "version",
|
||||||
Short: "Show version",
|
Short: "Show version",
|
||||||
Run: func(cmd *cobra.Command, args []string) {
|
Run: func(cmd *cobra.Command, args []string) {
|
||||||
fmt.Println("iptvc v" + app.VERSION)
|
fmt.Println("iptvc v" + app.Version())
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -7,9 +7,33 @@
|
|||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"axenov/iptv-checker/app"
|
||||||
"axenov/iptv-checker/cmd"
|
"axenov/iptv-checker/cmd"
|
||||||
|
"context"
|
||||||
|
"os"
|
||||||
|
"os/signal"
|
||||||
|
"syscall"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
var version string
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
cmd.Execute()
|
app.SetVersion(version)
|
||||||
|
|
||||||
|
ctx, cancel := context.WithCancel(context.Background())
|
||||||
|
defer cancel()
|
||||||
|
|
||||||
|
sigCh := make(chan os.Signal, 1)
|
||||||
|
signal.Notify(sigCh, os.Interrupt, syscall.SIGTERM)
|
||||||
|
go func() {
|
||||||
|
<-sigCh
|
||||||
|
cancel()
|
||||||
|
app.Shutdown()
|
||||||
|
os.Exit(130)
|
||||||
|
}()
|
||||||
|
|
||||||
|
if err := cmd.ExecuteContext(ctx); err != nil {
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
app.Shutdown()
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user