mirror of https://github.com/calzoneman/sync.git
132 lines
4.1 KiB
JavaScript
132 lines
4.1 KiB
JavaScript
var ChannelModule = require("./module");
|
|
var Flags = require("../flags");
|
|
var util = require("../utilities");
|
|
var InfoGetter = require("../get-info");
|
|
var db = require("../database");
|
|
import { Counter, Summary } from 'prom-client';
|
|
const LOGGER = require('@calzoneman/jsli')('channel/library');
|
|
|
|
const TYPE_UNCACHE = {
|
|
id: "string"
|
|
};
|
|
|
|
const TYPE_SEARCH_MEDIA = {
|
|
source: "string,optional",
|
|
query: "string"
|
|
};
|
|
|
|
function LibraryModule(_channel) {
|
|
ChannelModule.apply(this, arguments);
|
|
}
|
|
|
|
LibraryModule.prototype = Object.create(ChannelModule.prototype);
|
|
|
|
LibraryModule.prototype.onUserPostJoin = function (user) {
|
|
user.socket.typecheckedOn("uncache", TYPE_UNCACHE, this.handleUncache.bind(this, user));
|
|
user.socket.typecheckedOn("searchMedia", TYPE_SEARCH_MEDIA, this.handleSearchMedia.bind(this, user));
|
|
};
|
|
|
|
LibraryModule.prototype.cacheMedia = function (media) {
|
|
if (this.channel.is(Flags.C_REGISTERED) && !util.isLive(media.type)) {
|
|
db.channels.addToLibrary(this.channel.name, media);
|
|
}
|
|
};
|
|
|
|
LibraryModule.prototype.cacheMediaList = function (list) {
|
|
if (this.channel.is(Flags.C_REGISTERED)) {
|
|
LOGGER.info(
|
|
'Saving %d items to library for %s',
|
|
list.length,
|
|
this.channel.name
|
|
);
|
|
db.channels.addListToLibrary(this.channel.name, list).catch(error => {
|
|
LOGGER.error('Failed to add list to library: %s', error.stack);
|
|
});
|
|
}
|
|
};
|
|
|
|
LibraryModule.prototype.handleUncache = function (user, data) {
|
|
if (!this.channel.is(Flags.C_REGISTERED)) {
|
|
return;
|
|
}
|
|
|
|
if (!this.channel.modules.permissions.canUncache(user)) {
|
|
return;
|
|
}
|
|
|
|
const chan = this.channel;
|
|
chan.refCounter.ref("LibraryModule::handleUncache");
|
|
db.channels.deleteFromLibrary(chan.name, data.id, function (err, _res) {
|
|
if (chan.dead) {
|
|
return;
|
|
} else if (err) {
|
|
chan.refCounter.unref("LibraryModule::handleUncache");
|
|
return;
|
|
}
|
|
|
|
chan.logger.log("[library] " + user.getName() + " deleted " + data.id +
|
|
"from the library");
|
|
chan.refCounter.unref("LibraryModule::handleUncache");
|
|
});
|
|
};
|
|
|
|
const librarySearchQueryCount = new Counter({
|
|
name: 'cytube_library_search_query_count',
|
|
help: 'Counter for number of channel library searches',
|
|
labelNames: ['source']
|
|
});
|
|
const librarySearchResultSize = new Summary({
|
|
name: 'cytube_library_search_results_size',
|
|
help: 'Summary for number of channel library results returned',
|
|
labelNames: ['source']
|
|
});
|
|
LibraryModule.prototype.handleSearchMedia = function (user, data) {
|
|
var query = data.query.substring(0, 100);
|
|
var searchYT = function () {
|
|
librarySearchQueryCount.labels('yt').inc(1, new Date());
|
|
InfoGetter.Getters.ytSearch(query, function (e, vids) {
|
|
if (!e) {
|
|
librarySearchResultSize.labels('yt')
|
|
.observe(vids.length, new Date());
|
|
user.socket.emit("searchResults", {
|
|
source: "yt",
|
|
results: vids
|
|
});
|
|
}
|
|
});
|
|
};
|
|
|
|
if (data.source === "yt" || !this.channel.is(Flags.C_REGISTERED) ||
|
|
!this.channel.modules.permissions.canSeePlaylist(user)) {
|
|
searchYT();
|
|
} else {
|
|
librarySearchQueryCount.labels('library').inc(1, new Date());
|
|
|
|
db.channels.searchLibrary(this.channel.name, query, function (err, res) {
|
|
if (err) {
|
|
res = [];
|
|
}
|
|
|
|
librarySearchResultSize.labels('library')
|
|
.observe(res.length, new Date());
|
|
|
|
res.sort(function (a, b) {
|
|
var x = a.title.toLowerCase();
|
|
var y = b.title.toLowerCase();
|
|
return (x === y) ? 0 : (x < y ? -1 : 1);
|
|
});
|
|
|
|
res.forEach(function (r) {
|
|
r.duration = util.formatTime(r.seconds);
|
|
});
|
|
|
|
user.socket.emit("searchResults", {
|
|
source: "library",
|
|
results: res
|
|
});
|
|
});
|
|
}
|
|
};
|
|
|
|
module.exports = LibraryModule;
|