Hook new optdepend structures up

[pacman-ng.git] / src / pacman / util.c

/*
 *  util.c
 *
 *  Copyright (c) 2006-2012 Pacman Development Team <pacman-dev@archlinux.org>
 *  Copyright (c) 2002-2006 by Judd Vinet <jvinet@zeroflux.org>
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include <sys/types.h>
#include <sys/ioctl.h>
#include <sys/stat.h>
#include <time.h>

#include <stdio.h>
#include <stdlib.h>
#include <stdarg.h>
#include <stdint.h> /* intmax_t */
#include <string.h>
#include <errno.h>
#include <ctype.h>
#include <dirent.h>
#include <unistd.h>
#include <limits.h>
#include <wchar.h>
#ifdef HAVE_TERMIOS_H
#include <termios.h> /* tcflush */
#endif

#include <alpm.h>
#include <alpm_list.h>

/* pacman */
#include "util.h"
#include "conf.h"
#include "callback.h"


int trans_init(alpm_transflag_t flags, int check_valid)
{
        int ret;

        check_syncdbs(0, check_valid);

        ret = alpm_trans_init(config->handle, flags);
        if(ret == -1) {
                trans_init_error();
                return -1;
        }
        return 0;
}

void trans_init_error(void)
{
        alpm_errno_t err = alpm_errno(config->handle);
        pm_printf(ALPM_LOG_ERROR, _("failed to init transaction (%s)\n"),
                        alpm_strerror(err));
        if(err == ALPM_ERR_HANDLE_LOCK) {
                const char *lockfile = alpm_option_get_lockfile(config->handle);
                pm_printf(ALPM_LOG_ERROR, _("could not lock database: %s\n"),
                                        strerror(errno));
                if(access(lockfile, F_OK) == 0) {
                        fprintf(stderr, _("  if you're sure a package manager is not already\n"
                                                "  running, you can remove %s\n"), lockfile);
                }
        }
}

int trans_release(void)
{
        if(alpm_trans_release(config->handle) == -1) {
                pm_printf(ALPM_LOG_ERROR, _("failed to release transaction (%s)\n"),
                                alpm_strerror(alpm_errno(config->handle)));
                return -1;
        }
        return 0;
}

int needs_root(void)
{
        switch(config->op) {
                case PM_OP_DATABASE:
                        return 1;
                case PM_OP_UPGRADE:
                case PM_OP_REMOVE:
                        return !config->print;
                case PM_OP_SYNC:
                        return (config->op_s_clean || config->op_s_sync ||
                                        (!config->group && !config->op_s_info && !config->op_q_list &&
                                         !config->op_s_search && !config->print));
                default:
                        return 0;
        }
}

int check_syncdbs(size_t need_repos, int check_valid)
{
        int ret = 0;
        alpm_list_t *i;
        alpm_list_t *sync_dbs = alpm_get_syncdbs(config->handle);

        if(need_repos && sync_dbs == NULL) {
                pm_printf(ALPM_LOG_ERROR, _("no usable package repositories configured.\n"));
                return 1;
        }

        if(check_valid) {
                /* ensure all known dbs are valid */
                for(i = sync_dbs; i; i = alpm_list_next(i)) {
                        alpm_db_t *db = i->data;
                        if(alpm_db_get_valid(db)) {
                                pm_printf(ALPM_LOG_ERROR, _("database '%s' is not valid (%s)\n"),
                                                alpm_db_get_name(db), alpm_strerror(alpm_errno(config->handle)));
                                ret = 1;
                        }
                }
        }
        return ret;
}

/* discard unhandled input on the terminal's input buffer */
static int flush_term_input(void) {
#ifdef HAVE_TCFLUSH
        if(isatty(fileno(stdin))) {
                return tcflush(fileno(stdin), TCIFLUSH);
        }
#endif

        /* fail silently */
        return 0;
}

/* gets the current screen column width */
unsigned short getcols(void)
{
        const unsigned short default_tty = 80;
        const unsigned short default_notty = 0;
        unsigned short termwidth = 0;

        if(!isatty(fileno(stdout))) {
                return default_notty;
        }

#ifdef TIOCGSIZE
        struct ttysize win;
        if(ioctl(1, TIOCGSIZE, &win) == 0) {
                termwidth = win.ts_cols;
        }
#elif defined(TIOCGWINSZ)
        struct winsize win;
        if(ioctl(1, TIOCGWINSZ, &win) == 0) {
                termwidth = win.ws_col;
        }
#endif
        return termwidth == 0 ? default_tty : termwidth;
}

/* does the same thing as 'rm -rf' */
int rmrf(const char *path)
{
        int errflag = 0;
        struct dirent *dp;
        DIR *dirp;

        if(!unlink(path)) {
                return 0;
        } else {
                if(errno == ENOENT) {
                        return 0;
                } else if(errno == EPERM) {
                        /* fallthrough */
                } else if(errno == EISDIR) {
                        /* fallthrough */
                } else if(errno == ENOTDIR) {
                        return 1;
                } else {
                        /* not a directory */
                        return 1;
                }

                dirp = opendir(path);
                if(!dirp) {
                        return 1;
                }
                for(dp = readdir(dirp); dp != NULL; dp = readdir(dirp)) {
                        if(dp->d_name) {
                                if(strcmp(dp->d_name, "..") != 0 && strcmp(dp->d_name, ".") != 0) {
                                        char name[PATH_MAX];
                                        snprintf(name, PATH_MAX, "%s/%s", path, dp->d_name);
                                        errflag += rmrf(name);
                                }
                        }
                }
                closedir(dirp);
                if(rmdir(path)) {
                        errflag++;
                }
                return errflag;
        }
}

/** Parse the basename of a program from a path.
* @param path path to parse basename from
*
* @return everything following the final '/'
*/
const char *mbasename(const char *path)
{
        const char *last = strrchr(path, '/');
        if(last) {
                return last + 1;
        }
        return path;
}

/** Parse the dirname of a program from a path.
* The path returned should be freed.
* @param path path to parse dirname from
*
* @return everything preceding the final '/'
*/
char *mdirname(const char *path)
{
        char *ret, *last;

        /* null or empty path */
        if(path == NULL || path == '\0') {
                return strdup(".");
        }

        ret = strdup(path);
        last = strrchr(ret, '/');

        if(last != NULL) {
                /* we found a '/', so terminate our string */
                *last = '\0';
                return ret;
        }
        /* no slash found */
        free(ret);
        return strdup(".");
}

/* output a string, but wrap words properly with a specified indentation
 */
void indentprint(const char *str, size_t indent)
{
        wchar_t *wcstr;
        const wchar_t *p;
        int len, cidx;
        const unsigned short cols = getcols();

        if(!str) {
                return;
        }

        /* if we're not a tty, or our tty is not wide enough that wrapping even makes
         * sense, print without indenting */
        if(cols == 0 || indent > cols) {
                fputs(str, stdout);
                return;
        }

        len = strlen(str) + 1;
        wcstr = calloc(len, sizeof(wchar_t));
        len = mbstowcs(wcstr, str, len);
        p = wcstr;
        cidx = indent;

        if(!p || !len) {
                return;
        }

        while(*p) {
                if(*p == L' ') {
                        const wchar_t *q, *next;
                        p++;
                        if(p == NULL || *p == L' ') continue;
                        next = wcschr(p, L' ');
                        if(next == NULL) {
                                next = p + wcslen(p);
                        }
                        /* len captures # cols */
                        len = 0;
                        q = p;
                        while(q < next) {
                                len += wcwidth(*q++);
                        }
                        if(len > (cols - cidx - 1)) {
                                /* wrap to a newline and reindent */
                                printf("\n%-*s", (int)indent, "");
                                cidx = indent;
                        } else {
                                printf(" ");
                                cidx++;
                        }
                        continue;
                }
                printf("%lc", (wint_t)*p);
                cidx += wcwidth(*p);
                p++;
        }
        free(wcstr);
}

/* Trim whitespace and newlines from a string
 */
size_t strtrim(char *str)
{
        char *end, *pch = str;

        if(str == NULL || *str == '\0') {
                /* string is empty, so we're done. */
                return 0;
        }

        while(isspace((unsigned char)*pch)) {
                pch++;
        }
        if(pch != str) {
                size_t len = strlen(pch);
                if(len) {
                        memmove(str, pch, len + 1);
                } else {
                        *str = '\0';
                }
        }

        /* check if there wasn't anything but whitespace in the string. */
        if(*str == '\0') {
                return 0;
        }

        end = (str + strlen(str) - 1);
        while(isspace((unsigned char)*end)) {
                end--;
        }
        *++end = '\0';

        return end - pch;
}

/* Replace all occurances of 'needle' with 'replace' in 'str', returning
 * a new string (must be free'd) */
char *strreplace(const char *str, const char *needle, const char *replace)
{
        const char *p = NULL, *q = NULL;
        char *newstr = NULL, *newp = NULL;
        alpm_list_t *i = NULL, *list = NULL;
        size_t needlesz = strlen(needle), replacesz = strlen(replace);
        size_t newsz;

        if(!str) {
                return NULL;
        }

        p = str;
        q = strstr(p, needle);
        while(q) {
                list = alpm_list_add(list, (char *)q);
                p = q + needlesz;
                q = strstr(p, needle);
        }

        /* no occurences of needle found */
        if(!list) {
                return strdup(str);
        }
        /* size of new string = size of old string + "number of occurences of needle"
         * x "size difference between replace and needle" */
        newsz = strlen(str) + 1 +
                alpm_list_count(list) * (replacesz - needlesz);
        newstr = calloc(newsz, sizeof(char));
        if(!newstr) {
                return NULL;
        }

        p = str;
        newp = newstr;
        for(i = list; i; i = alpm_list_next(i)) {
                q = i->data;
                if(q > p) {
                        /* add chars between this occurence and last occurence, if any */
                        memcpy(newp, p, (size_t)(q - p));
                        newp += q - p;
                }
                memcpy(newp, replace, replacesz);
                newp += replacesz;
                p = q + needlesz;
        }
        alpm_list_free(list);

        if(*p) {
                /* add the rest of 'p' */
                strcpy(newp, p);
        }

        return newstr;
}

/** Splits a string into a list of strings using the chosen character as
 * a delimiter.
 *
 * @param str the string to split
 * @param splitchar the character to split at
 *
 * @return a list containing the duplicated strings
 */
alpm_list_t *strsplit(const char *str, const char splitchar)
{
        alpm_list_t *list = NULL;
        const char *prev = str;
        char *dup = NULL;

        while((str = strchr(str, splitchar))) {
                dup = strndup(prev, (size_t)(str - prev));
                if(dup == NULL) {
                        return NULL;
                }
                list = alpm_list_add(list, dup);

                str++;
                prev = str;
        }

        dup = strdup(prev);
        if(dup == NULL) {
                return NULL;
        }
        list = alpm_list_add(list, dup);

        return list;
}

static size_t string_length(const char *s)
{
        int len;
        wchar_t *wcstr;

        if(!s || s[0] == '\0') {
                return 0;
        }
        /* len goes from # bytes -> # chars -> # cols */
        len = strlen(s) + 1;
        wcstr = calloc(len, sizeof(wchar_t));
        len = mbstowcs(wcstr, s, len);
        len = wcswidth(wcstr, len);
        free(wcstr);

        return len;
}

void string_display(const char *title, const char *string)
{
        if(title) {
                printf("%s ", title);
        }
        if(string == NULL || string[0] == '\0') {
                printf(_("None"));
        } else {
                /* compute the length of title + a space */
                size_t len = string_length(title) + 1;
                indentprint(string, len);
        }
        printf("\n");
}

static void table_print_line(const alpm_list_t *line, short col_padding,
                size_t colcount, size_t *widths, int *has_data)
{
        size_t i, lastcol = 0;
        int need_padding = 0;
        const alpm_list_t *curcell;

        for(i = colcount; i > 0; i--) {
                if(has_data[i - 1]) {
                        lastcol = i - 1;
                        break;
                }
        }

        for(i = 0, curcell = line; curcell && i < colcount;
                        i++, curcell = alpm_list_next(curcell)) {
                const char *value;
                int cell_padding;

                if(!has_data[i]) {
                        continue;
                }

                value = curcell->data;
                /* silly printf requires padding size to be an int */
                cell_padding = (int)widths[i] - (int)string_length(value);
                if(cell_padding < 0) {
                        cell_padding = 0;
                }
                if(need_padding) {
                        printf("%*s", col_padding, "");
                }
                /* left-align all but the last column */
                if(i != lastcol) {
                        printf("%s%*s", value, cell_padding, "");
                } else {
                        printf("%*s%s", cell_padding, "", value);
                }
                need_padding = 1;
        }

        printf("\n");
}



/**
 * Find the max string width of each column. Also determines whether values
 * exist in the column and sets the value in has_data accordingly.
 * @param header a list of header strings
 * @param rows a list of lists of rows as strings
 * @param padding the amount of padding between columns
 * @param totalcols the total number of columns in the header and each row
 * @param widths a pointer to store width data
 * @param has_data a pointer to store whether column has data
 *
 * @return the total width of the table; 0 on failure
 */
static size_t table_calc_widths(const alpm_list_t *header,
                const alpm_list_t *rows, short padding, size_t totalcols,
                size_t **widths, int **has_data)
{
        const alpm_list_t *i;
        size_t curcol, totalwidth = 0, usefulcols = 0;
        size_t *colwidths;
        int *coldata;

        if(totalcols <= 0) {
                return 0;
        }

        colwidths = malloc(totalcols * sizeof(size_t));
        coldata = calloc(totalcols, sizeof(int));
        if(!colwidths || !coldata) {
                return 0;
        }
        /* header determines column count and initial values of longest_strs */
        for(i = header, curcol = 0; i; i = alpm_list_next(i), curcol++) {
                colwidths[curcol] = string_length(i->data);
                /* note: header does not determine whether column has data */
        }

        /* now find the longest string in each column */
        for(i = rows; i; i = alpm_list_next(i)) {
                /* grab first column of each row and iterate through columns */
                const alpm_list_t *j = i->data;
                for(curcol = 0; j; j = alpm_list_next(j), curcol++) {
                        const char *str = j->data;
                        size_t str_len = string_length(str);

                        if(str_len > colwidths[curcol]) {
                                colwidths[curcol] = str_len;
                        }
                        if(str_len > 0) {
                                coldata[curcol] = 1;
                        }
                }
        }

        for(i = header, curcol = 0; i; i = alpm_list_next(i), curcol++) {
                /* only include columns that have data */
                if(coldata[curcol]) {
                        usefulcols++;
                        totalwidth += colwidths[curcol];
                }
        }

        /* add padding between columns */
        if(usefulcols > 0) {
                totalwidth += padding * (usefulcols - 1);
        }

        *widths = colwidths;
        *has_data = coldata;
        return totalwidth;
}

/** Displays the list in table format
 *
 * @param title the tables title
 * @param header the column headers. column count is determined by the nr
 *               of headers
 * @param rows the rows to display as a list of lists of strings. the outer
 *             list represents the rows, the inner list the cells (= columns)
 *
 * @return -1 if not enough terminal cols available, else 0
 */
int table_display(const char *title, const alpm_list_t *header,
                const alpm_list_t *rows)
{
        const unsigned short padding = 2;
        const alpm_list_t *i;
        size_t *widths = NULL, totalcols, totalwidth;
        int *has_data = NULL;

        if(rows == NULL || header == NULL) {
                return 0;
        }

        totalcols = alpm_list_count(header);
        totalwidth = table_calc_widths(header, rows, padding, totalcols,
                        &widths, &has_data);
        /* return -1 if terminal is not wide enough */
        if(totalwidth > getcols()) {
                pm_printf(ALPM_LOG_WARNING,
                                _("insufficient columns available for table display\n"));
                return -1;
        }
        if(!totalwidth || !widths || !has_data) {
                return -1;
        }

        if(title != NULL) {
                printf("%s\n\n", title);
        }

        table_print_line(header, padding, totalcols, widths, has_data);
        printf("\n");

        for(i = rows; i; i = alpm_list_next(i)) {
                table_print_line(i->data, padding, totalcols, widths, has_data);
        }

        free(widths);
        free(has_data);
        return 0;
}

void list_display(const char *title, const alpm_list_t *list)
{
        const alpm_list_t *i;
        size_t len = 0;

        if(title) {
                len = string_length(title) + 1;
                printf("%s ", title);
        }

        if(!list) {
                printf("%s\n", _("None"));
        } else {
                const unsigned short maxcols = getcols();
                size_t cols = len;
                const char *str = list->data;
                fputs(str, stdout);
                cols += string_length(str);
                for(i = alpm_list_next(list); i; i = alpm_list_next(i)) {
                        str = i->data;
                        size_t s = string_length(str);
                        /* wrap only if we have enough usable column space */
                        if(maxcols > len && cols + s + 2 >= maxcols) {
                                size_t j;
                                cols = len;
                                printf("\n");
                                for(j = 1; j <= len; j++) {
                                        printf(" ");
                                }
                        } else if(cols != len) {
                                /* 2 spaces are added if this is not the first element on a line. */
                                printf("  ");
                                cols += 2;
                        }
                        fputs(str, stdout);
                        cols += s;
                }
                putchar('\n');
        }
}

void list_display_linebreak(const char *title, const alpm_list_t *list)
{
        size_t len = 0;

        if(title) {
                len = string_length(title) + 1;
                printf("%s ", title);
        }

        if(!list) {
                printf("%s\n", _("None"));
        } else {
                const alpm_list_t *i;
                /* Print the first element */
                indentprint((const char *)list->data, len);
                printf("\n");
                /* Print the rest */
                for(i = alpm_list_next(list); i; i = alpm_list_next(i)) {
                        size_t j;
                        for(j = 1; j <= len; j++) {
                                printf(" ");
                        }
                        indentprint((const char *)i->data, len);
                        printf("\n");
                }
        }
}

void signature_display(const char *title, alpm_siglist_t *siglist)
{
        size_t len = 0;

        if(title) {
                len = string_length(title) + 1;
                printf("%s ", title);
        }
        if(siglist->count == 0) {
                printf(_("None"));
        } else {
                size_t i;
                for(i = 0; i < siglist->count; i++) {
                        char *sigline;
                        const char *status, *validity, *name;
                        int ret;
                        alpm_sigresult_t *result = siglist->results + i;
                        /* Don't re-indent the first result */
                        if(i != 0) {
                                size_t j;
                                for(j = 1; j <= len; j++) {
                                        printf(" ");
                                }
                        }
                        switch(result->status) {
                                case ALPM_SIGSTATUS_VALID:
                                        status = _("Valid");
                                        break;
                                case ALPM_SIGSTATUS_KEY_EXPIRED:
                                        status = _("Key expired");
                                        break;
                                case ALPM_SIGSTATUS_SIG_EXPIRED:
                                        status = _("Expired");
                                        break;
                                case ALPM_SIGSTATUS_INVALID:
                                        status = _("Invalid");
                                        break;
                                case ALPM_SIGSTATUS_KEY_UNKNOWN:
                                        status = _("Key unknown");
                                        break;
                                case ALPM_SIGSTATUS_KEY_DISABLED:
                                        status = _("Key disabled");
                                        break;
                                default:
                                        status = _("Signature error");
                                        break;
                        }
                        switch(result->validity) {
                                case ALPM_SIGVALIDITY_FULL:
                                        validity = _("full trust");
                                        break;
                                case ALPM_SIGVALIDITY_MARGINAL:
                                        validity = _("marginal trust");
                                        break;
                                case ALPM_SIGVALIDITY_NEVER:
                                        validity = _("never trust");
                                        break;
                                case ALPM_SIGVALIDITY_UNKNOWN:
                                default:
                                        validity = _("unknown trust");
                                        break;
                        }
                        name = result->key.uid ? result->key.uid : result->key.fingerprint;
                        ret = pm_asprintf(&sigline, _("%s, %s from \"%s\""),
                                        status, validity, name);
                        if(ret == -1) {
                                pm_printf(ALPM_LOG_ERROR,  _("failed to allocate string\n"));
                                continue;
                        }
                        indentprint(sigline, len);
                        printf("\n");
                        free(sigline);
                }
        }
}

/* creates a header row for use with table_display */
static alpm_list_t *create_verbose_header(int dl_size)
{
        alpm_list_t *res = NULL;
        char *str;

        str = _("Name");
        res = alpm_list_add(res, str);
        str = _("Old Version");
        res = alpm_list_add(res, str);
        str = _("New Version");
        res = alpm_list_add(res, str);
        str = _("Net Change");
        res = alpm_list_add(res, str);
        if(dl_size) {
                str = _("Download Size");
                res = alpm_list_add(res, str);
        }

        return res;
}

/* returns package info as list of strings */
static alpm_list_t *create_verbose_row(pm_target_t *target, int dl_size)
{
        char *str;
        off_t size = 0;
        double human_size;
        const char *label;
        alpm_list_t *ret = NULL;

        /* a row consists of the package name, */
        if(target->install) {
                const alpm_db_t *db = alpm_pkg_get_db(target->install);
                if(db) {
                        pm_asprintf(&str, "%s/%s", alpm_db_get_name(db), alpm_pkg_get_name(target->install));
                } else {
                        pm_asprintf(&str, "%s", alpm_pkg_get_name(target->install));
                }
        } else {
                pm_asprintf(&str, "%s", alpm_pkg_get_name(target->remove));
        }
        ret = alpm_list_add(ret, str);

        /* old and new versions */
        pm_asprintf(&str, "%s",
                        target->remove != NULL ? alpm_pkg_get_version(target->remove) : "");
        ret = alpm_list_add(ret, str);

        pm_asprintf(&str, "%s",
                        target->install != NULL ? alpm_pkg_get_version(target->install) : "");
        ret = alpm_list_add(ret, str);

        /* and size */
        size -= target->remove ? alpm_pkg_get_isize(target->remove) : 0;
        size += target->install ? alpm_pkg_get_isize(target->install) : 0;
        human_size = humanize_size(size, 'M', 2, &label);
        pm_asprintf(&str, "%.2f %s", human_size, label);
        ret = alpm_list_add(ret, str);

        if(dl_size) {
                size = target->install ? alpm_pkg_download_size(target->install) : 0;
                human_size = humanize_size(size, 'M', 2, &label);
                if(size != 0) {
                        pm_asprintf(&str, "%.2f %s", human_size, label);
                } else {
                        str = strdup("");
                }
                ret = alpm_list_add(ret, str);
        }

        return ret;
}

/* prepare a list of pkgs to display */
static void _display_targets(alpm_list_t *targets, int verbose)
{
        char *str;
        const char *label;
        double size;
        off_t isize = 0, rsize = 0, dlsize = 0;
        alpm_list_t *i, *rows = NULL, *names = NULL;
        int show_dl_size = config->op == PM_OP_SYNC;

        if(!targets) {
                return;
        }

        /* gather package info */
        for(i = targets; i; i = alpm_list_next(i)) {
                pm_target_t *target = i->data;

                if(target->install) {
                        dlsize += alpm_pkg_download_size(target->install);
                        isize += alpm_pkg_get_isize(target->install);
                }
                if(target->remove) {
                        /* add up size of all removed packages */
                        rsize += alpm_pkg_get_isize(target->remove);
                }
        }

        /* form data for both verbose and non-verbose display */
        for(i = targets; i; i = alpm_list_next(i)) {
                pm_target_t *target = i->data;

                rows = alpm_list_add(rows, create_verbose_row(target, show_dl_size));
                if(target->install) {
                        pm_asprintf(&str, "%s-%s", alpm_pkg_get_name(target->install),
                                        alpm_pkg_get_version(target->install));
                } else if(isize == 0) {
                        pm_asprintf(&str, "%s-%s", alpm_pkg_get_name(target->remove),
                                        alpm_pkg_get_version(target->remove));
                } else {
                        pm_asprintf(&str, "%s-%s [removal]", alpm_pkg_get_name(target->remove),
                                        alpm_pkg_get_version(target->remove));
                }
                names = alpm_list_add(names, str);
        }

        /* print to screen */
        pm_asprintf(&str, _("Targets (%d):"), alpm_list_count(targets));

        printf("\n");
        if(verbose) {
                alpm_list_t *header = create_verbose_header(show_dl_size);
                if(table_display(str, header, rows) != 0) {
                        /* fallback to list display if table wouldn't fit */
                        list_display(str, names);
                }
                alpm_list_free(header);
        } else {
                list_display(str, names);
        }
        printf("\n");

        /* rows is a list of lists of strings, free inner lists here */
        for(i = rows; i; i = alpm_list_next(i)) {
                alpm_list_t *lp = i->data;
                FREELIST(lp);
        }
        alpm_list_free(rows);
        FREELIST(names);
        free(str);

        if(dlsize > 0 || config->op_s_downloadonly) {
                size = humanize_size(dlsize, 'M', 2, &label);
                printf(_("Total Download Size:    %.2f %s\n"), size, label);
        }
        if(!config->op_s_downloadonly) {
                if(isize > 0) {
                        size = humanize_size(isize, 'M', 2, &label);
                        printf(_("Total Installed Size:   %.2f %s\n"), size, label);
                }
                if(rsize > 0 && isize == 0) {
                        size = humanize_size(rsize, 'M', 2, &label);
                        printf(_("Total Removed Size:     %.2f %s\n"), size, label);
                }
                /* only show this net value if different from raw installed size */
                if(isize > 0 && rsize > 0) {
                        size = humanize_size(isize - rsize, 'M', 2, &label);
                        printf(_("Net Upgrade Size:       %.2f %s\n"), size, label);
                }
        }
}

static int target_cmp(const void *p1, const void *p2)
{
        const pm_target_t *targ1 = p1;
        const pm_target_t *targ2 = p2;
        /* explicit are always sorted after implicit (e.g. deps, pulled targets) */
        if(targ1->is_explicit != targ2->is_explicit) {
                return targ1->is_explicit > targ2->is_explicit;
        }
        const char *name1 = targ1->install ?
                alpm_pkg_get_name(targ1->install) : alpm_pkg_get_name(targ1->remove);
        const char *name2 = targ2->install ?
                alpm_pkg_get_name(targ2->install) : alpm_pkg_get_name(targ2->remove);
        return strcmp(name1, name2);
}

static int pkg_cmp(const void *p1, const void *p2)
{
        /* explicit cast due to (un)necessary removal of const */
        alpm_pkg_t *pkg1 = (alpm_pkg_t *)p1;
        alpm_pkg_t *pkg2 = (alpm_pkg_t *)p2;
        return strcmp(alpm_pkg_get_name(pkg1), alpm_pkg_get_name(pkg2));
}

void display_targets(void)
{
        alpm_list_t *i, *targets = NULL;
        alpm_db_t *db_local = alpm_get_localdb(config->handle);

        for(i = alpm_trans_get_add(config->handle); i; i = alpm_list_next(i)) {
                alpm_pkg_t *pkg = i->data;
                pm_target_t *targ = calloc(1, sizeof(pm_target_t));
                if(!targ) return;
                targ->install = pkg;
                targ->remove = alpm_db_get_pkg(db_local, alpm_pkg_get_name(pkg));
                if(alpm_list_find(config->explicit_adds, pkg, pkg_cmp)) {
                        targ->is_explicit = 1;
                }
                targets = alpm_list_add(targets, targ);
        }
        for(i = alpm_trans_get_remove(config->handle); i; i = alpm_list_next(i)) {
                alpm_pkg_t *pkg = i->data;
                pm_target_t *targ = calloc(1, sizeof(pm_target_t));
                if(!targ) return;
                targ->remove = pkg;
                if(alpm_list_find(config->explicit_removes, pkg, pkg_cmp)) {
                        targ->is_explicit = 1;
                }
                targets = alpm_list_add(targets, targ);
        }

        targets = alpm_list_msort(targets, alpm_list_count(targets), target_cmp);
        _display_targets(targets, config->verbosepkglists);
        FREELIST(targets);
}

static off_t pkg_get_size(alpm_pkg_t *pkg)
{
        switch(config->op) {
                case PM_OP_SYNC:
                        return alpm_pkg_download_size(pkg);
                case PM_OP_UPGRADE:
                        return alpm_pkg_get_size(pkg);
                default:
                        return alpm_pkg_get_isize(pkg);
        }
}

static char *pkg_get_location(alpm_pkg_t *pkg)
{
        alpm_list_t *servers;
        char *string = NULL;
        switch(config->op) {
                case PM_OP_SYNC:
                        servers = alpm_db_get_servers(alpm_pkg_get_db(pkg));
                        if(servers) {
                                pm_asprintf(&string, "%s/%s", servers->data,
                                                alpm_pkg_get_filename(pkg));
                                return string;
                        }
                case PM_OP_UPGRADE:
                        return strdup(alpm_pkg_get_filename(pkg));
                default:
                        pm_asprintf(&string, "%s-%s", alpm_pkg_get_name(pkg), alpm_pkg_get_version(pkg));
                        return string;
        }
}

/* a pow() implementation that is specialized for an integer base and small,
 * positive-only integer exponents. */
static double simple_pow(int base, int exp)
{
        double result = 1.0;
        for(; exp > 0; exp--) {
                result *= base;
        }
        return result;
}

/** Converts sizes in bytes into human readable units.
 *
 * @param bytes the size in bytes
 * @param target_unit '\0' or a short label. If equal to one of the short unit
 * labels ('B', 'K', ...) bytes is converted to target_unit; if '\0', the first
 * unit which will bring the value to below a threshold of 2048 will be chosen.
 * @param precision number of decimal places, ensures -0.00 gets rounded to
 * 0.00; -1 if no rounding desired
 * @param label will be set to the appropriate unit label
 *
 * @return the size in the appropriate unit
 */
double humanize_size(off_t bytes, const char target_unit, int precision,
                const char **label)
{
        static const char *labels[] = {"B", "KiB", "MiB", "GiB",
                "TiB", "PiB", "EiB", "ZiB", "YiB"};
        static const int unitcount = sizeof(labels) / sizeof(labels[0]);

        double val = (double)bytes;
        int index;

        for(index = 0; index < unitcount - 1; index++) {
                if(target_unit != '\0' && labels[index][0] == target_unit) {
                        break;
                } else if(target_unit == '\0' && val <= 2048.0 && val >= -2048.0) {
                        break;
                }
                val /= 1024.0;
        }

        if(label) {
                *label = labels[index];
        }

        /* fix FS#27924 so that it doesn't display negative zeroes */
        if(precision >= 0 && val < 0.0 &&
                        val > (-0.5 / simple_pow(10, precision))) {
                val = 0.0;
        }

        return val;
}

void print_packages(const alpm_list_t *packages)
{
        const alpm_list_t *i;
        if(!config->print_format) {
                config->print_format = strdup("%l");
        }
        for(i = packages; i; i = alpm_list_next(i)) {
                alpm_pkg_t *pkg = i->data;
                char *string = strdup(config->print_format);
                char *temp = string;
                /* %n : pkgname */
                if(strstr(temp, "%n")) {
                        string = strreplace(temp, "%n", alpm_pkg_get_name(pkg));
                        free(temp);
                        temp = string;
                }
                /* %v : pkgver */
                if(strstr(temp, "%v")) {
                        string = strreplace(temp, "%v", alpm_pkg_get_version(pkg));
                        free(temp);
                        temp = string;
                }
                /* %l : location */
                if(strstr(temp, "%l")) {
                        char *pkgloc = pkg_get_location(pkg);
                        string = strreplace(temp, "%l", pkgloc);
                        free(pkgloc);
                        free(temp);
                        temp = string;
                }
                /* %r : repo */
                if(strstr(temp, "%r")) {
                        const char *repo = "local";
                        alpm_db_t *db = alpm_pkg_get_db(pkg);
                        if(db) {
                                repo = alpm_db_get_name(db);
                        }
                        string = strreplace(temp, "%r", repo);
                        free(temp);
                        temp = string;
                }
                /* %s : size */
                if(strstr(temp, "%s")) {
                        char *size;
                        pm_asprintf(&size, "%jd", (intmax_t)pkg_get_size(pkg));
                        string = strreplace(temp, "%s", size);
                        free(size);
                        free(temp);
                }
                printf("%s\n",string);
                free(string);
        }
}

/**
 * Helper function for comparing depends using the alpm "compare func"
 * signature. The function descends through the structure in the following
 * comparison order: name, modifier (e.g., '>', '='), version, description.
 * @param d1 the first depend structure
 * @param d2 the second depend structure
 * @return -1, 0, or 1 if first is <, ==, or > second
 */
static int depend_cmp(const void *d1, const void *d2)
{
        const alpm_depend_t *dep1 = d1;
        const alpm_depend_t *dep2 = d2;
        int ret;

        ret = strcmp(dep1->name, dep2->name);
        if(ret == 0) {
                ret = dep1->mod - dep2->mod;
        }
        if(ret == 0) {
                if(dep1->version && dep2->version) {
                        ret = strcmp(dep1->version, dep2->version);
                } else if(!dep1->version && dep2->version) {
                        ret = -1;
                } else if(dep1->version && !dep2->version) {
                        ret = 1;
                }
        }
        if(ret == 0) {
                if(dep1->desc && dep2->desc) {
                        ret = strcmp(dep1->desc, dep2->desc);
                } else if(!dep1->desc && dep2->desc) {
                        ret = -1;
                } else if(dep1->desc && !dep2->desc) {
                        ret = 1;
                }
        }

        return ret;
}

void display_new_optdepends(alpm_pkg_t *oldpkg, alpm_pkg_t *newpkg)
{
        alpm_list_t *i, *old, *new, *optdeps, *optstrings = NULL;

        old = alpm_pkg_get_optdepends(oldpkg);
        new = alpm_pkg_get_optdepends(newpkg);
        optdeps = alpm_list_diff(new, old, depend_cmp);

        /* turn optdepends list into a text list */
        for(i = optdeps; i; i = alpm_list_next(i)) {
                alpm_depend_t *optdep = i->data;
                optstrings = alpm_list_add(optstrings, alpm_dep_compute_string(optdep));
        }

        if(optstrings) {
                printf(_("New optional dependencies for %s\n"), alpm_pkg_get_name(newpkg));
                list_display_linebreak("   ", optstrings);
        }

        alpm_list_free(optdeps);
        FREELIST(optstrings);
}

void display_optdepends(alpm_pkg_t *pkg)
{
        alpm_list_t *i, *optdeps, *optstrings = NULL;

        optdeps = alpm_pkg_get_optdepends(pkg);

        /* turn optdepends list into a text list */
        for(i = optdeps; i; i = alpm_list_next(i)) {
                alpm_depend_t *optdep = i->data;
                optstrings = alpm_list_add(optstrings, alpm_dep_compute_string(optdep));
        }

        if(optstrings) {
                printf(_("Optional dependencies for %s\n"), alpm_pkg_get_name(pkg));
                list_display_linebreak("   ", optstrings);
        }

        FREELIST(optstrings);
}

static void display_repo_list(const char *dbname, alpm_list_t *list)
{
        const char *prefix= "  ";

        printf(":: ");
        printf(_("Repository %s\n"), dbname);
        list_display(prefix, list);
}

void select_display(const alpm_list_t *pkglist)
{
        const alpm_list_t *i;
        int nth = 1;
        alpm_list_t *list = NULL;
        char *string = NULL;
        const char *dbname = NULL;

        for(i = pkglist; i; i = i->next) {
                alpm_pkg_t *pkg = i->data;
                alpm_db_t *db = alpm_pkg_get_db(pkg);

                if(!dbname)
                        dbname = alpm_db_get_name(db);
                if(strcmp(alpm_db_get_name(db), dbname) != 0) {
                        display_repo_list(dbname, list);
                        FREELIST(list);
                        dbname = alpm_db_get_name(db);
                }
                string = NULL;
                pm_asprintf(&string, "%d) %s", nth, alpm_pkg_get_name(pkg));
                list = alpm_list_add(list, string);
                nth++;
        }
        display_repo_list(dbname, list);
        FREELIST(list);
}

static int parseindex(char *s, int *val, int min, int max)
{
        char *endptr = NULL;
        int n = strtol(s, &endptr, 10);
        if(*endptr == '\0') {
                if(n < min || n > max) {
                        pm_printf(ALPM_LOG_ERROR,
                                        _("invalid value: %d is not between %d and %d\n"),
                                        n, min, max);
                        return -1;
                }
                *val = n;
                return 0;
        } else {
                pm_printf(ALPM_LOG_ERROR, _("invalid number: %s\n"), s);
                return -1;
        }
}

static int multiselect_parse(char *array, int count, char *response)
{
        char *str, *saveptr;

        for(str = response; ; str = NULL) {
                int include = 1;
                int start, end;
                size_t len;
                char *ends = NULL;
                char *starts = strtok_r(str, " ", &saveptr);

                if(starts == NULL) {
                        break;
                }
                len = strtrim(starts);
                if(len == 0)
                        continue;

                if(*starts == '^') {
                        starts++;
                        len--;
                        include = 0;
                } else if(str) {
                        /* if first token is including, we unselect all targets */
                        memset(array, 0, count);
                }

                if(len > 1) {
                        /* check for range */
                        char *p;
                        if((p = strchr(starts + 1, '-'))) {
                                *p = 0;
                                ends = p + 1;
                        }
                }

                if(parseindex(starts, &start, 1, count) != 0)
                        return -1;

                if(!ends) {
                        array[start-1] = include;
                } else {
                        int d;
                        if(parseindex(ends, &end, start, count) != 0) {
                                return -1;
                        }
                        for(d = start; d <= end; d++) {
                                array[d-1] = include;
                        }
                }
        }

        return 0;
}

int multiselect_question(char *array, int count)
{
        char *response, *lastchar;
        FILE *stream;
        size_t response_len = 64;

        if(config->noconfirm) {
                stream = stdout;
        } else {
                /* Use stderr so questions are always displayed when redirecting output */
                stream = stderr;
        }

        response = malloc(response_len);
        if(!response) {
                return -1;
        }
        lastchar = response + response_len - 1;
        /* sentinel byte to later see if we filled up the entire string */
        *lastchar = 1;

        while(1) {
                memset(array, 1, count);

                fprintf(stream, "\n");
                fprintf(stream, _("Enter a selection (default=all)"));
                fprintf(stream, ": ");
                fflush(stream);

                if(config->noconfirm) {
                        fprintf(stream, "\n");
                        break;
                }

                flush_term_input();

                if(fgets(response, response_len, stdin)) {
                        const size_t response_incr = 64;
                        size_t len;
                        /* handle buffer not being large enough to read full line case */
                        while(*lastchar == '\0' && lastchar[-1] != '\n') {
                                response_len += response_incr;
                                response = realloc(response, response_len);
                                if(!response) {
                                        return -1;
                                }
                                lastchar = response + response_len - 1;
                                /* sentinel byte */
                                *lastchar = 1;
                                if(fgets(response + response_len - response_incr - 1,
                                                        response_incr + 1, stdin) == 0) {
                                        free(response);
                                        return -1;
                                }
                        }

                        len = strtrim(response);
                        if(len > 0) {
                                if(multiselect_parse(array, count, response) == -1) {
                                        /* only loop if user gave an invalid answer */
                                        continue;
                                }
                        }
                        break;
                } else {
                        free(response);
                        return -1;
                }
        }

        free(response);
        return 0;
}

int select_question(int count)
{
        char response[32];
        FILE *stream;
        int preset = 1;

        if(config->noconfirm) {
                stream = stdout;
        } else {
                /* Use stderr so questions are always displayed when redirecting output */
                stream = stderr;
        }

        while(1) {
                fprintf(stream, "\n");
                fprintf(stream, _("Enter a number (default=%d)"), preset);
                fprintf(stream, ": ");

                if(config->noconfirm) {
                        fprintf(stream, "\n");
                        break;
                }

                flush_term_input();

                if(fgets(response, sizeof(response), stdin)) {
                        size_t len = strtrim(response);
                        if(len > 0) {
                                int n;
                                if(parseindex(response, &n, 1, count) != 0)
                                        continue;
                                return (n - 1);
                        }
                }
                break;
        }

        return (preset - 1);
}


/* presents a prompt and gets a Y/N answer */
static int question(short preset, char *fmt, va_list args)
{
        char response[32];
        FILE *stream;

        if(config->noconfirm) {
                stream = stdout;
        } else {
                /* Use stderr so questions are always displayed when redirecting output */
                stream = stderr;
        }

        /* ensure all text makes it to the screen before we prompt the user */
        fflush(stdout);
        fflush(stderr);

        vfprintf(stream, fmt, args);

        if(preset) {
                fprintf(stream, " %s ", _("[Y/n]"));
        } else {
                fprintf(stream, " %s ", _("[y/N]"));
        }

        if(config->noconfirm) {
                fprintf(stream, "\n");
                return preset;
        }

        fflush(stream);
        flush_term_input();

        if(fgets(response, sizeof(response), stdin)) {
                size_t len = strtrim(response);
                if(len == 0) {
                        return preset;
                }

                /* if stdin is piped, response does not get printed out, and as a result
                 * a \n is missing, resulting in broken output (FS#27909) */
                if(!isatty(fileno(stdin))) {
                        fprintf(stream, "%s\n", response);
                }

                if(strcasecmp(response, _("Y")) == 0 || strcasecmp(response, _("YES")) == 0) {
                        return 1;
                } else if(strcasecmp(response, _("N")) == 0 || strcasecmp(response, _("NO")) == 0) {
                        return 0;
                }
        }
        return 0;
}

int yesno(char *fmt, ...)
{
        int ret;
        va_list args;

        va_start(args, fmt);
        ret = question(1, fmt, args);
        va_end(args);

        return ret;
}

int noyes(char *fmt, ...)
{
        int ret;
        va_list args;

        va_start(args, fmt);
        ret = question(0, fmt, args);
        va_end(args);

        return ret;
}

int pm_printf(alpm_loglevel_t level, const char *format, ...)
{
        int ret;
        va_list args;

        /* print the message using va_arg list */
        va_start(args, format);
        ret = pm_vfprintf(stderr, level, format, args);
        va_end(args);

        return ret;
}

int pm_asprintf(char **string, const char *format, ...)
{
        int ret = 0;
        va_list args;

        /* print the message using va_arg list */
        va_start(args, format);
        if(vasprintf(string, format, args) == -1) {
                pm_printf(ALPM_LOG_ERROR,  _("failed to allocate string\n"));
                ret = -1;
        }
        va_end(args);

        return ret;
}

int pm_vasprintf(char **string, alpm_loglevel_t level, const char *format, va_list args)
{
        int ret = 0;
        char *msg = NULL;

        /* if current logmask does not overlap with level, do not print msg */
        if(!(config->logmask & level)) {
                return ret;
        }

        /* print the message using va_arg list */
        ret = vasprintf(&msg, format, args);

        /* print a prefix to the message */
        switch(level) {
                case ALPM_LOG_ERROR:
                        pm_asprintf(string, _("error: %s"), msg);
                        break;
                case ALPM_LOG_WARNING:
                        pm_asprintf(string, _("warning: %s"), msg);
                        break;
                case ALPM_LOG_DEBUG:
                        pm_asprintf(string, "debug: %s", msg);
                        break;
                case ALPM_LOG_FUNCTION:
                        pm_asprintf(string, "function: %s", msg);
                        break;
                default:
                        pm_asprintf(string, "%s", msg);
                        break;
        }
        free(msg);

        return ret;
}

int pm_vfprintf(FILE *stream, alpm_loglevel_t level, const char *format, va_list args)
{
        int ret = 0;

        /* if current logmask does not overlap with level, do not print msg */
        if(!(config->logmask & level)) {
                return ret;
        }

#if defined(PACMAN_DEBUG)
        /* If debug is on, we'll timestamp the output */
        if(config->logmask & ALPM_LOG_DEBUG) {
                time_t t;
                struct tm *tmp;
                char timestr[10] = {0};

                t = time(NULL);
                tmp = localtime(&t);
                strftime(timestr, 9, "%H:%M:%S", tmp);
                timestr[8] = '\0';

                fprintf(stream, "[%s] ", timestr);
        }
#endif

        /* print a prefix to the message */
        switch(level) {
                case ALPM_LOG_ERROR:
                        fprintf(stream, _("error: "));
                        break;
                case ALPM_LOG_WARNING:
                        fprintf(stream, _("warning: "));
                        break;
                case ALPM_LOG_DEBUG:
                        fprintf(stream, "debug: ");
                        break;
                case ALPM_LOG_FUNCTION:
                        fprintf(stream, "function: ");
                        break;
                default:
                        break;
        }

        /* print the message using va_arg list */
        ret = vfprintf(stream, format, args);
        return ret;
}

#ifndef HAVE_STRNDUP
/* A quick and dirty implementation derived from glibc */
static size_t strnlen(const char *s, size_t max)
{
    register const char *p;
    for(p = s; *p && max--; ++p);
    return (p - s);
}

char *strndup(const char *s, size_t n)
{
  size_t len = strnlen(s, n);
  char *new = (char *) malloc(len + 1);

  if(new == NULL)
    return NULL;

  new[len] = '\0';
  return (char *)memcpy(new, s, len);
}
#endif

/* vim: set ts=2 sw=2 noet: */