prompt.con commit completion: add --set-upstream-to and --unset-upstream (04308e9)
   1#include "cache.h"
   2#include "run-command.h"
   3#include "strbuf.h"
   4#include "prompt.h"
   5#include "compat/terminal.h"
   6
   7static char *do_askpass(const char *cmd, const char *prompt)
   8{
   9        struct child_process pass;
  10        const char *args[3];
  11        static struct strbuf buffer = STRBUF_INIT;
  12        int err = 0;
  13
  14        args[0] = cmd;
  15        args[1] = prompt;
  16        args[2] = NULL;
  17
  18        memset(&pass, 0, sizeof(pass));
  19        pass.argv = args;
  20        pass.out = -1;
  21
  22        if (start_command(&pass))
  23                return NULL;
  24
  25        if (strbuf_read(&buffer, pass.out, 20) < 0)
  26                err = 1;
  27
  28        close(pass.out);
  29
  30        if (finish_command(&pass))
  31                err = 1;
  32
  33        if (err) {
  34                error("unable to read askpass response from '%s'", cmd);
  35                strbuf_release(&buffer);
  36                return NULL;
  37        }
  38
  39        strbuf_setlen(&buffer, strcspn(buffer.buf, "\r\n"));
  40
  41        return strbuf_detach(&buffer, NULL);
  42}
  43
  44char *git_prompt(const char *prompt, int flags)
  45{
  46        char *r = NULL;
  47
  48        if (flags & PROMPT_ASKPASS) {
  49                const char *askpass;
  50
  51                askpass = getenv("GIT_ASKPASS");
  52                if (!askpass)
  53                        askpass = askpass_program;
  54                if (!askpass)
  55                        askpass = getenv("SSH_ASKPASS");
  56                if (askpass && *askpass)
  57                        r = do_askpass(askpass, prompt);
  58        }
  59
  60        if (!r)
  61                r = git_terminal_prompt(prompt, flags & PROMPT_ECHO);
  62        if (!r) {
  63                /* prompts already contain ": " at the end */
  64                die("could not read %s%s", prompt, strerror(errno));
  65        }
  66        return r;
  67}
  68
  69char *git_getpass(const char *prompt)
  70{
  71        return git_prompt(prompt, PROMPT_ASKPASS);
  72}