iptv-org-mirror/epg
1
0
Fork 0
mirror of https://github.com/iptv-org/epg.git synced 2025-05-09 08:30:06 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Install fuse.js package

Browse source
This commit is contained in:
freearhey 2025-01-13 02:29:58 +03:00
parent 2e64b51b79
commit c4189460cb
4 changed files with 41 additions and 52 deletions
Download patch file Download diff file Expand all files Collapse all files

19
scripts/commands/channels/edit.ts
View file

@ -5,7 +5,7 @@ import { Channel } from 'epg-grabber'
import nodeCleanup from 'node-cleanup'
import { program } from 'commander'
import inquirer, { QuestionCollection } from 'inquirer'
import sj from '@freearhey/search-js'
import Fuse from 'fuse.js'
program.argument('<filepath>', 'Path to *.channels.xml file to edit').parse(process.argv)
@ -25,15 +25,15 @@ async function main() {
const dataStorage = new Storage(DATA_DIR)
const channelsContent = await dataStorage.json('channels.json')
const channelsIndex = sj.createIndex(channelsContent)
const searchIndex = new Fuse(channelsContent, { keys: ['name', 'alt_names'], threshold: 0.4 })
for (const channel of channels.all()) {
if (channel.xmltv_id) continue
const question: QuestionCollection = {
name: 'option',
message: `"${channel.name}" (${channel.site_id}):`,
message: `Select xmltv_id for "${channel.name}" (${channel.site_id}):`,
type: 'list',
choices: getOptions(channelsIndex, channel),
choices: getOptions(searchIndex, channel),
pageSize: 10
}
@ -94,8 +94,7 @@ async function getInput(channel: Channel) {
}
function getOptions(index, channel: Channel) {
const query = prepareQuery(channel.name)
const similar = index.search(query).map(item => new ApiChannel(item))
const similar = index.search(channel.name).map(result => new ApiChannel(result.item))
const variants = new Collection()
similar.forEach((_channel: ApiChannel) => {
@ -110,11 +109,3 @@ function getOptions(index, channel: Channel) {
return variants.all()
}
function prepareQuery(channelName: string) {
return channelName
.replace(/\s(SD|TV|HD|SD\/HD|HDTV)$/i, '')
.replace(/(\(|\)|,)/gi, '')
.replace(/-/gi, ' ')
.replace(/\+/gi, '')
}