server.mjs 5.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320
  1. import fs from 'node:fs';
  2. import https from 'node:https';
  3. import net from 'node:net';
  4. const MPD_HOST =
  5. process.env.MPD_HOST ?? '127.0.0.1';
  6. const MPD_PORT =
  7. Number(process.env.MPD_PORT ?? '6600');
  8. const MPD_TIMEOUT_MS =
  9. Number(process.env.MPD_TIMEOUT_MS ?? '700');
  10. const POLL_INTERVAL_MS =
  11. Number(process.env.POLL_INTERVAL_MS ?? '1000');
  12. const TLS_CERT_FILE =
  13. process.env.TLS_CERT_FILE ?? '/var/run/tls/tls.crt';
  14. const TLS_KEY_FILE =
  15. process.env.TLS_KEY_FILE ?? '/var/run/tls/tls.key';
  16. const state = {
  17. available: false,
  18. current: null,
  19. previous: [],
  20. error: 'Waiting for MPD',
  21. updatedAt: null,
  22. };
  23. let pollRunning = false;
  24. const songDisplay = (fields) => {
  25. const artist =
  26. fields.Artist ??
  27. fields.AlbumArtist ??
  28. '';
  29. const title =
  30. fields.Title ??
  31. fields.Name ??
  32. fields.file ??
  33. 'Unknown';
  34. return artist
  35. ? `${artist} - ${title}`
  36. : title;
  37. };
  38. const parseSong = (raw) => {
  39. const fields = {};
  40. for (const line of raw.split(/\r?\n/)) {
  41. if (
  42. !line ||
  43. line === 'OK' ||
  44. line.startsWith('OK MPD ')
  45. ) {
  46. continue;
  47. }
  48. if (line.startsWith('ACK ')) {
  49. throw new Error(line);
  50. }
  51. const separator = line.indexOf(': ');
  52. if (separator === -1) {
  53. continue;
  54. }
  55. const key = line.slice(0, separator);
  56. const value = line.slice(separator + 2);
  57. if (fields[key] === undefined) {
  58. fields[key] = value;
  59. }
  60. }
  61. if (
  62. !fields.file &&
  63. !fields.Title &&
  64. !fields.Name
  65. ) {
  66. return null;
  67. }
  68. const display = songDisplay(fields);
  69. return {
  70. id: [
  71. fields.file ?? '',
  72. fields.Artist ?? '',
  73. fields.Title ?? '',
  74. fields.Name ?? '',
  75. ].join('\u0000'),
  76. display,
  77. artist: fields.Artist ?? null,
  78. title: fields.Title ?? null,
  79. file: fields.file ?? null,
  80. };
  81. };
  82. const getCurrentSong = () =>
  83. new Promise((resolve, reject) => {
  84. const socket = net.createConnection({
  85. host: MPD_HOST,
  86. port: MPD_PORT,
  87. });
  88. socket.setEncoding('utf8');
  89. socket.setTimeout(MPD_TIMEOUT_MS);
  90. let buffer = '';
  91. let commandSent = false;
  92. let finished = false;
  93. const fail = (error) => {
  94. if (finished) {
  95. return;
  96. }
  97. finished = true;
  98. socket.destroy();
  99. reject(error);
  100. };
  101. const complete = (song) => {
  102. if (finished) {
  103. return;
  104. }
  105. finished = true;
  106. socket.end();
  107. resolve(song);
  108. };
  109. socket.on('timeout', () => {
  110. fail(
  111. new Error(
  112. `MPD did not respond within ${MPD_TIMEOUT_MS} ms`,
  113. ),
  114. );
  115. });
  116. socket.on('error', fail);
  117. socket.on('data', (chunk) => {
  118. buffer += chunk;
  119. if (!commandSent) {
  120. const newline = buffer.indexOf('\n');
  121. if (newline === -1) {
  122. return;
  123. }
  124. const greeting =
  125. buffer.slice(0, newline).trim();
  126. if (!greeting.startsWith('OK MPD ')) {
  127. fail(
  128. new Error(
  129. `Invalid MPD greeting: ${greeting}`,
  130. ),
  131. );
  132. return;
  133. }
  134. buffer = buffer.slice(newline + 1);
  135. commandSent = true;
  136. socket.write('currentsong\n');
  137. }
  138. if (
  139. buffer === 'OK\n' ||
  140. buffer.endsWith('\nOK\n')
  141. ) {
  142. try {
  143. complete(parseSong(buffer));
  144. } catch (error) {
  145. fail(error);
  146. }
  147. }
  148. if (buffer.includes('\nACK ')) {
  149. fail(new Error(buffer.trim()));
  150. }
  151. });
  152. socket.on('close', () => {
  153. if (!finished) {
  154. fail(
  155. new Error(
  156. 'MPD connection closed unexpectedly',
  157. ),
  158. );
  159. }
  160. });
  161. });
  162. const pollMPD = async () => {
  163. if (pollRunning) {
  164. return;
  165. }
  166. pollRunning = true;
  167. try {
  168. const nextSong = await getCurrentSong();
  169. const currentId =
  170. state.current?.id ?? null;
  171. const nextId =
  172. nextSong?.id ?? null;
  173. if (currentId !== nextId) {
  174. if (state.current) {
  175. state.previous = [
  176. {
  177. display: state.current.display,
  178. artist: state.current.artist,
  179. title: state.current.title,
  180. file: state.current.file,
  181. },
  182. ...state.previous,
  183. ].slice(0, 5);
  184. }
  185. state.current = nextSong;
  186. }
  187. state.available = true;
  188. state.error = null;
  189. state.updatedAt =
  190. new Date().toISOString();
  191. } catch (error) {
  192. state.available = false;
  193. state.error =
  194. error instanceof Error
  195. ? error.message
  196. : String(error);
  197. state.updatedAt =
  198. new Date().toISOString();
  199. } finally {
  200. pollRunning = false;
  201. }
  202. };
  203. await pollMPD();
  204. setInterval(
  205. pollMPD,
  206. POLL_INTERVAL_MS,
  207. );
  208. const server = https.createServer(
  209. {
  210. cert: fs.readFileSync(TLS_CERT_FILE),
  211. key: fs.readFileSync(TLS_KEY_FILE),
  212. },
  213. (req, res) => {
  214. res.setHeader(
  215. 'Content-Type',
  216. 'application/json',
  217. );
  218. if (req.url === '/healthz') {
  219. res.writeHead(200);
  220. res.end(
  221. JSON.stringify({
  222. status: 'ok',
  223. }),
  224. );
  225. return;
  226. }
  227. if (req.url === '/api/status') {
  228. const current = state.current
  229. ? {
  230. display: state.current.display,
  231. artist: state.current.artist,
  232. title: state.current.title,
  233. file: state.current.file,
  234. }
  235. : null;
  236. res.writeHead(200);
  237. res.end(
  238. JSON.stringify({
  239. available: state.available,
  240. current,
  241. previous: state.previous,
  242. error: state.error,
  243. updatedAt: state.updatedAt,
  244. }),
  245. );
  246. return;
  247. }
  248. res.writeHead(404);
  249. res.end(
  250. JSON.stringify({
  251. error: 'Not found',
  252. }),
  253. );
  254. },
  255. );
  256. server.listen(9443, '0.0.0.0', () => {
  257. console.log(
  258. `MPD status server listening on :9443, MPD=${MPD_HOST}:${MPD_PORT}`,
  259. );
  260. });