/* * callback.c * * Copyright (c) 2002-2007 by Judd Vinet * * 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 . */ #include "config.h" #include #include #include #include #include #include #include #include /* pacman */ #include "callback.h" #include "util.h" #include "conf.h" /* TODO this should not have to be defined twice- trans.c & log.c */ #define LOG_STR_LEN 256 #define FILENAME_TRIM_LEN 23 /* download progress bar */ static float rate_last; static int xfered_last; static struct timeval initial_time; /* transaction progress bar */ static int prevpercent = 0; /* for less progressbar output */ /* delayed output during progress bar */ static int on_progress = 0; static alpm_list_t *output = NULL; /* Silly little helper function, determines if the caller needs a visual update * since the last time this function was called. * This is made for the two progress bar functions, to prevent flicker * * first_call indicates if this is the first time it is called, for * initialization purposes */ static float get_update_timediff(int first_call) { float retval = 0.0; static struct timeval last_time = {0, 0}; /* on first call, simply set the last time and return */ if(first_call) { gettimeofday(&last_time, NULL); } else { struct timeval this_time; float diff_sec, diff_usec; gettimeofday(&this_time, NULL); diff_sec = this_time.tv_sec - last_time.tv_sec; diff_usec = this_time.tv_usec - last_time.tv_usec; retval = diff_sec + (diff_usec / 1000000.0); /* return 0 and do not update last_time if interval was too short */ if(retval < UPDATE_SPEED_SEC) { retval = 0.0; } else { last_time = this_time; /* printf("\nupdate retval: %f\n", retval); DEBUG*/ } } return(retval); } /* refactored from cb_trans_progress */ static void fill_progress(const int graph_percent, const int display_percent, const int proglen) { const unsigned int hashlen = proglen - 8; const unsigned int hash = graph_percent * hashlen / 100; static unsigned int lasthash = 0, mouth = 0; unsigned int i; /* printf("\ndebug: proglen: %i\n", proglen); DEBUG*/ if(graph_percent == 0) { lasthash = 0; mouth = 0; } /* magic numbers, how I loathe thee */ if(proglen > 8) { printf(" ["); for(i = hashlen; i > 1; --i) { /* if special progress bar enabled */ if(config->chomp) { if(i > hashlen - hash) { printf("-"); } else if(i == hashlen - hash) { if(lasthash == hash) { if(mouth) { printf("\033[1;33mC\033[m"); } else { printf("\033[1;33mc\033[m"); } } else { lasthash = hash; mouth = mouth == 1 ? 0 : 1; if(mouth) { printf("\033[1;33mC\033[m"); } else { printf("\033[1;33mc\033[m"); } } } else if(i%3 == 0) { printf("\033[0;37mo\033[m"); } else { printf("\033[0;37m \033[m"); } } /* else regular progress bar */ else if(i > hashlen - hash) { printf("#"); } else { printf("-"); } } printf("]"); } /* print percent after progress bar */ if(proglen > 5) { printf(" %3d%%", display_percent); } if(graph_percent == 100) { printf("\n"); } else { printf("\r"); } fflush(stdout); } /* callback to handle messages/notifications from libalpm transactions */ void cb_trans_evt(pmtransevt_t event, void *data1, void *data2) { char str[LOG_STR_LEN] = ""; switch(event) { case PM_TRANS_EVT_CHECKDEPS_START: printf(_("checking dependencies...\n")); break; case PM_TRANS_EVT_FILECONFLICTS_START: if(config->noprogressbar) { printf(_("checking for file conflicts...\n")); } break; case PM_TRANS_EVT_RESOLVEDEPS_START: printf(_("resolving dependencies...\n")); break; case PM_TRANS_EVT_INTERCONFLICTS_START: printf(_("looking for inter-conflicts...\n")); break; case PM_TRANS_EVT_ADD_START: if(config->noprogressbar) { printf(_("installing %s...\n"), alpm_pkg_get_name(data1)); } break; case PM_TRANS_EVT_ADD_DONE: snprintf(str, LOG_STR_LEN, "installed %s (%s)\n", alpm_pkg_get_name(data1), alpm_pkg_get_version(data1)); alpm_logaction(str); break; case PM_TRANS_EVT_REMOVE_START: if(config->noprogressbar) { printf(_("removing %s...\n"), alpm_pkg_get_name(data1)); } break; case PM_TRANS_EVT_REMOVE_DONE: snprintf(str, LOG_STR_LEN, "removed %s (%s)\n", alpm_pkg_get_name(data1), alpm_pkg_get_version(data1)); alpm_logaction(str); break; case PM_TRANS_EVT_UPGRADE_START: if(config->noprogressbar) { printf(_("upgrading %s...\n"), alpm_pkg_get_name(data1)); } break; case PM_TRANS_EVT_UPGRADE_DONE: snprintf(str, LOG_STR_LEN, "upgraded %s (%s -> %s)\n", (char *)alpm_pkg_get_name(data1), (char *)alpm_pkg_get_version(data2), (char *)alpm_pkg_get_version(data1)); alpm_logaction(str); break; case PM_TRANS_EVT_INTEGRITY_START: printf(_("checking package integrity...\n")); break; case PM_TRANS_EVT_DELTA_INTEGRITY_START: printf(_("checking delta integrity...\n")); break; case PM_TRANS_EVT_DELTA_PATCHES_START: printf(_("applying deltas...\n")); break; case PM_TRANS_EVT_DELTA_PATCH_START: printf(_("generating %s with %s... "), (char *)data1, (char *)data2); break; case PM_TRANS_EVT_DELTA_PATCH_DONE: printf(_("success!\n")); break; case PM_TRANS_EVT_DELTA_PATCH_FAILED: printf(_("failed.\n")); break; case PM_TRANS_EVT_SCRIPTLET_INFO: printf("%s", (char*)data1); break; case PM_TRANS_EVT_PRINTURI: printf("%s/%s\n", (char*)data1, (char*)data2); break; case PM_TRANS_EVT_RETRIEVE_START: printf(_(":: Retrieving packages from %s...\n"), (char*)data1); break; /* all the simple done events, with fallthrough for each */ case PM_TRANS_EVT_FILECONFLICTS_DONE: case PM_TRANS_EVT_EXTRACT_DONE: case PM_TRANS_EVT_CHECKDEPS_DONE: case PM_TRANS_EVT_RESOLVEDEPS_DONE: case PM_TRANS_EVT_INTERCONFLICTS_DONE: case PM_TRANS_EVT_INTEGRITY_DONE: case PM_TRANS_EVT_DELTA_INTEGRITY_DONE: case PM_TRANS_EVT_DELTA_PATCHES_DONE: /* nothing */ break; } fflush(stdout); } /* callback to handle questions from libalpm transactions (yes/no) */ /* TODO this is one of the worst ever functions written. void *data ? wtf */ void cb_trans_conv(pmtransconv_t event, void *data1, void *data2, void *data3, int *response) { char str[LOG_STR_LEN] = ""; switch(event) { case PM_TRANS_CONV_INSTALL_IGNOREPKG: if(data2) { /* TODO we take this route based on data2 being not null? WTF */ snprintf(str, LOG_STR_LEN, _(":: %s requires installing %s from IgnorePkg/IgnoreGroup. Install anyway? [Y/n] "), alpm_pkg_get_name(data1), alpm_pkg_get_name(data2)); *response = yesno(str); } else { snprintf(str, LOG_STR_LEN, _(":: %s is in IgnorePkg/IgnoreGroup. Install anyway? [Y/n] "), alpm_pkg_get_name(data1)); *response = yesno(str); } break; case PM_TRANS_CONV_REMOVE_HOLDPKG: snprintf(str, LOG_STR_LEN, _(":: %s is designated as a HoldPkg. Remove anyway? [Y/n] "), alpm_pkg_get_name(data1)); *response = yesno(str); break; case PM_TRANS_CONV_REPLACE_PKG: snprintf(str, LOG_STR_LEN, _(":: Replace %s with %s/%s? [Y/n] "), alpm_pkg_get_name(data1), (char *)data3, alpm_pkg_get_name(data2)); *response = yesno(str); break; case PM_TRANS_CONV_CONFLICT_PKG: snprintf(str, LOG_STR_LEN, _(":: %s conflicts with %s. Remove %s? [Y/n] "), (char *)data1, (char *)data2, (char *)data2); *response = yesno(str); break; case PM_TRANS_CONV_LOCAL_NEWER: if(!config->op_s_downloadonly) { snprintf(str, LOG_STR_LEN, _(":: %s-%s: local version is newer. Upgrade anyway? [Y/n] "), alpm_pkg_get_name(data1), alpm_pkg_get_version(data1)); *response = yesno(str); } else { *response = 1; } break; case PM_TRANS_CONV_CORRUPTED_PKG: if(!config->noconfirm) { snprintf(str, LOG_STR_LEN, _(":: File %s is corrupted. Do you want to delete it? [Y/n] "), (char *)data1); *response = yesno(str); } else { *response = 1; } break; } } /* callback to handle display of transaction progress */ void cb_trans_progress(pmtransprog_t event, const char *pkgname, int percent, int howmany, int remain) { float timediff; /* size of line to allocate for text printing (e.g. not progressbar) */ const int infolen = 50; int tmp, digits, oprlen, textlen, pkglen; char *opr = NULL; wchar_t *wcopr = NULL; if(config->noprogressbar) { return; } if(percent == 0) { timediff = get_update_timediff(1); } else { timediff = get_update_timediff(0); } if(percent > 0 && percent < 100 && !timediff) { /* only update the progress bar when * a) we first start * b) we end the progress * c) it has been long enough since the last call */ return; } /* if no pkgname, percent is too high or unchanged, then return */ if(!pkgname || percent == prevpercent) { return; } prevpercent=percent; /* set text of message to display */ switch (event) { case PM_TRANS_PROGRESS_ADD_START: opr = _("installing"); break; case PM_TRANS_PROGRESS_UPGRADE_START: opr = _("upgrading"); break; case PM_TRANS_PROGRESS_REMOVE_START: opr = _("removing"); break; case PM_TRANS_PROGRESS_CONFLICTS_START: opr = _("checking for file conflicts"); break; } /* convert above strings to wide chars */ oprlen = strlen(opr); wcopr = calloc(oprlen, sizeof(wchar_t)); if(!wcopr) { fprintf(stderr, "malloc failure: could not allocate %zd bytes\n", strlen(opr) * sizeof(wchar_t)); return; } oprlen = mbstowcs(wcopr, opr, oprlen); /* find # of digits in package counts to scale output */ digits = 1; tmp = howmany; while((tmp /= 10)) { ++digits; } /* determine room left for non-digits text [not ( 1/12) part] */ textlen = infolen - 3 - (2 * digits); /* room left for package name */ pkglen = textlen - oprlen - 1; switch (event) { case PM_TRANS_PROGRESS_ADD_START: case PM_TRANS_PROGRESS_UPGRADE_START: case PM_TRANS_PROGRESS_REMOVE_START: /* old way of doing it, but ISO C does not recognize it printf("(%2$*1$d/%3$*1$d) %4$s %6$-*5$.*5$s", digits, remain, howmany, opr, pkglen, pkgname);*/ printf("(%*d/%*d) %s %-*.*s", digits, remain, digits, howmany, opr, pkglen, pkglen, pkgname); break; case PM_TRANS_PROGRESS_CONFLICTS_START: /* old way of doing it, but ISO C does not recognize it printf("(%2$*1$d/%3$*1$d) %5$-*4$s", digits, remain, howmany, textlen, opr);*/ printf("(%*d/%*d) %-*s", digits, remain, digits, howmany, textlen, opr); break; } free(wcopr); /* call refactored fill progress function */ fill_progress(percent, percent, getcols() - infolen); if(percent == 100) { alpm_list_t *i = NULL; on_progress = 0; for(i = output; i; i = i->next) { printf("%s", (char *)i->data); } fflush(stdout); FREELIST(output); } else { on_progress = 1; } } /* callback to handle display of download progress */ void cb_dl_progress(const char *filename, int file_xfered, int file_total, int list_xfered, int list_total) { const int infolen = 50; char *fname, *p; float rate = 0.0, timediff = 0.0, f_xfered = 0.0; unsigned int eta_h = 0, eta_m = 0, eta_s = 0; int graph_percent = 0, display_percent = 0; char rate_size = 'K', xfered_size = 'K'; int xfered = 0, total = 0; /* Need this variable when TotalDownload is set to know if we should * reset xfered_last and rate_last. */ static int has_init = 0; if(config->noprogressbar) { return; } /* Choose how to display the amount downloaded, rate, ETA, and * percentage depending on the TotalDownload option. */ if (config->totaldownload && list_total > 0) { xfered = list_xfered; total = list_total; } else { xfered = file_xfered; total = file_total; } /* this is basically a switch on file_xferred: 0, file_total, and * anything else */ if(file_xfered == 0) { /* set default starting values, but only once for TotalDownload */ if (!(config->totaldownload && list_total > 0) || (config->totaldownload && list_total > 0 && !has_init)) { gettimeofday(&initial_time, NULL); timediff = get_update_timediff(1); xfered_last = 0; rate_last = 0.0; has_init = 1; } rate = 0.0; eta_s = 0; } else if(file_xfered == file_total) { /* compute final values */ struct timeval current_time; float diff_sec, diff_usec; gettimeofday(¤t_time, NULL); diff_sec = current_time.tv_sec - initial_time.tv_sec; diff_usec = current_time.tv_usec - initial_time.tv_usec; timediff = diff_sec + (diff_usec / 1000000.0); rate = xfered / (timediff * 1024.0); /* round elapsed time to the nearest second */ eta_s = (int)(timediff + 0.5); } else { /* compute current average values */ timediff = get_update_timediff(0); if(timediff < UPDATE_SPEED_SEC) { /* return if the calling interval was too short */ return; } rate = (xfered - xfered_last) / (timediff * 1024.0); /* average rate to reduce jumpiness */ rate = (rate + 2*rate_last) / 3; eta_s = (total - xfered) / (rate * 1024.0); rate_last = rate; xfered_last = xfered; } /* fix up time for display */ eta_h = eta_s / 3600; eta_s -= eta_h * 3600; eta_m = eta_s / 60; eta_s -= eta_m * 60; fname = strdup(filename); /* strip package or DB extension for cleaner look */ if((p = strstr(fname, PKGEXT)) || (p = strstr(fname, DBEXT))) { *p = '\0'; } if(strlen(fname) > FILENAME_TRIM_LEN) { strcpy(fname + FILENAME_TRIM_LEN -3,"..."); } /* Awesome formatting for progress bar. We need a mess of Kb->Mb->Gb stuff * here. We'll use limit of 2048 for each until we get some empirical */ /* rate_size = 'K'; was set above */ if(rate > 2048.0) { rate /= 1024.0; rate_size = 'M'; if(rate > 2048.0) { rate /= 1024.0; rate_size = 'G'; /* we should not go higher than this for a few years (9999.9 Gb/s?)*/ } } f_xfered = xfered / 1024.0; /* convert to K by default */ /* xfered_size = 'K'; was set above */ if(f_xfered > 2048.0) { f_xfered /= 1024.0; xfered_size = 'M'; if(f_xfered > 2048.0) { f_xfered /= 1024.0; xfered_size = 'G'; /* I should seriously hope that archlinux packages never break * the 9999.9GB mark... we'd have more serious problems than the progress * bar in pacman */ } } printf(" %-*s %6.1f%c %#6.1f%c/s %02u:%02u:%02u", FILENAME_TRIM_LEN, fname, f_xfered, xfered_size, rate, rate_size, eta_h, eta_m, eta_s); free(fname); /* The progress bar is based on the file percent regardless of the * TotalDownload option. */ graph_percent = (int)((float)file_xfered) / ((float)file_total) * 100; display_percent = (int)((float)xfered) / ((float)total) * 100; fill_progress(graph_percent, display_percent, getcols() - infolen); return; } /* Callback to handle notifications from the library */ void cb_log(pmloglevel_t level, char *fmt, va_list args) { if(!strlen(fmt)) { return; } if(on_progress) { char *string = NULL; pm_vasprintf(&string, level, fmt, args); if(string != NULL) { output = alpm_list_add(output, string); } } else { pm_vfprintf(stdout, level, fmt, args); } } /* vim: set ts=2 sw=2 noet: */