git.maemo.org Git - modest/blob - tests/check_folder-transfer.c

   1 /* Copyright (c) 2006, Nokia Corporation
   2  * All rights reserved.
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions are
   6  * met:
   7  *
   8  * * Redistributions of source code must retain the above copyright
   9  *   notice, this list of conditions and the following disclaimer.
  10  * * Redistributions in binary form must reproduce the above copyright
  11  *   notice, this list of conditions and the following disclaimer in the
  12  *   documentation and/or other materials provided with the distribution.
  13  * * Neither the name of the Nokia Corporation nor the names of its
  14  *   contributors may be used to endorse or promote products derived from
  15  *   this software without specific prior written permission.
  16  *
  17  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS
  18  * IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
  19  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
  20  * PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER
  21  * OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  22  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  23  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  24  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
  25  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
  26  * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  27  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  28  */
  29
  30 #include <glib.h>
  31 #include <string.h>
  32
  33 #include <tny-list.h>
  34 #include <tny-iterator.h>
  35 #include <tny-simple-list.h>
  36 #include <tny-account-store.h>
  37 #include <tny-store-account.h>
  38 #include <tny-folder.h>
  39 #include <tny-folder-store.h>
  40 #include <modest-tny-platform-factory.h>
  41
  42
  43 #include <modest-account-mgr.h>
  44 #include <modest-mail-operation.h>
  45
  46 static gchar *cachedir=NULL;
  47 static gboolean move=FALSE;
  48 static const gchar *src_name = NULL;
  49 static const gchar *dst_name = NULL;
  50
  51 static void
  52 find_folders (TnyFolderStore *store, TnyFolderStoreQuery *query,
  53               TnyFolder **folder_src, TnyFolder **folder_dst)
  54 {
  55         TnyIterator *iter;
  56         TnyList *folders;
  57
  58         if ((*folder_src != NULL) && (*folder_dst != NULL))
  59                 return;
  60
  61         folders = tny_simple_list_new ();
  62         tny_folder_store_get_folders (store, folders, query, NULL);
  63         iter = tny_list_create_iterator (folders);
  64
  65         while (!tny_iterator_is_done (iter) && (!*folder_src || !*folder_dst))
  66         {
  67                 TnyFolderStore *folder = (TnyFolderStore*) tny_iterator_get_current (iter);
  68                 const gchar *folder_name = NULL;
  69
  70                 folder_name = tny_folder_get_name (TNY_FOLDER (folder));
  71
  72                 if (strcmp (folder_name, src_name) == 0)
  73                     *folder_src = g_object_ref (folder);
  74
  75                 if (!strcmp (folder_name, dst_name))
  76                     *folder_dst = g_object_ref (folder);
  77
  78                 find_folders (folder, query, folder_src, folder_dst);
  79
  80                 g_object_unref (G_OBJECT (folder));
  81
  82                 tny_iterator_next (iter);
  83         }
  84
  85          g_object_unref (G_OBJECT (iter));
  86          g_object_unref (G_OBJECT (folders));
  87 }
  88
  89 static const GOptionEntry options[] = {
  90                 { "from",  'f', 0, G_OPTION_ARG_STRING, &src_name,
  91                   "Source folder", NULL},
  92                 { "to", 't', 0, G_OPTION_ARG_STRING, &dst_name,
  93                   "Destination folder", NULL},
  94                 { "cachedir", 'c', 0, G_OPTION_ARG_STRING, &cachedir,
  95                   "Cache directory", NULL },
  96                 { "move", 'm', 0, G_OPTION_ARG_NONE, &move,
  97                   "Move the messages instead of copy them", NULL },
  98                 { NULL }
  99 };
 100
 101 int
 102 main (int argc, char **argv)
 103 {
 104         GOptionContext *context;
 105         TnyList *accounts, *src_headers;
 106         TnyStoreAccount *account;
 107         TnyIterator *iter;
 108         TnyFolder *folder_src = NULL, *folder_dst = NULL;
 109         TnyPlatformFactory *fact = NULL;
 110         ModestAccountMgr *acc_mgr = NULL;
 111         ModestMailOperation *mail_op = NULL;
 112         TnyAccountStore *account_store = NULL;
 113         guint src_num_headers = 0, dst_num_headers = 0;
 114         GError *err;
 115
 116         g_type_init ();
 117         g_thread_init (NULL);
 118
 119         context = g_option_context_new ("Test");
 120         g_option_context_add_main_entries (context, options, "Modest");
 121         if (!g_option_context_parse (context, &argc, &argv, &err)) {
 122                 g_printerr ("Error in command line parameter(s): '%s', exiting\n",
 123                             err ? err->message : "");
 124                 return 0;
 125         }
 126         g_option_context_free (context);
 127
 128         fact = TNY_PLATFORM_FACTORY (modest_tny_platform_factory_get_instance ());
 129         acc_mgr = MODEST_ACCOUNT_MGR (modest_tny_platform_factory_get_account_mgr_instance (MODEST_TNY_PLATFORM_FACTORY (fact)));
 130         account_store = tny_platform_factory_new_account_store (fact);
 131
 132         if (cachedir)
 133                 g_print ("Using %s as cache directory\n", cachedir);
 134
 135         if (!src_name || !dst_name) {
 136                 g_printerr ("Error in command line parameter(s), specify source and target folders\n");
 137                 return 0;
 138         }
 139
 140         /* Get accounts */
 141         accounts = tny_simple_list_new ();
 142
 143         tny_account_store_get_accounts (account_store, accounts,
 144               TNY_ACCOUNT_STORE_STORE_ACCOUNTS);
 145
 146         iter = tny_list_create_iterator (accounts);
 147         account = (TnyStoreAccount*) tny_iterator_get_current (iter);
 148
 149         g_object_unref (G_OBJECT (iter));
 150         g_object_unref (G_OBJECT (accounts));
 151
 152         /* Find the two folders */
 153         find_folders (TNY_FOLDER_STORE (account), NULL,
 154                       &folder_src, &folder_dst);
 155
 156         if (!folder_src || !folder_dst)
 157                 goto cleanup;
 158
 159         /* Refresh folders */
 160         tny_folder_refresh (folder_src, NULL);
 161         src_num_headers = tny_folder_get_all_count (folder_src);
 162
 163         tny_folder_refresh (folder_dst, NULL);
 164         dst_num_headers = tny_folder_get_all_count (folder_dst);
 165
 166         /* Get all the headers of the source & target folder */
 167         src_headers = tny_simple_list_new ();
 168         tny_folder_get_headers (folder_src, src_headers, TRUE, NULL);
 169
 170         mail_op = modest_mail_operation_new ();
 171
 172         if (move)
 173                 modest_mail_operation_move_folder (mail_op,
 174                                                    folder_src,
 175                                                    TNY_FOLDER_STORE (folder_dst));
 176         else
 177                 modest_mail_operation_copy_folder (mail_op,
 178                                                    folder_src,
 179                                                    TNY_FOLDER_STORE (folder_dst));
 180
 181         g_object_unref (G_OBJECT (src_headers));
 182         g_object_unref (G_OBJECT (mail_op));
 183
 184  cleanup:
 185         g_object_unref (account);
 186
 187         return 0;
 188 }