michael/auto-audio-recorder
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Client Created

Browse Source
This commit is contained in:
Michael Peters 2019-09-18 23:05:44 -04:00
parent 237b7a617d
commit 0ddf4480c7
15 changed files with 2011 additions and 34 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
.gitignore vendored
View File

@ -1,6 +1,9 @@
# Audio Files
audio
# Intermediate Files
intermediate
# Logs
logs
*.log

211
client/actions.js Normal file
View File

@ -0,0 +1,211 @@
const child_process = require('child_process');
const request = require('request');
const fs = require('fs');
const LOG = require('../logger/logger.js')('action');
let Actions = {};
// Notes:
// While it may be nice to be able to use the webview tag
// in electron, it probably does not play DRM controlled
// content (i.e. Spotify) so this needs to connect to a
// firefox extension. This can be done by running a web
// server and sending messages to the client from the
// firefox client over said server.
//
// An alternative to this may be to use this:
// https://github.com/castlabs/electron-releases/
// Recording A Song Outline
// Song starts playing on firefox browser
// firefox extension sends web request to the recording client (song started (name, artists, album name, album art (small)))
// recording client starts recording (using gstreamer, song.guid + .wav)
// recording client downloads basic album art
// firefox extension looks up the better album art image
// firefox extension sends web request to the recording client (new song data with same guid)
// recording client downloads song full album art
// Song finishes playing on firefox browser
// firefox extension sends web request to the recording client (song stopped)
// recording client stops recording
// ffmpeg conversion begins to convert song.guid + .wav -> song.filename + .mp3 with album art and mp3 metadata
// Key Objects
// Requirements:
// gstreamer: gst-launch-1.0
// ffmpeg: ffmpeg
/* Song
* {
* guid: string ("11bf5b37-e0b8-42e0-8dcf-dc8c4aefc000"),
* filename: string ("Low-11bf5b37"),
* name: string ("Low")
* artists: list ([ "Michael", "Eric" ])
* duration: string ("3:15")
* album: {
* name: string ("Best Yoinks 2019")
* art: string ("https://spotify.com/art/872364")
* artFull: string ("https://spotify.com/art/2398423428374") [optional]
* }
* }
*/
// Removes Windows special characters from the song name for the file name
Actions.getSongFilename = function(song) {
return song.name.replace(/[\\\/:*?\"<>|]+/g, '') + ' ' + song.guid;
}
// uses a logger to log the output of a process
Actions.registerLogger = function(process, log) {
process.stdout.on('data', (chunk) => {
log.debug(chunk.toString('utf-8'));
});
process.stderr.on('data', (chunk) => {
log.error(chunk.toString('utf-8'));
});
}
// Starts a process and handles general errors
Actions.spawnProcess = function(name, command, args) {
let process = child_process.spawn(command, args);
LOG.info(`spawned ${name} ${process.pid}`);
process.on('close', (code, signal) => {
LOG.silly(`${name} ${process.pid} closed, code: ${code}, signal: ${signal}`);
});
process.on('exit', (code, signal) => {
LOG.debug(`${name} ${process.pid} exited, code: ${code}, signal: ${signal}`);
});
process.on('disconnect', () => {
LOG.silly(`${name} ${process.pid} disconnected from parent`);
});
process.on('error', (err) => {
LOG.silly(`${name} ${process.pid} error`, err);
});
return process;
}
/**
* Starts recording audio output
* @param {Song} song The song to start recording (used for file name)
* @returns the child process that was spawned
*/
Actions.startRecording = function(song) {
let device = 'alsa_output.pci-0000_00_1f.3.analog-stereo.monitor';
let output = '../intermediate/' + song.guid + '.wav';
// -v for verbose output
let recording_process = Actions.spawnProcess(
'recording process',
'gst-launch-1.0',
[ '-v', 'pulsesrc', 'device', '=', device, '!', 'wavenc', '!', 'filesink', 'location', '=', output ]
);
return recording_process;
}
/**
* Stops the recording process
* @param {*} recordingProcess The recording child process
*/
Actions.stopRecording = async function(recordingProcess) {
return new Promise((resolve, reject) => {
LOG.debug(`killing process ${recordingProcess.pid}`);
recordingProcess.on('exit', (code, signal) => {
resolve();
});
recordingProcess.kill('SIGINT');
});
}
/**
* Downloads the album art for a song. If song.album.art_full exists, uses that, else song.album.art
* @param {Song} song The song to download album art for
*/
Actions.downloadAlbumArt = async function(song) {
return new Promise((resolve, reject) => {
try {
let url = song.album.artFull || song.album.art || null;
if (url) {
let writeStream = fs.createWriteStream('../intermediate/' + song.guid + '.jpg'); // Assume .jpg image
writeStream.on('finish', () => { resolve(); });
request.get(url).pipe(writeStream);
} else {
fs.copyFile('../resources/default-album-art.jpg', '../intermediate/' + song.guid + '.jpg', (err) => {
if (err) {
reject(err);
} else {
resolve();
}
});
}
} catch (e) {
reject(e);
}
});
}
Actions.convertToMp3 = function(song) {
if (
!fs.existsSync('../intermediate/' + song.guid + '.wav') ||
!fs.existsSync('../intermediate/' + song.guid + '.jpg')
) {
LOG.warn(`Cannot convert ${song.name} to mp3: [${song.guid}.wav/${song.guid}.jpg] does not exist`);
return;
}
// Album Art: https://stackoverflow.com/questions/18710992/how-to-add-album-art-with-ffmpeg
let conversionProcess = Actions.spawnProcess(
'conversion process',
'ffmpeg', [
'-hide_banner',
'-i', '../intermediate/' + song.guid + '.wav',
'-f', 'mp3',
'../intermediate/' + song.guid + '.mp3'
]
);
return conversionProcess;
}
Actions.addMetadata = function(song) {
if (!fs.existsSync('../intermediate/' + song.guid + '.mp3')) {
LOG.warn(`Cannot add metadata to ${song.name}: [${song.guid}.mp3] does not exist`);
return;
}
let metadataProcess = Actions.spawnProcess(
'metadata process',
'ffmpeg', [
'-hide_banner',
'-i', '../intermediate/' + song.guid + '.mp3',
'-i', '../intermediate/' + song.guid + '.jpg',
'-map', '0:0',
'-map', '1:0',
'-c', 'copy',
'-id3v2_version', '3',
'-metadata:s:v', 'title=\'Album Cover\'',
'-metadata:s:v', 'comment=\'Cover (front)\'',
'-metadata', `title=\'${song.name}\'`,
'-metadata', `author=\'${song.artists.join(' / ')}\'`,
'-metadata', `album=\'${song.album.name}\'`,
'-b:a', '192k',
'../audio/' + song.filename + '.mp3'
]
);
return metadataProcess;
}
Actions.removeIntermediate = function(song) {
fs.unlink('../intermediate/' + song.guid + '.wav', (err) => {
if (err) {
LOG.error(`error removing ${song.guid}.wav`, err);
}
});
fs.unlink('../intermediate/' + song.guid + '.jpg', (err) => {
if (err) {
LOG.error(`error removing ${song.guid}.jpg`, err);
}
});
fs.unlink('../intermediate/' + song.guid + '.mp3', (err) => {
if (err) {
LOG.error(`error removing ${song.guid}.mp3`, err);
}
});
}
module.exports = Actions;

97
client/diactrics.js Normal file
View File

@ -0,0 +1,97 @@
// https://gist.github.com/yeah/1283961
var diacriticsMap = [
{'base':'A', 'letters':/[\u0041\u24B6\uFF21\u00C0\u00C1\u00C2\u1EA6\u1EA4\u1EAA\u1EA8\u00C3\u0100\u0102\u1EB0\u1EAE\u1EB4\u1EB2\u0226\u01E0\u01DE\u1EA2\u00C5\u01FA\u01CD\u0200\u0202\u1EA0\u1EAC\u1EB6\u1E00\u0104\u023A\u2C6F]/g},
{'base':'AA','letters':/[\uA732]/g},
{'base':'AE','letters':/[\u00C4\u00C6\u01FC\u01E2]/g},
{'base':'AO','letters':/[\uA734]/g},
{'base':'AU','letters':/[\uA736]/g},
{'base':'AV','letters':/[\uA738\uA73A]/g},
{'base':'AY','letters':/[\uA73C]/g},
{'base':'B', 'letters':/[\u0042\u24B7\uFF22\u1E02\u1E04\u1E06\u0243\u0182\u0181]/g},
{'base':'C', 'letters':/[\u0043\u24B8\uFF23\u0106\u0108\u010A\u010C\u00C7\u1E08\u0187\u023B\uA73E]/g},
{'base':'D', 'letters':/[\u0044\u24B9\uFF24\u1E0A\u010E\u1E0C\u1E10\u1E12\u1E0E\u0110\u018B\u018A\u0189\uA779]/g},
{'base':'DZ','letters':/[\u01F1\u01C4]/g},
{'base':'Dz','letters':/[\u01F2\u01C5]/g},
{'base':'E', 'letters':/[\u0045\u24BA\uFF25\u00C8\u00C9\u00CA\u1EC0\u1EBE\u1EC4\u1EC2\u1EBC\u0112\u1E14\u1E16\u0114\u0116\u00CB\u1EBA\u011A\u0204\u0206\u1EB8\u1EC6\u0228\u1E1C\u0118\u1E18\u1E1A\u0190\u018E]/g},
{'base':'F', 'letters':/[\u0046\u24BB\uFF26\u1E1E\u0191\uA77B]/g},
{'base':'G', 'letters':/[\u0047\u24BC\uFF27\u01F4\u011C\u1E20\u011E\u0120\u01E6\u0122\u01E4\u0193\uA7A0\uA77D\uA77E]/g},
{'base':'H', 'letters':/[\u0048\u24BD\uFF28\u0124\u1E22\u1E26\u021E\u1E24\u1E28\u1E2A\u0126\u2C67\u2C75\uA78D]/g},
{'base':'I', 'letters':/[\u0049\u24BE\uFF29\u00CC\u00CD\u00CE\u0128\u012A\u012C\u0130\u00CF\u1E2E\u1EC8\u01CF\u0208\u020A\u1ECA\u012E\u1E2C\u0197]/g},
{'base':'J', 'letters':/[\u004A\u24BF\uFF2A\u0134\u0248]/g},
{'base':'K', 'letters':/[\u004B\u24C0\uFF2B\u1E30\u01E8\u1E32\u0136\u1E34\u0198\u2C69\uA740\uA742\uA744\uA7A2]/g},
{'base':'L', 'letters':/[\u004C\u24C1\uFF2C\u013F\u0139\u013D\u1E36\u1E38\u013B\u1E3C\u1E3A\u0141\u023D\u2C62\u2C60\uA748\uA746\uA780]/g},
{'base':'LJ','letters':/[\u01C7]/g},
{'base':'Lj','letters':/[\u01C8]/g},
{'base':'M', 'letters':/[\u004D\u24C2\uFF2D\u1E3E\u1E40\u1E42\u2C6E\u019C]/g},
{'base':'N', 'letters':/[\u004E\u24C3\uFF2E\u01F8\u0143\u00D1\u1E44\u0147\u1E46\u0145\u1E4A\u1E48\u0220\u019D\uA790\uA7A4]/g},
{'base':'NJ','letters':/[\u01CA]/g},
{'base':'Nj','letters':/[\u01CB]/g},
{'base':'O', 'letters':/[\u004F\u24C4\uFF2F\u00D2\u00D3\u00D4\u1ED2\u1ED0\u1ED6\u1ED4\u00D5\u1E4C\u022C\u1E4E\u014C\u1E50\u1E52\u014E\u022E\u0230\u022A\u1ECE\u0150\u01D1\u020C\u020E\u01A0\u1EDC\u1EDA\u1EE0\u1EDE\u1EE2\u1ECC\u1ED8\u01EA\u01EC\u00D8\u01FE\u0186\u019F\uA74A\uA74C]/g},
{'base':'OE','letters':/[\u00D6\u0152]/g},
{'base':'OI','letters':/[\u01A2]/g},
{'base':'OO','letters':/[\uA74E]/g},
{'base':'OU','letters':/[\u0222]/g},
{'base':'P', 'letters':/[\u0050\u24C5\uFF30\u1E54\u1E56\u01A4\u2C63\uA750\uA752\uA754]/g},
{'base':'Q', 'letters':/[\u0051\u24C6\uFF31\uA756\uA758\u024A]/g},
{'base':'R', 'letters':/[\u0052\u24C7\uFF32\u0154\u1E58\u0158\u0210\u0212\u1E5A\u1E5C\u0156\u1E5E\u024C\u2C64\uA75A\uA7A6\uA782]/g},
{'base':'S', 'letters':/[\u0053\u24C8\uFF33\u1E9E\u015A\u1E64\u015C\u1E60\u0160\u1E66\u1E62\u1E68\u0218\u015E\u2C7E\uA7A8\uA784]/g},
{'base':'T', 'letters':/[\u0054\u24C9\uFF34\u1E6A\u0164\u1E6C\u021A\u0162\u1E70\u1E6E\u0166\u01AC\u01AE\u023E\uA786]/g},
{'base':'TZ','letters':/[\uA728]/g},
{'base':'U', 'letters':/[\u0055\u24CA\uFF35\u00D9\u00DA\u00DB\u0168\u1E78\u016A\u1E7A\u016C\u01DB\u01D7\u01D5\u01D9\u1EE6\u016E\u0170\u01D3\u0214\u0216\u01AF\u1EEA\u1EE8\u1EEE\u1EEC\u1EF0\u1EE4\u1E72\u0172\u1E76\u1E74\u0244]/g},
{'base':'UE','letters':/[\u00DC]/g},
{'base':'V', 'letters':/[\u0056\u24CB\uFF36\u1E7C\u1E7E\u01B2\uA75E\u0245]/g},
{'base':'VY','letters':/[\uA760]/g},
{'base':'W', 'letters':/[\u0057\u24CC\uFF37\u1E80\u1E82\u0174\u1E86\u1E84\u1E88\u2C72]/g},
{'base':'X', 'letters':/[\u0058\u24CD\uFF38\u1E8A\u1E8C]/g},
{'base':'Y', 'letters':/[\u0059\u24CE\uFF39\u1EF2\u00DD\u0176\u1EF8\u0232\u1E8E\u0178\u1EF6\u1EF4\u01B3\u024E\u1EFE]/g},
{'base':'Z', 'letters':/[\u005A\u24CF\uFF3A\u0179\u1E90\u017B\u017D\u1E92\u1E94\u01B5\u0224\u2C7F\u2C6B\uA762]/g},
{'base':'a', 'letters':/[\u0061\u24D0\uFF41\u1E9A\u00E0\u00E1\u00E2\u1EA7\u1EA5\u1EAB\u1EA9\u00E3\u0101\u0103\u1EB1\u1EAF\u1EB5\u1EB3\u0227\u01E1\u01DF\u1EA3\u00E5\u01FB\u01CE\u0201\u0203\u1EA1\u1EAD\u1EB7\u1E01\u0105\u2C65\u0250]/g},
{'base':'aa','letters':/[\uA733]/g},
{'base':'ae','letters':/[\u00E4\u00E6\u01FD\u01E3]/g},
{'base':'ao','letters':/[\uA735]/g},
{'base':'au','letters':/[\uA737]/g},
{'base':'av','letters':/[\uA739\uA73B]/g},
{'base':'ay','letters':/[\uA73D]/g},
{'base':'b', 'letters':/[\u0062\u24D1\uFF42\u1E03\u1E05\u1E07\u0180\u0183\u0253]/g},
{'base':'c', 'letters':/[\u0063\u24D2\uFF43\u0107\u0109\u010B\u010D\u00E7\u1E09\u0188\u023C\uA73F\u2184]/g},
{'base':'d', 'letters':/[\u0064\u24D3\uFF44\u1E0B\u010F\u1E0D\u1E11\u1E13\u1E0F\u0111\u018C\u0256\u0257\uA77A]/g},
{'base':'dz','letters':/[\u01F3\u01C6]/g},
{'base':'e', 'letters':/[\u0065\u24D4\uFF45\u00E8\u00E9\u00EA\u1EC1\u1EBF\u1EC5\u1EC3\u1EBD\u0113\u1E15\u1E17\u0115\u0117\u00EB\u1EBB\u011B\u0205\u0207\u1EB9\u1EC7\u0229\u1E1D\u0119\u1E19\u1E1B\u0247\u025B\u01DD]/g},
{'base':'f', 'letters':/[\u0066\u24D5\uFF46\u1E1F\u0192\uA77C]/g},
{'base':'g', 'letters':/[\u0067\u24D6\uFF47\u01F5\u011D\u1E21\u011F\u0121\u01E7\u0123\u01E5\u0260\uA7A1\u1D79\uA77F]/g},
{'base':'h', 'letters':/[\u0068\u24D7\uFF48\u0125\u1E23\u1E27\u021F\u1E25\u1E29\u1E2B\u1E96\u0127\u2C68\u2C76\u0265]/g},
{'base':'hv','letters':/[\u0195]/g},
{'base':'i', 'letters':/[\u0069\u24D8\uFF49\u00EC\u00ED\u00EE\u0129\u012B\u012D\u00EF\u1E2F\u1EC9\u01D0\u0209\u020B\u1ECB\u012F\u1E2D\u0268\u0131]/g},
{'base':'j', 'letters':/[\u006A\u24D9\uFF4A\u0135\u01F0\u0249]/g},
{'base':'k', 'letters':/[\u006B\u24DA\uFF4B\u1E31\u01E9\u1E33\u0137\u1E35\u0199\u2C6A\uA741\uA743\uA745\uA7A3]/g},
{'base':'l', 'letters':/[\u006C\u24DB\uFF4C\u0140\u013A\u013E\u1E37\u1E39\u013C\u1E3D\u1E3B\u017F\u0142\u019A\u026B\u2C61\uA749\uA781\uA747]/g},
{'base':'lj','letters':/[\u01C9]/g},
{'base':'m', 'letters':/[\u006D\u24DC\uFF4D\u1E3F\u1E41\u1E43\u0271\u026F]/g},
{'base':'n', 'letters':/[\u006E\u24DD\uFF4E\u01F9\u0144\u00F1\u1E45\u0148\u1E47\u0146\u1E4B\u1E49\u019E\u0272\u0149\uA791\uA7A5]/g},
{'base':'nj','letters':/[\u01CC]/g},
{'base':'o', 'letters':/[\u006F\u24DE\uFF4F\u00F2\u00F3\u00F4\u1ED3\u1ED1\u1ED7\u1ED5\u00F5\u1E4D\u022D\u1E4F\u014D\u1E51\u1E53\u014F\u022F\u0231\u022B\u1ECF\u0151\u01D2\u020D\u020F\u01A1\u1EDD\u1EDB\u1EE1\u1EDF\u1EE3\u1ECD\u1ED9\u01EB\u01ED\u00F8\u01FF\u0254\uA74B\uA74D\u0275]/g},
{'base':'oe','letters': /[\u00F6\u0153]/g},
{'base':'oi','letters':/[\u01A3]/g},
{'base':'ou','letters':/[\u0223]/g},
{'base':'oo','letters':/[\uA74F]/g},
{'base':'p','letters':/[\u0070\u24DF\uFF50\u1E55\u1E57\u01A5\u1D7D\uA751\uA753\uA755]/g},
{'base':'q','letters':/[\u0071\u24E0\uFF51\u024B\uA757\uA759]/g},
{'base':'r','letters':/[\u0072\u24E1\uFF52\u0155\u1E59\u0159\u0211\u0213\u1E5B\u1E5D\u0157\u1E5F\u024D\u027D\uA75B\uA7A7\uA783]/g},
{'base':'s','letters':/[\u0073\u24E2\uFF53\u015B\u1E65\u015D\u1E61\u0161\u1E67\u1E63\u1E69\u0219\u015F\u023F\uA7A9\uA785\u1E9B]/g},
{'base':'ss','letters':/[\u00DF]/g},
{'base':'t','letters':/[\u0074\u24E3\uFF54\u1E6B\u1E97\u0165\u1E6D\u021B\u0163\u1E71\u1E6F\u0167\u01AD\u0288\u2C66\uA787]/g},
{'base':'tz','letters':/[\uA729]/g},
{'base':'u','letters':/[\u0075\u24E4\uFF55\u00F9\u00FA\u00FB\u0169\u1E79\u016B\u1E7B\u016D\u01DC\u01D8\u01D6\u01DA\u1EE7\u016F\u0171\u01D4\u0215\u0217\u01B0\u1EEB\u1EE9\u1EEF\u1EED\u1EF1\u1EE5\u1E73\u0173\u1E77\u1E75\u0289]/g},
{'base':'ue','letters':/[\u00FC]/g},
{'base':'v','letters':/[\u0076\u24E5\uFF56\u1E7D\u1E7F\u028B\uA75F\u028C]/g},
{'base':'vy','letters':/[\uA761]/g},
{'base':'w','letters':/[\u0077\u24E6\uFF57\u1E81\u1E83\u0175\u1E87\u1E85\u1E98\u1E89\u2C73]/g},
{'base':'x','letters':/[\u0078\u24E7\uFF58\u1E8B\u1E8D]/g},
{'base':'y','letters':/[\u0079\u24E8\uFF59\u1EF3\u00FD\u0177\u1EF9\u0233\u1E8F\u00FF\u1EF7\u1E99\u1EF5\u01B4\u024F\u1EFF]/g},
{'base':'z','letters':/[\u007A\u24E9\uFF5A\u017A\u1E91\u017C\u017E\u1E93\u1E95\u01B6\u0225\u0240\u2C6C\uA763]/g}
];
for(var i=0; i<diacriticsMap.length; i++) {
permalink = stringWithDiacritics.replace(diacriticsMap[i].letters, diacriticsMap[i].base);
}

69
client/index.css Normal file
View File

@ -0,0 +1,69 @@
* {
font-family: Arial, Helvetica, sans-serif;
box-sizing: border-box;
}
body {
background-color: #222;
color: #fff;
margin: 0;
}
h1, h2 {
margin: 0 0 10px 0;
}
.row {
display: flex;
}
#connection {
margin: 10px;
}
#current-song {
margin: 10px;
}
#current-song div.info {
display: flex;
align-items: center;
line-height: 1;
}
#current-song div.info {
margin-right: 10px;
}
#current-song div.info img.art {
width: 80px;
height: 80px;
border-radius: 10px;
margin-right: 10px;
}
#current-song div.info div.description > div {
white-space: nowrap;
overflow: hidden;
text-overflow: ellipsis;
margin-bottom: 10px;
}
#current-song div.info div.description > div:last-child {
margin-bottom: 0;
}
#current-song div.duration {
display: flex;
align-items: center;
justify-content: center;
}
#current-song div.duration > * {
padding: 10px;
}
#current-song div.guid {
margin-top: 10px;
text-align: center;
}

32
client/index.html Normal file
View File

@ -0,0 +1,32 @@
<!DOCTYPE html>
<html>
<head>
<meta charset="utf-8">
<title>Spotify Audio Recorder</title>
<link rel="stylesheet" type="text/css" href="index.css">
<script src="index.js"></script>
</head>
<body>
<div id="connection">
Extension Status: <span id="extension-status">Disconnected</span>
</div>
<div id="current-song">
<div class="row">
<div class="info">
<img class="art" src="../resources/default-album-art.jpg">
<div class="description">
<div class="title">Waiting for song...</div>
<div class="artists"></div>
<div class="album"></div>
</div>
</div>
</div>
<div class="duration">
<div class="current">?</div>
<progress value="0" max="180"></progress>
<div class="total">?</div>
</div>
<div class="guid">deadbeef-lead-leaf-ca1f-deadb00b5a1e</div>
</div>
</body>
</html>

133
client/index.js Normal file
View File

@ -0,0 +1,133 @@
const electron = require('electron');
const electronConsole = electron.remote.getGlobal('console');
const LOG = require('../logger/logger.js')('index', electronConsole);
let UI = {
DEFAULT_ALBUM_ART: '../resources/default-album-art.jpg',
durationTimer: 0,
durationTimerInterval: null,
}
UI.padLeft = function (s, l, c) {
s = '' + s;
l = l || 0;
c = c || ' ';
while (s.length < l) {
s = c + s;
}
return s;
}
UI.secondsToTimeString = function(seconds) {
return `${Math.floor(seconds / 60)}:${UI.padLeft(seconds % 60, 2, '0')}`;
}
UI.getElements = function() {
let qs = (s) => { return document.querySelector(s); };
return {
extension: {
status: qs('#extension-status'),
},
guid: qs('#current-song .guid'),
title: qs('#current-song .title'),
artists: qs('#current-song .artists'),
album: {
name: qs('#current-song .album'),
art: qs('#current-song .art'),
},
duration: {
current: qs('#current-song .duration .current'),
total: qs('#current-song .duration .total'),
progress: qs('#current-song .duration progress'),
},
recording: {
status: qs('#recording .status'),
},
conversion: {
output: qs('#conversion .output'),
},
};
}
UI.updateSongData = function(song) {
let elements = UI.getElements();
elements.guid.innerText = song.guid;
elements.title.innerText = song.name;
elements.artists.innerText = song.artists.join(', ');
elements.album.name.innerText = song.album.name;
elements.album.art.src = song.album.artFull || song.album.art || UI.DEFAULT_ALBUM_ART;
if (song.duration) {
elements.duration.total.innerText = UI.secondsToTimeString(song.duration);
elements.duration.progress.max = song.duration;
} else {
elements.duration.total.innerText = '?';
elements.duration.progress.max = 180;
}
}
UI.onNewSong = function(song) {
UI.updateSongData(song);
let elements = UI.getElements();
elements.duration.current.innerText = UI.secondsToTimeString(0);
}
UI.onUpdateSong = function(song) {
UI.updateSongData(song);
}
UI.onDurationTick = function(duration) {
let elements = UI.getElements();
elements.duration.current.innerText = UI.secondsToTimeString(duration);
elements.duration.progress.value = duration;
}
UI.onAdvertisement = function () {
let elements = UI.getElements();
elements.guid.innerText = '';
elements.title.innerText = 'Advertisement';
elements.artists.innerText = '';
elements.album.name.innerText = '';
elements.album.art.src = UI.DEFAULT_ALBUM_ART;
elements.duration.current.innerText = '?';
elements.duration.progress.value = 0;
elements.duration.progress.max = 1;
elements.duration.total.innerText = '?';
}
UI.onExtensionConnect = function() {
let elements = UI.getElements();
elements.extension.status.innerText = 'Connected';
}
UI.onExtensionDisconnect = function() {
let elements = UI.getElements();
elements.extension.status.innerText = 'Disconnected';
}
UI.init = function () {
electron.ipcRenderer.on('new-song', (event, song) => { UI.onNewSong(song); });
electron.ipcRenderer.on('update-song', (event, song) => { UI.onUpdateSong(song); });
electron.ipcRenderer.on('advertisement', (event) => { UI.onAdvertisement(); });
electron.ipcRenderer.on('duration-tick', (event, duration) => { UI.onDurationTick(duration); });
electron.ipcRenderer.on('extension-connect', (event) => { UI.onExtensionConnect(); });
electron.ipcRenderer.on('extension-disconnect', (event) => { UI.onExtensionDisconnect(); });
let restartButton = document.getElementById('restart');
let skipButton = document.getElementById('skip');
restartButton.addEventListener('click', (event) => {
electron.ipcRenderer.send('restart-song');
});
skipButton.addEventListener('click', (event) => {
electron.ipcRenderer.send('skip-song');
});
}
document.addEventListener('DOMContentLoaded', UI.init);

185
client/main.js Normal file
View File

@ -0,0 +1,185 @@
const electron = require('electron');
const http = require('http');
const Actions = require('./actions.js');
const LOG = require('../logger/logger.js')('main');
let window = null;
let port = 50233;
electron.app.on('ready', (launchInfo) => {
window = new electron.BrowserWindow({
width: 400,
height: 200,
webPreferences: {
nodeIntegration: true
}
});
window.setMenuBarVisibility(false);
LOG.debug('window being created...');
window.loadFile('index.html');
window.on('closed', () => { window = null });
LOG.debug('window created');
});
electron.app.on('window-all-closed', () => {
electron.app.quit();
});
electron.ipcMain.on('restart-song', () => {
LOG.debug('got restart song request');
});
electron.ipcMain.on('skip-song', () => {
LOG.debug('got skip song request');
});
let wasAd = false;
let curentSong = null;
const RECLOG = require('../logger/logger.js')('rec');
const CVRTLOG = require('../logger/logger.js')('cvrt');
const METALOG = require('../logger/logger.js')('meta');
let recordingProcess = null;
let durationInterval = null;
async function stopRecordingStartConversion() {
LOG.debug(`${currentSong.filename} stopping recording...`);
await Actions.stopRecording(recordingProcess);
let conversionSong = currentSong;
LOG.debug(`${conversionSong.filename} starting conversion...`);
let conversionProcess = Actions.convertToMp3(conversionSong);
Actions.registerLogger(conversionProcess, CVRTLOG);
conversionProcess.on('exit', (code, signal) => {
LOG.info(`${conversionSong.filename} completed conversion`);
LOG.debug(`${conversionSong.filename} starting adding metadata...`);
let metadataProcess = Actions.addMetadata(conversionSong);
Actions.registerLogger(metadataProcess, METALOG);
metadataProcess.on('exit', (code, signal) => {
LOG.info(`${conversionSong.filename} completed adding metadata`);
Actions.removeIntermediate(conversionSong);
});
LOG.info(`${conversionSong.filename} started adding metadata`);
});
LOG.info(`${conversionSong.filename} started conversion`);
}
async function onNewSong(song) {
wasAd = false;
LOG.debug(`${song.filename} new song`);
if (recordingProcess != null) {
await stopRecordingStartConversion();
}
LOG.debug(`${song.filename} starting recording...`);
recordingProcess = Actions.startRecording(song);
Actions.registerLogger(recordingProcess, RECLOG);
recordingProcess.on('exit', (code, signal) => {
LOG.info(`${song.filename} recording stopped`);
recordingProcess = null;
});
LOG.info(`${song.filename} started recording`);
window.webContents.send('new-song', song);
currentSong = song;
let duration = 0;
clearInterval(durationInterval);
durationInterval = setInterval(() => {
duration += 1;
window.webContents.send('duration-tick', duration);
if (duration >= 60 * 60) {
LOG.debug('song lasted more than an hour, marking as an advertisement');
onAdvertisement();
}
}, 1000);
LOG.debug(`${song.filename} starting album art download`);
await Actions.downloadAlbumArt(song);
LOG.info(`${song.filename} album art downloaded`);
}
async function onUpdateSong(song) {
LOG.debug(`${song.filename} update song`)
window.webContents.send('update-song', song);
LOG.debug(`${song.filename} starting full album art download`);
await Actions.downloadAlbumArt(song);
LOG.info(`${song.filename} full album art downloaded`);
}
async function onAdvertisement() {
if (!wasAd) {
LOG.debug(`advertisement`);
}
wasAd = true;
clearInterval(durationInterval);
if (recordingProcess != null) {
await stopRecordingStartConversion();
}
window.webContents.send('advertisement');
}
let heartbeatTimeout = null;
function onHeartbeat() {
if (heartbeatTimeout == null) {
LOG.debug('extension connected');
window.webContents.send('extension-connect');
}
clearTimeout(heartbeatTimeout);
heartbeatTimeout = setTimeout(() => {
LOG.debug('extension disconnected');
window.webContents.send('extension-disconnect');
heartbeatTimeout = null;
}, 8000);
}
http.createServer((req, res) => {
let body = '';
req.on('data', (chunk) => {
body += chunk;
});
req.on('end', () => {
let data = null;
try {
data = JSON.parse(body);
} catch (e) {
LOG.error('unable to parse request json', { body: body, error: e });
res.statusCode = 400;
res.end();
return;
}
if (data.song) {
data.song.filename = Actions.getSongFilename(data.song);
}
switch (data.type) {
case 'new-song':
onNewSong(data.song);
break;
case 'update-song':
onUpdateSong(data.song);
break;
case 'advertisement':
onAdvertisement();
case 'heartbeat':
onHeartbeat();
break;
default:
res.statusCode = 400;
res.end();
return;
}
res.statusCode = 200;
res.end();
});
}).listen(port);

1070
client/package-lock.json generated Normal file
View File

File diff suppressed because it is too large Load Diff

16
client/package.json Normal file
View File

@ -0,0 +1,16 @@
{
"name": "client",
"version": "1.0.0",
"description": "",
"main": "main.js",
"scripts": {
"start": "electron ."
},
"author": "",
"license": "ISC",
"dependencies": {
"electron": "^6.0.9",
"request": "^2.88.0",
"uuid": "^3.3.3"
}
}

31
command-line/main.js
View File

@ -1,31 +0,0 @@
const child_process = require('child_process');
// Requirements:
// gstreamer: gst-launch-1.0
// ffmpeg: ffmpeg
// gstreamer = audio recording
// see `pacmd list` for audio sources
// -v for verbose output
// $ gst-launch-1.0 -v pulsesrc device = "alsa_output.pci-0000_00_1f.3.analog-stereo.monitor" ! wavenc ! filesink location="output.wav"
function start_recording() {
let device = 'alsa_output.pci-0000_00_1f.3.analog-stereo.monitor';
let output = 'output.wav';
let recording_process = child_process.spawn(
'gst-launch-1.0', [ '-v', 'pulsesrc', 'device', '=', device, '!', 'wavenc', '!', 'filesink', 'location', '=', output ]
);
return recording_process;
}
async function stop_recording() {
}
// ffmpeg = file conversion (wav -> mp3)
// $ ffmpeg -i output.wav -b:a 192k output.mp3
// metadata can be added as follows
// $ ffmpeg -i output.wav \
// -metadata title="Song Name" \
// -metadata author="Song Artist" \
// -metadata album="Song Album" \
// -b:a 192k output.mp3

59
extension-firefox/background.js Normal file
View File

@ -0,0 +1,59 @@
let url = 'http://localhost:50233';
function send(data) {
console.log('sending', data);
let xhr = new XMLHttpRequest();
xhr.open('POST', 'http://localhost:50233');
xhr.setRequestHeader('Content-Type', 'application/json');
xhr.setRequestHeader('Access-Control-Allow-Origin', '*')
xhr.send(JSON.stringify(data));
xhr.addEventListener('abort', (event) => {
console.error('XHR Abort', event);
});
xhr.addEventListener('error', (event) => {
console.error('XHR Error', event);
});
xhr.addEventListener('timeout', (event) => {
console.error('XHR Timeout', event);
});
xhr.addEventListener('load', (event) => {
if (xhr.status != 200) {
console.error('bad xhr response status: ' + xhr.status);
}
});
}
function sendNewSong(song) {
send({ type: 'new-song', song: song });
}
function sendUpdateSong(song) {
send({ type: 'update-song', song: song });
}
function sendAdvertisement() {
send({ type: 'advertisement' });
}
function sendHeartbeat() {
send({ type: 'heartbeat' });
}
browser.runtime.onMessage.addListener((message) => {
message = JSON.parse(message);
switch (message.type) {
case 'new-song':
sendNewSong(message.song);
break;
case 'update-song':
sendUpdateSong(message.song);
break;
case 'advertisement':
sendAdvertisement();
break;
case 'heartbeat':
sendHeartbeat();
break;
}
});

113
extension-firefox/content.js Normal file
View File

@ -0,0 +1,113 @@
// See https://stackoverflow.com/questions/105034/create-guid-uuid-in-javascript
function uuidv4() {
return ([1e7]+-1e3+-4e3+-8e3+-1e11).replace(/[018]/g, c =>
(c ^ crypto.getRandomValues(new Uint8Array(1))[0] & 15 >> c / 4).toString(16)
);
}
function onNewSong(song) {
browser.runtime.sendMessage(JSON.stringify({ type: 'new-song', song: song }));
}
function onUpdateSong(song) {
browser.runtime.sendMessage(JSON.stringify({ type: 'update-song', song: song }));
}
function onAdvertisement() {
browser.runtime.sendMessage(JSON.stringify({ type: 'advertisement' }));
}
function onHeartbeat() {
browser.runtime.sendMessage(JSON.stringify({ type: 'heartbeat' }));
}
let song = { name: '', artists: [] };
// Initial Song Data
function getNameElement() { return document.querySelector('.Root__now-playing-bar .now-playing-bar-container div.track-info__name span a'); }
function getArtistElements() { return document.querySelectorAll('.Root__now-playing-bar .now-playing-bar-container div.track-info__artists span a'); }
function getAlbumArtElement() { return document.querySelector('.Root__now-playing-bar .now-playing-bar-container .cover-art .cover-art-image'); }
// After album is loaded
function getDurationElement() { return document.querySelector('.Root__now-playing-bar .now-playing-bar .playback-bar .playback-bar__progress-time:last-child'); }
function getAlbumNameElement() { return document.querySelector('.TrackListHeader .mo-info-name'); }
function getAlbumArtFullElement() { return document.querySelector('.TrackListHeader .cover-art .cover-art-image'); }
function checkSong() {
let nameElement = getNameElement();
let artistsElement = getArtistElements();
let albumArtElement = getAlbumArtElement();
let name = null;
let artists = [];
let albumArtLink = null;
name = nameElement ? nameElement.innerText : 'Unknown Name';
if (
name == 'Advertisement' ||
document.title.toLowerCase().indexOf('advertisement') != -1 ||
document.title.toLowerCase().indexOf('spotify') != -1
) {
console.log('advertisement');
onAdvertisement();
return;
}
for (let i = 0; i < artistsElement.length; ++i) {
let artistElement = artistsElement[i];
artists.push(artistElement.innerText);
}
let albumArtLinkStyle = albumArtElement.getAttribute('style');
albumArtLink = albumArtLinkStyle.substring('background-image: url("'.length, albumArtLinkStyle.length - '");'.length);
if (name != song.name || artists.join(', ') != song.artists.join(', ')) {
// new song
song = {
guid: uuidv4(),
name: name,
artists: artists,
album: {
name: 'Unknown Album',
art: albumArtLink
}
};
console.log('new song', song);
onNewSong(song);
// Click the album link
nameElement.click();
setTimeout(() => {
let durationElement = getDurationElement();
let albumNameElement = getAlbumNameElement();
let albumArtFullElement = getAlbumArtFullElement();
let duration = durationElement ? durationElement.innerText : null;
if (duration) {
let s = duration.split(':');
duration = parseInt(s[0]) * 60 + parseInt(s[1]);
}
let albumName = albumNameElement ? albumNameElement.getAttribute('title') : 'Unknown Album';
let albumArtFullLinkStyle = albumArtFullElement.getAttribute('style');
let albumArtFullLink = albumArtLinkStyle.substring('background-image: url("'.length, albumArtFullLinkStyle.length - 3);
song.duration = duration;
song.album.name = albumName;
song.album.artFull = albumArtFullLink;
onUpdateSong(song);
}, 5000); // give 5 seconds to load
}
}
console.log('Spotify Song Data Sender Extension Running');
// Send heartbeats every 3 seconds as soon as the content script is loaded
setInterval(() => { onHeartbeat(); }, 3000);
onHeartbeat();
// Check for new songs every 300 ms
setInterval(() => { checkSong(); }, 300);

18
extension-firefox/manifest.json Normal file
View File

@ -0,0 +1,18 @@
{
"manifest_version": 2,
"name": "Spotify Song Information Sender",
"version": "1.0",
"description": "Sends spotify song data to a server as you listen.",
"permissions": [
"*://localhost/*"
],
"background": {
"scripts": [ "background.js" ]
},
"content_scripts": [
{
"matches": [ "*://open.spotify.com/*" ],
"js": [ "content.js" ]
}
]
}

8
logger/logger.js
View File

@ -17,7 +17,8 @@ function padCenter(x, p, n) {
return x;
}
function createLogger(name) {
function createLogger(name, processConsole) {
processConsole = processConsole || console;
let logger = {};
function colorize(level, text) {
let map = {
@ -41,7 +42,8 @@ function createLogger(name) {
message = null;
}
if (message) {
out += `${prefix}: ${message}`;
out += message.split('\n').map(o => `${prefix}: ${o}`).join('\n');
// out += `${prefix}: ${message}`;
}
function handleData(data) {
if (data) {
@ -68,7 +70,7 @@ function createLogger(name) {
} else {
handleData(data);
}
console.log(out);
processConsole.log(out);
}
logger.error = (message, data) => { logger.log('error', message, data); };
logger.warn = (message, data) => { logger.log('warn', message, data); };

BIN
resources/default-album-art.jpg Normal file
View File

Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 35 KiB