X-Git-Url: http://review.tizen.org/git/?a=blobdiff_plain;f=tools%2Fcommon_write.c;h=884b4179639656e8b6d88117f767db25e7409a96;hb=4ccae3859358e9518cb67b101d0b62138f7554eb;hp=78623388511600202b8cc3466e70207080843e2c;hpb=b33428b853d91ff15eee32fd1407ae0924a56bb7;p=platform%2Fupstream%2Flibsolv.git

diff --git a/tools/common_write.c b/tools/common_write.c
index 7862338..884b417 100644
--- a/tools/common_write.c
+++ b/tools/common_write.c
@@ -17,82 +17,281 @@
 #include "repo_write.h"
 #include "common_write.h"
 
-static char *verticals[] = {
-  "authors",
-  "description",
-  "messagedel",
-  "messageins",
-  "eula",
-  "diskusage",
+static Id verticals[] = {
+  SOLVABLE_AUTHORS,
+  SOLVABLE_DESCRIPTION,
+  SOLVABLE_MESSAGEDEL,
+  SOLVABLE_MESSAGEINS,
+  SOLVABLE_EULA,
+  SOLVABLE_DISKUSAGE,
+  SOLVABLE_FILELIST,
   0
 };
 
-static unsigned char *filter;
-static int nfilter;
-
-static void
-create_filter(Pool *pool)
-{
-  char **s;
-  Id id;
-  for (s = verticals; *s; s++)
-    {
-      id = str2id(pool, *s, 1);
-      if (id >= nfilter)
-	{
-	  filter = sat_realloc(filter, id + 16);
-	  memset(filter + nfilter, 0, id + 16 - nfilter);
-	  nfilter = id + 16;
-	}
-      filter[id] = 1;
-    }
-}
+static char *languagetags[] = {
+  "solvable:summary:",
+  "solvable:description:",
+  "solvable:messageins:",
+  "solvable:messagedel:",
+  "solvable:eula:",
+  0
+};
 
 static int test_separate = 0;
 
 static int
 keyfilter_solv(Repo *data, Repokey *key, void *kfdata)
 {
+  int i;
+  const char *keyname;
   if (test_separate && key->storage != KEY_STORAGE_SOLVABLE)
     return KEY_STORAGE_DROPPED;
-  if (key->name < nfilter && filter[key->name])
-    return KEY_STORAGE_VERTICAL_OFFSET;
+  for (i = 0; verticals[i]; i++)
+    if (key->name == verticals[i])
+      return KEY_STORAGE_VERTICAL_OFFSET;
+  keyname = id2str(data->pool, key->name);
+  for (i = 0; languagetags[i] != 0; i++)
+    if (!strncmp(keyname, languagetags[i], strlen(languagetags[i])))
+      return KEY_STORAGE_VERTICAL_OFFSET;
   return KEY_STORAGE_INCORE;
 }
 
 static int
 keyfilter_attr(Repo *data, Repokey *key, void *kfdata)
 {
+  int i;
+  const char *keyname;
   if (key->storage == KEY_STORAGE_SOLVABLE)
     return KEY_STORAGE_DROPPED;
-  if (key->name < nfilter && filter[key->name])
-    return KEY_STORAGE_VERTICAL_OFFSET;
+  for (i = 0; verticals[i]; i++)
+    if (key->name == verticals[i])
+      return KEY_STORAGE_VERTICAL_OFFSET;
+  keyname = id2str(data->pool, key->name);
+  for (i = 0; languagetags[i] != 0; i++)
+    if (!strncmp(keyname, languagetags[i], strlen(languagetags[i])))
+      return KEY_STORAGE_VERTICAL_OFFSET;
+  return KEY_STORAGE_INCORE;
+}
+
+static int
+keyfilter_language(Repo *repo, Repokey *key, void *kfdata)
+{
+  const char *name, *p;
+  char *lang = kfdata, *bname;
+  int i;
+  Id id;
+
+  name = id2str(repo->pool, key->name);
+  p = strrchr(name, ':');
+  if (!p || strcmp(p + 1, lang) != 0)
+    return KEY_STORAGE_DROPPED;
+  /* find base name id */
+  bname = strdup(name);
+  bname[p - name] = 0;
+  id = str2id(repo->pool, bname, 1);
+  for (i = 0; verticals[i]; i++)
+    if (id == verticals[i])
+      return KEY_STORAGE_VERTICAL_OFFSET;
+  return KEY_STORAGE_INCORE;
+}
+
+static int
+keyfilter_DU(Repo *repo, Repokey *key, void *kfdata)
+{
+  int i;
+  if (key->name != SOLVABLE_DISKUSAGE)
+    return KEY_STORAGE_DROPPED;
+  for (i = 0; verticals[i]; i++)
+    if (key->name == verticals[i])
+      return KEY_STORAGE_VERTICAL_OFFSET;
   return KEY_STORAGE_INCORE;
 }
 
+static int
+keyfilter_FL(Repo *repo, Repokey *key, void *kfdata)
+{
+  int i;
+  if (key->name != SOLVABLE_FILELIST)
+    return KEY_STORAGE_DROPPED;
+  for (i = 0; verticals[i]; i++)
+    if (key->name == verticals[i])
+      return KEY_STORAGE_VERTICAL_OFFSET;
+  return KEY_STORAGE_INCORE;
+}
+
+struct keyfilter_other_data {
+  char **languages;
+  int nlanguages;
+};
+
+static int
+keyfilter_other(Repo *repo, Repokey *key, void *kfdata)
+{
+  const char *name, *p;
+  struct keyfilter_other_data *kd = kfdata;
+  int i;
+
+  if (key->name == SOLVABLE_FILELIST || key->name == SOLVABLE_DISKUSAGE)
+    return KEY_STORAGE_DROPPED;
+
+  name = id2str(repo->pool, key->name);
+  p = strrchr(name, ':');
+  if (p)
+    {
+      for (i = 0; i < kd->nlanguages; i++)
+	if (!strcmp(p + 1, kd->languages[i]))
+	  return KEY_STORAGE_DROPPED;
+    }
+  for (i = 0; verticals[i]; i++)
+    if (key->name == verticals[i])
+      return KEY_STORAGE_VERTICAL_OFFSET;
+  return KEY_STORAGE_INCORE;
+}
+
+/*
+ * Write <repo> to stdout
+ * If <attrname> is given, write attributes to <attrname>
+ * If <basename> is given, split attributes
+ */
+
+#define REPODATAFILE_BLOCK 15
+
 int
-tool_write(Repo *repo, const char *basename, int separate)
+tool_write(Repo *repo, const char *basename, const char *attrname)
 {
-  Pool *pool = repo->pool;
-  Repodatafile fileinfoa[1];
-  Repodatafile *fileinfo = 0;
-  int nsubfiles = 0;
-
-  create_filter(pool);
-  memset (fileinfoa, 0, sizeof fileinfoa);
-  if (separate)
+  Repodata *data;
+  Repokey *key;
+  Repodatafile *fileinfos = 0;
+  int nfileinfos = 0;
+  char **languages = 0;
+  int nlanguages = 0;
+  int i, j, k, l;
+
+  fileinfos = sat_zextend(fileinfos, nfileinfos, 1, sizeof(Repodatafile), REPODATAFILE_BLOCK);
+  pool_addfileprovides_ids(repo->pool, 0, &fileinfos[nfileinfos].addedfileprovides);
+  if (repo->rpmdbcookie)
+    fileinfos[nfileinfos].rpmdbcookie = repo->rpmdbcookie;
+  if (fileinfos[nfileinfos].addedfileprovides || fileinfos[nfileinfos].rpmdbcookie)
+    nfileinfos++;
+
+  if (basename)
     {
+      struct keyfilter_other_data kd;
+      char fn[4096];
+      FILE *fp;
+      int has_DU = 0;
+      int has_FL = 0;
+
+      /* find languages and other info */
+      for (i = 0, data = repo->repodata; i < repo->nrepodata; i++, data++)
+	{
+	  for (j = 1, key = data->keys + j; j < data->nkeys; j++, key++)
+	    {
+	      const char *keyname = id2str(repo->pool, key->name);
+	      if (key->name == SOLVABLE_DISKUSAGE)
+		has_DU = 1;
+	      if (key->name == SOLVABLE_FILELIST)
+		has_FL = 1;
+	      for (k = 0; languagetags[k] != 0; k++)
+		if (!strncmp(keyname, languagetags[k], strlen(languagetags[k])))
+		  break;
+	      if (!languagetags[k])
+		continue;
+	      l = strlen(languagetags[k]);
+	      if (strlen(keyname + l) > 5)
+		continue;
+	      for (k = 0; k < nlanguages; k++)
+		if (!strcmp(languages[k], keyname + l))
+		  break;
+	      if (k < nlanguages)
+		continue;
+	      languages = sat_realloc2(languages, nlanguages + 1, sizeof(char *));
+	      languages[nlanguages++] = strdup(keyname + l);
+	    }
+	}
+      fileinfos = sat_zextend(fileinfos, nfileinfos, nlanguages + 2, sizeof(Repodatafile), REPODATAFILE_BLOCK);
+      /* write language subfiles */
+      for (i = 0; i < nlanguages; i++)
+        {
+	  sprintf(fn, "%s.%s.solv", basename, languages[i]);
+	  if (!(fp = fopen(fn, "w")))
+	    {
+	      perror(fn);
+	      exit(1);
+	    }
+          repo_write(repo, fp, keyfilter_language, languages[i], fileinfos + nfileinfos, 0);
+	  fileinfos[nfileinfos].location = strdup(fn);
+	  fclose(fp);
+	  nfileinfos++;
+        }
+      /* write DU subfile */
+      if (has_DU)
+	{
+	  sprintf(fn, "%s.DU.solv", basename);
+	  if (!(fp = fopen(fn, "w")))
+	    {
+	      perror(fn);
+	      exit(1);
+	    }
+	  repo_write(repo, fp, keyfilter_DU, 0, fileinfos + nfileinfos, 0);
+	  fileinfos[nfileinfos].location = strdup(fn);
+	  fclose(fp);
+	  nfileinfos++;
+	}
+      /* write filelist */
+      if (has_FL)
+	{
+	  sprintf(fn, "%s.FL.solv", basename);
+	  if (!(fp = fopen(fn, "w")))
+	    {
+	      perror(fn);
+	      exit(1);
+	    }
+	  repo_write(repo, fp, keyfilter_FL, 0, fileinfos + nfileinfos, 0);
+	  fileinfos[nfileinfos].location = strdup(fn);
+	  fclose(fp);
+	  nfileinfos++;
+	}
+      /* write everything else */
+      sprintf(fn, "%s.solv", basename);
+      if (!(fp = fopen(fn, "w")))
+	{
+	  perror(fn);
+	  exit(1);
+	}
+      kd.languages = languages;
+      kd.nlanguages = nlanguages;
+      repo_write(repo, fp, keyfilter_other, &kd, nfileinfos ? fileinfos : 0, nfileinfos);
+      fclose(fp);
+      for (i = 0; i < nlanguages; i++)
+	free(languages[i]);
+      sat_free(languages);
+      for (i = 0; i < nfileinfos; i++)
+	{
+	  sat_free(fileinfos[i].addedfileprovides);
+	  sat_free(fileinfos[i].location);
+	  sat_free(fileinfos[i].keys);
+	}
+      sat_free(fileinfos);
+      return 0;
+    }
+  if (attrname)
+    {
+      fileinfos = sat_zextend(fileinfos, nfileinfos, 1, sizeof(Repodatafile), REPODATAFILE_BLOCK);
       test_separate = 1;
-      fileinfo = fileinfoa;
-      FILE *fp = fopen ("test.attr", "w");
-      repo_write(repo, fp, keyfilter_attr, 0, fileinfo, 0);
-      fclose (fp);
-      fileinfo->location = strdup ("test.attr");
-      fileinfo++;
-
-      nsubfiles = fileinfo - fileinfoa;
-      fileinfo = fileinfoa;
+      FILE *fp = fopen (attrname, "w");
+      repo_write(repo, fp, keyfilter_attr, 0, fileinfos + nfileinfos, 0);
+      fileinfos[nfileinfos].location = strdup(attrname);
+      fclose(fp);
+      nfileinfos++;
+    }
+  repo_write(repo, stdout, keyfilter_solv, 0, nfileinfos ? fileinfos : 0, nfileinfos);
+  for (i = 0; i < nfileinfos; i++)
+    {
+      sat_free(fileinfos[i].addedfileprovides);
+      sat_free(fileinfos[i].location);
+      sat_free(fileinfos[i].keys);
     }
-  repo_write(repo, stdout, keyfilter_solv, 0, fileinfo, nsubfiles);
+  sat_free(fileinfos);
   return 0;
 }