gio/tests/gio-du.c

   1 #include <gio/gio.h>
   2 #include <locale.h>
   3
   4 static gboolean option_use_async;
   5 static gint     option_format_size;
   6
   7 static gint     outstanding_asyncs;
   8
   9 static void
  10 print_result (const gchar *filename,
  11               guint64      disk_usage,
  12               guint64      num_dirs,
  13               guint64      num_files,
  14               GError      *error,
  15               gchar        nl)
  16 {
  17   if (!error)
  18     {
  19       if (option_format_size)
  20         {
  21           GFormatSizeFlags format_flags;
  22           gchar *str;
  23
  24           format_flags = (option_format_size > 1) ? G_FORMAT_SIZE_LONG_FORMAT : G_FORMAT_SIZE_DEFAULT;
  25           str = g_format_size_full (disk_usage, format_flags);
  26           g_print ("%s: %s (%"G_GUINT64_FORMAT" dirs, %"G_GUINT64_FORMAT" files)%c",
  27                    filename, str, num_dirs, num_files, nl);
  28           g_free (str);
  29         }
  30       else
  31         g_print ("%s: %"G_GUINT64_FORMAT" (%"G_GUINT64_FORMAT" dirs, %"G_GUINT64_FORMAT" files)%c",
  32                  filename, disk_usage, num_dirs, num_files, nl);
  33     }
  34   else
  35     {
  36       g_printerr ("%s: %s\n", filename, error->message);
  37       g_error_free (error);
  38     }
  39 }
  40
  41 static void
  42 async_ready_func (GObject      *source,
  43                   GAsyncResult *result,
  44                   gpointer      user_data)
  45 {
  46   gchar *filename = user_data;
  47   GError *error = NULL;
  48   guint64 disk_usage;
  49   guint64 num_dirs;
  50   guint64 num_files;
  51
  52   g_file_measure_disk_usage_finish (G_FILE (source), result, &disk_usage, &num_dirs, &num_files, &error);
  53   print_result (filename, disk_usage, num_dirs, num_files, error, '\n');
  54   outstanding_asyncs--;
  55   g_free (filename);
  56 }
  57
  58 static void
  59 report_progress (gboolean reporting,
  60                  guint64  disk_usage,
  61                  guint64  num_dirs,
  62                  guint64  num_files,
  63                  gpointer user_data)
  64 {
  65   const gchar *filename = user_data;
  66
  67   if (!reporting)
  68     g_printerr ("%s: warning: does not support progress reporting\n", filename);
  69
  70   print_result (filename, disk_usage, num_dirs, num_files, NULL, '\r');
  71 }
  72
  73 int
  74 main (int argc, char **argv)
  75 {
  76   GFileMeasureProgressCallback progress = NULL;
  77   GFileMeasureFlags flags = 0;
  78   gint i;
  79
  80 #ifdef G_OS_WIN32
  81   argv = g_win32_get_command_line ();
  82 #endif
  83
  84   setlocale (LC_ALL, "");
  85
  86
  87
  88   for (i = 1; argv[i] && argv[i][0] == '-'; i++)
  89     {
  90       if (g_str_equal (argv[i], "--"))
  91         break;
  92
  93       if (g_str_equal (argv[i], "--help"))
  94         {
  95           g_print ("usage: du [--progress] [--async] [-x] [-h] [-h] [--apparent-size] [--any-error] [--] files...\n");
  96 #ifdef G_OS_WIN32
  97           g_strfreev (argv);
  98 #endif
  99           return 0;
 100         }
 101       else if (g_str_equal (argv[i], "-x"))
 102         flags |= G_FILE_MEASURE_NO_XDEV;
 103       else if (g_str_equal (argv[i], "-h"))
 104         option_format_size++;
 105       else if (g_str_equal (argv[i], "--apparent-size"))
 106         flags |= G_FILE_MEASURE_APPARENT_SIZE;
 107       else if (g_str_equal (argv[i], "--any-error"))
 108         flags |= G_FILE_MEASURE_REPORT_ANY_ERROR;
 109       else if (g_str_equal (argv[i], "--async"))
 110         option_use_async = TRUE;
 111       else if (g_str_equal (argv[i], "--progress"))
 112         progress = report_progress;
 113       else
 114         {
 115           g_printerr ("unrecognised flag %s\n", argv[i]);
 116         }
 117     }
 118
 119   if (!argv[i])
 120     {
 121       g_printerr ("usage: du [--progress] [--async] [-x] [-h] [-h] [--apparent-size] [--any-error] [--] files...\n");
 122 #ifdef G_OS_WIN32
 123       g_strfreev (argv);
 124 #endif
 125       return 1;
 126     }
 127
 128   while (argv[i])
 129   {
 130     GFile *file = g_file_new_for_commandline_arg (argv[i]);
 131
 132     if (option_use_async)
 133     {
 134       g_file_measure_disk_usage_async (file, flags, G_PRIORITY_DEFAULT, NULL,
 135                                        progress, argv[i], async_ready_func, argv[i]);
 136       outstanding_asyncs++;
 137     }
 138     else
 139     {
 140       GError *error = NULL;
 141       guint64 disk_usage;
 142       guint64 num_dirs;
 143       guint64 num_files;
 144
 145       g_file_measure_disk_usage (file, flags, NULL, progress, argv[i],
 146                                  &disk_usage, &num_dirs, &num_files, &error);
 147       print_result (argv[i], disk_usage, num_dirs, num_files, error, '\n');
 148     }
 149
 150     g_object_unref (file);
 151
 152     i++;
 153   }
 154
 155   while (outstanding_asyncs)
 156     g_main_context_iteration (NULL, TRUE);
 157
 158 #ifdef G_OS_WIN32
 159   g_strfreev (argv);
 160 #endif
 161
 162   return 0;
 163 }