From a1d801ba14ea334cdef5f156aceb007bc588dfd4 Mon Sep 17 00:00:00 2001
From: yanyongyu <yanyongyu_1@126.com>
Date: Tue, 12 Jan 2021 18:02:05 +0800
Subject: [PATCH] :alembic: new fastapi config settings

---
 nonebot/drivers/fastapi.py | 22 +++++++++++++++++-----
 1 file changed, 17 insertions(+), 5 deletions(-)

diff --git a/nonebot/drivers/fastapi.py b/nonebot/drivers/fastapi.py
index 8a19a5c7..24a430cc 100644
--- a/nonebot/drivers/fastapi.py
+++ b/nonebot/drivers/fastapi.py
@@ -14,18 +14,28 @@ import logging
 from typing import Optional, Callable
 
 import uvicorn
+from pydantic import BaseSettings
 from fastapi.responses import Response
 from fastapi import Body, status, Request, FastAPI, HTTPException
 from starlette.websockets import WebSocketDisconnect, WebSocket as FastAPIWebSocket
 
 from nonebot.log import logger
 from nonebot.typing import overrides
-from nonebot.config import Env, Config
 from nonebot.utils import DataclassEncoder
 from nonebot.exception import RequestDenied
+from nonebot.config import Env, Config as NoneBotConfig
 from nonebot.drivers import Driver as BaseDriver, WebSocket as BaseWebSocket
 
 
+class Config(BaseSettings):
+    fastapi_openapi_url: Optional[str] = None
+    fastapi_docs_url: Optional[str] = None
+    fastapi_redoc_url: Optional[str] = None
+
+    class Config:
+        extra = "ignore"
+
+
 class Driver(BaseDriver):
     """
     FastAPI 驱动框架
@@ -38,14 +48,16 @@ class Driver(BaseDriver):
       * ``/{adapter name}/ws/``: WebSocket 上报
     """
 
-    def __init__(self, env: Env, config: Config):
+    def __init__(self, env: Env, config: NoneBotConfig):
         super().__init__(env, config)
 
+        self.fastapi_config = Config(**config.dict())
+
         self._server_app = FastAPI(
             debug=config.debug,
-            openapi_url=None,
-            docs_url=None,
-            redoc_url=None,
+            openapi_url=self.fastapi_config.fastapi_openapi_url,
+            docs_url=self.fastapi_config.fastapi_docs_url,
+            redoc_url=self.fastapi_config.fastapi_redoc_url,
         )
 
         self._server_app.post("/{adapter}/")(self._handle_http)