[hacktree] Make link-file work

[Colin Walters <walters src gnome org>]

commit 4dd6800389114c9d711c9c6dab2f3ba247b05437
Author: Colin Walters <walters verbum org>
Date:   Tue Oct 11 22:05:16 2011 -0400

    Make link-file work

 Makefile-hacktree.am            |    3 +
 src/ht-builtin-link-file.c      |   59 ++++++++++++++++
 src/ht-builtins.h               |    1 +
 src/libhacktree/hacktree-repo.c |  145 ++++++++++++++++++++++++++------------
 src/libhacktree/hacktree-repo.h |    6 +-
 src/main.c                      |    1 +
 6 files changed, 166 insertions(+), 49 deletions(-)
---
diff --git a/Makefile-hacktree.am b/Makefile-hacktree.am
index dcbc95b..308f3c0 100644
--- a/Makefile-hacktree.am
+++ b/Makefile-hacktree.am
@@ -3,6 +3,8 @@ noinst_LTLIBRARIES += libhtutil.la
 libhtutil_la_SOURCES = \
 	src/libhtutil/ht-unix-utils.c \
 	src/libhtutil/ht-unix-utils.h \
+	src/libhtutil/ht-gio-utils.c \
+	src/libhtutil/ht-gio-utils.h \
 	src/libhtutil/htutil.h \
 	$(NULL)
 libhtutil_la_CFLAGS = -I$(srcdir)/src/libhtutil -DLOCALEDIR=\"$(datadir)/locale\" $(GIO_UNIX_CFLAGS)
@@ -23,6 +25,7 @@ bin_PROGRAMS += hacktree
 hacktree_SOURCES = src/main.c \
 	src/ht-builtins.h \
 	src/ht-builtin-init.c \
+	src/ht-builtin-link-file.c \
 	$(NULL)
 hacktree_CFLAGS = -I$(srcdir)/src -I$(srcdir)/src/libhacktree -I$(srcdir)/src/libhtutil -DLOCALEDIR=\"$(datadir)/locale\" $(GIO_UNIX_CFLAGS)
 hacktree_LDADD = libhtutil.la libhacktree.la $(GIO_UNIX_LIBS)
diff --git a/src/ht-builtin-link-file.c b/src/ht-builtin-link-file.c
new file mode 100644
index 0000000..5944565
--- /dev/null
+++ b/src/ht-builtin-link-file.c
@@ -0,0 +1,59 @@
+/* -*- mode: C; c-file-style: "gnu"; indent-tabs-mode: nil; -*-
+ *
+ * Copyright (C) 2011 Colin Walters <walters verbum 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, write to the Free Software
+ * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
+ *
+ * Author: Colin Walters <walters verbum org>
+ */
+
+#include "config.h"
+
+#include "ht-builtins.h"
+#include "hacktree.h"
+
+#include <glib/gi18n.h>
+
+static char *repo_path;
+static GOptionEntry options[] = {
+  { "repo", 0, 0, G_OPTION_ARG_FILENAME, &repo_path, "Repository path", NULL },
+  { NULL }
+};
+
+gboolean
+hacktree_builtin_link_file (int argc, const char **argv, const char *prefix, GError **error)
+{
+  gboolean ret = FALSE;
+  HacktreeRepo *repo = NULL;
+  int i;
+
+  if (repo_path == NULL)
+    repo_path = ".";
+
+  repo = hacktree_repo_new (repo_path);
+  if (!hacktree_repo_check (repo, error))
+    goto out;
+
+  for (i = 0; i < argc; i++)
+    {
+      if (!hacktree_repo_link_file (repo, argv[i], error))
+        goto out;
+    }
+ 
+  ret = TRUE;
+ out:
+  g_clear_object (&repo);
+  return ret;
+}
diff --git a/src/ht-builtins.h b/src/ht-builtins.h
index c40dc73..fee9777 100644
--- a/src/ht-builtins.h
+++ b/src/ht-builtins.h
@@ -37,6 +37,7 @@ typedef struct {
 } HacktreeBuiltin;
 
 gboolean hacktree_builtin_init (int argc, const char **argv, const char *prefix, GError **error);
+gboolean hacktree_builtin_link_file (int argc, const char **argv, const char *prefix, GError **error);
 
 G_END_DECLS
 
diff --git a/src/libhacktree/hacktree-repo.c b/src/libhacktree/hacktree-repo.c
index c51ec5d..bb0a9b8 100644
--- a/src/libhacktree/hacktree-repo.c
+++ b/src/libhacktree/hacktree-repo.c
@@ -21,7 +21,7 @@
 
 #include "config.h"
 
-#include "hacktree-repo.h"
+#include "hacktree.h"
 #include "htutil.h"
 
 enum {
@@ -47,6 +47,12 @@ struct _HacktreeRepoPrivate {
 static void
 hacktree_repo_finalize (GObject *object)
 {
+  HacktreeRepo *self = HACKTREE_REPO (object);
+  HacktreeRepoPrivate *priv = GET_PRIVATE (self);
+
+  g_free (priv->path);
+  g_free (priv->objects_path);
+
   G_OBJECT_CLASS (hacktree_repo_parent_class)->finalize (object);
 }
 
@@ -90,6 +96,27 @@ hacktree_repo_get_property(GObject         *object,
     }
 }
 
+static GObject *
+hacktree_repo_constructor (GType                  gtype,
+                           guint                  n_properties,
+                           GObjectConstructParam *properties)
+{
+  GObject *object;
+  GObjectClass *parent_class;
+  HacktreeRepoPrivate *priv;
+
+  parent_class = G_OBJECT_CLASS (hacktree_repo_parent_class);
+  object = parent_class->constructor (gtype, n_properties, properties);
+
+  priv = GET_PRIVATE (object);
+
+  g_assert (priv->path != NULL);
+
+  priv->objects_path = g_build_filename (priv->path, HACKTREE_REPO_DIR, "objects", NULL);
+
+  return object;
+}
+
 static void
 hacktree_repo_class_init (HacktreeRepoClass *klass)
 {
@@ -97,6 +124,9 @@ hacktree_repo_class_init (HacktreeRepoClass *klass)
 
   g_type_class_add_private (klass, sizeof (HacktreeRepoPrivate));
 
+  object_class->constructor = hacktree_repo_constructor;
+  object_class->get_property = hacktree_repo_get_property;
+  object_class->set_property = hacktree_repo_set_property;
   object_class->finalize = hacktree_repo_finalize;
 
   g_object_class_install_property (object_class,
@@ -113,9 +143,6 @@ hacktree_repo_init (HacktreeRepo *self)
 {
   HacktreeRepoPrivate *priv = GET_PRIVATE (self);
   
-  g_assert (priv->path);
-
-  priv->objects_path = g_build_filename (priv->path, "objects", NULL);
 }
 
 HacktreeRepo*
@@ -128,6 +155,13 @@ gboolean
 hacktree_repo_check (HacktreeRepo *self, GError **error)
 {
   HacktreeRepoPrivate *priv = GET_PRIVATE (self);
+
+  if (!g_file_test (priv->objects_path, G_FILE_TEST_IS_DIR))
+    {
+      g_set_error (error, G_IO_ERROR, G_IO_ERROR_FAILED,
+                   "Couldn't find objects directory '%s'", priv->objects_path);
+      return FALSE;
+    }
   
   priv->inited = TRUE;
   return TRUE;
@@ -188,13 +222,23 @@ stat_and_compute_checksum (int dirfd, const char *path,
   char *symlink_target = NULL;
   char *device_id = NULL;
 
-  if (fstat (fd, out_stbuf) < 0)
+  basename = g_path_get_basename (path);
+
+  if (fstatat (dirfd, basename, out_stbuf, AT_SYMLINK_NOFOLLOW) < 0)
     {
       ht_util_set_error_from_errno (error, errno);
       goto out;
     }
 
-  basename = g_path_get_basename (path);
+  if (!S_ISLNK(out_stbuf->st_mode))
+    {
+      fd = ht_util_open_file_read_at (dirfd, basename, error);
+      if (fd < 0)
+        {
+          ht_util_set_error_from_errno (error, errno);
+          goto out;
+        }
+    }
 
   stat_string = stat_to_string (out_stbuf);
 
@@ -237,12 +281,6 @@ stat_and_compute_checksum (int dirfd, const char *path,
     {
       guint8 buf[8192];
 
-      fd = ht_util_open_file_read_at (dirfd, basename, error);
-      if (fd < 0)
-        {
-          ht_util_set_error_from_errno (error, errno);
-          goto out;
-        }
       while ((bytes_read = read (fd, buf, sizeof (buf))) > 0)
         g_checksum_update (content_sha256, buf, bytes_read);
       if (bytes_read < 0)
@@ -262,7 +300,7 @@ stat_and_compute_checksum (int dirfd, const char *path,
         }
       g_checksum_update (content_sha256, symlink_target, strlen (symlink_target));
     }
-  else if (S_ISDEV(out_stbuf->st_mode))
+  else if (S_ISCHR(out_stbuf->st_mode) || S_ISBLK(out_stbuf->st_mode))
     {
       device_id = g_strdup_printf ("%d", out_stbuf->st_rdev);
       g_checksum_update (content_sha256, device_id, strlen (device_id));
@@ -282,6 +320,7 @@ stat_and_compute_checksum (int dirfd, const char *path,
   g_checksum_update (content_and_meta_sha256, xattrs_canonicalized, strlen (stat_string));
 
   *out_checksum = content_and_meta_sha256;
+  ret = TRUE;
  out:
   if (fd >= 0)
     close (fd);
@@ -297,63 +336,73 @@ stat_and_compute_checksum (int dirfd, const char *path,
 }
 
 static char *
-get_object_dir_for_checksum (HacktreeRepo *self,
-                             GChecksum    *checksum,
-                             GError      **error)
+prepare_dir_for_checksum_get_object_path (HacktreeRepo *self,
+                                          GChecksum    *checksum,
+                                          GError      **error)
 {
   HacktreeRepoPrivate *priv = GET_PRIVATE (self);
-  char *checksum_prefix;
-  char *path;
-  gboolean ret = FALSE;
+  char *checksum_prefix = NULL;
+  char *checksum_dir = NULL;
+  char *object_path = NULL;
   GError *temp_error = NULL;
 
   checksum_prefix = g_strdup (g_checksum_get_string (checksum));
-  checksum_prefix[3] = '\0';
-  path = g_build_filename (priv->objects_path, checksum_prefix, NULL);
+  checksum_prefix[2] = '\0';
+  checksum_dir = g_build_filename (priv->objects_path, checksum_prefix, NULL);
 
-  if (!ht_util_ensure_directory (path, FALSE, error))
+  if (!ht_util_ensure_directory (checksum_dir, FALSE, error))
     goto out;
   
-  ret = TRUE;
+  object_path = g_build_filename (checksum_dir, checksum_prefix + 3, NULL);
  out:
   g_free (checksum_prefix);
-  if (ret)
-    return path;
-  g_free (path);
-  return NULL;
+  g_free (checksum_dir);
+  return object_path;
 }
 
 static gboolean
-import_one_file (HacktreeRepo *self, const char *path, GError **error)
+link_one_file (HacktreeRepo *self, const char *path, GError **error)
 {
   HacktreeRepoPrivate *priv = GET_PRIVATE (self);
-  char *basename = NULL;
-  char *dirname = NULL;
+  char *src_basename = NULL;
+  char *src_dirname = NULL;
+  char *dest_basename = NULL;
+  char *dest_dirname = NULL;
   GChecksum *id = NULL;
-  DIR *dir = NULL;
+  DIR *src_dir = NULL;
+  DIR *dest_dir = NULL;
   gboolean ret = FALSE;
   int fd;
   struct stat stbuf;
   char *dest_path = NULL;
   char *checksum_prefix;
 
-  basename = g_path_get_dirname (path);
-  dirname = g_path_get_dirname (path);
+  src_basename = g_path_get_basename (path);
+  src_dirname = g_path_get_dirname (path);
 
-  dir = opendir (dirname);
-  if (dir == NULL)
+  src_dir = opendir (src_dirname);
+  if (src_dir == NULL)
     {
       ht_util_set_error_from_errno (error, errno);
       goto out;
     }
 
-  if (!stat_and_compute_checksum (dirfd (dir), path, &id, &stbuf, error))
+  if (!stat_and_compute_checksum (dirfd (src_dir), path, &id, &stbuf, error))
     goto out;
-  dest_path = get_object_dir_for_checksum (self, id, error);
+  dest_path = prepare_dir_for_checksum_get_object_path (self, id, error);
   if (!dest_path)
     goto out;
+
+  dest_basename = g_path_get_basename (dest_path);
+  dest_dirname = g_path_get_dirname (dest_path);
+  dest_dir = opendir (dest_dirname);
+  if (dest_dir == NULL)
+    {
+      ht_util_set_error_from_errno (error, errno);
+      goto out;
+    }
   
-  if (linkat (dirfd, basename, -1, dest_path, 0) < 0)
+  if (linkat (dirfd (src_dir), src_basename, dirfd (dest_dir), dest_basename, 0) < 0)
     {
       ht_util_set_error_from_errno (error, errno);
       goto out;
@@ -363,21 +412,25 @@ import_one_file (HacktreeRepo *self, const char *path, GError **error)
  out:
   if (id != NULL)
     g_checksum_free (id);
-  if (dir != NULL)
-    closedir (dir);
-  g_free (basename);
-  g_free (dirname);
+  if (src_dir != NULL)
+    closedir (src_dir);
+  if (dest_dir != NULL)
+    closedir (dest_dir);
+  g_free (src_basename);
+  g_free (src_dirname);
+  g_free (dest_basename);
+  g_free (dest_dirname);
   return ret;
 }
 
 gboolean
-hacktree_repo_import (HacktreeRepo *self,
-                      const char   *path,
-                      GError      **error)
+hacktree_repo_link_file (HacktreeRepo *self,
+                         const char   *path,
+                         GError      **error)
 {
   HacktreeRepoPrivate *priv = GET_PRIVATE (self);
 
   g_return_val_if_fail (priv->inited, FALSE);
 
-  return import_one_file (self, path, error);
+  return link_one_file (self, path, error);
 }
diff --git a/src/libhacktree/hacktree-repo.h b/src/libhacktree/hacktree-repo.h
index 2d7e59a..715560e 100644
--- a/src/libhacktree/hacktree-repo.h
+++ b/src/libhacktree/hacktree-repo.h
@@ -53,9 +53,9 @@ HacktreeRepo* hacktree_repo_new (const char *path);
 
 gboolean      hacktree_repo_check (HacktreeRepo  *repo, GError **error);
 
-gboolean      hacktree_repo_import (HacktreeRepo *repo,
-                                    const char   *path,
-                                    GError      **error);
+gboolean      hacktree_repo_link_file (HacktreeRepo *repo,
+                                       const char   *path,
+                                       GError      **error);
 
 G_END_DECLS
 
diff --git a/src/main.c b/src/main.c
index 1223bac..df4022e 100644
--- a/src/main.c
+++ b/src/main.c
@@ -30,6 +30,7 @@
 
 static HacktreeBuiltin builtins[] = {
   { "init", hacktree_builtin_init, 0 },
+  { "link-file", hacktree_builtin_link_file, 0 },
   { NULL }
 };