Refactoring: code organization

package.json

@@ -21,7 +21,7 @@
     "onCommand:extension.elixirRunTestSuite",
     "onCommand:extension.elixirRunTestAtCursor"
   ],
-  "main": "./extension.js",
+  "main": "./src/extension.js",
   "contributes": {
     "commands": [
       {

src/commands/index.js

@@ -0,0 +1,6 @@
+const jumpToTest = require("./jumpToTest");
+const runTestAtCursor = require("./runTestAtCursor");
+const runTestFile = require("./runTestFile");
+const runTestSuite = require("./runTestSuite");
+
+module.exports = [jumpToTest, runTestAtCursor, runTestFile, runTestSuite];

src/commands/jumpToTest.js

@@ -0,0 +1,102 @@
+const vscode = require("vscode");
+
+function handler() {
+  const activeFile = vscode.window.activeTextEditor;
+
+  if (!activeFile) {
+    return;
+  }
+
+  const openedFilename = activeFile.document.fileName;
+  const isCodeFile = /(.*(\/.*\/))(.*)(\.\w+)$/;
+  const openedFile = openedFilename.match(isCodeFile);
+
+  if (!openedFile) {
+    return;
+  }
+
+  const fullDir = openedFile[1];
+  const context =
+    openedFile[2] == "/lib/"
+      ? "/test/"
+      : openedFile[2].replace("/test/", "/lib/");
+  const fileName = openedFile[3];
+  const isTestFile = fileName.includes("_test");
+
+  if (isTestFile === true) {
+    var strippedFileName = fileName.replace("_test", "");
+    var fileToOpen = `**${context}${strippedFileName}.ex`;
+  } else {
+    var fileToOpen = `**${context}${fileName}_test.exs`;
+  }
+
+  vscode.workspace.findFiles(fileToOpen, "**/.elixir_ls/**").then(files => {
+    if (!files.length) {
+      let dir = fullDir.replace("lib", "test");
+      let file = `${fileName}_test.exs`;
+
+      askToCreateANewFile(dir, file);
+    } else {
+      let file = files[0].fsPath;
+      openFile(file);
+    }
+  });
+}
+
+function askToCreateANewFile(dir, file) {
+  return showConfirmationDialog(
+    `Create the test file at ${dir}?`,
+    "Create"
+  ).then(answer => {
+    if (answer == "Create") {
+      createNewTestFile(dir, file).then(() => {
+        openFile(`${dir}${file}`);
+      });
+    } else {
+      return;
+    }
+  });
+}
+
+function showConfirmationDialog(text, button) {
+  return vscode.window.showWarningMessage(text, { modal: true }, button);
+}
+
+function openFile(file) {
+  return vscode.workspace
+    .openTextDocument(vscode.Uri.file(file))
+    .then(vscode.window.showTextDocument);
+}
+
+async function createNewTestFile(dir, file) {
+  const uriDir = vscode.Uri.file(dir);
+  const uriFile = vscode.Uri.file(`${dir}${file}`);
+  const ws = new vscode.WorkspaceEdit();
+
+  const originalFile = vscode.window.activeTextEditor.document.fileName;
+  const originalFileUri = vscode.Uri.file(originalFile);
+  const moduleName = await getModuleName(originalFileUri);
+
+  await vscode.workspace.fs.createDirectory(uriDir);
+
+  ws.createFile(uriFile);
+  ws.insert(
+    uriFile,
+    new vscode.Position(0, 0),
+    `defmodule ${moduleName}Test do\nend`
+  );
+
+  await vscode.workspace.applyEdit(ws);
+}
+
+async function getModuleName(uriFile) {
+  const content = (await vscode.workspace.fs.readFile(uriFile)).toString();
+  const moduleDefinition = content.match(/defmodule (.*) do/);
+
+  return moduleDefinition[1];
+}
+
+module.exports = {
+  name: "extension.elixirJumpToTest",
+  handler
+};

src/commands/runTestAtCursor.js

@@ -0,0 +1,37 @@
+const vscode = require("vscode");
+
+function handler() {
+  const activeFile = vscode.window.activeTextEditor;
+  if (!activeFile) {
+    return;
+  }
+
+  const openedFilename = activeFile.document.fileName;
+  /* 
+  We do a +1 here because the `line` returned is zero based.
+  Ref: https://code.visualstudio.com/api/references/vscode-api#Position
+  */
+  const cursorLine = activeFile.selection.active.line + 1;
+
+  const isTestFile = openedFilename.includes("_test.exs");
+  const isUmbrella = openedFilename.includes("/apps/");
+
+  if (isTestFile === true) {
+    const testPathFilter = isUmbrella ? /.*\/(apps\/.*)$/ : /.*\/(test\/.*)$/;
+    let terminal =
+      vscode.window.activeTerminal || vscode.window.createTerminal();
+    terminal.sendText(
+      `mix test ${openedFilename.match(testPathFilter)[1]}:${cursorLine}`
+    );
+    terminal.show();
+  } else {
+    vscode.window.showInformationMessage(
+      `The current file is not a test file.`
+    );
+  }
+}
+
+module.exports = {
+  name: "extension.elixirRunTestAtCursor",
+  handler
+};

src/commands/runTestFile.js

@@ -0,0 +1,30 @@
+const vscode = require("vscode");
+
+function handler() {
+  const activeFile = vscode.window.activeTextEditor;
+  if (!activeFile) {
+    return;
+  }
+
+  const openedFilename = activeFile.document.fileName;
+
+  const isTestFile = openedFilename.includes("_test.exs");
+  const isUmbrella = openedFilename.includes("/apps/");
+
+  if (isTestFile === true) {
+    const testPathFilter = isUmbrella ? /.*\/(apps\/.*)$/ : /.*\/(test\/.*)$/;
+    let terminal =
+      vscode.window.activeTerminal || vscode.window.createTerminal();
+    terminal.sendText(`mix test ${openedFilename.match(testPathFilter)[1]}`);
+    terminal.show();
+  } else {
+    vscode.window.showInformationMessage(
+      `The current file is not a test file.`
+    );
+  }
+}
+
+module.exports = {
+  name: "extension.elixirRunTestFile",
+  handler
+};

src/commands/runTestSuite.js

@@ -0,0 +1,12 @@
+const vscode = require("vscode");
+
+function handler() {
+  let terminal = vscode.window.activeTerminal || vscode.window.createTerminal();
+  terminal.sendText(`mix test`);
+  terminal.show();
+}
+
+module.exports = {
+  name: "extension.elixirRunTestSuite",
+  handler
+};

src/extension.js

@@ -0,0 +1,19 @@
+const vscode = require("vscode");
+const commands = require("./commands");
+
+function activate(context) {
+  commands.forEach(command => {
+    const disposable = vscode.commands.registerCommand(
+      command.name,
+      command.handler
+    );
+    context.subscriptions.push(disposable);
+  });
+}
+
+function deactivate() {}
+
+module.exports = {
+  activate,
+  deactivate
+};