From f57d5d21dd2b3eed5c1d309215499e9970be07d9 Mon Sep 17 00:00:00 2001
From: Hiyorimi <playittodeath@gmail.com>
Date: Mon, 16 May 2016 11:21:15 +0300
Subject: [PATCH] Added dir and maximum number of messages to backup as a cli
 options

---
 lib/cli_parser.rb        | 11 ++++++++++-
 telegram-history-dump.rb | 12 ++++++++++++
 2 files changed, 22 insertions(+), 1 deletion(-)

diff --git a/lib/cli_parser.rb b/lib/cli_parser.rb
index 9534846..f78438e 100644
--- a/lib/cli_parser.rb
+++ b/lib/cli_parser.rb
@@ -1,6 +1,6 @@
 require 'optparse'
 
-Options = Struct.new(:cfgfile, :kill_tg)
+Options = Struct.new(:cfgfile, :kill_tg, :userdir, :backlog_limit)
 
 class CliParser
   def self.parse(options)
@@ -22,6 +22,15 @@ def self.parse(options)
         puts opts
         exit
       end
+
+      opts.on("-dDIR", "--dir=DIR", String, "Subdirectory for logs") do |userdir|
+        args.userdir = userdir
+      end
+
+      opts.on("-lLIMIT", "--limit=LIMIT", Integer, "Maximum number of messages to backup for each target (0 means unlimited)") do |backlog_limit|
+        args.backlog_limit = backlog_limit
+      end
+
     end
 
     opt_parser.parse!(options)
diff --git a/telegram-history-dump.rb b/telegram-history-dump.rb
index 9008b94..3f2d659 100755
--- a/telegram-history-dump.rb
+++ b/telegram-history-dump.rb
@@ -196,6 +196,18 @@ def save_progress
 
 FileUtils.mkdir_p(get_backup_dir)
 
+if !cli_opts[:userdir].nil?
+        if  !cli_opts.userdir.empty? || !cli_opts.userdir.nil?
+                $config['backup_dir'] = File.join($config['backup_dir'],cli_opts.userdir)
+        end
+end
+
+if !cli_opts[:backlog_limit].nil?
+        if  !cli_opts.backlog_limit.empty? || !cli_opts.backlog_limit.nil?
+                $config['backlog_limit'] = cli_opts.backlog_limit
+        end
+end
+
 $dumper = JsonDumper.new
 $progress = {}
 $progress_snapshot = {}