From 8ccfe0a4e0a7a6b0bb9137eaa1df5363d15c3a0a Mon Sep 17 00:00:00 2001
From: yangjian <zmister@qq.com>
Date: Wed, 28 Oct 2020 21:25:34 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E5=90=8E=E5=8F=B0=E7=AE=A1?=
 =?UTF-8?q?=E7=90=86=E6=96=B0=E7=89=88=E6=9C=AC=E9=9D=99=E9=BB=98=E6=A3=80?=
 =?UTF-8?q?=E6=B5=8B=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app_admin/urls.py                  |  1 +
 app_admin/views.py                 | 12 ++++++++--
 requirements.txt                   |  3 ++-
 template/app_admin/admin_base.html | 35 ++++++++++++++++++++++++++++++
 4 files changed, 48 insertions(+), 3 deletions(-)

diff --git a/app_admin/urls.py b/app_admin/urls.py
index 6d16f39..2add56c 100644
--- a/app_admin/urls.py
+++ b/app_admin/urls.py
@@ -19,5 +19,6 @@ urlpatterns = [
     path('forget_pwd/',views.forget_pwd,name='forget_pwd'), # 忘记密码
     path('send_email_vcode/',views.send_email_vcode,name='send_email_vcode'), # 忘记密码发送邮件验证码
     path('admin_register_code/',views.admin_register_code,name='register_code_manage'), # 注册邀请码管理
+    path('check_update/',views.check_update,name='check_update'), # 检测版本更新
 
 ]
\ No newline at end of file
diff --git a/app_admin/views.py b/app_admin/views.py
index 41a4845..3eba309 100644
--- a/app_admin/views.py
+++ b/app_admin/views.py
@@ -8,7 +8,8 @@ from django.core.paginator import Paginator,PageNotAnInteger,EmptyPage,InvalidPa
 from app_admin.decorators import superuser_only,open_register
 from django.core.exceptions import ObjectDoesNotExist
 from django.db.models import Q
-import json,datetime,hashlib,random
+import datetime
+import requests
 from app_doc.models import *
 from app_admin.models import *
 from app_admin.utils import *
@@ -760,4 +761,11 @@ def admin_setting(request):
                 name='attachment_size',
                 defaults={'value': attachment_size, 'types': 'doc'}
             )
-            return render(request, 'app_admin/admin_setting.html', locals())
\ No newline at end of file
+            return render(request, 'app_admin/admin_setting.html', locals())
+
+
+# 检测版本更新
+def check_update(request):
+    url = 'https://gitee.com/api/v5/repos/zmister/MrDoc/tags'
+    resp = requests.get(url,timeout=5).json()
+    return JsonResponse({'status':True,'data':resp[-1]})
diff --git a/requirements.txt b/requirements.txt
index 834ac5b..4867ca2 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,4 +4,5 @@ lxml
 pillow==6.2.2
 pyppeteer==0.0.25
 loguru==0.4.1
-djangorestframework==3.9.1
\ No newline at end of file
+djangorestframework==3.9.1
+requests
\ No newline at end of file
diff --git a/template/app_admin/admin_base.html b/template/app_admin/admin_base.html
index 1152ac0..dd6a1f1 100644
--- a/template/app_admin/admin_base.html
+++ b/template/app_admin/admin_base.html
@@ -75,6 +75,7 @@
             {% if debug %}
                 <blockquote class="layui-elem-quote">当前站点处于Debug调试模式</blockquote>
             {% endif %}
+            <blockquote id="check_update_div" class="layui-elem-quote" style="border-left: 5px solid #ff213b !important;display: none;">检测到新版本发布，请及时更新！下载地址：<a href="https://github.com/zmister2016/MrDoc" target="_blank">GitHub</a>、<a href="https://gitee.com/zmister/MrDoc" target="_blank">码云</a></blockquote>
             {% block content %}{% endblock %}
         </div>
       </div>
@@ -109,6 +110,40 @@
           content: $('#dashang_img')
         });
   })
+
+</script>
+<script async>
+  // 检测版本更新
+  checkUpdate = function(){
+    $.get("{% url 'check_update' %}",function(r){
+      console.log(r)
+      if(r.status){
+        var new_version = r.data.name.slice(1,5)
+        var current_version = '{{mrdoc_version}}'
+        // 如果存在新版本，顶部提示div
+        if(new_version > current_version){
+          $("#check_update_div").show()
+        }
+      }      
+    })
+  }
+  
+  var check_update = window.localStorage.getItem('check_update')
+  // console.log(check_update)
+  // 如果没有检测过，立即检测
+  if(check_update == null){
+    checkUpdate();
+    window.localStorage.setItem('check_update',new Date().getTime())
+  }else{
+    // 如果上次检测时间距今超过5天，再次检测
+    var time_diff = (new Date().getTime() - check_update)/86400000
+    // console.log(time_diff)
+    if(time_diff > 5){
+      checkUpdate()
+      window.localStorage.setItem('check_update',new Date().getTime())
+    }
+  }
+  
 </script>
 {% block custom_script %}
 {% endblock %}