stagelinq 1.0.0

package/.gitattributes

@@ -0,0 +1,2 @@
+# Set the default behavior, in case people don't have core.autocrlf set.
+* text=auto eol=lf

package/.prettierignore

@@ -0,0 +1,7 @@
+www/
+node_modules/
+.DS_Store
+.yarn
+.github
+*.md
+*.yml

package/.prettierrc.js

@@ -0,0 +1,10 @@
+module.exports = {
+	trailingComma: 'es5',
+	tabWidth: 2,
+	semi: true,
+	singleQuote: true,
+	useTabs: false,
+	bracketSpacing: true,
+	endOfLine: 'lf',
+	printWidth: 120,
+};

package/.vscode/launch.json

@@ -0,0 +1,40 @@
+{
+	// Use IntelliSense to learn about possible attributes.
+	// Hover to view descriptions of existing attributes.
+	// For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
+	"version": "0.2.0",
+	"configurations": [
+		{
+			"type": "node",
+			"request": "launch",
+			"name": "Launch JS",
+			"program": "${workspaceFolder}/dist/cli.js",
+			"internalConsoleOptions": "openOnSessionStart",
+			"sourceMaps": true,
+			"smartStep": true,
+			"outputCapture": "std",
+			"skipFiles": [
+				"${workspaceFolder}/node_modules/**/*.js",
+				"${workspaceFolder}/lib/**/*.js",
+				"<node_internals>/**/*.js"
+			]
+		},
+		{
+			"type": "node",
+			"request": "launch",
+			"name": "Launch TS",
+			"program": "${workspaceFolder}/cli/index.ts",
+			"preLaunchTask": "tsc: build - tsconfig.json",
+			"internalConsoleOptions": "openOnSessionStart",
+			"outFiles": ["${workspaceFolder}/dist/**/*.js"],
+			"sourceMaps": true,
+			"smartStep": true,
+			"outputCapture": "std",
+			"skipFiles": [
+				"${workspaceFolder}/node_modules/**/*.js",
+				"${workspaceFolder}/lib/**/*.js",
+				"<node_internals>/**/*.js"
+			]
+		}
+	]
+}

package/.vscode/tasks.json

@@ -0,0 +1,24 @@
+{
+	"version": "2.0.0",
+	"tasks": [
+		{
+			"type": "typescript",
+			"tsconfig": "tsconfig.json",
+			"option": "watch",
+			"presentation": {
+				"echo": true,
+				"reveal": "silent",
+				"focus": false,
+				"panel": "shared"
+			},
+			"isBackground": true,
+			"runOptions": { "runOn": "folderOpen" },
+			"problemMatcher": ["$tsc-watch"],
+			"group": {
+				"kind": "build",
+				"isDefault": true
+			},
+			"label": "tsc: watch - tsconfig.json"
+		}
+	]
+}

package/Databases/Databases.ts

@@ -0,0 +1,69 @@
+import { EventEmitter } from 'stream';
+import { FileTransfer } from '../services';
+import { getTempFilePath } from '../albumArt';
+import { NetworkDevice } from '../network';
+import * as fs from 'fs';
+import { Logger } from '../LogEmitter';
+import { ConnectionInfo, Source } from '../types';
+
+export declare interface Databases {
+  on(event: 'dbDownloaded', listener: (sourceName: string, dbPath: string) => void): this;
+  on(event: 'dbDownloading', listener: (sourceName: string, dbPath: string) => void): this;
+  on(event: 'dbProgress', listener: (sourceName: string, total: number, bytesDownloaded: number, percentComplete: number) => void): this;
+}
+
+export class Databases extends EventEmitter {
+
+  sources: Map<string, string> = new Map();
+
+  constructor() {
+    super();
+  }
+
+  async downloadSourcesFromDevice(connectionInfo: ConnectionInfo, networkDevice: NetworkDevice) {
+    const service = await networkDevice.connectToService(FileTransfer);
+    const sources = await service.getSources();
+    const output: string[] = [];
+    for (const source of sources) {
+      const dbConnectionName = `${connectionInfo.address}_${connectionInfo.port}_${source.name}`;
+      if (this.sources.has(dbConnectionName)) {
+        Logger.debug(`Already seen ${source} on ${connectionInfo.address}:${connectionInfo.port}`);
+      } else {
+        await this.downloadDb(dbConnectionName, service, source);
+        output.push(dbConnectionName);
+      }
+    }
+    return output;
+  }
+
+  /**
+   * Download databases from this network source.
+   */
+  async downloadDb(sourceId: string, service: FileTransfer, source: Source) {
+    const dbPath = getTempFilePath(`${sourceId}/m.db`);
+
+    // Read database from source
+    Logger.debug(`Reading database ${sourceId}`);
+    this.emit('dbDownloading', sourceId, dbPath);
+
+    service.on('fileTransferProgress', (progress) => {
+      this.emit('dbProgress', sourceId, progress.total, progress.bytesDownloaded, progress.percentComplete);
+    });
+
+    // Save database to a file
+    const file = await service.getFile(source.database.location);
+    Logger.debug(`Saving ${sourceId} to ${dbPath}`);
+    fs.writeFileSync(dbPath, Buffer.from(file));
+    this.sources.set(sourceId, dbPath);
+
+    Logger.debug(`Downloaded ${sourceId} to ${dbPath}`);
+    this.emit('dbDownloaded', sourceId, dbPath);
+  }
+
+  getDbPath(dbSourceName: string) {
+    if (!this.sources.has(dbSourceName))
+      throw new Error(`Database name ${dbSourceName} does not exist`);
+    return this.sources.get(dbSourceName);
+  }
+
+}

package/Databases/DbConnection.ts

@@ -0,0 +1,42 @@
+import Database = require('better-sqlite3');
+
+
+export class DbConnection {
+
+  private db: Database.Database;
+  private dbPath: string;
+
+  constructor(dbPath: string) {
+    this.dbPath = dbPath;
+    console.debug(`Opening ${this.dbPath}`);
+    this.db = new Database(this.dbPath);
+  }
+
+  /**
+   * Execute a SQL query.
+   *
+   * @param query SQL query to execute
+   * @param params Parameters for BetterSqlite3 result.all.
+   * @returns
+   */
+  querySource<T>(query: string, ...params: any[]): T[] {
+    console.debug(`Querying ${this.dbPath}: `);
+    const result = this.db.prepare(query);
+    return result.all(params);
+  }
+
+  /**
+   * Return track's DB entry.
+   *
+   * @param trackPath Path of track on the source's filesystem.
+   * @returns
+   */
+  getTrackInfo(trackPath: string) {
+    return this.querySource(`SELECT * FROM Track WHERE path = '${trackPath}'`);
+  }
+
+  close() {
+    console.debug(`Closing ${this.dbPath}`);
+    this.db.close();
+  }
+}

package/Databases/index.ts

@@ -0,0 +1,2 @@
+export * from './Databases';
+export * from './DbConnection';