src/common/tags.c

   1 /*
   2  * Claws Mail -- a GTK based, lightweight, and fast e-mail client
   3  * Copyright (C) 2007-2012 The Claws Mail Team
   4  *
   5  * This program is free software; you can redistribute it and/or modify
   6  * it under the terms of the GNU General Public License as published by
   7  * the Free Software Foundation; either version 3 of the License, or
   8  * (at your option) any later version.
   9  *
  10  * This program is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  * GNU General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License
  16  * along with this program. If not, see <http://www.gnu.org/licenses/>.
  17  *
  18  */
  19
  20 #ifdef HAVE_CONFIG_H
  21 #  include "config.h"
  22 #include "claws-features.h"
  23 #endif
  24
  25 #include <glib.h>
  26 #include <glib/gi18n.h>
  27 #include <stdio.h>
  28 #include <stdlib.h>
  29 #include <string.h>
  30 #include <errno.h>
  31 #include <sys/types.h>
  32 #if HAVE_SYS_WAIT_H
  33 #  include <sys/wait.h>
  34 #endif
  35 #include <signal.h>
  36 #include <unistd.h>
  37
  38 #include "defs.h"
  39 #include "utils.h"
  40 #include "tags.h"
  41 #include "file-utils.h"
  42
  43 static GHashTable *tags_table = NULL;
  44 static GHashTable *tags_reverse_table = NULL;
  45
  46 static int tag_max_id = 0;
  47
  48 void tags_read_tags(void)
  49 {
  50         gchar *file = g_strconcat(get_rc_dir(), G_DIR_SEPARATOR_S,
  51                         TAGS_RC, NULL);
  52         gchar tmp[255];
  53         gint id;
  54         FILE *fp = claws_fopen(file, "rb");
  55
  56         g_free(file);
  57
  58         if (tags_table == NULL)
  59                 tags_table = g_hash_table_new_full(
  60                                 g_direct_hash, g_direct_equal,
  61                                 NULL, g_free);
  62         if (tags_reverse_table == NULL)
  63                 tags_reverse_table = g_hash_table_new_full(
  64                                 g_str_hash, g_str_equal,
  65                                 g_free, NULL);
  66
  67         if (!fp)
  68                 return;
  69         if (fscanf(fp, "max_id %d\n", &tag_max_id) != 1) {
  70                 claws_fclose(fp);
  71                 return;
  72         }
  73         while (claws_fgets(tmp, sizeof(tmp), fp) != NULL) {
  74                 gchar *sep = strchr(tmp, '\t');
  75                 gchar *tag_name = sep?(sep+1):NULL;
  76
  77                 if (!tag_name || !sep)
  78                         continue;
  79                 g_strstrip(tag_name);
  80                 *(sep) = '\0';
  81                 if (IS_NOT_RESERVED_TAG(tag_name)) {
  82                         id = atoi(tmp);
  83                         g_hash_table_insert(tags_table,
  84                                             GINT_TO_POINTER(id), g_strdup(tag_name));
  85                         g_hash_table_insert(tags_reverse_table,
  86                                             g_strdup(tag_name), GINT_TO_POINTER(id));
  87                 }
  88         }
  89
  90         claws_fclose(fp);
  91 }
  92
  93 typedef struct _TagWriteData
  94 {
  95         FILE *fp;
  96         gboolean error;
  97 } TagWriteData;
  98
  99 static void tag_write(gpointer key, gpointer value, gpointer user_data)
 100 {
 101         TagWriteData *data = (TagWriteData *)user_data;
 102         const gchar *str = value;
 103         gint id = GPOINTER_TO_INT(key);
 104
 105         if (data->error)
 106                 return;
 107
 108         if (fprintf(data->fp, "%d\t%s\n", id, str) <= 0) {
 109                 FILE_OP_ERROR("tagsrc", "fprintf");
 110                 data->error = TRUE;
 111         }
 112 }
 113
 114 void tags_write_tags(void)
 115 {
 116         gchar *file = g_strconcat(get_rc_dir(), G_DIR_SEPARATOR_S,
 117                         TAGS_RC, ".tmp", NULL);
 118         gchar *file_new = g_strconcat(get_rc_dir(), G_DIR_SEPARATOR_S,
 119                         TAGS_RC, NULL);
 120         TagWriteData data;
 121
 122         FILE *fp = claws_fopen(file, "wb");
 123
 124         if (!fp) {
 125                 FILE_OP_ERROR(file, "claws_fopen");
 126                 g_free(file);
 127                 g_free(file_new);
 128                 return;
 129         }
 130
 131         data.fp = fp;
 132         data.error = FALSE;
 133
 134         if (fprintf(data.fp, "max_id %d\n", tag_max_id) <= 0) {
 135                 FILE_OP_ERROR("tagsrc", "fprintf");
 136                 data.error = TRUE;
 137         } else {
 138                 g_hash_table_foreach(tags_table, tag_write, &data);
 139         }
 140
 141         if (data.error) {
 142                 claws_fclose(fp);
 143                 g_free(file);
 144                 g_free(file_new);
 145                 return;
 146         }
 147
 148         if (claws_safe_fclose(fp) == EOF) {
 149                 FILE_OP_ERROR(file, "claws_fclose");
 150                 g_free(file);
 151                 g_free(file_new);
 152                 return;
 153         }
 154
 155         if (rename_force(file, file_new) < 0) {
 156                 FILE_OP_ERROR(file, "rename_force");
 157         }
 158
 159         g_free(file);
 160         g_free(file_new);
 161 }
 162
 163 gint tags_add_tag(const gchar *tag)
 164 {
 165         if (!tag || !(*tag))
 166                 return -1;
 167
 168         if (g_hash_table_lookup(tags_reverse_table, tag))
 169                 return -1;
 170
 171         if (IS_NOT_RESERVED_TAG(tag)) {
 172                 tag_max_id++;
 173                 g_hash_table_insert(tags_table, GINT_TO_POINTER(tag_max_id),
 174                         g_strdup(tag));
 175                 g_hash_table_insert(tags_reverse_table, g_strdup(tag),
 176                         GINT_TO_POINTER(tag_max_id));
 177
 178                 return tag_max_id;
 179         } else {
 180                 return -1;
 181         }
 182 }
 183
 184 void tags_remove_tag(gint id)
 185 {
 186         gchar *old_tag = g_hash_table_lookup(tags_table, GINT_TO_POINTER(id));
 187
 188         if (old_tag) {
 189                 g_hash_table_remove(tags_reverse_table, old_tag);
 190         }
 191         g_hash_table_remove(tags_table, GINT_TO_POINTER(id));
 192 }
 193
 194 /* extern decl. to avoid including ../prefs_filtering.h */
 195 extern void prefs_filtering_rename_tag(const gchar *old_tag, const gchar *new_tag);
 196
 197 void tags_update_tag(gint id, const gchar *tag)
 198 {
 199         gchar *old_tag = g_hash_table_lookup(tags_table, GINT_TO_POINTER(id));
 200
 201         if (IS_NOT_RESERVED_TAG(tag)) {
 202                 if (old_tag) {
 203                         prefs_filtering_rename_tag(old_tag, tag);
 204                         g_hash_table_remove(tags_reverse_table, old_tag);
 205                 }
 206
 207                 g_hash_table_replace(tags_table, GINT_TO_POINTER(id),
 208                         g_strdup(tag));
 209                 g_hash_table_insert(tags_reverse_table, g_strdup(tag),
 210                         GINT_TO_POINTER(id));
 211         }
 212 }
 213
 214 const gchar *tags_get_tag(gint id)
 215 {
 216         return (const gchar *)g_hash_table_lookup(tags_table,
 217                                 GINT_TO_POINTER(id));
 218 }
 219
 220 gint tags_get_id_for_str(const gchar *str)
 221 {
 222         gpointer id_ptr;
 223         if ((id_ptr = g_hash_table_lookup(tags_reverse_table, str)) != NULL)
 224                 return GPOINTER_TO_INT(id_ptr);
 225         else
 226                 return -1;
 227 }
 228
 229 typedef struct _TagListData {
 230         GSList *list;
 231 } TagListData;
 232
 233 static void tag_add_list(gpointer key, gpointer value, gpointer user_data)
 234 {
 235         TagListData *data = (TagListData *)user_data;
 236
 237         data->list = g_slist_prepend(data->list, GINT_TO_POINTER(key));
 238 }
 239
 240 GSList *tags_get_list(void)
 241 {
 242         TagListData data;
 243         data.list = NULL;
 244
 245         g_hash_table_foreach(tags_table, tag_add_list, &data);
 246
 247         data.list = g_slist_reverse(data.list);
 248
 249         return data.list;
 250 }
 251
 252 guint tags_get_size(void)
 253 {
 254         return g_hash_table_size(tags_table);
 255 }