now-cli/lib/git.js

// Native
const path = require("path");
const url = require("url");
const childProcess = require("child_process");

// Packages
const fs = require("fs-promise");
const download = require("download");
const tmp = require("tmp-promise");
const isURL = require("is-url");

const cloneRepo = (parts, tmpDir) =>
  new Promise((resolve, reject) => {
    let host;

    switch (parts.type) {
      case "GitLab":
        host = `gitlab.com`;
        break;
      case "Bitbucket":
        host = `bitbucket.org`;
        break;
      default:
        host = `github.com`;
    }

    const url = `https://${host}/${parts.main}`;
    const ref = parts.ref ||
      (parts.type === "Bitbucket" ? "default" : "master");
    const cmd = `git clone ${url} --single-branch ${ref}`;

    childProcess.exec(cmd, { cwd: tmpDir.path }, (err, stdout) => {
      if (err) {
        reject(err);
      }

      resolve(stdout);
    });
  });

const renameRepoDir = async (pathParts, tmpDir) => {
  const tmpContents = await fs.readdir(tmpDir.path);

  const oldTemp = path.join(tmpDir.path, tmpContents[0]);
  const newTemp = path.join(tmpDir.path, pathParts.main.replace("/", "-"));

  await fs.rename(oldTemp, newTemp);
  tmpDir.path = newTemp;

  return tmpDir;
};

const downloadRepo = async repoPath => {
  const pathParts = gitPathParts(repoPath);

  const tmpDir = await tmp.dir({
    // We'll remove it manually once deployment is done
    keep: true,
    // Recursively remove directory when calling respective method
    unsafeCleanup: true
  });

  let gitInstalled = true;

  try {
    await cloneRepo(pathParts, tmpDir);
  } catch (err) {
    gitInstalled = false;
  }

  if (gitInstalled) {
    return await renameRepoDir(pathParts, tmpDir);
  }

  let url;

  switch (pathParts.type) {
    case "GitLab": {
      const ref = pathParts.ref ? `?ref=${pathParts.ref}` : "";
      url = `https://gitlab.com/${pathParts.main}/repository/archive.tar` + ref;
      break;
    }
    case "Bitbucket":
      url = `https://bitbucket.org/${pathParts.main}/get/${pathParts.ref || "default"}.zip`;
      break;
    default:
      url = `https://api.github.com/repos/${pathParts.main}/tarball/${pathParts.ref}`;
  }

  try {
    await download(url, tmpDir.path, {
      extract: true
    });
  } catch (err) {
    tmpDir.cleanup();
    return false;
  }

  return await renameRepoDir(pathParts, tmpDir);
};

const capitalizePlatform = name => {
  const names = {
    github: "GitHub",
    gitlab: "GitLab",
    bitbucket: "Bitbucket"
  };

  return names[name];
};

const splittedURL = fullURL => {
  const parsedURL = url.parse(fullURL);
  const pathParts = parsedURL.path.split("/");

  pathParts.shift();

  // Set path to repo...
  const main = pathParts[0] + "/" + pathParts[1];

  // ...and then remove it from the parts
  pathParts.splice(0, 2);

  // Assign Git reference
  let ref = pathParts.length >= 2 ? pathParts[1] : "";

  // Firstly be sure that we haven know the ref type
  if (pathParts[0]) {
    // Then shorten the SHA of the commit
    if (pathParts[0] === "commit" || pathParts[0] === "commits") {
      ref = ref.substring(0, 7);
    }
  }

  // We're deploying master by default,
  // so there's no need to indicate it explicitly
  if (ref === "master") {
    ref = "";
  }

  return {
    main,
    ref,
    type: capitalizePlatform(parsedURL.host.split(".")[0])
  };
};

const gitPathParts = main => {
  let ref = "";

  if (isURL(main)) {
    return splittedURL(main);
  }

  if (main.split("/")[1].includes("#")) {
    const parts = main.split("#");

    ref = parts[1];
    main = parts[0];
  }

  return {
    main,
    ref,
    type: capitalizePlatform("github")
  };
};

const isRepoPath = path => {
  if (!path) {
    return false;
  }

  const allowedHosts = ["github.com", "gitlab.com", "bitbucket.org"];

  if (isURL(path)) {
    const urlParts = url.parse(path);
    const slashSplitted = urlParts.path.split("/").filter(n => n);
    const notBare = slashSplitted.length >= 2;

    if (allowedHosts.includes(urlParts.host) && notBare) {
      return true;
    }

    return "no-valid-url";
  }

  return /[^\s\\]\/[^\s\\]/g.test(path);
};

const fromGit = async (path, debug) => {
  let tmpDir = false;

  try {
    tmpDir = await downloadRepo(path);
  } catch (err) {
    if (debug) {
      console.log(`Could not download "${path}" repo from GitHub`);
    }
  }

  return tmpDir;
};

module.exports = {
  gitPathParts,
  isRepoPath,
  fromGit
};