[service] Verify

app.py

@@ -21,21 +21,21 @@
 services_path='services'
 if not os.path.exists(services_path):
  logger.critical(f"{os.path.join(os.getcwd(),services_path)} 不存在")
- raise
+ raise OSError(f"{os.path.join(os.getcwd(),services_path)} 不存在")
 
 services = os.listdir(services_path)
 for service in services:
  try:
  entrypoint = importlib.import_module(f"{services_path}.{service}").entrypoint
- matedata = entrypoint(settings)
+ metadata = entrypoint(settings)
  logger.warn("#"*50+f"""
 加载{service}
-\t作者{matedata['Author']}
-\t版本{matedata['Version']}
-\t描述{matedata['Describe']}
+\t作者{metadata['Author']}
+\t版本{metadata['Version']}
+\t描述{metadata['Describe']}
 """+"#"*50)
- application.include_router(matedata['Router'])
- matedata['Init']()
+ application.include_router(metadata['Router'])
+ metadata['Init']()
  except Exception as e:
  logger.warn(f"加载{service}出错")
  logger.warn(e)

services/verify/__init__.py

@@ -0,0 +1,31 @@
+"""
+
+"""
+
+from fastapi import APIRouter
+from dynaconf import Dynaconf
+
+from . import router
+
+router.init(APIRouter(prefix="/verify"), APIRouter(prefix="/verify"))
+
+
+def entrypoint(settings: Dynaconf):
+ return {
+ "Author": "Bernie H.",
+ "Version": "1.0.0",
+ "Describe": "验证微服务，主要用于验证用户的邮箱等。",
+ "InterRouter": router.inter_router,
+ "ExposeRouter": router.public_router,
+ "Init": lambda: init(settings),
+ }
+
+
+g_settings = None
+
+
+def init(settings: Dynaconf):
+ print("Verify service is starting...")
+ global g_settings
+ g_settings = settings
+

services/verify/db.py

@@ -0,0 +1,24 @@
+from pony.orm import Database, Required, PrimaryKey
+
+""" Create Database """
+db = Database()
+
+
+class OutsideVerifyCode(db.Entity):
+ code = PrimaryKey(str)
+ session = Required(str)
+ expired = Required(int)
+ callbackURI = Required(str)
+
+
+class InsideVerifyCode(db.Entity):
+ code = PrimaryKey(str)
+ session = Required(str)
+ expired = Required(int)
+
+
+db.bind(provider="sqlite", filename="verify.db")
+
+db.generate_mapping(create_tables=True)
+
+from pony.orm import db_session

services/verify/router.py

@@ -0,0 +1,160 @@
+"""
+[service] Verify Service
+
+* this: 本服务
+* s1: 调用本服务的服务
+* client: 用户的客户端
+
+1. [s1] 请求内网api "/newVerify" 来进行一个新的验证 (需提供一个回调地址、一个 session ID)
+2. [client] 通过路由请求内部api "/verifyOutsideCode" 进行验证
+3. [this] 将 [client] 重定向到 "<CALLBACK> ?s= <SUCCESS> &v= <INSIDE-CODE>" (`SUCCESS` 为 `0/1`, 代表验证失败/成功, `INSIDE-CODE` 为内部验证码)
+4. [s1] 请求内网api "/verifyInsideCode" 验证 `INSIDE-CODE`
+5. [s1] 根据返回字段 `ok` 判断验证是否成功, `sess` 来恢复会话
+
+"""
+
+import fastapi
+from fastapi import Request, Response
+from starlette.responses import RedirectResponse
+
+import verify
+import send
+
+
+inter_router = None
+public_router = None
+
+
+def init(ri: fastapi.APIRouter, rp: fastapi.APIRouter):
+ global inter_router
+ global public_router
+ inter_router = ri
+ public_router = rp
+
+ inter_router.add_route("/newVerifyCode", newVerifyCode, methods=["GET"])
+ inter_router.add_route("/verifyOutsideCode", verifyOutsideCode, methods=["GET"])
+ inter_router.add_route("/verifyInsideCode", verifyInsideCode, methods=["GET"])
+
+ public_router.add_route("/verify", verifyOutsideCode, methods=["GET"])
+
+
+def _getVerifyUri(code: str):
+ """
+ [Tool] 从验证码生成验证链接
+ """
+ return f"https://openteens.org/verify/verify?code={code}"
+
+def newVerify():
+ """
+ [内部] 进行一个新的验证
+
+ Request: 
+ method: str (one of [email])
+ target: str (e.g. "[email protected]")
+ session: str
+ callbackURI: str (建议使用带https的绝对路径)
+
+ Response:
+ ..codeblock:: json
+ {
+ code: int
+ }
+ """
+ request = Request()
+
+ method = request.args.get("method")
+ target = request.args.get("target")
+ session = request.args.get("session")
+ callbackURI = request.args.get("callbackURI")
+
+ code = verify.genOutsideCode(session, callbackURI)
+
+ if method == "email":
+ send.sendEmail(target, _getVerifyUri(code))
+ else:
+ return {"code": -1}
+
+ return {"code": 0}
+
+
+def newVerifyCode():
+ """
+ [内部] 进行一个新的验证（返回验证链接）
+
+ Request: 
+ session: str
+ callbackURI: str (建议使用带https的绝对路径)
+
+ Response:
+ ..codeblock:: json
+ {
+ code: int,
+ verifyUri: str
+ }
+ """
+ request = Request()
+
+ session = request.args.get("session")
+ callbackURI = request.args.get("callbackURI")
+
+ code = verify.genOutsideCode(session, callbackURI)
+
+ return {"code": 0, "verifyUri": _getVerifyUri(code)}
+
+
+def verifyOutsideCode():
+ """
+ [外部] 验证外部验证码
+
+ Request:
+ code: str (外部验证码)
+
+ Response:
+ <Redirect> (重定向到回调地址)
+ """
+ request = Request()
+
+ code = request.args.get("code")
+
+ result = verify.verifyOutsideCode(code)
+ if result:
+ sess = result["session"]
+ callbackURI = result["callbackURI"]
+
+ insideCode = verify.genInsideCode(sess)
+ redirectURI = f"{callbackURI}?s=1&v={insideCode}"
+ else:
+ redirectURI = "ERROR: Invalid Verification Link"
+
+ return RedirectResponse(redirectURI)
+
+
+def verifyInsideCode():
+ """
+ [内部] 验证内部验证码
+
+ Request:
+ code: str (内部验证码)
+
+ Response:
+ ..codeblock:: json
+ {
+ code: int,
+ ok: bool,
+ sess: str
+ }
+ """
+ request = Request()
+
+ code = request.args.get("code")
+
+ result = verify.verifyInsideCode(code)
+ if result:
+ return {"code": 0, "ok": True, "sess": result["session"]}
+ else:
+ return {"code": -1}
+
+
+if __name__ == "__main__":
+ init(fastapi.APIRouter())
+ # router.run("localhost", 5003)