output: Rename caption to title.
[pspp] / src / output / odt.c
index f02ae3defd559fbf54fb3504b78490d07b9182aa..21aaeb6c1a723ebe9927a0d02beb85fc67456c24 100644 (file)
@@ -1,5 +1,5 @@
 /* PSPP - a program for statistical analysis.
-   Copyright (C) 2009, 2010 Free Software Foundation, Inc.
+   Copyright (C) 2009-2014 Free Software Foundation, Inc.
 
    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
@@ -21,7 +21,9 @@
 #include <errno.h>
 #include <libgen.h>
 #include <libxml/xmlwriter.h>
+#ifdef HAVE_PWD_H
 #include <pwd.h>
+#endif
 #include <sys/stat.h>
 #include <sys/types.h>
 #include <time.h>
 
 #include "libpspp/assertion.h"
 #include "libpspp/cast.h"
+#include "libpspp/message.h"
 #include "libpspp/str.h"
+#include "libpspp/temp-file.h"
 #include "libpspp/version.h"
+#include "libpspp/zip-writer.h"
 #include "output/driver-provider.h"
+#include "output/message-item.h"
 #include "output/options.h"
 #include "output/tab.h"
 #include "output/table-item.h"
 #include "output/text-item.h"
 
 #include "gl/xalloc.h"
-#include "gl/error.h"
 
 #include "gettext.h"
 #define _(msgid) gettext (msgid)
 
-#define _xml(X) (const xmlChar *)(X)
+#define _xml(X) (CHAR_CAST (const xmlChar *, X))
 
 struct odt_driver
 {
   struct output_driver driver;
 
+  struct zip_writer *zip;     /* ZIP file writer. */
   char *file_name;            /* Output file name. */
-  bool debug;
 
-  /* The name of the temporary directory used to construct the ODF */
-  char *dirname;
+  /* content.xml */
+  xmlTextWriterPtr content_wtr; /* XML writer. */
+  FILE *content_file;           /* Temporary file. */
 
-  /* Writer for the content.xml file */
-  xmlTextWriterPtr content_wtr;
-
-  /* Writer fot the manifest.xml file */
-  xmlTextWriterPtr manifest_wtr;
+  /* manifest.xml */
+  xmlTextWriterPtr manifest_wtr; /* XML writer. */
+  FILE *manifest_file;           /* Temporary file. */
 
   /* Number of tables so far. */
   int table_num;
+
+  /* Name of current command. */
+  char *command_name;
+
+  /* Number of footnotes so far. */
+  int n_footnotes;
 };
 
+static const struct output_driver_class odt_driver_class;
+
 static struct odt_driver *
 odt_driver_cast (struct output_driver *driver)
 {
-  assert (driver->class == &odt_class);
+  assert (driver->class == &odt_driver_class);
   return UP_CAST (driver, struct odt_driver, driver);
 }
 
 /* Create the "mimetype" file needed by ODF */
 static bool
-create_mimetype (const char *dirname)
+create_mimetype (struct zip_writer *zip)
 {
   FILE *fp;
-  struct string filename;
-  ds_init_cstr (&filename, dirname);
-  ds_put_cstr (&filename, "/mimetype");
-  fp = fopen (ds_cstr (&filename), "w");
 
+  fp = create_temp_file ();
   if (fp == NULL)
     {
-      error (0, errno, _("failed to create output file %s"),
-             ds_cstr (&filename));
-      ds_destroy (&filename);
+      msg_error (errno, _("error creating temporary file"));
       return false;
     }
-  ds_destroy (&filename);
 
   fprintf (fp, "application/vnd.oasis.opendocument.text");
-  fclose (fp);
+  zip_writer_add (zip, fp, "mimetype");
+  close_temp_file (fp);
 
   return true;
 }
 
-/* Create a new XML file called FILENAME in the temp directory, and return a writer for it */
-static xmlTextWriterPtr
-create_writer (const struct odt_driver *driver, const char *filename)
+/* Creates a new temporary file and stores it in *FILE, then creates an XML
+   writer for it and stores it in *W. */
+static void
+create_writer (FILE **file, xmlTextWriterPtr *w)
 {
-  char *copy = NULL;
-  xmlTextWriterPtr w;
-  struct string str;
-  ds_init_cstr (&str, driver->dirname);
-  ds_put_cstr (&str, "/");
-  ds_put_cstr (&str, filename);
-
-  /* dirname modifies its argument, so we must copy it */
-  copy = xstrdup (ds_cstr (&str));
-  mkdir (dirname (copy), 0700);
-  free (copy);
-
-  w = xmlNewTextWriterFilename (ds_cstr (&str), 0);
-
-  ds_destroy (&str);
+  /* XXX this can fail */
+  *file = create_temp_file ();
+  *w = xmlNewTextWriter (xmlOutputBufferCreateFile (*file, NULL));
 
-  xmlTextWriterStartDocument (w, NULL, "UTF-8", NULL);
-
-  return w;
+  xmlTextWriterStartDocument (*w, NULL, "UTF-8", NULL);
 }
 
 
@@ -137,7 +131,10 @@ register_file (struct odt_driver *odt, const char *filename)
 static void
 write_style_data (struct odt_driver *odt)
 {
-  xmlTextWriterPtr w = create_writer (odt, "styles.xml");
+  xmlTextWriterPtr w;
+  FILE *file;
+
+  create_writer (&file, &w);
   register_file (odt, "styles.xml");
 
   xmlTextWriterStartElement (w, _xml ("office:document-styles"));
@@ -224,12 +221,17 @@ write_style_data (struct odt_driver *odt)
 
   xmlTextWriterEndDocument (w);
   xmlFreeTextWriter (w);
+  zip_writer_add (odt->zip, file, "styles.xml");
+  close_temp_file (file);
 }
 
 static void
 write_meta_data (struct odt_driver *odt)
 {
-  xmlTextWriterPtr w = create_writer (odt, "meta.xml");
+  xmlTextWriterPtr w;
+  FILE *file;
+
+  create_writer (&file, &w);
   register_file (odt, "meta.xml");
 
   xmlTextWriterStartElement (w, _xml ("office:document-meta"));
@@ -249,74 +251,71 @@ write_meta_data (struct odt_driver *odt)
 
   {
     char buf[30];
-    struct passwd *pw = getpwuid (getuid ());
     time_t t = time (NULL);
     struct tm *tm =  localtime (&t);
 
     strftime (buf, 30, "%Y-%m-%dT%H:%M:%S", tm);
 
-    xmlTextWriterStartElement (w, _xml ("meta:initial-creator"));
-    xmlTextWriterWriteString (w, _xml (strtok (pw->pw_gecos, ",")));
-    xmlTextWriterEndElement (w);
-
     xmlTextWriterStartElement (w, _xml ("meta:creation-date"));
     xmlTextWriterWriteString (w, _xml (buf));
     xmlTextWriterEndElement (w);
 
-    xmlTextWriterStartElement (w, _xml ("dc:creator"));
-    xmlTextWriterWriteString (w, _xml (strtok (pw->pw_gecos, ",")));
-
-    xmlTextWriterEndElement (w);
-
     xmlTextWriterStartElement (w, _xml ("dc:date"));
     xmlTextWriterWriteString (w, _xml (buf));
     xmlTextWriterEndElement (w);
   }
 
+#ifdef HAVE_PWD_H
+  {
+    struct passwd *pw = getpwuid (getuid ());
+    if (pw != NULL)
+      {
+        xmlTextWriterStartElement (w, _xml ("meta:initial-creator"));
+        xmlTextWriterWriteString (w, _xml (strtok (pw->pw_gecos, ",")));
+        xmlTextWriterEndElement (w);
+
+        xmlTextWriterStartElement (w, _xml ("dc:creator"));
+        xmlTextWriterWriteString (w, _xml (strtok (pw->pw_gecos, ",")));
+        xmlTextWriterEndElement (w);
+      }
+  }
+#endif
+
   xmlTextWriterEndElement (w);
   xmlTextWriterEndElement (w);
   xmlTextWriterEndDocument (w);
   xmlFreeTextWriter (w);
-}
-
-enum
-{
-  output_file_arg,
-  boolean_arg,
-};
-
-static struct driver_option *
-opt (struct output_driver *d, struct string_map *options, const char *key,
-     const char *default_value)
-{
-  return driver_option_get (d, options, key, default_value);
+  zip_writer_add (odt->zip, file, "meta.xml");
+  close_temp_file (file);
 }
 
 static struct output_driver *
-odt_create (const char *name, enum output_device_type device_type,
-            struct string_map *o)
+odt_create (const char *file_name, enum settings_output_devices device_type,
+            struct string_map *o UNUSED)
 {
   struct output_driver *d;
   struct odt_driver *odt;
+  struct zip_writer *zip;
+
+  zip = zip_writer_create (file_name);
+  if (zip == NULL)
+    return NULL;
 
   odt = xzalloc (sizeof *odt);
   d = &odt->driver;
-  output_driver_init (d, &odt_class, name, device_type);
-
-  odt->file_name = parse_string (opt (d, o, "output-file", "pspp.odt"));
-  odt->debug = parse_boolean (opt (d, o, "debug", "false"));
+  output_driver_init (d, &odt_driver_class, file_name, device_type);
 
-  odt->dirname = xstrdup ("odt-XXXXXX");
-  mkdtemp (odt->dirname);
+  odt->zip = zip;
+  odt->file_name = xstrdup (file_name);
 
-  if (!create_mimetype (odt->dirname))
+  if (!create_mimetype (zip))
     {
       output_driver_destroy (d);
       return NULL;
     }
 
   /* Create the manifest */
-  odt->manifest_wtr = create_writer (odt, "META-INF/manifest.xml");
+  create_writer (&odt->manifest_file, &odt->manifest_wtr);
 
   xmlTextWriterStartElement (odt->manifest_wtr, _xml("manifest:manifest"));
   xmlTextWriterWriteAttribute (odt->manifest_wtr, _xml("xmlns:manifest"),
@@ -333,7 +332,7 @@ odt_create (const char *name, enum output_device_type device_type,
   write_meta_data (odt);
   write_style_data (odt);
 
-  odt->content_wtr = create_writer (odt, "content.xml");
+  create_writer (&odt->content_file, &odt->content_wtr);
   register_file (odt, "content.xml");
 
 
@@ -359,6 +358,8 @@ odt_create (const char *name, enum output_device_type device_type,
   xmlTextWriterEndElement (odt->manifest_wtr);
   xmlTextWriterEndDocument (odt->manifest_wtr);
   xmlFreeTextWriter (odt->manifest_wtr);
+  zip_writer_add (odt->zip, odt->manifest_file, "META-INF/manifest.xml");
+  close_temp_file (odt->manifest_file);
 
   return d;
 }
@@ -370,56 +371,100 @@ odt_destroy (struct output_driver *driver)
 
   if (odt->content_wtr != NULL)
     {
-      struct string zip_cmd;
-
       xmlTextWriterEndElement (odt->content_wtr); /* office:text */
       xmlTextWriterEndElement (odt->content_wtr); /* office:body */
       xmlTextWriterEndElement (odt->content_wtr); /* office:document-content */
 
       xmlTextWriterEndDocument (odt->content_wtr);
       xmlFreeTextWriter (odt->content_wtr);
+      zip_writer_add (odt->zip, odt->content_file, "content.xml");
+      close_temp_file (odt->content_file);
 
-      /* Zip up the directory */
-      ds_init_empty (&zip_cmd);
-      ds_put_format (&zip_cmd,
-                     "cd %s ; rm -f ../%s; zip -q -X ../%s mimetype; zip -q -X -u -r ../%s .",
-                     odt->dirname, odt->file_name, odt->file_name, odt->file_name);
-      system (ds_cstr (&zip_cmd));
-      ds_destroy (&zip_cmd);
+      zip_writer_close (odt->zip);
     }
+  
+  free (odt->file_name);
+  free (odt->command_name);
+  free (odt);
+}
+
+static void
+write_xml_with_line_breaks (struct odt_driver *odt, const char *line_)
+{
+  xmlTextWriterPtr writer = odt->content_wtr;
 
-  if ( !odt->debug )
+  if (!strchr (line_, '\n'))
+    xmlTextWriterWriteString (writer, _xml(line_));
+  else
     {
-      /* Remove the temp dir */
-      struct string rm_cmd;
+      char *line = xstrdup (line_);
+      char *newline;
+      char *p;
+
+      for (p = line; *p; p = newline + 1)
+        {
+          newline = strchr (p, '\n');
 
-      ds_init_empty (&rm_cmd);
-      ds_put_format (&rm_cmd, "rm -r %s", odt->dirname);
-      system (ds_cstr (&rm_cmd));
-      ds_destroy (&rm_cmd);
+          if (!newline)
+            {
+              xmlTextWriterWriteString (writer, _xml(p));
+              free (line);
+              return;
+            }
+
+          if (newline > p && newline[-1] == '\r')
+            newline[-1] = '\0';
+          else
+            *newline = '\0';
+          xmlTextWriterWriteString (writer, _xml(p));
+          xmlTextWriterWriteElement (writer, _xml("text:line-break"), _xml(""));
+        }
     }
+}
+
+static void
+write_footnote (struct odt_driver *odt, const char *footnote)
+{
+  char marker[16];
+
+  xmlTextWriterStartElement (odt->content_wtr, _xml("text:note"));
+  xmlTextWriterWriteAttribute (odt->content_wtr, _xml("text:note-class"),
+                               _xml("footnote"));
+
+  xmlTextWriterStartElement (odt->content_wtr, _xml("text:note-citation"));
+  str_format_26adic (++odt->n_footnotes, false, marker, sizeof marker);
+  if (strlen (marker) > 1)
+    xmlTextWriterWriteFormatAttribute (odt->content_wtr, _xml("text:label"),
+                                       "(%s)", marker);
   else
-    fprintf (stderr, "Not removing directory %s\n", odt->dirname);
+    xmlTextWriterWriteAttribute (odt->content_wtr, _xml("text:label"),
+                                 _xml(marker));
+  xmlTextWriterEndElement (odt->content_wtr);
 
-  free (odt->dirname);
-  free (odt);
+  xmlTextWriterStartElement (odt->content_wtr, _xml("text:note-body"));
+  xmlTextWriterStartElement (odt->content_wtr, _xml("text:p"));
+  write_xml_with_line_breaks (odt, footnote);
+  xmlTextWriterEndElement (odt->content_wtr);
+  xmlTextWriterEndElement (odt->content_wtr);
+
+  xmlTextWriterEndElement (odt->content_wtr);
 }
 
 static void
-odt_submit_table (struct odt_driver *odt, struct table_item *item)
+write_table (struct odt_driver *odt, const struct table_item *item)
 {
   const struct table *tab = table_item_get_table (item);
-  const char *caption = table_item_get_caption (item);
+  const char *title = table_item_get_title (item);
   int r, c;
 
   /* Write a heading for the table */
-  if (caption != NULL)
+  if (title != NULL)
     {
       xmlTextWriterStartElement (odt->content_wtr, _xml("text:h"));
-      xmlTextWriterWriteFormatAttribute (odt->content_wtr, _xml("text:level"),
-                                         "%d", 2);
+      xmlTextWriterWriteFormatAttribute (odt->content_wtr,
+                                         _xml("text:outline-level"), "%d", 2);
       xmlTextWriterWriteString (odt->content_wtr,
-                                _xml (table_item_get_caption (item)) );
+                                _xml (table_item_get_title (item)) );
       xmlTextWriterEndElement (odt->content_wtr);
     }
 
@@ -450,6 +495,7 @@ odt_submit_table (struct odt_driver *odt, struct table_item *item)
       for (c = 0 ; c < table_nc (tab) ; ++c)
        {
           struct table_cell cell;
+          size_t i;
 
           table_get_cell (tab, c, r, &cell);
 
@@ -471,17 +517,32 @@ odt_submit_table (struct odt_driver *odt, struct table_item *item)
                   odt->content_wtr, _xml("table:number-rows-spanned"),
                   "%d", rowspan);
 
-             xmlTextWriterStartElement (odt->content_wtr, _xml("text:p"));
+              for (i = 0; i < cell.n_contents; i++)
+                {
+                  const struct cell_contents *contents = &cell.contents[i];
+                  int j;
+
+                  if (contents->text)
+                    {
+                      xmlTextWriterStartElement (odt->content_wtr, _xml("text:p"));
+
+                      if ( r < table_ht (tab) || c < table_hl (tab) )
+                        xmlTextWriterWriteAttribute (odt->content_wtr, _xml("text:style-name"), _xml("Table_20_Heading"));
+                      else
+                        xmlTextWriterWriteAttribute (odt->content_wtr, _xml("text:style-name"), _xml("Table_20_Contents"));
 
-             if ( r < table_ht (tab) || c < table_hl (tab) )
-               xmlTextWriterWriteAttribute (odt->content_wtr, _xml("text:style-name"), _xml("Table_20_Heading"));
-             else
-               xmlTextWriterWriteAttribute (odt->content_wtr, _xml("text:style-name"), _xml("Table_20_Contents"));
+                      write_xml_with_line_breaks (odt, contents->text);
 
-             xmlTextWriterWriteString (odt->content_wtr, _xml(cell.contents));
+                      for (j = 0; j < contents->n_footnotes; j++)
+                        write_footnote (odt, contents->footnotes[j]);
 
-             xmlTextWriterEndElement (odt->content_wtr); /* text:p */
-             xmlTextWriterEndElement (odt->content_wtr); /* table:table-cell */
+                      xmlTextWriterEndElement (odt->content_wtr); /* text:p */
+                    }
+                  else if (contents->table)
+                    write_table (odt, contents->table);
+
+                }
+              xmlTextWriterEndElement (odt->content_wtr); /* table:table-cell */
            }
          else
            {
@@ -501,31 +562,48 @@ odt_submit_table (struct odt_driver *odt, struct table_item *item)
   xmlTextWriterEndElement (odt->content_wtr); /* table */
 }
 
+static void
+odt_output_text (struct odt_driver *odt, const char *text)
+{
+  xmlTextWriterStartElement (odt->content_wtr, _xml("text:p"));
+  xmlTextWriterWriteString (odt->content_wtr, _xml(text));
+  xmlTextWriterEndElement (odt->content_wtr);
+}
+
 /* Submit a table to the ODT driver */
 static void
 odt_submit (struct output_driver *driver,
             const struct output_item *output_item)
 {
   struct odt_driver *odt = odt_driver_cast (driver);
+
+  output_driver_track_current_command (output_item, &odt->command_name);
+
   if (is_table_item (output_item))
-    odt_submit_table (odt, to_table_item (output_item));
+    write_table (odt, to_table_item (output_item));
   else if (is_text_item (output_item))
     {
-      const struct text_item *text_item = to_text_item (output_item);
-      const char *text = text_item_get_text (text_item);
+      struct text_item *text_item = to_text_item (output_item);
 
-      /* XXX apply different styles based on text_item's type.  */
-      xmlTextWriterStartElement (odt->content_wtr, _xml("text:p"));
-      xmlTextWriterWriteString (odt->content_wtr, _xml(text));
-      xmlTextWriterEndElement (odt->content_wtr);
+      if (text_item_get_type (text_item) != TEXT_ITEM_COMMAND_CLOSE)
+        odt_output_text (odt, text_item_get_text (text_item));
+    }
+  else if (is_message_item (output_item))
+    {
+      const struct message_item *message_item = to_message_item (output_item);
+      const struct msg *msg = message_item_get_msg (message_item);
+      char *s = msg_to_string (msg, odt->command_name);
+      odt_output_text (odt, s);
+      free (s);
     }
 }
 
-/* ODT driver class. */
-const struct output_driver_class odt_class =
+struct output_driver_factory odt_driver_factory =
+  { "odt", "pspp.odf", odt_create };
+
+static const struct output_driver_class odt_driver_class =
 {
   "odf",
-  odt_create,
   odt_destroy,
   odt_submit,
   NULL,