f93ed0bc840d7e4c98e6ae636ca85bf4bc8d7d71
[pspp] / src / output / journal.c
1 /* PSPP - a program for statistical analysis.
2    Copyright (C) 2007, 2010, 2012, 2013 Free Software Foundation, Inc.
3
4    This program is free software: you can redistribute it and/or modify
5    it under the terms of the GNU General Public License as published by
6    the Free Software Foundation, either version 3 of the License, or
7    (at your option) any later version.
8
9    This program is distributed in the hope that it will be useful,
10    but WITHOUT ANY WARRANTY; without even the implied warranty of
11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12    GNU General Public License for more details.
13
14    You should have received a copy of the GNU General Public License
15    along with this program.  If not, see <http://www.gnu.org/licenses/>. */
16
17 #include <config.h>
18
19 #include "output/journal.h"
20
21 #include <errno.h>
22 #include <stdio.h>
23 #include <stdlib.h>
24 #include <stdbool.h>
25
26 #include "data/file-name.h"
27 #include "libpspp/cast.h"
28 #include "libpspp/message.h"
29 #include "libpspp/str.h"
30 #include "output/driver-provider.h"
31 #include "output/message-item.h"
32 #include "output/text-item.h"
33
34 #include "gl/fwriteerror.h"
35 #include "gl/xalloc.h"
36
37 #include "gettext.h"
38 #define _(msgid) gettext (msgid)
39
40 struct journal_driver
41   {
42     struct output_driver driver;
43     FILE *file;
44
45     /* Name of journal file. */
46     char *file_name;
47     bool destroyed;
48   };
49
50 static const struct output_driver_class journal_class;
51
52 /* Journal driver, if journaling is enabled. */
53 static struct journal_driver journal;
54
55
56 static struct journal_driver *
57 journal_driver_cast (struct output_driver *driver)
58 {
59   assert (driver->class == &journal_class);
60   return UP_CAST (driver, struct journal_driver, driver);
61 }
62
63 static void
64 journal_close (void)
65 {
66   if (journal.file != NULL)
67     {
68       if (fwriteerror (journal.file))
69         msg_error (errno, _("error writing output file `%s'"),
70                    journal.file_name);
71
72       }
73   journal.file = NULL;
74 }
75
76 static void
77 journal_destroy (struct output_driver *driver)
78 {
79   struct journal_driver *j = journal_driver_cast (driver);
80
81   if (!j->destroyed)
82     journal_close ();
83
84   j->destroyed = true;
85 }
86
87 static void
88 journal_output (struct journal_driver *j, const char *s)
89 {
90   if (j->file == NULL)
91     return;
92
93   fprintf (j->file, "%s\n", s);
94
95   /* Flush the journal in case the syntax we're about to write
96      causes a crash.  Having the syntax already written to disk
97      makes postmortem analysis of the problem possible. */
98   fflush (j->file);
99 }
100
101 static void
102 journal_submit (struct output_driver *driver, const struct output_item *item)
103 {
104   struct journal_driver *j = journal_driver_cast (driver);
105
106   if (is_text_item (item))
107     {
108       const struct text_item *text_item = to_text_item (item);
109       enum text_item_type type = text_item_get_type (text_item);
110
111       if (type == TEXT_ITEM_SYNTAX)
112         {
113           char *text = text_item_get_plain_text (text_item);
114           journal_output (j, text);
115           free (text);
116         }
117     }
118   else if (is_message_item (item))
119     {
120       const struct message_item *message_item = to_message_item (item);
121       char *s = msg_to_string (message_item_get_msg (message_item));
122       journal_output (j, s);
123       free (s);
124     }
125 }
126
127 static const struct output_driver_class journal_class =
128   {
129     "journal",
130     journal_destroy,
131     journal_submit,
132     NULL                        /* flush */
133   };
134
135 \f
136
137 /* Enables journaling. */
138 void
139 journal_init (void)
140 {
141   /* Create journal driver. */
142   output_driver_init (&journal.driver, &journal_class, "journal",
143                       SETTINGS_DEVICE_UNFILTERED);
144   journal.file = NULL;
145
146   /* Register journal driver. */
147   output_driver_register (&journal.driver);
148
149   journal_enable ();
150   journal.destroyed = false;
151 }
152
153 /* Disables journaling. */
154 void
155 journal_disable (void)
156 {
157   journal_close ();
158 }
159
160
161 /* Enable journaling. */
162 void
163 journal_enable (void)
164 {
165   if (journal.file == NULL)
166     {
167       journal.file = fopen (journal_get_file_name (), "a");
168       if (journal.file == NULL)
169         {
170           msg_error (errno, _("error opening output file `%s'"),
171                      journal_get_file_name ());
172           journal_close ();
173         }
174     }
175 }
176
177
178 /* Returns true if journaling is enabled, false otherwise. */
179 bool
180 journal_is_enabled (void)
181 {
182   return journal.file != NULL ;
183 }
184
185 /* Sets the name of the journal file to FILE_NAME. */
186 void
187 journal_set_file_name (const char *file_name)
188 {
189   journal_close ();
190   free (journal.file_name);
191   journal.file_name = xstrdup (file_name);
192 }
193
194 /* Returns the name of the journal file.  The caller must not modify or free
195    the returned string. */
196 const char *
197 journal_get_file_name (void)
198 {
199   if (journal.file_name == NULL)
200     {
201       const char *output_path = default_output_path ();
202       journal.file_name = xasprintf ("%s%s", output_path, "pspp.jnl");
203     }
204   return journal.file_name;
205 }