From 9842338d2536d1e841349a8d517700f9b137efca Mon Sep 17 00:00:00 2001
From: Wisp X <wisp-x@qq.com>
Date: Mon, 20 Dec 2021 22:05:35 +0800
Subject: [PATCH] =?UTF-8?q?:sparkles:=20=E5=85=B6=E4=BB=96=E6=94=B9?=
 =?UTF-8?q?=E8=BF=9B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 application/index/controller/Auth.php   | 4 ++++
 application/index/view/auth/login.html  | 2 ++
 application/index/view/common/base.html | 2 ++
 3 files changed, 8 insertions(+)

diff --git a/application/index/controller/Auth.php b/application/index/controller/Auth.php
index 05bf5e7a0..cb94b678c 100644
--- a/application/index/controller/Auth.php
+++ b/application/index/controller/Auth.php
@@ -48,6 +48,10 @@ public function register()
             Session::flash('success', '注册成功');
             $this->redirect(url('auth/login'));
         }
+
+        if ($this->getConfig('close_register')) {
+            abort(404, '系统已关闭注册');
+        }
         return $this->fetch();
     }
 
diff --git a/application/index/view/auth/login.html b/application/index/view/auth/login.html
index 346ca1af3..72ef91ba3 100644
--- a/application/index/view/auth/login.html
+++ b/application/index/view/auth/login.html
@@ -31,7 +31,9 @@
             </form>
             <p class="mdui-clearfix mdui-m-t-1">
               <a class="mdui-float-left" href="{:url('auth/forgot')}">忘记密码？</a>
+              {if !$config.close_register}
               <a class="mdui-float-right" href="{:url('auth/register')}">注册账号</a>
+              {/if}
             </p>
           </div>
         </div>
diff --git a/application/index/view/common/base.html b/application/index/view/common/base.html
index 554a05df8..5da47699f 100644
--- a/application/index/view/common/base.html
+++ b/application/index/view/common/base.html
@@ -110,11 +110,13 @@
       <i class="mdui-list-item-icon mdui-icon material-icons iconfont icon-login mdui-text-color-teal"></i>
       <div class="mdui-list-item-content">登录</div>
     </a>
+    {if !$config.close_register}
     <a class="mdui-list-item mdui-ripple {if $uri eq 'auth/register'}mdui-list-item-active{/if}" href="{:url('auth/register')}">
       <i class="mdui-list-item-icon mdui-icon material-icons mdui-text-color-purple">&#xe7f0;</i>
       <div class="mdui-list-item-content">注册</div>
     </a>
     {/if}
+    {/if}
   </div>
 </div>
 {/block}