iptv-org-mirror/database
1
0
Fork 0
mirror of https://github.com/iptv-org/database.git synced 2025-05-09 19:20:01 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update scripts

Browse source
This commit is contained in:
freearhey 2023-10-08 02:13:58 +03:00
parent 0feb6b1e89
commit f0cbc45e14
4 changed files with 29 additions and 34 deletions
Download patch file Download diff file Expand all files Collapse all files

25
scripts/db/update.ts
View file

@ -1,4 +1,4 @@
import { CSV, IssueLoader, CSVParser } from '../core'
import { CSV, IssueLoader, CSVParser, IDCreator } from '../core'
import { Channel, Blocked, Issue } from '../models'
import { DATA_DIR } from '../constants'
import { Storage, Collection } from '@freearhey/core'
@ -8,6 +8,7 @@ let channels = new Collection()
const processedIssues = new Collection()
async function main() {
const idCreator = new IDCreator()
const dataStorage = new Storage(DATA_DIR)
const parser = new CSVParser()
@ -20,8 +21,8 @@ async function main() {
const loader = new IssueLoader()
await removeChannels({ loader })
await editChannels({ loader })
await addChannels({ loader })
await editChannels({ loader, idCreator })
await addChannels({ loader, idCreator })
await blockChannels({ loader })
await unblockChannels({ loader })
@ -65,7 +66,7 @@ async function removeChannels({ loader }: { loader: IssueLoader }) {
})
}
async function editChannels({ loader }: { loader: IssueLoader }) {
async function editChannels({ loader, idCreator }: { loader: IssueLoader; idCreator: IDCreator }) {
const issues = await loader.load({ labels: ['channels:edit,approved'] })
issues.forEach((issue: Issue) => {
const data = issue.data
@ -80,7 +81,7 @@ async function editChannels({ loader }: { loader: IssueLoader }) {
if (data.has('name') || data.has('country')) {
const name = data.get('name') || found.name
const country = data.get('country') || found.country
channelId = generateChannelId(name, country)
channelId = idCreator.create(name, country)
}
found.update({
@ -107,13 +108,13 @@ async function editChannels({ loader }: { loader: IssueLoader }) {
})
}
async function addChannels({ loader }: { loader: IssueLoader }) {
async function addChannels({ loader, idCreator }: { loader: IssueLoader; idCreator: IDCreator }) {
const issues = await loader.load({ labels: ['channels:add,approved'] })
issues.forEach((issue: Issue) => {
const data = issue.data
if (data.missing('name') || data.missing('country')) return
const channelId = generateChannelId(data.get('name'), data.get('country'))
const channelId = idCreator.create(data.get('name'), data.get('country'))
const found: Channel = channels.first((channel: Channel) => channel.id === channelId)
if (found) return
@ -182,13 +183,3 @@ async function blockChannels({ loader }: { loader: IssueLoader }) {
processedIssues.push(issue)
})
}
function generateChannelId(name: string, country: string): string {
const slug = name
.replace(/\+/gi, 'Plus')
.replace(/^@/gi, 'At')
.replace(/[^a-z\d]+/gi, '')
country = country.toLowerCase()
return `${slug}.${country}`
}