Expand VS Code generated SSH config paths

vscode-extension/extension.js

@@ -84,7 +84,11 @@ async function configureHost() {
   });
   const sshDir = path.join(os.homedir(), '.ssh');
   fs.mkdirSync(sshDir, { recursive: true });
-  const generatedPath = config.get('generatedSshConfig') || path.join(sshDir, 'config.dosh');
+  const generatedPath = expandHomePath(
+    config.get('generatedSshConfig') || path.join(sshDir, 'config.dosh'),
+    os.homedir(),
+    process.platform
+  );
   const mainConfig = path.join(sshDir, 'config');
   ensureInclude(mainConfig, sshIncludeTarget(mainConfig, generatedPath, process.platform));
   upsertBlock(generatedPath, alias, block);
@@ -116,6 +120,10 @@ function sshBlock(options) {
 }
 
 function ensureInclude(configPath, includeFile) {
+  const dir = path.dirname(configPath);
+  if (dir && dir !== '.') {
+    fs.mkdirSync(dir, { recursive: true });
+  }
   const raw = fs.existsSync(configPath) ? fs.readFileSync(configPath, 'utf8') : '';
   if (raw.split(/\r?\n/).some((line) => line.trim().toLowerCase() === `include ${includeFile}`.toLowerCase())) {
     return;
@@ -136,6 +144,10 @@ function sshIncludeTarget(mainConfigPath, generatedPath, platform = process.plat
 }
 
 function upsertBlock(configPath, alias, block) {
+  const dir = path.dirname(configPath);
+  if (dir && dir !== '.') {
+    fs.mkdirSync(dir, { recursive: true });
+  }
   const raw = fs.existsSync(configPath) ? fs.readFileSync(configPath, 'utf8') : '';
   const begin = `# BEGIN DOSH ${alias}`;
   const end = `# END DOSH ${alias}`;
@@ -173,7 +185,11 @@ async function showSshConfig() {
   const api = requireVscode();
   const config = api.workspace.getConfiguration('dosh');
   const sshDir = path.join(os.homedir(), '.ssh');
-  const generatedPath = config.get('generatedSshConfig') || path.join(sshDir, 'config.dosh');
+  const generatedPath = expandHomePath(
+    config.get('generatedSshConfig') || path.join(sshDir, 'config.dosh'),
+    os.homedir(),
+    process.platform
+  );
   if (!fs.existsSync(generatedPath)) {
     api.window.showWarningMessage('No generated Dosh SSH config yet.');
     return;
@@ -213,6 +229,17 @@ function normalizeCase(value, platform) {
   return platform === 'win32' ? value.toLowerCase() : value;
 }
 
+function expandHomePath(value, home = os.homedir(), platform = process.platform) {
+  if (value === '~') {
+    return home;
+  }
+  if (value.startsWith('~/') || value.startsWith('~\\')) {
+    const pathApi = platform === 'win32' ? path.win32 : path;
+    return pathApi.join(home, value.slice(2));
+  }
+  return value;
+}
+
 function windowsCommandWord(value) {
   let out = '"';
   let backslashes = 0;
@@ -241,10 +268,13 @@ function deactivate() {}
 module.exports = {
   activate,
   deactivate,
+  expandHomePath,
+  ensureInclude,
   sshBlock,
   sshConfigPathWord,
   shellQuote,
   sshIncludeTarget,
   sshConfigWord,
+  upsertBlock,
   windowsCommandWord
 };