wip

scripts/commands/generate-playlists.js

@@ -10,8 +10,9 @@ async function main() {
   await generator.generate('languages', streams)
   await generator.generate('regions', streams)
   await generator.generate('index_m3u', streams)
+  await generator.generate('index_nsfw_m3u', streams)
+  await generator.generate('index_category_m3u', streams)
 
-  // await generateIndexNSFW()
   // await generateIndexCategory()
   // await generateIndexCountry()
   // await generateIndexLanguage()
@@ -62,31 +63,6 @@ async function loadStreams() {
   })
 }
 
-// async function generateIndexNSFW() {
-//   logger.info(`Generating index.nsfw.m3u...`)
-
-//   await generator.generate(
-//     `${PUBLIC_PATH}/index.nsfw.m3u`,
-//     {},
-//     {
-//       includeNSFW: true,
-//       onLoad: function (items) {
-//         return items.map(item => {
-//           if (!item.categories || !item.categories.length) {
-//             item.group_title = 'Other'
-//           }
-
-//           return item
-//         })
-//       },
-//       sortBy: item => {
-//         if (item.group_title === 'Other') return '_'
-//         return item.group_title || ''
-//       }
-//     }
-//   )
-// }
-
 // async function generateIndexCategory() {
 //   logger.info(`Generating index.category.m3u...`)
 

scripts/generators/index.js

@@ -3,3 +3,5 @@ exports.countries = require('./countries')
 exports.languages = require('./languages')
 exports.regions = require('./regions')
 exports.index_m3u = require('./index_m3u')
+exports.index_nsfw_m3u = require('./index_nsfw_m3u')
+exports.index_category_m3u = require('./index_category_m3u')

scripts/generators/index_category_m3u.js

@@ -0,0 +1,22 @@
+const api = require('../core/api')
+const _ = require('lodash')
+
+module.exports = async function (streams = []) {
+	streams = _.filter(streams, s => !s.channel || s.channel.is_nsfw === false)
+	let items = []
+	streams.forEach(stream => {
+		if (!stream.categories.length) return items.push(stream)
+
+		stream.categories.forEach(category => {
+			const item = _.cloneDeep(stream)
+			item.group_title = category.name
+			items.push(item)
+		})
+	})
+	items = _.sortBy(items, i => {
+		if (i.group_title === 'Undefined') return '_'
+		return i.group_title
+	})
+
+	return { filepath: 'index.category.m3u', items }
+}

scripts/generators/index_nsfw_m3u.js

@@ -0,0 +1,6 @@
+const api = require('../core/api')
+const _ = require('lodash')
+
+module.exports = async function (streams = []) {
+	return { filepath: 'index.nsfw.m3u', items: streams }
+}