Code

Move struct io's argv and dir members to struct view
[tig.git] / tig.c
diff --git a/tig.c b/tig.c
index 46f289ec1335debbcff6bc0ba014455bf7d06c77..3121a3a3847e9ab2deb45a5d27ec6c2990d7af2f 100644 (file)
--- a/tig.c
+++ b/tig.c
@@ -140,13 +140,6 @@ static struct ref_list *get_ref_list(const char *id);
 static void foreach_ref(bool (*visitor)(void *data, const struct ref *ref), void *data);
 static int load_refs(void);
 
-enum format_flags {
-       FORMAT_ALL,             /* Perform replacement in all arguments. */
-       FORMAT_NONE             /* No replacement should be performed. */
-};
-
-static bool format_argv(const char *dst[], const char *src[], enum format_flags flags);
-
 enum input_status {
        INPUT_OK,
        INPUT_SKIP,
@@ -685,15 +678,18 @@ argv_free(const char *argv[])
 
        for (argc = 0; argv[argc]; argc++)
                free((void *) argv[argc]);
+       argv[0] = NULL;
 }
 
-static void
-argv_copy(const char *dst[], const char *src[])
+static bool
+argv_copy(const char *dst[], const char *src[], bool allocate)
 {
        int argc;
 
        for (argc = 0; src[argc]; argc++)
-               dst[argc] = src[argc];
+               if (!(dst[argc] = allocate ? strdup(src[argc]) : src[argc]))
+                       return FALSE;
+       return TRUE;
 }
 
 
@@ -711,12 +707,9 @@ enum io_type {
 };
 
 struct io {
-       enum io_type type;      /* The requested type of pipe. */
-       const char *dir;        /* Directory from which to execute. */
        pid_t pid;              /* PID of spawned process. */
        int pipe;               /* Pipe end for reading or writing. */
        int error;              /* Error status. */
-       const char *argv[SIZEOF_ARG];   /* Shell command arguments. */
        char *buf;              /* Read buffer. */
        size_t bufalloc;        /* Allocated buffer size. */
        size_t bufsize;         /* Buffer content size. */
@@ -736,26 +729,9 @@ io_reset(struct io *io)
 }
 
 static void
-io_init(struct io *io, const char *dir, enum io_type type)
+io_init(struct io *io)
 {
        io_reset(io);
-       io->type = type;
-       io->dir = dir;
-}
-
-static void
-io_prepare(struct io *io, const char *dir, enum io_type type, const char *argv[])
-{
-       io_init(io, dir, type);
-       argv_copy(io->argv, argv);
-}
-
-static bool
-io_format(struct io *io, const char *dir, enum io_type type,
-         const char *argv[], enum format_flags flags)
-{
-       io_init(io, dir, type);
-       return format_argv(io->argv, argv, flags);
 }
 
 static bool
@@ -765,7 +741,7 @@ io_open(struct io *io, const char *fmt, ...)
        bool fits;
        va_list args;
 
-       io_init(io, NULL, IO_FD);
+       io_init(io);
 
        va_start(args, fmt);
        fits = vsnprintf(name, sizeof(name), fmt, args) < sizeof(name);
@@ -818,35 +794,32 @@ io_done(struct io *io)
 }
 
 static bool
-io_start(struct io *io)
+io_start(struct io *io, enum io_type type, const char *dir, const char *argv[])
 {
        int pipefds[2] = { -1, -1 };
 
-       if (io->type == IO_FD)
-               return TRUE;
-
-       if ((io->type == IO_RD || io->type == IO_WR) && pipe(pipefds) < 0) {
+       if ((type == IO_RD || type == IO_WR) && pipe(pipefds) < 0) {
                io->error = errno;
                return FALSE;
-       } else if (io->type == IO_AP) {
+       } else if (type == IO_AP) {
                pipefds[1] = io->pipe;
        }
 
        if ((io->pid = fork())) {
                if (io->pid == -1)
                        io->error = errno;
-               if (pipefds[!(io->type == IO_WR)] != -1)
-                       close(pipefds[!(io->type == IO_WR)]);
+               if (pipefds[!(type == IO_WR)] != -1)
+                       close(pipefds[!(type == IO_WR)]);
                if (io->pid != -1) {
-                       io->pipe = pipefds[!!(io->type == IO_WR)];
+                       io->pipe = pipefds[!!(type == IO_WR)];
                        return TRUE;
                }
 
        } else {
-               if (io->type != IO_FG) {
+               if (type != IO_FG) {
                        int devnull = open("/dev/null", O_RDWR);
-                       int readfd  = io->type == IO_WR ? pipefds[0] : devnull;
-                       int writefd = (io->type == IO_RD || io->type == IO_AP)
+                       int readfd  = type == IO_WR ? pipefds[0] : devnull;
+                       int writefd = (type == IO_RD || type == IO_AP)
                                                        ? pipefds[1] : devnull;
 
                        dup2(readfd,  STDIN_FILENO);
@@ -860,63 +833,51 @@ io_start(struct io *io)
                                close(pipefds[1]);
                }
 
-               if (io->dir && *io->dir && chdir(io->dir) == -1)
+               if (dir && *dir && chdir(dir) == -1)
                        exit(errno);
 
-               execvp(io->argv[0], (char *const*) io->argv);
+               execvp(argv[0], (char *const*) argv);
                exit(errno);
        }
 
-       if (pipefds[!!(io->type == IO_WR)] != -1)
-               close(pipefds[!!(io->type == IO_WR)]);
+       if (pipefds[!!(type == IO_WR)] != -1)
+               close(pipefds[!!(type == IO_WR)]);
        return FALSE;
 }
 
 static bool
 io_run(struct io *io, const char **argv, const char *dir, enum io_type type)
 {
-       io_prepare(io, dir, type, argv);
-       return io_start(io);
+       io_init(io);
+       return io_start(io, type, dir, argv);
 }
 
-static int
-io_complete(struct io *io)
-{
-       return io_start(io) && io_done(io);
-}
-
-static int
-io_run_bg(const char **argv)
+static bool
+io_complete(enum io_type type, const char **argv, const char *dir, int fd)
 {
        struct io io = {};
 
-       io_prepare(&io, NULL, IO_BG, argv);
-       return io_complete(&io);
+       io_init(&io);
+       io.pipe = fd;
+       return io_start(&io, type, dir, argv) && io_done(&io);
 }
 
 static bool
-io_run_fg(const char **argv, const char *dir)
+io_run_bg(const char **argv)
 {
-       struct io io = {};
-
-       io_prepare(&io, dir, IO_FG, argv);
-       return io_complete(&io);
+       return io_complete(IO_BG, argv, NULL, -1);
 }
 
 static bool
-io_run_append(const char **argv, int fd)
+io_run_fg(const char **argv, const char *dir)
 {
-       struct io io = {};
-
-       io_prepare(&io, NULL, IO_AP, argv);
-       io.pipe = fd;
-       return io_complete(&io);
+       return io_complete(IO_FG, argv, dir, -1);
 }
 
 static bool
-io_run_rd(struct io *io, const char **argv, const char *dir)
+io_run_append(const char **argv, int fd)
 {
-       return io_format(io, dir, IO_RD, argv, FORMAT_NONE) && io_start(io);
+       return io_complete(IO_AP, argv, NULL, -1);
 }
 
 static bool
@@ -1053,8 +1014,8 @@ io_run_buf(const char **argv, char buf[], size_t bufsize)
 {
        struct io io = {};
 
-       io_prepare(&io, NULL, IO_RD, argv);
-       return io_start(&io) && io_read_buf(&io, buf, bufsize);
+       io_init(&io);
+       return io_start(&io, IO_RD, NULL, argv) && io_read_buf(&io, buf, bufsize);
 }
 
 static int
@@ -1064,9 +1025,6 @@ io_load(struct io *io, const char *separators,
        char *name;
        int state = OK;
 
-       if (!io_start(io))
-               return ERR;
-
        while (state == OK && (name = io_get(io, '\n', TRUE))) {
                char *value;
                size_t namelen;
@@ -1101,7 +1059,9 @@ io_run_load(const char **argv, const char *separators,
 {
        struct io io = {};
 
-       io_prepare(&io, NULL, IO_RD, argv);
+       io_init(&io);
+       if (!io_start(&io, IO_RD, NULL, argv))
+               return ERR;
        return io_load(&io, separators, read_property);
 }
 
@@ -1754,7 +1714,7 @@ add_run_request(enum keymap keymap, int key, int argc, const char **argv)
        req->key = key;
        req->argv[0] = NULL;
 
-       if (!format_argv(req->argv, argv, FORMAT_NONE))
+       if (!argv_copy(req->argv, argv, TRUE))
                return REQ_NONE;
 
        return REQ_NONE + ++run_requests;
@@ -2269,6 +2229,8 @@ struct view {
        bool has_scrolled;      /* View was scrolled. */
 
        /* Loading */
+       const char *argv[SIZEOF_ARG];   /* Shell command arguments. */
+       const char *dir;        /* Directory from which to execute. */
        struct io io;
        struct io *pipe;
        time_t start_time;
@@ -3196,11 +3158,10 @@ format_arg(const char *name)
 }
 
 static bool
-format_argv(const char *dst_argv[], const char *src_argv[], enum format_flags flags)
+format_argv(const char *dst_argv[], const char *src_argv[], bool replace)
 {
        char buf[SIZEOF_STR];
        int argc;
-       bool noreplace = flags == FORMAT_NONE;
 
        argv_free(dst_argv);
 
@@ -3213,7 +3174,7 @@ format_argv(const char *dst_argv[], const char *src_argv[], enum format_flags fl
                        int len = next - arg;
                        const char *value;
 
-                       if (!next || noreplace) {
+                       if (!next || !replace) {
                                len = strlen(arg);
                                value = "";
 
@@ -3228,7 +3189,7 @@ format_argv(const char *dst_argv[], const char *src_argv[], enum format_flags fl
                        if (!string_format_from(buf, &bufpos, "%.*s%s", len, arg, value))
                                return FALSE;
 
-                       arg = next && !noreplace ? strchr(next, ')') + 1 : NULL;
+                       arg = next && replace ? strchr(next, ')') + 1 : NULL;
                }
 
                dst_argv[argc] = strdup(buf);
@@ -3287,12 +3248,29 @@ setup_update(struct view *view, const char *vid)
        view->start_time = time(NULL);
 }
 
+static bool
+prepare_io(struct view *view, const char *dir, const char *argv[], bool replace)
+{
+       io_init(&view->io);
+       view->dir = dir;
+       return format_argv(view->argv, argv, replace);
+}
+
 static bool
 prepare_update(struct view *view, const char *argv[], const char *dir)
 {
        if (view->pipe)
                end_update(view, TRUE);
-       return io_format(&view->io, dir, IO_RD, argv, FORMAT_NONE);
+       return prepare_io(view, dir, argv, FALSE);
+}
+
+static bool
+start_update(struct view *view, const char **argv, const char *dir)
+{
+       if (view->pipe)
+               io_done(view->pipe);
+       return prepare_io(view, dir, argv, FALSE) &&
+              io_start(&view->io, IO_RD, dir, view->argv);
 }
 
 static bool
@@ -3300,6 +3278,7 @@ prepare_update_file(struct view *view, const char *name)
 {
        if (view->pipe)
                end_update(view, TRUE);
+       argv_free(view->argv);
        return io_open(&view->io, "%s/%s", opt_cdup[0] ? opt_cdup : ".", name);
 }
 
@@ -3313,7 +3292,7 @@ begin_update(struct view *view, bool refresh)
                if (view->ops->prepare) {
                        if (!view->ops->prepare(view))
                                return FALSE;
-               } else if (!io_format(&view->io, NULL, IO_RD, view->ops->argv, FORMAT_ALL)) {
+               } else if (!prepare_io(view, NULL, view->ops->argv, TRUE)) {
                        return FALSE;
                }
 
@@ -3324,7 +3303,7 @@ begin_update(struct view *view, bool refresh)
                string_copy_rev(view->ref, view->id);
        }
 
-       if (!io_start(&view->io))
+       if (view->argv[0] && !io_start(&view->io, IO_RD, view->dir, view->argv))
                return FALSE;
 
        setup_update(view, view->id);
@@ -3610,7 +3589,7 @@ open_run_request(enum request request)
                return;
        }
 
-       if (format_argv(argv, req->argv, FORMAT_ALL))
+       if (format_argv(argv, req->argv, TRUE))
                open_external_viewer(argv, NULL);
        argv_free(argv);
 }
@@ -4527,7 +4506,6 @@ tree_read_date(struct view *view, char *text, bool *read_date)
                        "git", "log", "--no-color", "--pretty=raw",
                                "--cc", "--raw", view->id, "--", path, NULL
                };
-               struct io io = {};
 
                if (!view->lines) {
                        tree_entry(view, LINE_TREE_HEAD, opt_path, NULL, NULL);
@@ -4535,13 +4513,11 @@ tree_read_date(struct view *view, char *text, bool *read_date)
                        return TRUE;
                }
 
-               if (!io_run_rd(&io, log_file, opt_cdup)) {
+               if (!start_update(view, log_file, opt_cdup)) {
                        report("Failed to load tree data");
                        return TRUE;
                }
 
-               io_done(view->pipe);
-               view->io = io;
                *read_date = TRUE;
                return FALSE;
 
@@ -4825,7 +4801,7 @@ tree_prepare(struct view *view)
                opt_path[0] = 0;
        }
 
-       return io_format(&view->io, opt_cdup, IO_RD, view->ops->argv, FORMAT_ALL);
+       return prepare_io(view, opt_cdup, view->ops->argv, TRUE);
 }
 
 static const char *tree_argv[SIZEOF_ARG] = {
@@ -4922,7 +4898,7 @@ blame_open(struct view *view)
                };
 
                if (!string_format(path, "%s:%s", opt_ref, opt_file) ||
-                   !io_run_rd(&view->io, blame_cat_file_argv, opt_cdup))
+                   !start_update(view, blame_cat_file_argv, opt_cdup))
                        return FALSE;
        }
 
@@ -5016,18 +4992,15 @@ blame_read_file(struct view *view, const char *line, bool *read_file)
                        "git", "blame", "--incremental",
                                *opt_ref ? opt_ref : "--incremental", "--", opt_file, NULL
                };
-               struct io io = {};
 
                if (view->lines == 0 && !view->prev)
                        die("No blame exist for %s", view->vid);
 
-               if (view->lines == 0 || !io_run_rd(&io, blame_argv, opt_cdup)) {
+               if (view->lines == 0 || !start_update(view, blame_argv, opt_cdup)) {
                        report("Failed to load blame data");
                        return TRUE;
                }
 
-               io_done(view->pipe);
-               view->io = io;
                *read_file = FALSE;
                return FALSE;
 
@@ -5455,7 +5428,7 @@ branch_open(struct view *view)
                        "--simplify-by-decoration", "--all", NULL
        };
 
-       if (!io_run_rd(&view->io, branch_log, NULL)) {
+       if (!start_update(view, branch_log, NULL)) {
                report("Failed to load branch data");
                return TRUE;
        }