]> git.ipfire.org Git - thirdparty/git.git/blobdiff - contrib/scalar/scalar.c
Merge branch 'vd/scalar-enables-fsmonitor'
[thirdparty/git.git] / contrib / scalar / scalar.c
index 07c3f7dd6b6672616a43df83b7324e69b4f83f71..642d16124eb20c28de20e8601e99f1b347c7fa75 100644 (file)
@@ -14,8 +14,6 @@
 #include "dir.h"
 #include "packfile.h"
 #include "help.h"
-#include "archive.h"
-#include "object-store.h"
 
 static void setup_enlistment_directory(int argc, const char **argv,
                                       const char * const *usagestr,
@@ -285,99 +283,6 @@ static int unregister_dir(void)
        return res;
 }
 
-static int add_directory_to_archiver(struct strvec *archiver_args,
-                                         const char *path, int recurse)
-{
-       int at_root = !*path;
-       DIR *dir = opendir(at_root ? "." : path);
-       struct dirent *e;
-       struct strbuf buf = STRBUF_INIT;
-       size_t len;
-       int res = 0;
-
-       if (!dir)
-               return error_errno(_("could not open directory '%s'"), path);
-
-       if (!at_root)
-               strbuf_addf(&buf, "%s/", path);
-       len = buf.len;
-       strvec_pushf(archiver_args, "--prefix=%s", buf.buf);
-
-       while (!res && (e = readdir(dir))) {
-               if (!strcmp(".", e->d_name) || !strcmp("..", e->d_name))
-                       continue;
-
-               strbuf_setlen(&buf, len);
-               strbuf_addstr(&buf, e->d_name);
-
-               if (e->d_type == DT_REG)
-                       strvec_pushf(archiver_args, "--add-file=%s", buf.buf);
-               else if (e->d_type != DT_DIR)
-                       warning(_("skipping '%s', which is neither file nor "
-                                 "directory"), buf.buf);
-               else if (recurse &&
-                        add_directory_to_archiver(archiver_args,
-                                                  buf.buf, recurse) < 0)
-                       res = -1;
-       }
-
-       closedir(dir);
-       strbuf_release(&buf);
-       return res;
-}
-
-#ifndef WIN32
-#include <sys/statvfs.h>
-#endif
-
-static int get_disk_info(struct strbuf *out)
-{
-#ifdef WIN32
-       struct strbuf buf = STRBUF_INIT;
-       char volume_name[MAX_PATH], fs_name[MAX_PATH];
-       DWORD serial_number, component_length, flags;
-       ULARGE_INTEGER avail2caller, total, avail;
-
-       strbuf_realpath(&buf, ".", 1);
-       if (!GetDiskFreeSpaceExA(buf.buf, &avail2caller, &total, &avail)) {
-               error(_("could not determine free disk size for '%s'"),
-                     buf.buf);
-               strbuf_release(&buf);
-               return -1;
-       }
-
-       strbuf_setlen(&buf, offset_1st_component(buf.buf));
-       if (!GetVolumeInformationA(buf.buf, volume_name, sizeof(volume_name),
-                                  &serial_number, &component_length, &flags,
-                                  fs_name, sizeof(fs_name))) {
-               error(_("could not get info for '%s'"), buf.buf);
-               strbuf_release(&buf);
-               return -1;
-       }
-       strbuf_addf(out, "Available space on '%s': ", buf.buf);
-       strbuf_humanise_bytes(out, avail2caller.QuadPart);
-       strbuf_addch(out, '\n');
-       strbuf_release(&buf);
-#else
-       struct strbuf buf = STRBUF_INIT;
-       struct statvfs stat;
-
-       strbuf_realpath(&buf, ".", 1);
-       if (statvfs(buf.buf, &stat) < 0) {
-               error_errno(_("could not determine free disk size for '%s'"),
-                           buf.buf);
-               strbuf_release(&buf);
-               return -1;
-       }
-
-       strbuf_addf(out, "Available space on '%s': ", buf.buf);
-       strbuf_humanise_bytes(out, st_mult(stat.f_bsize, stat.f_bavail));
-       strbuf_addf(out, " (mount flags 0x%lx)\n", stat.f_flag);
-       strbuf_release(&buf);
-#endif
-       return 0;
-}
-
 /* printf-style interface, expects `<key>=<value>` argument */
 static int set_config(const char *fmt, ...)
 {
@@ -628,83 +533,6 @@ cleanup:
        return res;
 }
 
-static void dir_file_stats_objects(const char *full_path, size_t full_path_len,
-                                  const char *file_name, void *data)
-{
-       struct strbuf *buf = data;
-       struct stat st;
-
-       if (!stat(full_path, &st))
-               strbuf_addf(buf, "%-70s %16" PRIuMAX "\n", file_name,
-                           (uintmax_t)st.st_size);
-}
-
-static int dir_file_stats(struct object_directory *object_dir, void *data)
-{
-       struct strbuf *buf = data;
-
-       strbuf_addf(buf, "Contents of %s:\n", object_dir->path);
-
-       for_each_file_in_pack_dir(object_dir->path, dir_file_stats_objects,
-                                 data);
-
-       return 0;
-}
-
-static int count_files(char *path)
-{
-       DIR *dir = opendir(path);
-       struct dirent *e;
-       int count = 0;
-
-       if (!dir)
-               return 0;
-
-       while ((e = readdir(dir)) != NULL)
-               if (!is_dot_or_dotdot(e->d_name) && e->d_type == DT_REG)
-                       count++;
-
-       closedir(dir);
-       return count;
-}
-
-static void loose_objs_stats(struct strbuf *buf, const char *path)
-{
-       DIR *dir = opendir(path);
-       struct dirent *e;
-       int count;
-       int total = 0;
-       unsigned char c;
-       struct strbuf count_path = STRBUF_INIT;
-       size_t base_path_len;
-
-       if (!dir)
-               return;
-
-       strbuf_addstr(buf, "Object directory stats for ");
-       strbuf_add_absolute_path(buf, path);
-       strbuf_addstr(buf, ":\n");
-
-       strbuf_add_absolute_path(&count_path, path);
-       strbuf_addch(&count_path, '/');
-       base_path_len = count_path.len;
-
-       while ((e = readdir(dir)) != NULL)
-               if (!is_dot_or_dotdot(e->d_name) &&
-                   e->d_type == DT_DIR && strlen(e->d_name) == 2 &&
-                   !hex_to_bytes(&c, e->d_name, 1)) {
-                       strbuf_setlen(&count_path, base_path_len);
-                       strbuf_addstr(&count_path, e->d_name);
-                       total += (count = count_files(count_path.buf));
-                       strbuf_addf(buf, "%s : %7d files\n", e->d_name, count);
-               }
-
-       strbuf_addf(buf, "Total: %d loose objects", total);
-
-       strbuf_release(&count_path);
-       closedir(dir);
-}
-
 static int cmd_diagnose(int argc, const char **argv)
 {
        struct option options[] = {
@@ -714,106 +542,19 @@ static int cmd_diagnose(int argc, const char **argv)
                N_("scalar diagnose [<enlistment>]"),
                NULL
        };
-       struct strbuf zip_path = STRBUF_INIT;
-       struct strvec archiver_args = STRVEC_INIT;
-       char **argv_copy = NULL;
-       int stdout_fd = -1, archiver_fd = -1;
-       time_t now = time(NULL);
-       struct tm tm;
-       struct strbuf buf = STRBUF_INIT;
+       struct strbuf diagnostics_root = STRBUF_INIT;
        int res = 0;
 
        argc = parse_options(argc, argv, NULL, options,
                             usage, 0);
 
-       setup_enlistment_directory(argc, argv, usage, options, &zip_path);
-
-       strbuf_addstr(&zip_path, "/.scalarDiagnostics/scalar_");
-       strbuf_addftime(&zip_path,
-                       "%Y%m%d_%H%M%S", localtime_r(&now, &tm), 0, 0);
-       strbuf_addstr(&zip_path, ".zip");
-       switch (safe_create_leading_directories(zip_path.buf)) {
-       case SCLD_EXISTS:
-       case SCLD_OK:
-               break;
-       default:
-               error_errno(_("could not create directory for '%s'"),
-                           zip_path.buf);
-               goto diagnose_cleanup;
-       }
-       stdout_fd = dup(1);
-       if (stdout_fd < 0) {
-               res = error_errno(_("could not duplicate stdout"));
-               goto diagnose_cleanup;
-       }
-
-       archiver_fd = xopen(zip_path.buf, O_CREAT | O_WRONLY | O_TRUNC, 0666);
-       if (archiver_fd < 0 || dup2(archiver_fd, 1) < 0) {
-               res = error_errno(_("could not redirect output"));
-               goto diagnose_cleanup;
-       }
-
-       init_zip_archiver();
-       strvec_pushl(&archiver_args, "scalar-diagnose", "--format=zip", NULL);
-
-       strbuf_reset(&buf);
-       strbuf_addstr(&buf, "Collecting diagnostic info\n\n");
-       get_version_info(&buf, 1);
+       setup_enlistment_directory(argc, argv, usage, options, &diagnostics_root);
+       strbuf_addstr(&diagnostics_root, "/.scalarDiagnostics");
 
-       strbuf_addf(&buf, "Enlistment root: %s\n", the_repository->worktree);
-       get_disk_info(&buf);
-       write_or_die(stdout_fd, buf.buf, buf.len);
-       strvec_pushf(&archiver_args,
-                    "--add-virtual-file=diagnostics.log:%.*s",
-                    (int)buf.len, buf.buf);
-
-       strbuf_reset(&buf);
-       strbuf_addstr(&buf, "--add-virtual-file=packs-local.txt:");
-       dir_file_stats(the_repository->objects->odb, &buf);
-       foreach_alt_odb(dir_file_stats, &buf);
-       strvec_push(&archiver_args, buf.buf);
-
-       strbuf_reset(&buf);
-       strbuf_addstr(&buf, "--add-virtual-file=objects-local.txt:");
-       loose_objs_stats(&buf, ".git/objects");
-       strvec_push(&archiver_args, buf.buf);
-
-       if ((res = add_directory_to_archiver(&archiver_args, ".git", 0)) ||
-           (res = add_directory_to_archiver(&archiver_args, ".git/hooks", 0)) ||
-           (res = add_directory_to_archiver(&archiver_args, ".git/info", 0)) ||
-           (res = add_directory_to_archiver(&archiver_args, ".git/logs", 1)) ||
-           (res = add_directory_to_archiver(&archiver_args, ".git/objects/info", 0)))
-               goto diagnose_cleanup;
-
-       strvec_pushl(&archiver_args, "--prefix=",
-                    oid_to_hex(the_hash_algo->empty_tree), "--", NULL);
-
-       /* `write_archive()` modifies the `argv` passed to it. Let it. */
-       argv_copy = xmemdupz(archiver_args.v,
-                            sizeof(char *) * archiver_args.nr);
-       res = write_archive(archiver_args.nr, (const char **)argv_copy, NULL,
-                           the_repository, NULL, 0);
-       if (res) {
-               error(_("failed to write archive"));
-               goto diagnose_cleanup;
-       }
-
-       if (!res)
-               fprintf(stderr, "\n"
-                      "Diagnostics complete.\n"
-                      "All of the gathered info is captured in '%s'\n",
-                      zip_path.buf);
-
-diagnose_cleanup:
-       if (archiver_fd >= 0) {
-               close(1);
-               dup2(stdout_fd, 1);
-       }
-       free(argv_copy);
-       strvec_clear(&archiver_args);
-       strbuf_release(&zip_path);
-       strbuf_release(&buf);
+       res = run_git("diagnose", "--mode=all", "-s", "%Y%m%d_%H%M%S",
+                     "-o", diagnostics_root.buf, NULL);
 
+       strbuf_release(&diagnostics_root);
        return res;
 }