Merge pull request #23 from aditya7iyengar/feature/askpass-support

Add support for askpass

.gitignore

@@ -22,3 +22,4 @@ erl_crash.dump
 # Ignore package tarball (built via "mix hex.build").
 adify-*.tar
 
+.askpass.sh

lib/adify/system_info.ex

@@ -113,13 +113,51 @@ defmodule Adify.SystemInfo do
       iex> {:error, output}  = Adify.SystemInfo.cmd(cmd, [], ".")
       iex> output =~ "bad_command"
       true
+
+      # When the command has sudo
+      iex> cmd = "sudo ls"
+      iex> {:ok, output} = Adify.SystemInfo.cmd(cmd, [], ".", true)
+      iex> output =~ "password"
+      true
   """
   @spec cmd(String.t(), [{String.t(), String.t()}], Path.t()) ::
           {:ok, String.t()} | {:error, String.t()}
-  def cmd(cmd, env \\ [], cd \\ ".") do
+  def cmd(cmd, env \\ [], cd \\ ".", mock_sudo \\ false) do
+    case String.contains?(cmd, "sudo") do
+      true -> cmd_sudo(cmd, env, cd, mock_sudo)
+      false -> cmd_udo(cmd, env, cd)
+    end
+  end
+
+  defp cmd_sudo(cmd, env, cd, mock_sudo) do
+    create_or_update_askpass()
+
+    cmd = String.replace(cmd, "sudo", "")
+
+    env = env ++ [{"SUDO_ASKPASS", ".askpass.sh"}]
+
+    case mock_sudo do
+      true -> {:ok, "password mocked"}
+      _ ->
+        case System.cmd("sudo", ["-A", "sh", "-c", cmd], env: env, cd: cd, stderr_to_stdout: true) do
+          {output, 0} -> {:ok, output}
+          {output, _} -> {:error, output}
+        end
+    end
+  end
+
+  defp cmd_udo(cmd, env, cd) do
     case System.cmd("sh", ["-c", cmd], env: env, cd: cd, stderr_to_stdout: true) do
       {output, 0} -> {:ok, output}
       {output, _} -> {:error, output}
     end
   end
+
+  defp create_or_update_askpass do
+    askpass_path = Path.join(:code.priv_dir(:adify), "askpass.sh")
+    content = File.read!(askpass_path)
+    cmd("touch .askpass.sh")
+    cmd("echo '#{content}' > .askpass.sh")
+    cmd("chmod 777 .askpass.sh")
+  end
 end

priv/askpass.sh

@@ -0,0 +1,2 @@
+#!/bin/bash
+zenity --password --title=Authentication