import HEAD into svn+ssh://svn.samba.org/home/svn/samba/trunk
[metze/old/v3-2-winbind-ndr.git] / source / intl / lang_tdb.c
1 /* 
2    Unix SMB/CIFS implementation.
3    tdb based replacement for gettext 
4    Copyright (C) Andrew Tridgell 2001
5    
6    This program is free software; you can redistribute it and/or modify
7    it under the terms of the GNU General Public License as published by
8    the Free Software Foundation; either version 2 of the License, or
9    (at your option) any later version.
10    
11    This program is distributed in the hope that it will be useful,
12    but WITHOUT ANY WARRANTY; without even the implied warranty of
13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14    GNU General Public License for more details.
15    
16    You should have received a copy of the GNU General Public License
17    along with this program; if not, write to the Free Software
18    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
19 */
20
21 #include "includes.h"
22
23 static TDB_CONTEXT *tdb;
24
25 /* the currently selected language */
26 static char *current_lang;
27
28
29 /* load a msg file into the tdb */
30 static BOOL load_msg(const char *msg_file)
31 {
32         char **lines;
33         int num_lines, i;
34         char *msgid, *msgstr;
35         TDB_DATA key, data;
36
37         lines = file_lines_load(msg_file, &num_lines);
38
39         if (!lines) {
40                 return False;
41         }
42
43         if (tdb_lockall(tdb) != 0) return False;
44
45         /* wipe the db */
46         tdb_traverse(tdb, tdb_traverse_delete_fn, NULL);
47
48         msgid = NULL;
49         
50         for (i=0;i<num_lines;i++) {
51                 if (strncmp(lines[i], "msgid \"", 7) == 0) {
52                         msgid = lines[i] + 7;
53                 }
54                 if (msgid && strncmp(lines[i], "msgstr \"", 8) == 0) {
55                         msgstr = lines[i] + 8;
56                         trim_char(msgid, '\0', '\"');
57                         trim_char(msgstr, '\0', '\"');
58                         if (*msgstr == 0) {
59                                 msgstr = msgid;
60                         }
61                         all_string_sub(msgid, "\\n", "\n", 0);
62                         all_string_sub(msgstr, "\\n", "\n", 0);
63                         key.dptr = msgid;
64                         key.dsize = strlen(msgid)+1;
65                         data.dptr = msgstr;
66                         data.dsize = strlen(msgstr)+1;
67                         tdb_store(tdb, key, data, 0);
68                         msgid = NULL;
69                 }
70         }
71
72         file_lines_free(lines);
73         tdb_unlockall(tdb);
74
75         return True;
76 }
77
78
79 /* work out what language to use from locale variables */
80 static const char *get_lang(void)
81 {
82         const char *vars[] = {"LANGUAGE", "LC_ALL", "LC_LANG", "LANG", NULL};
83         int i;
84         char *p;
85
86         for (i=0; vars[i]; i++) {
87                 if ((p = getenv(vars[i]))) {
88                         return p;
89                 }
90         }
91
92         return NULL;
93 }
94
95 /* initialise the message translation subsystem. If the "lang" argument
96    is NULL then get the language from the normal environment variables */
97 BOOL lang_tdb_init(const char *lang)
98 {
99         char *path = NULL;
100         char *msg_path = NULL;
101         struct stat st;
102         static int initialised;
103         time_t loadtime;
104         BOOL result = False;
105
106         /* we only want to init once per process, unless given
107            an override */
108         if (initialised && !lang) 
109                 return True;
110
111         if (initialised) {
112                 /* we are re-initialising, free up any old init */
113                 if (tdb) {
114                         tdb_close(tdb);
115                         tdb = NULL;
116                 }
117                 SAFE_FREE(current_lang);
118         }
119
120         initialised = 1;
121
122         if (!lang) {
123                 /* no lang given, use environment */
124                 lang = get_lang();
125         }
126
127         /* if no lang then we don't translate */
128         if (!lang) 
129                 return True;
130
131         asprintf(&msg_path, "%s.msg", lib_path((const char *)lang));
132         if (stat(msg_path, &st) != 0) {
133                 /* the msg file isn't available */
134                 DEBUG(10, ("lang_tdb_init: %s: %s\n", msg_path, 
135                            strerror(errno)));
136                 goto done;
137         }
138         
139         asprintf(&path, "%s%s.tdb", lock_path("lang_"), lang);
140
141         DEBUG(10, ("lang_tdb_init: loading %s\n", path));
142
143         tdb = tdb_open_log(path, 0, TDB_DEFAULT, O_RDWR|O_CREAT, 0644);
144         if (!tdb) {
145                 tdb = tdb_open_log(path, 0, TDB_DEFAULT, O_RDONLY, 0);
146                 if (!tdb) {
147                         DEBUG(10, ("lang_tdb_init: %s: %s\n", path,
148                                    strerror(errno)));
149                         goto done;
150                 }
151                 current_lang = strdup(lang);
152                 result = True;
153                 goto done;
154         }
155
156         loadtime = tdb_fetch_int32(tdb, "/LOADTIME/");
157
158         if (loadtime == -1 || loadtime < st.st_mtime) {
159                 load_msg(msg_path);
160                 tdb_store_int32(tdb, "/LOADTIME/", (int)time(NULL));
161         }
162
163         current_lang = strdup(lang);
164         result = True;
165
166  done:
167         SAFE_FREE(msg_path);
168         SAFE_FREE(path);
169
170         return result;
171 }
172
173 /* translate a msgid to a message string in the current language 
174    returns a string that must be freed by calling lang_msg_free()
175 */
176 const char *lang_msg(const char *msgid)
177 {
178         TDB_DATA key, data;
179         const char *p;
180         char *q, *msgid_quoted;
181         int count;
182
183         lang_tdb_init(NULL);
184
185         if (!tdb) return msgid;
186
187         /* Due to the way quotes in msgids are escaped in the msg file we
188            must replace " with \" before doing a lookup in the tdb. */
189
190         count = 0;
191
192         for(p = msgid; *p; p++) {
193                 if (*p == '\"')
194                         count++;
195         }
196
197         if (!(msgid_quoted = malloc(strlen(msgid) + count + 1)))
198                 return msgid;
199
200         /* string_sub() is unsuitable here as it replaces some punctuation
201            chars with underscores. */
202
203         for(p = msgid, q = msgid_quoted; *p; p++) {
204                 if (*p == '\"') {
205                         *q = '\\';
206                         q++;
207                 }
208                 *q = *p;
209                 q++;
210         }
211
212         *q = 0;
213
214         key.dptr = (char *)msgid_quoted;
215         key.dsize = strlen(msgid_quoted)+1;
216         
217         data = tdb_fetch(tdb, key);
218
219         free(msgid_quoted);
220
221         /* if the message isn't found then we still need to return a pointer
222            that can be freed. Pity. */
223         if (!data.dptr)
224                 return strdup(msgid);
225
226         return (const char *)data.dptr;
227 }
228
229
230 /* free up a string from lang_msg() */
231 void lang_msg_free(const char *msgstr)
232 {
233         if (!tdb) return;
234         free((void *)msgstr);
235 }
236
237
238 /*
239   when the _() translation macro is used there is no obvious place to free
240   the resulting string and there is no easy way to give a static pointer.
241   All we can do is rotate between some static buffers and hope a single d_printf() 
242   doesn't have more calls to _() than the number of buffers 
243 */
244 const char *lang_msg_rotate(const char *msgid)
245 {
246 #define NUM_LANG_BUFS 16
247         char *msgstr;
248         static pstring bufs[NUM_LANG_BUFS];
249         static int next;
250
251         msgstr = (char *)lang_msg(msgid);
252         if (!msgstr) return msgid;
253
254         pstrcpy(bufs[next], msgstr);
255         msgstr = bufs[next];
256
257         next = (next+1) % NUM_LANG_BUFS;
258         
259         return msgstr;
260 }
261
262
263 /* 
264    return the current language - needed for language file mappings 
265 */
266 char *lang_tdb_current(void)
267 {
268         return current_lang;
269 }