git.roojs.org Git - app.mailtrimmer/blob - src/strip.vala

   1 /**
   2
   3  ** check left to do:
   4   - range scans on maildir
   5   - see how replacing the links works in the resulting email via thunderbird etc..
   6   - some checksum issues (see dupelicates?? suspect 0byte issues?)
   7
   8
   9
  10
  11   needs to scan 2 things
  12   a) our mailfort email database
  13        point it at the top directory, containing YEAR/MONTH/DAY.... directories.
  14        scan each file (over a year old...)
  15        extract out the attachment, and replace with HTML
  16        DATABASE? - mysql or sqlite? -
  17            filesize / name / date / checksum / mimetype -- into mailfort should be OK.
  18   b) the imap user emails
  19            loop through user's directories
  20            check age of email .. over 1 years..
  21            ?? how to prevent 'repeat' scanning of emails?
  22               ??? hidden '.' files containing last scan date?
  23
  24            check if file exists in our DB.. - replace the link...
  25            otherwise generate a file. + add to DB...
  26
  27    c) retreival system
  28      -> URL -> get file
  29    d) redirect system.
  30      -> URL -> redirect to correct server
  31
  32
  33 More notes on our Mailfort DB sync:
  34 * some of these attachments are already in the database...
  35  - so we need to update the DB..
  36  - probably worth putting the code in a stored procedure..
  37
  38  -- key scenarios
  39    * first scan (and extract)
  40    * rescan (as I messed up the first time - fix the DB...)
  41    * email scan - attachments might not have related messages.
  42
  43
  44  - {id} attachment_init(
  45                 {exim_msg_id}
  46                 {chksum}
  47                 {filename),
  48         )
  49         // creates or returns id (can look for existing messages?
  50         // can do a merge?? - copy 'old' record data into 'new'....  "prefer checksummed"
  51
  52         attachment_update(
  53                 {id}
  54                 {exim_msg_id}
  55                 {mailfort_msg_sig}
  56                 {file_size}
  57                 {created} // message date..
  58                 {chksum}
  59                 {filename),
  60         {mime_type}
  61         )
  62         attachment_update_store(
  63                         {id}
  64                         {stored_filename}
  65         )
  66
  67
  68 */
  69
  70 // valac --pkg gmime --vapi
  71 /*
  72
  73 // http://www.fromdual.com/mysql-vala-program-example << check mysql if this does not work.
  74
  75  valac  -g --vapidir=. --thread  strip.vala   --vapidir=../vapi \
  76      --pkg glib-2.0 --pkg mysql --pkg gio-2.0 --pkg posix --pkg gmime-2.6 \
  77       --Xcc=-lmysqlclient  -v \
  78        -o /tmp/strip
  79 */
  80
  81 public class StripApplication : GLib.Application {
  82
  83         public static string? opt_path = null;
  84         public static string? opt_file = null;
  85         public static string? opt_target_path = null;
  86         public static string? opt_db_host = "127.0.0.1";
  87         public static string? opt_db_name = null;
  88         public static string? opt_db_user = null;
  89         public static string? opt_db_pass = null;
  90
  91         public static int    opt_limit = -1;
  92
  93         public static int    opt_age_newest = 1;
  94         public static int    opt_age_oldest = 6;
  95
  96
  97         public static bool      opt_is_extracting = false;
  98         public static bool      opt_is_replacing = false;
  99         public static bool      opt_scan_maildir  = false;
 100         public static bool      opt_scan_mailfort  = false;
 101         public static bool              opt_dump = false;
 102         public static bool              opt_debug = false;
 103
 104         public static bool opt_debug_sql = false;
 105         public static string? opt_replace_link = null;
 106
 107
 108         public const GLib.OptionEntry[] options = {
 109
 110                 { "debug", 0, 0, OptionArg.NONE, ref opt_debug, "show debug messages for components", null },
 111                 { "debug-sql", 0, 0, OptionArg.NONE, ref opt_debug_sql, "debug the SQL statements", null },
 112
 113                 { "path", 0, 0, OptionArg.STRING, ref opt_path, "Directory where email to be parsed is", null },
 114                 { "file", 0, 0, OptionArg.STRING, ref opt_file, "A specific file to be parsed", null },
 115
 116                 { "target-path", 0, 0, OptionArg.STRING, ref opt_target_path, "Directory where attachments are to be put", null },
 117
 118                 { "link", 0, 0, OptionArg.STRING, ref opt_replace_link, "url for the replement link: eg. http://www.mysite.com/xxxx/%s", null },
 119
 120                 { "host", 0, 0, OptionArg.STRING, ref opt_db_host, "Mysql host (default localhost)", null },
 121                 { "name", 0, 0, OptionArg.STRING, ref opt_db_name, "Mysql database name REQUIRED", null },
 122                 { "user", 0, 0, OptionArg.STRING, ref opt_db_user, "Mysql database user REQUIRED", null },
 123                 { "pass", 0, 0, OptionArg.STRING, ref opt_db_pass, "Mysql database password (default empty)", null },
 124
 125                 { "extract", 0, 0, OptionArg.NONE, ref opt_is_extracting, "Should attachments be extracted (default NO)", null },
 126                 { "replace", 0, 0, OptionArg.NONE, ref opt_is_replacing, "Should attachments be replaced (default NO)", null },
 127                 { "dump", 0, 0, OptionArg.NONE, ref opt_dump, "Print the replaced mail contents to stdout", null },
 128
 129                 { "limit", 0, 0, OptionArg.INT, ref opt_limit, "stop after X number of messages with attachments have been processed", null },
 130                 { "newest", 0, 0, OptionArg.INT, ref opt_age_newest, "do not replace messages newer that X months (default is 1 months)", null },
 131                 { "oldest", 0, 0, OptionArg.INT, ref opt_age_oldest, "do not replace messages older than X (default is 6 months)", null },
 132
 133                 { "scan-maildir", 0, 0, OptionArg.NONE, ref opt_scan_maildir, "scan an maildir tree", null },
 134                 { "scan-mailfort", 0, 0, OptionArg.NONE, ref opt_scan_mailfort, "scan a mailfort tree", null },
 135                 { null }
 136         };
 137     public StripApplication( string[] args )
 138     {
 139                  Object(
 140             application_id: "org.roojs.mailstripper",
 141             flags: ApplicationFlags.FLAGS_NONE
 142          );
 143
 144
 145          var opt_context =  new GLib.OptionContext ("Mail Stripper");
 146
 147          try {
 148
 149             opt_context.set_help_enabled (true);
 150             opt_context.add_main_entries (options, null);
 151             opt_context.parse ( ref  args);
 152             //opt_detach = !optx_no_detach;
 153
 154
 155
 156              // options that have to be set.. bee or hive... (or stop all)
 157             if ((!opt_scan_mailfort && !opt_scan_maildir) || (opt_scan_mailfort && opt_scan_maildir))  {
 158                stdout.printf ("You must specify the type of directory tree to scan - either imap or mailfort\n%s",
 159                    opt_context.get_help(true, null));
 160                GLib.Process.exit(Posix.EXIT_FAILURE);
 161             }
 162
 163                          if ((opt_db_name == null || opt_db_name.length < 1 || opt_db_user == null || opt_db_user.length < 1))  {
 164                stdout.printf ("You must specify the database name / user \n%s",
 165                    opt_context.get_help(true, null));
 166                GLib.Process.exit(Posix.EXIT_FAILURE);
 167             }
 168                          if ((opt_path == null || opt_path.length < 1)   )  {
 169                stdout.printf ("You must specify the scan start path\n%s",
 170                    opt_context.get_help(true, null));
 171                GLib.Process.exit(Posix.EXIT_FAILURE);
 172             }
 173                         if (opt_replace_link == null || (opt_replace_link.length < 1))  {
 174                stdout.printf ("You must specify the link to use in the replacement \n%s",
 175                    opt_context.get_help(true, null));
 176                GLib.Process.exit(Posix.EXIT_FAILURE);
 177             }
 178             if ((opt_is_replacing || opt_is_extracting ) && (opt_target_path == null || opt_target_path.length < 1)) {
 179                       stdout.printf ("You must specify a target path to put attachments\n%s",
 180                    opt_context.get_help(true, null));
 181                GLib.Process.exit(Posix.EXIT_FAILURE);
 182             }
 183
 184
 185          } catch (GLib.OptionError e) {
 186             stdout.printf ("error: %s\n", e.message);
 187             stdout.printf ("Run '%s --help' to see a full list of available command line options.\n%s",
 188                       args[0], opt_context.get_help(true, null));
 189             GLib.Process.exit(Posix.EXIT_FAILURE);
 190          }
 191         }
 192
 193     public static int main(string[] args)
 194     {
 195
 196                 var application = new StripApplication(  args);
 197
 198                 GLib.Log.set_always_fatal(LogLevelFlags.LEVEL_ERROR | LogLevelFlags.LEVEL_CRITICAL);
 199
 200            if (opt_debug || opt_debug_sql) {
 201                         GLib.Log.set_handler(null,
 202                         GLib.LogLevelFlags.LEVEL_DEBUG | GLib.LogLevelFlags.LEVEL_WARNING | GLib.LogLevelFlags.LEVEL_INFO,
 203                         (dom, lvl, msg) => {
 204                                         print("%s\n", msg);
 205                                 }
 206                         );
 207                 }
 208
 209         GMime.init(0);
 210                 if (StripApplication.opt_is_replacing) {
 211                         StripApplication.opt_is_extracting = true;
 212                 }
 213
 214                 GLib.debug("scanning folder: %s", opt_path );
 215
 216                 var strip = new Strip( opt_path );
 217
 218
 219                 strip.mysql  = new Mysql.Database();
 220                 if (!strip.mysql.real_connect(
 221                                 opt_db_host,
 222                                 opt_db_user ,
 223                                 opt_db_pass == null ? "" : opt_db_pass, //passwd
 224                                 opt_db_name, //DB
 225                                 3306, // not changable...?
 226                                 null
 227                         )
 228                 ) {
 229                         stdout.printf("ERROR %u: Connection failed: %s\n",
 230                                 strip.mysql.errno(), strip.mysql.error()
 231                         );
 232
 233                         return 1;
 234                 }
 235         if (opt_file != null) {
 236                 strip.base_dir = opt_path;
 237                 strip.scan_file( GLib.Path.get_dirname(opt_file),  GLib.Path.get_basename(opt_file));
 238                 return 0;
 239         }
 240
 241                 strip.scan_dir(opt_path, "");
 242
 243
 244
 245         return 0;
 246     }
 247 }
 248
 249 public class Strip : GLib.Object {
 250
 251
 252
 253         public string base_dir = "";
 254
 255         public Mysql.Database mysql;
 256
 257         int processed = 0;
 258
 259     uint64 used_space_before = 0;
 260     uint64 used_space_after = 0;
 261
 262
 263     public Strip(string base_dir)
 264     {
 265         this.base_dir = base_dir;
 266     }
 267
 268     public void handle_part(GMime.Object parent, GMime.Object mime_obj)
 269     {
 270                 if (mime_obj is GMime.Part) {
 271                    var  p = (GMime.Part)mime_obj;
 272                         var ct = p.get_content_type();
 273                         var cd = p.get_content_disposition();
 274
 275                         var sid = p.get_header("X-strip-id");
 276                     if (sid != null && sid.length > 0) {
 277                         this.update_attachment_db(p);
 278                             GLib.debug("Skip attachment replace - it's already been done");
 279                         return;
 280                         }
 281
 282                         if (cd == null || cd.get_disposition().down() != "attachment") {
 283                                 return;
 284                         }
 285                         if (ct.get_media_type() == "text") {
 286                                 return;
 287                         }
 288                         if (ct.to_string() == "application/pgp-encrypted") {
 289                                 return;
 290                         }
 291                         if (ct.to_string() == "application/pgp-keys") {
 292                                 return;
 293                         }
 294                         if (p.get_filename() == null) {
 295                                 return;
 296                         }
 297                          // print("got part %s\n", ct.to_string());
 298                          if (parent is GMime.Multipart) {
 299
 300                                 this.replace_attachment(((GMime.Multipart)parent), p);
 301                                 // remove it !?
 302
 303                           }
 304
 305
 306                         return;
 307                 }
 308                 if (mime_obj is GMime.Multipart) {
 309
 310
 311                         var  mp = (GMime.Multipart)mime_obj;
 312                         //var ct = mp.get_content_type();
 313
 314                         //print("got multi-part %s\n", ct.to_string());
 315                         for (var i = 0; i< mp.get_count(); i++) {
 316                           var mo = mp.get_part(i);
 317                           this.handle_part(mime_obj,mo);
 318                         }
 319                    // ((GMime.Multipart)mime_obj).foreach((sub_obj) => {
 320                    //     Strip.handle_part(sub_obj);
 321                 //
 322                    // });
 323
 324
 325                         return;
 326                 }
 327
 328                 if (mime_obj is GMime.MessagePart) {
 329                         var msg = ((GMime.MessagePart)mime_obj).get_message();
 330                         msg.foreach((subobj) => {
 331                          this.handle_part(msg,subobj);
 332                     });
 333
 334                         //print("got message-part\n");
 335                         return;
 336                 }
 337
 338                 if (mime_obj is GMime.Message) {
 339                         var mp = ((GMime.Message) mime_obj).get_mime_part();
 340
 341                         if (!(mp is GMime.Multipart)) {
 342                                 //GLib.debug("get mimepart does not return a Multipart?");
 343                                 return;
 344                         }
 345
 346                         var mpc = ((GMime.Multipart)mp).get_count();
 347
 348                         //GLib.debug("Message has %d parts", mpc);
 349                         for (var i =0 ; i < mpc; i++) {
 350                                 //GLib.debug("Getting part %d", i);
 351                                 var submime_obj = ((GMime.Multipart)mp).get_part(i);
 352                         this.handle_part(mp,submime_obj);
 353                     }
 354                         print("got message??\n");
 355                         return;
 356                 }
 357
 358                 print("got something else\n");
 359
 360
 361     }
 362     public void update_attachment_db(GMime.Part attachment)
 363     {
 364         // only called when we have an sid...
 365         var sid = attachment.get_header("X-strip-id");
 366         if (sid == null || sid.length < 1) {
 367                 GLib.debug("Strange - update attachment db called ?");
 368                 return;
 369         }
 370
 371         // initialize it with known data..
 372         // that should wipe out dupes.
 373         var filesize = this.query("SELECT filesize FROM Attachment WHERE id = %d".printf(
 374                         int.parse(sid)));
 375                 if (int.parse(filesize) < 1) {
 376                 GLib.debug("Could not get filesize from id :%s = %s", sid,filesize);
 377                 Posix.exit(0);
 378                 return;
 379         }
 380
 381         var chksum = this.query("SELECT  checksum FROM Attachment WHERE id = %d".printf(
 382                         int.parse(sid)
 383                 ));
 384         var mime_filename = this.query("SELECT  mime_filename FROM Attachment WHERE id = %d".printf(
 385                         int.parse(sid)));
 386
 387         this.query("""
 388              SELECT
 389                  attachment_init(
 390                      '%s', '%s', '%s', %d
 391                  ) as id
 392
 393           """.printf(
 394                           this.mysql_escape(this.active_message_exim_id),
 395                           this.mysql_escape(chksum),
 396                           this.mysql_escape(mime_filename),
 397                           int.parse(filesize)
 398                 ));
 399         this.query("""
 400                  SELECT attachment_update(
 401                       %d, -- in_id INT(11),
 402                       '%s', -- in_mime_type varchar(255),
 403                       '%s', -- in_created DATETIME,
 404                       '%s' -- in_mailfort_sig varchar(64)
 405                  )
 406               """.printf(
 407                         int.parse(sid),
 408                         "", // this will be ignored..
 409                                 this.created_date,
 410                                 this.mysql_escape(this.active_message_x_mailfort_sig)
 411
 412               )
 413                 );
 414                 this.mysql.store_result();
 415
 416
 417
 418     }
 419
 420
 421     public void replace_attachment(GMime.Multipart parent, GMime.Part attachment)
 422     {
 423         var sid = attachment.get_header("X-strip-id");
 424         if (sid != null && sid.length > 0) {
 425                 GLib.debug("Skip attachment replace - it's already been done");
 426                 return;
 427         }
 428
 429         var c = attachment.get_content_object();
 430
 431         var filename = attachment.get_filename().replace("/", "-").replace("\n", "").replace("\t", " ");
 432         var fn = GLib.Environment.get_tmp_dir() +
 433                         "/"+ this.active_name + "."+   filename;
 434
 435             var outfile = new GMime.StreamFile.for_path(fn, "w");
 436             outfile.set_owner(true);
 437             var file_size = (int) c.write_to_stream(outfile);
 438             var chksum = this.md5_file(fn);
 439             outfile.flush();
 440             outfile = null;
 441
 442         if (file_size == 0) {
 443
 444                 GLib.debug("ERROR - file size of write to stream returned 0?");
 445                 Posix.unlink(fn);
 446                 return;
 447         }
 448
 449
 450
 451
 452         var mime_type= attachment.get_content_type().to_string();
 453         // at this point we have to do our database magic...
 454         //filesize / name / date / checksum / mimetype -- into mailfort should be OK.
 455
 456         var file_id = this.query("""
 457                 SELECT
 458
 459                 attachment_init(
 460                                 '%s', -- in_msgid VARCHAR(32),
 461                                 '%s', -- in_checksum VARCHAR(64),
 462                                 '%s', -- in_mime_filename varchar(255)
 463                                 %d -- filesize
 464                         ) as id
 465
 466           """.printf(
 467                         this.mysql_escape(this.active_message_exim_id),
 468                         chksum,
 469                         this.mysql_escape( attachment.get_filename() ), // what is thsi is invalid?
 470                          file_size)
 471                 );
 472
 473
 474                 if (file_id.length < 1) {
 475                         GLib.debug("ERROR - CALL to attachment_init failed");
 476                 Posix.unlink(fn);
 477                 return;
 478
 479                 }
 480
 481                 if (int.parse(file_id) < 1) {
 482                         GLib.debug("ERROR - CALL to attachment_init failed - returned 0?");
 483                 Posix.unlink(fn);
 484                 return;
 485
 486                 }
 487
 488
 489                 GLib.debug("fn = %s, m5=%s, id= %s", filename, mime_type, this.active_message_id);
 490                 this.query("""
 491
 492                         SELECT attachment_update(
 493                                 %d, -- in_id INT(11),
 494                                 '%s', -- in_mime_type varchar(255),
 495                                 '%s', -- in_created DATETIME,
 496                                 '%s' -- in_mailfort_sig varchar(64)
 497
 498                                 ) as result
 499       """.printf(
 500                 int.parse(file_id),
 501                         this.mysql_escape(mime_type),
 502                         this.created_date,
 503                         this.mysql_escape(this.active_message_x_mailfort_sig)
 504                 ));
 505                  this.mysql.store_result();
 506
 507
 508                 this.used_space_after += file_size;
 509
 510                 var target_fn = "";
 511
 512             if (StripApplication.opt_is_extracting) {
 513                         target_fn = StripApplication.opt_target_path + "/" + this.created_dir +"/"+ file_id  + "-" + filename;
 514                 }
 515
 516             var stored =  "/" + this.created_dir +"/"+ file_id  + "-" + filename;
 517                  this.query("""
 518
 519                         SELECT attachment_update_store(
 520                                 %d, -- in_id INT(11),
 521                                 '%s'  -- in_store_filename varchar(255),
 522
 523
 524                                 ) as result
 525       """.printf(
 526                 int.parse(file_id),
 527                          this.mysql_escape( stored)
 528                 ));
 529
 530         var rep = new GMime.Part.with_type("text","html");
 531         // we have to set up a redirect server - to redirect hpasite... to their internal service..
 532         rep.set_filename(filename);
 533         string txt = "<html><body>"+
 534             "<a href=\"" + StripApplication.opt_replace_link + "/" +
 535                         file_id + "/" + this.created_dir + "/"+chksum+"/"+ GLib.Uri.escape_string( filename) +"\">" +
 536             GLib.Uri.escape_string( filename) + // fixme needs html escaping...
 537             "</a>" +
 538             "</body></html>";
 539
 540         rep.get_content_type().set_parameter("charset", "utf-8");
 541                 rep.set_header("X-strip-id", file_id);
 542                 rep.set_header("X-strip-content-name",  filename);
 543                 rep.set_header("X-strip-path", this.created_dir + "/" + file_id + "-" + filename);
 544                 rep.set_header("X-strip-content-type", mime_type);
 545         var stream =  new GMime.StreamMem.with_buffer(txt.data);
 546         var con = new GMime.DataWrapper.with_stream(stream,GMime.ContentEncoding.DEFAULT);
 547
 548         rep.set_content_object(con);
 549         GLib.debug("Replacing Attachment with HTML");
 550         parent.replace(parent.index_of(attachment), rep);
 551                 this.has_replaced = true;
 552
 553                 if (StripApplication.opt_is_extracting && target_fn.length > 0) {
 554                         var dir = GLib.Path.get_dirname(target_fn);
 555                         if (!FileUtils.test (dir, FileTest.IS_DIR)) {
 556                                 GLib.DirUtils.create_with_parents(dir, 0755);
 557                         }
 558                         GLib.debug("Creating file %s", target_fn);
 559                         if (!FileUtils.test (target_fn, FileTest.EXISTS)) {
 560                                 Posix.link(fn, target_fn);
 561                         }
 562                 } else {
 563                         GLib.debug("Skipping extraction %s", target_fn);
 564                 }
 565                 Posix.unlink(fn);
 566
 567
 568
 569     }
 570     public string query(string str)
 571     {
 572             return this.real_query(true, str);
 573     }
 574     public string execute(string str)
 575     {
 576             return this.real_query(false, str);
 577     }
 578     public string real_query(bool need_return, string str)
 579     {
 580                 GLib.debug("Before Query : %u  : %s\n", this.mysql.errno(), this.mysql.error());
 581
 582
 583         if (StripApplication.opt_debug_sql) {
 584                 GLib.debug("SQL: %s\n", str);
 585                 }
 586
 587
 588
 589         var rc=  this.mysql.query(str);
 590         if ( rc != 0 ) {
 591
 592                     GLib.debug("ERROR %u: Query failed: %s\n", this.mysql.errno(), this.mysql.error());
 593                                 Posix.exit(1);
 594                 }
 595
 596
 597         var rs = mysql.use_result();
 598
 599         var got_row = false;
 600                 string[] row;
 601                 string ret = "";
 602                 while( (row = rs.fetch_row()) != null) {
 603                         got_row = true;
 604                         ret = row[0];
 605
 606                 }
 607                 if (!need_return) {
 608                         return "";
 609                 }
 610                 if (!got_row) {
 611                          GLib.debug("ERROR : no rows returned");
 612                         Posix.exit(1);
 613                         return "";
 614                 }
 615                 GLib.debug("got %s", ret);
 616                 return ret;
 617
 618
 619         }
 620
 621     public string mysql_escape(string str)
 622     {
 623             unichar[] value_escaped = new unichar[str.length * 2 + 1];
 624                 this.mysql.real_escape_string ((string) value_escaped, str, str.length);
 625                 return (string) value_escaped;
 626     }
 627
 628     public string  md5_file(string fn) {
 629               Checksum checksum = new Checksum (ChecksumType.MD5);
 630
 631               FileStream stream = FileStream.open (fn, "rb");
 632               uint8 fbuf[100];
 633               size_t size;
 634
 635               while ((size = stream.read (fbuf)) > 0) {
 636                       checksum.update (fbuf, size);
 637               }
 638
 639               unowned string digest = checksum.get_string ();
 640               return digest;
 641     }
 642
 643         string active_path = "";
 644     string active_name = "";
 645     string active_message_id = "";
 646     string active_message_x_mailfort_sig = "";
 647     string active_message_exim_id = "";
 648     bool has_replaced = false;
 649     string created_date = ""; // should be YYYY-mm-dd
 650     string created_dir = ""; // should be YYY/mm/dd
 651
 652     public void scan_file(string path, string name)
 653     {
 654                 GLib.debug("Scan: %s/%s", path,name);
 655
 656                 this.has_replaced = false;
 657         this.active_path = path;
 658         this.active_name = name;
 659         this.active_message_id = "";
 660
 661                 var mailtime = new DateTime.now_local();
 662                 if (StripApplication.opt_scan_mailfort) {
 663                     this.created_dir = this.active_path.substring(this.base_dir.length + 1 );
 664                         this.created_date = this.created_dir.replace("/", "-");
 665                         var bits = this.created_date.split("-");
 666                         mailtime = new DateTime.local(int.parse(bits[0]),int.parse(bits[1]),int.parse(bits[2]),0,0,0);
 667
 668                         var oldest = new  DateTime.now_local();
 669                         oldest = oldest.add_months(-1 * StripApplication.opt_age_oldest);
 670                         var tspan = mailtime.difference(oldest) / GLib.TimeSpan.DAY;
 671
 672                         if (tspan < 0) {
 673                                 GLib.debug("skip file is %d days older than %d months", (int)tspan, StripApplication.opt_age_oldest);
 674                                 return;
 675                         }
 676
 677                         var newest = new  DateTime.now_local();
 678                         newest = newest.add_months(-1 * StripApplication.opt_age_newest);
 679                         tspan = mailtime.difference(newest) / GLib.TimeSpan.DAY;
 680                         if (tspan > 0) {
 681                                 GLib.debug("skip file is %d days newer than %d months", (int)tspan, StripApplication.opt_age_newest);
 682                                 return;
 683                         }
 684
 685                 }
 686
 687
 688                 var fileinfo = File.new_for_path(path +"/" + name)
 689                                         .query_info(GLib.FileAttribute.STANDARD_SIZE+","+GLib.FileAttribute.TIME_MODIFIED
 690                                                 ,GLib.FileQueryInfoFlags.NONE,null);
 691         var file_size = (int) fileinfo.get_size();
 692                 var mod_time = fileinfo.get_modification_time();
 693
 694
 695
 696                 if (!StripApplication.opt_scan_mailfort) {
 697
 698                 // it's a mail directory...
 699                 // use the last modification time? as the default...
 700                  mailtime = new DateTime.from_timeval_utc(mod_time);
 701                  this.created_dir = mailtime.format("%Y/%m/%d");
 702                          this.created_date =  mailtime.format("%Y-%m-%d %H:%M:%S");
 703
 704         }
 705                 // check on age of file...
 706
 707
 708
 709
 710
 711         this.used_space_before += file_size;
 712
 713         var stream = new GMime.StreamFs.for_path (path +"/" + name,Posix.O_RDONLY, 0);
 714         //stream.set_owner(true);
 715         var parser = new GMime.Parser.with_stream(stream);
 716         var message = parser.construct_message();
 717
 718                 if (message == null) {
 719                         GLib.debug("Could not parse file? %s/%s", path,name);
 720                 this.used_space_after += file_size;
 721                 return;
 722                 }
 723
 724
 725                 // check : - is message over a year old?
 726                 // get various msg info..
 727                 this.active_message_id = message.get_message_id();
 728                 this.active_message_x_mailfort_sig = message.get_header("x-mailfort-sig");
 729                 var recvd = message.get_header("received");
 730                 this.active_message_exim_id = "";
 731                 if (recvd != null && recvd.length > 1) {
 732                         GLib.debug("RECV: %s", recvd);
 733                         var lines = recvd.split("\t");
 734                         for (var i = 0; i < lines.length;i++) {
 735                                 var bits = lines[i].strip().split(" ");
 736                                 if (bits[0] == "id") {
 737                                         this.active_message_exim_id = bits[1].replace(";","");
 738
 739                                 }
 740
 741                                 if (lines[i].contains(";")) {
 742                                         var dbits = lines[i].strip().split(";");
 743                                         GLib.debug("Reading time from : %s", dbits[1]);
 744                                         var timez = GMime.utils_header_decode_date(dbits[1], null);
 745                                         if (timez != 0) {
 746                                                 mailtime = new DateTime.from_unix_utc(timez);
 747                                                 this.created_date = mailtime.format("%Y-%m-%d %H:%M:%S");
 748                                                 GLib.debug("Time is %s",this.created_date);
 749                                                 // if it's not mailfort we can use that date to determine where to store it...
 750                                                 if (!StripApplication.opt_scan_mailfort) {
 751                                                         this.created_dir = mailtime.format("%Y/%m/%d");
 752                                                 }
 753                                         } else {
 754                                                 GLib.debug("Could not read time from headers?");
 755                                         }
 756                                 }
 757
 758                         }
 759                 }
 760
 761                 var oldest = new  DateTime.now_local();
 762                 oldest = oldest.add_months(-1 * StripApplication.opt_age_oldest);
 763                 var rtspan = mailtime.difference(oldest) / GLib.TimeSpan.DAY;
 764                 GLib.debug("Checking oldest %d days difference", (int)rtspan   );
 765                 if (rtspan < 0) {
 766                         GLib.debug("skip(2) file is %d days older than %d months", (int)rtspan, StripApplication.opt_age_oldest);
 767                         return;
 768                 }
 769                 var newest = new  DateTime.now_local();
 770                 newest = newest.add_months(-1 * StripApplication.opt_age_newest);
 771                 rtspan = mailtime.difference(newest) / GLib.TimeSpan.DAY;
 772                 if (rtspan > 0) {
 773                         GLib.debug("skip(2) file is %d days newer than %d months : %s", (int)rtspan, StripApplication.opt_age_newest,
 774                                 mailtime.format("%Y-%m-%d %H:%M:%S"));
 775                         return;
 776                 }
 777
 778
 779
 780                 /*
 781                 GLib.debug("Message DATA:\n mid: %s\nmailfort: %s \nexim_id: %s",
 782                         this.active_message_id,
 783                         this.active_message_x_mailfort_sig,
 784                         this.active_message_exim_id
 785                 );
 786                  */
 787
 788                 // DATE?
 789
 790                 var mp = message.get_mime_part();
 791
 792                 if (!(mp is GMime.Multipart)) {
 793                         //GLib.debug("get mimepart does not return a Multipart?");
 794                 this.used_space_after += file_size;
 795                         return;
 796                 }
 797
 798                 var mpc = ((GMime.Multipart)mp).get_count();
 799
 800                 //GLib.debug("Message has %d parts", mpc);
 801                 for (var i =0 ; i < mpc; i++) {
 802                         //GLib.debug("Getting part %d", i);
 803                         var mime_obj = ((GMime.Multipart)mp).get_part(i);
 804             this.handle_part(mp,mime_obj);
 805         }
 806
 807         parser= null;
 808
 809       //  stream.set_owner(false);
 810             //stream.close();
 811         stream = null;//.close();
 812
 813
 814                 if (!this.has_replaced) {
 815                         this.used_space_after += file_size;
 816                         GLib.debug("skpping write file - no replacement occured");
 817                         return;
 818                 }
 819                 string tmpfile = "";
 820                 GMime.Stream outstream = new GMime.StreamNull();
 821                 if (StripApplication.opt_is_replacing) {
 822
 823                         tmpfile = GLib.Environment.get_tmp_dir() +"/" + name;
 824                 outstream = new GMime.StreamFile.for_path (tmpfile,"w");
 825                 ((GMime.StreamFile)outstream).set_owner(true);
 826         }
 827                 if (StripApplication.opt_dump) {
 828                         outstream = new GMime.StreamMem();
 829         }
 830
 831         file_size = (int) message.write_to_stream(outstream);
 832         if (StripApplication.opt_is_replacing) {
 833                 ((GMime.StreamFile)outstream).set_owner(false);
 834         }
 835                 if (StripApplication.opt_dump) {
 836                         var ua = ((GMime.StreamMem)outstream).get_byte_array().data;
 837                         print("%s\n", (string) ua);
 838                 }
 839         message = null;
 840         outstream.flush();
 841         outstream.close();
 842         GLib.debug("finished writing output %d", file_size);
 843
 844         //
 845
 846
 847         this.used_space_after += file_size;
 848
 849
 850         if (StripApplication.opt_is_replacing) {
 851                 Posix.unlink(path +"/" + name);
 852                 GLib.debug("copy tmp file %s to %s" , tmpfile, path +"/" + name);
 853                 Posix.link(tmpfile, path +"/" + name);
 854                 Posix.unlink(tmpfile);
 855                 var nf = File.new_for_path(path +"/" + name);
 856                 var newfileinfo = nf.query_info(GLib.FileAttribute.TIME_MODIFIED,GLib.FileQueryInfoFlags.NONE,null);
 857                 newfileinfo.set_modification_time(mod_time);
 858                 nf.set_attributes_from_info(newfileinfo,FileQueryInfoFlags.NONE);
 859                 }
 860         this.processed++;
 861
 862         if (StripApplication.opt_limit > -1 && this.processed >= StripApplication.opt_limit) {
 863                 GLib.debug("Reached replacement limit");
 864                 Posix.exit(1);
 865         }
 866
 867
 868
 869
 870     }
 871
 872
 873     public void scan_dir(string basepath, string subpath)
 874     {
 875
 876
 877         // determine if path is to old to scan..
 878         if (subpath.length > 0 && StripApplication.opt_scan_mailfort) {
 879                         var year =  int.parse(subpath.substring(1,4));  // "/2000"
 880                         var month = subpath.length > 5 ? int.parse(subpath.substring(6,2)) : 999; // "/2000/12"
 881                         var day = subpath.length > 8 ? int.parse(subpath.substring(9,2)) : 999; // "/2000/12/01"
 882
 883                 var oldest = new  DateTime.now_local();
 884                         oldest = oldest.add_months(-1 * StripApplication.opt_age_oldest);
 885
 886                         //GLib.debug("Checking directory %s is older than min: %d/%d/%d", subpath, oldest.get_year() , oldest.get_month(), oldest.get_day_of_month() );
 887
 888                         if (year < oldest.get_year()) {
 889                                 GLib.debug("Skip directory %s is older than min year: %d", subpath, oldest.get_year());
 890                                 return;
 891                         }
 892                         if (year == oldest.get_year() &&  month < oldest.get_month()) {
 893                                 GLib.debug("Skip directory %s is older than min month: %d/%d", subpath, oldest.get_year() , oldest.get_month() );
 894                                 return;
 895                         }
 896                 if (year == oldest.get_year() &&  month == oldest.get_month() && day < oldest.get_day_of_month()) {
 897                                 GLib.debug("Skip directory %s is older than min day: %d/%d/%d", subpath, oldest.get_year() , oldest.get_month(), oldest.get_day_of_month() );
 898                                 return;
 899                         }
 900
 901                 var newest = new  DateTime.now_local();
 902                         newest = newest.add_months(-1 * StripApplication.opt_age_newest);
 903
 904                         //GLib.debug("Checking directory %s is newer than max: %d/%d/%d", subpath, newest.get_year() , newest.get_month(), newest.get_day_of_month() );
 905
 906                         if (year > newest.get_year()) {
 907                                 GLib.debug("Skip directory %s is newer than max year: %d", subpath, newest.get_year());
 908                                 return;
 909                         }
 910                         if (year == newest.get_year() &&  month != 999 && month > newest.get_month()) {
 911                                 GLib.debug("Skip directory %s is newer than max month: %d/%d", subpath, newest.get_year() , newest.get_month() );
 912                                 return;
 913                         }
 914                 if (year == newest.get_year() &&  month == newest.get_month() &&  day != 999 && day > newest.get_day_of_month()) {
 915                                 GLib.debug("Skip directory %s is newer than max day: %d/%d/%d", subpath, newest.get_year() , newest.get_month(), newest.get_day_of_month() );
 916                                 return;
 917                         }
 918
 919
 920
 921         }
 922
 923
 924         var f = File.new_for_path(basepath + subpath);
 925                 FileEnumerator file_enum;
 926         var cancellable = new Cancellable ();
 927         try {
 928             file_enum = f.enumerate_children(
 929                 FileAttribute.STANDARD_DISPLAY_NAME + "," +   FileAttribute.STANDARD_TYPE,
 930                         FileQueryInfoFlags.NOFOLLOW_SYMLINKS,  // FileQueryInfoFlags.NONE,
 931                         cancellable
 932                 );
 933         } catch (Error e) {
 934                 GLib.debug("Got error scanning dir? %s", e.message);
 935             // FIXME - show error..
 936             return;
 937         }
 938         FileInfo next_file;
 939
 940         while (cancellable.is_cancelled () == false ) {
 941             try {
 942                 next_file = file_enum.next_file (cancellable);
 943             } catch(Error e) {
 944                 GLib.debug("error getting next file? %s", e.message);
 945                 break;
 946             }
 947
 948             if (next_file == null) {
 949                 break;
 950             }
 951
 952
 953                 if (next_file.get_is_symlink()) {
 954                 next_file = null;
 955                 continue;
 956             }
 957
 958
 959             if (next_file.get_file_type() != FileType.DIRECTORY) {
 960
 961                 if (next_file.get_display_name()[0] == ',') {
 962                         continue;
 963                 }
 964                 // other files to ignore?
 965
 966                 this.scan_file(basepath + subpath , next_file.get_display_name());
 967                                 if(this.has_replaced) {
 968                          this.report_state("After scanning %s/%s".printf(basepath + subpath , next_file.get_display_name()));
 969                         }
 970                 continue;
 971             }
 972
 973
 974             //stdout.printf("Monitor.monitor: got file %s : type :%u\n",
 975             //        next_file.get_display_name(), next_file.get_file_type());
 976
 977
 978
 979             var ds = next_file.get_display_name();
 980             // not really needed?? - we are storing attachments in a seperate location now...
 981             if (ds[0] == '.') {
 982                 next_file = null;
 983                 continue;
 984             }
 985             if (ds == "attachments") {
 986                         continue;
 987                 }
 988
 989
 990             var sp = subpath+"/"+next_file.get_display_name();
 991             // skip modules.
 992             //print("got a file : " + sp);
 993
 994             next_file = null;
 995
 996
 997             this.scan_dir(basepath,sp);
 998
 999         }
1000
1001
1002     }
1003     void report_state(string msg)
1004     {
1005         // Saved: 2G  Original 10G : 20%
1006         GLib.debug("Saved : %s (%.1f%%) | Original %s | %s",
1007                         GLib.format_size(this.used_space_before - this.used_space_after),
1008                         100f * ((1f * (this.used_space_before - this.used_space_after)) / (this.used_space_before * 1f)),
1009                         GLib.format_size(this.used_space_before),
1010                         msg
1011                 );
1012
1013         }
1014
1015
1016
1017 }