From d3b7051c2015858055ca8d235d2026a235b1af0f Mon Sep 17 00:00:00 2001
From: qux-bbb <1147635419@qq.com>
Date: Fri, 12 Aug 2022 23:20:25 +0800
Subject: [PATCH] =?UTF-8?q?=E5=8F=AF=E8=BE=93=E5=87=BA=E8=87=AA=E8=BA=AB?=
 =?UTF-8?q?=E7=89=88=E6=9C=AC=E4=BF=A1=E6=81=AF?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 setup.py            | 2 +-
 xanalyzer/VERSION   | 1 +
 xanalyzer/config.py | 1 +
 xanalyzer/main.py   | 5 +++++
 4 files changed, 8 insertions(+), 1 deletion(-)
 create mode 100644 xanalyzer/VERSION

diff --git a/setup.py b/setup.py
index 9485b71..de2df60 100644
--- a/setup.py
+++ b/setup.py
@@ -3,7 +3,7 @@
 
 setup(
     name='xanalyzer',
-    version='0.2.1',
+    version=open('xanalyzer/VERSION', 'r').read(),
     packages=find_packages(),
     package_data={
         'xanalyzer': [
diff --git a/xanalyzer/VERSION b/xanalyzer/VERSION
new file mode 100644
index 0000000..7dff5b8
--- /dev/null
+++ b/xanalyzer/VERSION
@@ -0,0 +1 @@
+0.2.1
\ No newline at end of file
diff --git a/xanalyzer/config.py b/xanalyzer/config.py
index 0905b91..e9aac0d 100644
--- a/xanalyzer/config.py
+++ b/xanalyzer/config.py
@@ -10,6 +10,7 @@ class Config:
     # UserDB.TXT文件默认是GBK编码, 需要简单处理一下, 去掉乱码, 新的UserDB.TXT加了新的规则, peutils解析不了
     peid_signature_path = home_dir / 'data' / 'UserDB.TXT'
     tools_info_path = home_dir / 'data' / 'tools_info.json'
+    VERSION = open(home_dir / 'VERSION', 'r').read().strip()
 
     conf = {}
 
diff --git a/xanalyzer/main.py b/xanalyzer/main.py
index 1b05096..8234869 100644
--- a/xanalyzer/main.py
+++ b/xanalyzer/main.py
@@ -18,9 +18,14 @@ def main():
     group = parser.add_mutually_exclusive_group(required=True)
     group.add_argument('-f', '--file', nargs='+', help='analyze one or more files, can be a folder path')
     group.add_argument('-u', '--url', help='analyze the url')
+    group.add_argument('--version', action='store_true', help='print version info')
     parser.add_argument('-s', '--save', action='store_true', help='save log and data')
     args = parser.parse_args()
 
+    if args.version:
+        print(Config.VERSION)
+        return
+
     Config.init(args.save)
     init_log()