Compare commits

...

3 Commits

11 changed files with 227 additions and 92 deletions

8
app.js
View File

@ -4,15 +4,17 @@ const helmet = require("helmet");
const logger = require("./services/logging"); const logger = require("./services/logging");
const rateLimit = require("express-rate-limit"); const rateLimit = require("express-rate-limit");
const sessions = require("./services/session-manager"); const sessions = require("./services/session-manager");
const PlexTracker = require("./services/trackers/PlexTracker"); const PlexTracker = require("./services/trackers/plex-tracker");
const SpotifyTracker = require("./services/trackers/SpotifyTracker"); const SpotifyTracker = require("./services/trackers/spotify-tracker");
const Recorder = require("./services/recorder"); const Recorder = require("./services/recorder");
const MalojaScrobbler = require("./services/scrobblers/maloja-scrobbler");
const maloja = new MalojaScrobbler(config.maloja);
const spotify = new SpotifyTracker(config.spotify); const spotify = new SpotifyTracker(config.spotify);
(async () => await spotify.loadCredentials())(); (async () => await spotify.loadCredentials())();
const plex = new PlexTracker(config.plex); const plex = new PlexTracker(config.plex);
const recorder = new Recorder(sessions, [plex, spotify], config.scrobble, logger); const recorder = new Recorder(sessions, [plex, spotify], [maloja], config.scrobble, logger);
setInterval(() => recorder.record(), 5000); setInterval(() => recorder.record(), 5000);

View File

@ -2,10 +2,28 @@ const schema = {
type: 'object', type: 'object',
required: [], required: [],
properties: { properties: {
maloja: {
type: 'object',
required: ['name', 'url', 'token'],
properties: {
name: {
type: 'string'
},
url: {
type: 'string'
},
token: {
type: 'string'
},
}
},
plex: { plex: {
type: 'object', type: 'object',
required: ['url', 'token'], required: ['name', 'url', 'token', 'scrobblers'],
properties: { properties: {
name: {
type: 'string'
},
url: { url: {
type: 'string' type: 'string'
}, },
@ -55,6 +73,13 @@ const schema = {
}, },
} }
} }
},
scrobblers: {
type: 'array',
items: {
type: 'string'
},
minItems: 1
} }
} }
}, },
@ -78,8 +103,11 @@ const schema = {
spotify: { spotify: {
type: 'object', type: 'object',
required: ['client_id', 'client_secret', 'redirect_uri'], required: ['name', 'client_id', 'client_secret', 'redirect_uri', 'scrobblers'],
properties: { properties: {
name: {
type: 'string'
},
client_id: { client_id: {
type: 'string' type: 'string'
}, },
@ -88,6 +116,13 @@ const schema = {
}, },
redirect_uri: { redirect_uri: {
type: 'string' type: 'string'
},
scrobblers: {
type: 'array',
items: {
type: 'string'
},
minItems: 1
} }
} }
}, },

View File

@ -5,9 +5,11 @@ const yaml = require("js-yaml");
const configurationBase = { const configurationBase = {
plex: { plex: {
name: null,
url: null, url: null,
token: null, token: null,
filters: [] // { library, ip, deviceId, platform, product } filters: [], // { library, ip, deviceId, platform, product }
scrobblers: []
}, },
scrobble: { scrobble: {
minimum: { minimum: {
@ -16,9 +18,11 @@ const configurationBase = {
}, },
}, },
spotify: { spotify: {
name: null,
client_id: null, client_id: null,
client_secret: null, client_secret: null,
redirect_uri: null redirect_uri: null,
scrobblers: []
}, },
web: { web: {
host: null, host: null,
@ -27,7 +31,6 @@ const configurationBase = {
}; };
const configurationFile = yaml.load(fs.readFileSync('config/config.yml'), yaml.JSON_SCHEMA); const configurationFile = yaml.load(fs.readFileSync('config/config.yml'), yaml.JSON_SCHEMA);
const configuration = { ...configurationBase, ...configurationFile }
const ajv = new Ajv({ allErrors: true }); const ajv = new Ajv({ allErrors: true });
const schema = require("./config.schema"); const schema = require("./config.schema");
@ -40,4 +43,5 @@ if (!valid) {
(async () => { await new Promise(resolve => setTimeout(resolve, 1000)); exit(1); })(); (async () => { await new Promise(resolve => setTimeout(resolve, 1000)); exit(1); })();
} }
const configuration = { ...configurationBase, ...configurationFile }
module.exports = configuration; module.exports = configuration;

View File

@ -2,8 +2,10 @@ class Session {
#id = null; #id = null;
#started = null; #started = null;
#current = null; #current = null;
#lastScrobble = null; lastScrobbleTimestamp = 0;
#pauseDuration = 0; lastUpdateTimestamp = 0;
pauseDuration = 0;
playDuration = 0;
constructor(id) { constructor(id) {
this.#id = id; this.#id = id;
@ -25,22 +27,6 @@ class Session {
get started() { get started() {
return this.#started; return this.#started;
} }
get lastScrobbleTimestamp() {
return this.#lastScrobble;
}
set lastScrobbleTimestamp(value) {
this.#lastScrobble = value;
}
get pauseDuration() {
return this.#pauseDuration;
}
set pauseDuration(value) {
this.#pauseDuration = value;
}
} }
module.exports = Session; module.exports = Session;

View File

@ -9,8 +9,9 @@ class Song {
session = null; session = null;
state = null; state = null;
source = null; source = null;
provider = null;
constructor(id, name, album, artists, year, duration, progress, session, state, source) { constructor(id, name, album, artists, year, duration, progress, session, state, source, provider) {
this.id = id; this.id = id;
this.name = name; this.name = name;
this.album = album; this.album = album;
@ -21,6 +22,7 @@ class Song {
this.session = session; this.session = session;
this.state = state; this.state = state;
this.source = source; this.source = source;
this.provider = provider;
} }
} }

View File

@ -1,98 +1,130 @@
const AggregateTracker = require("./trackers/AggregateTracker"); const AggregateTracker = require("./trackers/aggregate-tracker");
const Session = require("../models/session"); const Session = require("../models/session");
class Recorder { class Recorder {
#sessions = null; #sessions = null;
#trackers = null; #trackers = null;
#originalTrackers = [];
#scrobblers = [];
#config = null; #config = null;
#logger = null; #logger = null;
#lastTick = null;
constructor(sessions, trackers, config, logger) { constructor(sessions, trackers, scrobblers, config, logger) {
this.#sessions = sessions; this.#sessions = sessions;
this.#trackers = new AggregateTracker(trackers); this.#trackers = new AggregateTracker('aggregate', trackers);
this.#originalTrackers = trackers;
this.#scrobblers = scrobblers;
this.#config = config; this.#config = config;
this.#logger = logger; this.#logger = logger;
this.#lastTick = Date.now();
} }
async record() { async record() {
const now = Date.now(); const now = Date.now();
const timeDiff = now - this.#lastTick;
const media = await this.#trackers.poll(); const media = await this.#trackers.poll();
const data = media.map(m => this.#fetchSession(m)); const contexts = media.map(m => this.#fetchContext(m));
// Find sessions that ended and that are deemable of a scrobble. // Find sessions that ended and that are deemable of a scrobble.
const sessionIds = this.#sessions.getSessionIds(); const sessionIds = this.#sessions.getSessionIds();
const stopped = sessionIds.filter(sessionId => !data.some(d => sessionId == d.session.id)).map(s => this.#sessions.get(s)); const stopped = sessionIds.filter(sessionId => !contexts.some(context => sessionId == context.session.id))
const sessionEnded = stopped.filter(s => this.#canScrobble(s, null, s.playing, now)); .map(sessionId => this.#sessions.get(sessionId))
.map(session => this.#fetchContext(session.playing));
const contextEnded = stopped.filter(context => this.#canScrobble(context.session, null, context.session.playing));
for (let context of contextEnded)
context.session.playDuration = now - (context.session.lastScrobbleTimestamp || context.session.started) - context.session.pauseDuration;
// Find ongoing sessions that have moved on to the next song. // Find ongoing sessions that have moved on to the next song.
const finishedPlaying = data.filter(d => this.#listen(d.session, d.media, d.session.playing, now, timeDiff)); const finishedPlaying = contexts.filter(context => this.#listen(context, now));
const scrobbling = finishedPlaying.concat(sessionEnded); // Scrobble
for (let track of scrobbling) const scrobbling = finishedPlaying.concat(contextEnded);
this.#scrobble(track); for (let context of scrobbling) {
await this.#scrobble(context);
if (context.session.playing == null)
continue;
context.session.playDuration = context.extraDuration;
context.session.pauseDuration = 0;
}
// Remove dead sessions. // Remove dead sessions.
for (let sessionId of stopped) for (let context of stopped) {
this.#sessions.remove(sessionId); this.#sessions.remove(context.session.id);
}
this.#lastTick = now;
} }
#fetchSession(media) { #fetchContext(media) {
const tracker = this.#originalTrackers.find(t => t.provider == media.provider && t.name == media.source);
let session = this.#sessions.get(media.session); let session = this.#sessions.get(media.session);
if (session == null) { if (session == null) {
session = new Session(media.session); session = new Session(media.session);
this.#sessions.add(session); this.#sessions.add(session);
} }
return { session: session, media: media } return { session, media, tracker, extraDuration: 0 }
} }
#listen(session, current, previous, timestamp, timeDiff) { #listen(context, timestamp) {
const session = context.session;
const current = context.media;
const previous = context.session.playing;
session.playing = current; session.playing = current;
if (previous == null) { if (!previous) {
this.#logger.info(current, "A new session has started."); this.#logger.info(current, "A new session has started.");
session.lastUpdateTimestamp = timestamp;
return false; return false;
} }
if (session.playing.state == "paused" || previous.state == "paused") { const updated = current.progress != previous.progress || current.id != previous.id || current.state != previous.state;
session.pauseDuration += timeDiff - (session.playing.progress - previous.progress); if (!updated)
return false;
const timeDiff = timestamp - session.lastUpdateTimestamp;
const progressDiff = Math.max(0, Math.min(current.progress - previous.progress, timeDiff));
session.playDuration += progressDiff;
session.pauseDuration += timeDiff - progressDiff;
const canScrobble = this.#canScrobble(session, current, previous);
if (canScrobble || current.id != previous.id) {
context.extraDuration = Math.min(current.progress, timeDiff - (previous.duration - previous.progress));
session.lastScrobbleTimestamp = timestamp;
} }
if (this.#canScrobble(session, current, previous, timestamp)) { session.lastUpdateTimestamp = timestamp;
session.pauseDuration = 0; return canScrobble;
session.lastScrobbleTimestamp = timestamp - (timeDiff - (previous.duration - previous.progress));
return true;
} else if (current.progress < previous.progress && session.playing.id != previous.id) {
session.pauseDuration = 0;
if (current.progress < timeDiff)
session.lastScrobbleTimestamp = timestamp - session.playing.progress;
else
session.lastScrobbleTimestamp = timestamp;
}
return false;
} }
#canScrobble(session, current, previous, timestamp) { #canScrobble(session, current, previous) {
if (previous == null) if (previous == null)
return false; return false;
const scrobbleDuration = this.#config.minimum.duration || 240; const scrobbleDuration = this.#config.minimum.duration || 240;
const scrobblePercent = this.#config.minimum.percent || 50; const scrobblePercent = this.#config.minimum.percent || 50;
const durationPlayed = timestamp - (session.lastScrobbleTimestamp || session.started) - session.pauseDuration;
const newPlayback = current == null || current.progress < previous.progress; const newPlayback = current == null || current.progress < previous.progress;
const canBeScrobbled = durationPlayed > scrobbleDuration * 1000 || durationPlayed / previous.duration > scrobblePercent / 100.0; const canBeScrobbled = session.playDuration > scrobbleDuration * 1000 || session.playDuration / previous.duration > scrobblePercent / 100.0;
return newPlayback && canBeScrobbled; return newPlayback && canBeScrobbled;
} }
#scrobble(media) { async #scrobble(context) {
this.#logger.info(media, "Scrobble"); this.#logger.info(context, "Scrobble");
for (var scrobblerName of context.tracker.scrobblerNames) {
const scrobbler = this.#scrobblers.find(s => s.name == scrobblerName);
if (scrobbler == null) {
this.#logger.error(`Cannot find scrobbler by name of '${scrobblerName}'.`);
continue;
}
try {
await scrobbler.scrobble(context.media, Date.now() - Math.min(context.media.duration, context.session.playDuration));
} catch (ex) {
this.#logger.error(ex, "Could not send to maloja.");
}
}
} }
} }

View File

@ -0,0 +1,43 @@
const axios = require("axios");
class MalojaScrobbler {
#config = null;
#counter = 0;
constructor(config) {
this.#config = config;
if (!config.name)
throw new Error("Invalid name for Maloja scrobber.");
if (!config.url)
throw new Error(`Invalid url for Maloja scrobbler '${this.name}'.`);
if (!config.token)
throw new Error(`Invalid token for Maloja scrobbler '${this.name}'.`)
}
get counter() {
return this.#counter;
}
get name() {
return this.#config.name;
}
async scrobble(song, progress, start) {
const url = new URL(this.#config.url);
url.pathname += "/apis/mlj_1/newscrobble";
url.search = "?key=" + this.#config.token;
await axios.post(url.toString(), {
title: song.name,
album: song.album,
artists: song.artists,
duration: Math.round(progress / 1000),
length: Math.round(song.duration / 1000),
//time: start
});
this.#counter++;
}
}
module.exports = MalojaScrobbler;

View File

@ -1,17 +0,0 @@
class AggregateTracker {
#trackers = []
constructor(trackers) {
this.#trackers = trackers;
}
async poll() {
let media = []
for (let tracker of this.#trackers)
media = media.concat(await tracker.poll());
return media;
}
}
module.exports = AggregateTracker;

View File

@ -0,0 +1,30 @@
class AggregateTracker {
#name = null;
#trackers = []
provider = null;
constructor(name, trackers) {
this.#name = name;
this.#trackers = trackers;
}
get name() {
return this.#name;
}
get scrobblerNames() {
return this.#trackers.map(t => t.scrobblerNames)
.flat()
.filter((v, i, a) => a.indexOf(v) == i);
}
async poll() {
let media = []
for (let tracker of this.#trackers)
media = media.concat(await tracker.poll());
return media;
}
}
module.exports = AggregateTracker;

View File

@ -4,11 +4,20 @@ const Song = require("../../models/song");
class PlexTracker { class PlexTracker {
#config = null; #config = null;
#cache = []; #cache = [];
provider = "plex";
constructor(config) { constructor(config) {
this.#config = config; this.#config = config;
} }
get name() {
return this.#config.name;
}
get scrobblerNames() {
return this.#config.scrobblers;
}
async poll(useCache = false) { async poll(useCache = false) {
if (!this.#config.token || !this.#config.url) if (!this.#config.token || !this.#config.url)
return []; return [];
@ -28,7 +37,7 @@ class PlexTracker {
} }
const filtered = response.data.MediaContainer?.Metadata.filter(m => this.#filter(m)); const filtered = response.data.MediaContainer?.Metadata.filter(m => this.#filter(m));
this.#cache = filtered.map(m => this.#transform(m)); this.#cache = filtered.map(m => this.#transform(m, this.#config.name));
return this.#cache; return this.#cache;
} }
@ -52,9 +61,10 @@ class PlexTracker {
return false; return false;
} }
#transform(data) { #transform(data, source) {
const id = data.guid.substring(data.guid.lastIndexOf('/') + 1); const id = data.guid.substring(data.guid.lastIndexOf('/') + 1);
return new Song(id, data.title, data.parentTitle, data.grandparentTitle, data.parentYear, data.duration, data.viewOffset, data.sessionKey, data.Player.state, "plex"); const artists = data.grandparentTitle.split(',').map(a => a.trim());
return new Song(id, data.title, data.parentTitle, artists, data.parentYear, data.duration, data.viewOffset, data.sessionKey, data.Player.state, source, "plex");
} }
} }

View File

@ -7,16 +7,24 @@ const Song = require("../../models/song");
class SpotifyTracker { class SpotifyTracker {
#config = null; #config = null;
#token = null; #token = null;
#cache = null; #cache = [];
#auth = null; #auth = null;
provider = "spotify";
constructor(config, token = null) { constructor(config, token = null) {
this.#config = config; this.#config = config;
this.#token = token; this.#token = token;
this.#cache = null;
this.#auth = new Buffer.from(config.client_id + ':' + config.client_secret).toString('base64'); this.#auth = new Buffer.from(config.client_id + ':' + config.client_secret).toString('base64');
} }
get name() {
return this.#config.name;
}
get scrobblerNames() {
return this.#config.scrobblers;
}
async poll(useCache = false) { async poll(useCache = false) {
if (this.#token == null) if (this.#token == null)
return []; return [];
@ -40,7 +48,7 @@ class SpotifyTracker {
return this.#cache; return this.#cache;
} }
this.#cache = [this.#transform(response.data)]; this.#cache = [this.#transform(response.data, this.#config.name)];
return this.#cache; return this.#cache;
} catch (ex) { } catch (ex) {
logger.error(ex, "Failed to get currently playing data from Spotify."); logger.error(ex, "Failed to get currently playing data from Spotify.");
@ -58,12 +66,12 @@ class SpotifyTracker {
this.#token = JSON.parse(content); this.#token = JSON.parse(content);
} }
#transform(data) { #transform(data, source) {
const item = data.item; const item = data.item;
const artists = item.artists.map(a => a.name); const artists = item.artists.map(a => a.name);
const year = null; const year = null;
const state = data.is_playing ? "playing" : "paused"; const state = data.is_playing ? "playing" : "paused";
return new Song(item.id, item.name, item.album.name, artists, year, item.duration_ms, data.progress_ms, "spotify", state, "spotify"); return new Song(item.id, item.name, item.album.name, artists, year, item.duration_ms, data.progress_ms, "spotify", state, source, "spotify");
} }
async #refreshTokenIfNeeded() { async #refreshTokenIfNeeded() {