/*
 * "git add" builtin command
 *
 * Copyright (C) 2006 Linus Torvalds
 */
#include "cache.h"
#include "builtin.h"
#include "dir.h"
#include "exec_cmd.h"
#include "cache-tree.h"
#include "run-command.h"
#include "parse-options.h"

static const char * const builtin_add_usage[] = {
        "git add [options] [--] <filepattern>...",
        NULL
};
static int patch_interactive = 0, add_interactive = 0;
static int take_worktree_changes;

static void prune_directory(struct dir_struct *dir, const char **pathspec, int prefix)
{
        char *seen;
        int i, specs;
        struct dir_entry **src, **dst;

        for (specs = 0; pathspec[specs];  specs++)
                /* nothing */;
        seen = xcalloc(specs, 1);

        src = dst = dir->entries;
        i = dir->nr;
        while (--i >= 0) {
                struct dir_entry *entry = *src++;
                if (match_pathspec(pathspec, entry->name, entry->len,
                                   prefix, seen))
                        *dst++ = entry;
        }
        dir->nr = dst - dir->entries;

        for (i = 0; i < specs; i++) {
                if (!seen[i] && !file_exists(pathspec[i]))
                        die("pathspec '%s' did not match any files",
                                        pathspec[i]);
        }
        free(seen);
}

static void fill_directory(struct dir_struct *dir, const char **pathspec,
                int ignored_too)
{
        const char *path, *base;
        int baselen;

        /* Set up the default git porcelain excludes */
        memset(dir, 0, sizeof(*dir));
        if (!ignored_too) {
                dir->collect_ignored = 1;
                setup_standard_excludes(dir);
        }

        /*
         * Calculate common prefix for the pathspec, and
         * use that to optimize the directory walk
         */
        baselen = common_prefix(pathspec);
        path = ".";
        base = "";
        if (baselen)
                path = base = xmemdupz(*pathspec, baselen);

        /* Read the directory and prune it */
        read_directory(dir, path, base, baselen, pathspec);
        if (pathspec)
                prune_directory(dir, pathspec, baselen);
}

static void refresh(int verbose, const char **pathspec)
{
        char *seen;
        int i, specs;

        for (specs = 0; pathspec[specs];  specs++)
                /* nothing */;
        seen = xcalloc(specs, 1);
        refresh_index(&the_index, verbose ? REFRESH_SAY_CHANGED : REFRESH_QUIET,
                      pathspec, seen);
        for (i = 0; i < specs; i++) {
                if (!seen[i])
                        die("pathspec '%s' did not match any files", pathspec[i]);
        }
        free(seen);
}

static const char **validate_pathspec(int argc, const char **argv, const char *prefix)
{
        const char **pathspec = get_pathspec(prefix, argv);

        return pathspec;
}

int interactive_add(int argc, const char **argv, const char *prefix)
{
        int status, ac;
        const char **args;
        const char **pathspec = NULL;

        if (argc) {
                pathspec = validate_pathspec(argc, argv, prefix);
                if (!pathspec)
                        return -1;
        }

        args = xcalloc(sizeof(const char *), (argc + 4));
        ac = 0;
        args[ac++] = "add--interactive";
        if (patch_interactive)
                args[ac++] = "--patch";
        args[ac++] = "--";
        if (argc) {
                memcpy(&(args[ac]), pathspec, sizeof(const char *) * argc);
                ac += argc;
        }
        args[ac] = NULL;

        status = run_command_v_opt(args, RUN_GIT_CMD);
        free(args);
        return status;
}

static struct lock_file lock_file;

static const char ignore_error[] =
"The following paths are ignored by one of your .gitignore files:\n";

static int verbose = 0, show_only = 0, ignored_too = 0, refresh_only = 0;
static int ignore_add_errors, addremove;

static struct option builtin_add_options[] = {
        OPT__DRY_RUN(&show_only),
        OPT__VERBOSE(&verbose),
        OPT_GROUP(""),
        OPT_BOOLEAN('i', "interactive", &add_interactive, "interactive picking"),
        OPT_BOOLEAN('p', "patch", &patch_interactive, "interactive patching"),
        OPT_BOOLEAN('f', "force", &ignored_too, "allow adding otherwise ignored files"),
        OPT_BOOLEAN('u', "update", &take_worktree_changes, "update tracked files"),
        OPT_BOOLEAN('A', "all", &addremove, "add all, noticing removal of tracked files"),
        OPT_BOOLEAN( 0 , "refresh", &refresh_only, "don't add, only refresh the index"),
        OPT_BOOLEAN( 0 , "ignore-errors", &ignore_add_errors, "just skip files which cannot be added because of errors"),
        OPT_END(),
};

static int add_config(const char *var, const char *value, void *cb)
{
        if (!strcasecmp(var, "add.ignore-errors")) {
                ignore_add_errors = git_config_bool(var, value);
                return 0;
        }
        return git_default_config(var, value, cb);
}

static int add_files(struct dir_struct *dir, int flags)
{
        int i, exit_status = 0;

        if (dir->ignored_nr) {
                fprintf(stderr, ignore_error);
                for (i = 0; i < dir->ignored_nr; i++)
                        fprintf(stderr, "%s\n", dir->ignored[i]->name);
                fprintf(stderr, "Use -f if you really want to add them.\n");
                die("no files added");
        }

        for (i = 0; i < dir->nr; i++)
                if (add_file_to_cache(dir->entries[i]->name, flags)) {
                        if (!ignore_add_errors)
                                die("adding files failed");
                        exit_status = 1;
                }
        return exit_status;
}

int cmd_add(int argc, const char **argv, const char *prefix)
{
        int exit_status = 0;
        int newfd;
        const char **pathspec;
        struct dir_struct dir;
        int flags;
        int add_new_files;
        int require_pathspec;

        argc = parse_options(argc, argv, builtin_add_options,
                          builtin_add_usage, 0);
        if (patch_interactive)
                add_interactive = 1;
        if (add_interactive)
                exit(interactive_add(argc, argv, prefix));

        git_config(add_config, NULL);

        if (addremove && take_worktree_changes)
                die("-A and -u are mutually incompatible");
        if (addremove && !argc) {
                static const char *here[2] = { ".", NULL };
                argc = 1;
                argv = here;
        }

        add_new_files = !take_worktree_changes && !refresh_only;
        require_pathspec = !take_worktree_changes;

        newfd = hold_locked_index(&lock_file, 1);

        flags = ((verbose ? ADD_CACHE_VERBOSE : 0) |
                 (show_only ? ADD_CACHE_PRETEND : 0) |
                 (ignore_add_errors ? ADD_CACHE_IGNORE_ERRORS : 0));

        if (require_pathspec && argc == 0) {
                fprintf(stderr, "Nothing specified, nothing added.\n");
                fprintf(stderr, "Maybe you wanted to say 'git add .'?\n");
                return 0;
        }
        pathspec = get_pathspec(prefix, argv);

        /*
         * If we are adding new files, we need to scan the working
         * tree to find the ones that match pathspecs; this needs
         * to be done before we read the index.
         */
        if (add_new_files)
                fill_directory(&dir, pathspec, ignored_too);

        if (read_cache() < 0)
                die("index file corrupt");

        if (refresh_only) {
                refresh(verbose, pathspec);
                goto finish;
        }

        if (take_worktree_changes || addremove)
                exit_status |= add_files_to_cache(prefix, pathspec, flags);

        if (add_new_files)
                exit_status |= add_files(&dir, flags);

 finish:
        if (active_cache_changed) {
                if (write_cache(newfd, active_cache, active_nr) ||
                    commit_locked_index(&lock_file))
                        die("Unable to write new index file");
        }

        return exit_status;
}