session login

src/client/app.tsx

@@ -1,8 +1,8 @@
-import React, { createContext, useRef, useState } from 'react';
+import React, { createContext, useEffect, useRef, useState } from 'react';
 import ReactDom from 'react-dom';
 import {BrowserRouter, Redirect, Route, Switch as RouterSwitch} from 'react-router-dom';
 import { Gallery, ContentAbout, LoginPage, NotFoundPage} from './page/mod';
-import {UserContext} from './state';
+import {getInitialValue, UserContext} from './state';
 
 import './css/style.css';
 
@@ -10,6 +10,14 @@ const FooProfile = ()=><div>test profile</div>;
 const App = () => {
     const [user,setUser] = useState("");
     const [userPermission,setUserPermission] = useState<string[]>([]);
+    (async ()=>{
+        const {username,permission} = await getInitialValue();
+        if(username !== user){
+            setUser(username);
+            setUserPermission(permission);
+        }
+    })();
+    //useEffect(()=>{});
     return (
     <UserContext.Provider value={{
         username:user,

src/client/page/login.tsx

@@ -47,7 +47,7 @@ export const LoginPage = ()=>{
             <div style={{minHeight:theme.spacing(2)}}></div>
         <form style={{display:'flex', flexFlow:'column', alignItems:'stretch'}}>
             <TextField label="username" onChange={(e)=>setUserLoginInfo({...userLoginInfo,username:e.target.value ?? "",})}></TextField>
-            <TextField label="password" type="password" 
+            <TextField label="password" type="password"  onKeyDown={(e)=>{if(e.key === 'Enter') doLogin();}}
                 onChange={(e)=>setUserLoginInfo({...userLoginInfo,password:e.target.value ?? "",})}/>
             <div style={{minHeight:theme.spacing(2)}}></div>
             <div style={{display:'flex'}}>

src/client/state.tsx

@@ -1,8 +1,47 @@
 import React, { createContext, useRef, useState } from 'react';
 
 export const UserContext = createContext({
-    username:"",
+    username: "",
-    permission:["openContent"],
+    permission: [] as string[],
-    setUsername:(s:string)=>{},
+    setUsername: (s: string) => { },
-    setPermission:(permission:string[])=>{}
+    setPermission: (permission: string[]) => { }
 });
+
+type LoginLocalStorage = {
+    username: string,
+    permission: string[],
+    refreshExpired: number
+};
+
+let localObj: LoginLocalStorage|null = null;
+
+export const getInitialValue = async () => {
+    if(localObj === null){
+        const storagestr = window.localStorage.getItem("UserLoginContext") as string | null;
+        const storage = storagestr !== null ? JSON.parse(storagestr) as LoginLocalStorage | null : null;
+        localObj = storage;
+    }
+    if (localObj !== null && localObj.refreshExpired > Math.floor(Date.now() / 1000)) {
+        return {
+            username: localObj.username,
+            permission: localObj.permission,
+        }
+    }
+    const res = await fetch('/user/refresh', {
+        method: 'POST',
+    });
+    if (res.status !== 200) throw new Error("Maybe Network Error")
+    const r = await res.json() as LoginLocalStorage & { refresh: boolean };
+    if (r.refresh) {
+        localObj = {
+            username: r.username,
+            permission: r.permission,
+            refreshExpired: r.refreshExpired
+        }
+        window.localStorage.setItem("UserLoginContext", JSON.stringify(localObj));
+    }
+    return {
+        username: r.username,
+        permission: r.permission
+    }
+}

src/login.ts

@@ -1,106 +1,253 @@
-import {sign, decode, verify} from 'jsonwebtoken';
+import { decode, sign, TokenExpiredError, verify } from "jsonwebtoken";
-import Koa from 'koa';
+import Koa from "koa";
-import Router from 'koa-router';
+import Router from "koa-router";
-import { sendError } from './route/error_handler';
+import { sendError } from "./route/error_handler";
-import Knex from 'knex'
+import Knex from "knex";
-import { createKnexUserController } from './db/mod';
+import { createKnexUserController } from "./db/mod";
-import { request } from 'http';
+import { request } from "http";
-import { get_setting } from './setting';
+import { get_setting } from "./setting";
-import { IUser } from './model/mod';
+import { IUser, UserAccessor } from "./model/mod";
 
 type PayloadInfo = {
-    username:string,
+  username: string;
-    permission:string[]
+  permission: string[];
-}
+};
 
 export type UserState = {
-    user:PayloadInfo
+  user: PayloadInfo;
 };
 
-const isUserState = (obj:object|string):obj is PayloadInfo =>{
+const isUserState = (obj: object | string): obj is PayloadInfo => {
-    if(typeof obj ==="string") return false;
+  if (typeof obj === "string") return false;
-    return 'username' in obj && 'permission' in obj && (obj as {permission:unknown}).permission instanceof Array;
+  return "username" in obj && "permission" in obj &&
-}
+    (obj as { permission: unknown }).permission instanceof Array;
+};
+type RefreshPayloadInfo = { username: string };
+const isRefreshToken = (obj: object | string): obj is RefreshPayloadInfo => {
+  if (typeof obj === "string") return false;
+  return "username" in obj &&
+    typeof (obj as { username: unknown }).username === "string";
+};
 
-export const loginTokenName = 'access_token'
+export const accessTokenName = "access_token";
+export const refreshTokenName = "refresh_token";
+const accessExpiredTime = 60 * 60; //1 hour
+const refreshExpiredTime = 60 * 60 * 24 * 14; // 14 day;
 
-export const getAdminCookieValue = ()=>{
+export const getAdminAccessTokenValue = () => {
+  const {jwt_secretkey} = get_setting();
+  return publishAccessToken(jwt_secretkey,"admin",[],accessExpiredTime);
+};
+export const getAdminRefreshTokenValue = () => {
+    const {jwt_secretkey} = get_setting();
+    return publishRefreshToken(jwt_secretkey,"admin",refreshExpiredTime);
+};
+const publishAccessToken = (
+  secretKey: string,
+  username: string,
+  permission: string[],
+  expiredtime: number,
+) => {
+  const payload = sign(
+    {
+      username: username,
+      permission: permission,
+    },
+    secretKey,
+    { expiresIn: expiredtime },
+  );
+  return payload;
+};
+const publishRefreshToken = (
+  secretKey: string,
+  username: string,
+  expiredtime: number,
+) => {
+  const payload = sign(
+    { username: username },
+    secretKey,
+    { expiresIn: expiredtime },
+  );
+  return payload;
+};
+const setToken = (
+  ctx: Koa.Context,
+  token_name: string,
+  token_payload: string|null,
+  expiredtime: number,
+) => {
+  const setting = get_setting();
+  if(token_payload === null && !!!ctx.cookies.get(token_name)){
+      return;
+  }
+  ctx.cookies.set(token_name, token_payload, {
+    httpOnly: true,
+    secure: !setting.localmode,
+    sameSite: "strict",
+    expires: new Date(Date.now() + expiredtime),
+  });
+};
+export const createLoginMiddleware = (knex: Knex) => {
+  const userController = createKnexUserController(knex);
+  return async (ctx: Koa.Context, next: Koa.Next) => {
     const setting = get_setting();
     const secretKey = setting.jwt_secretkey;
-    return sign({
+    const body = ctx.request.body;
-        username: "admin",
+    //check format
-        permission: [],
+    if (!("username" in body) || !("password" in body)) {
-    },secretKey,{expiresIn:'3d'});
+      return sendError(
-}
+        400,
-
+        "invalid form : username or password is not found in query.",
-export const createLoginMiddleware = (knex: Knex)=>{
+      );
-    const userController = createKnexUserController(knex);
-    return async (ctx: Koa.Context,next: Koa.Next)=>{
-        const setting = get_setting();
-        const secretKey = setting.jwt_secretkey;
-        const body = ctx.request.body;
-        if(!('username' in body)||!('password' in body)){
-            return sendError(400,"invalid form : username or password is not found in query.");
-        }
-        const username = body['username'];
-        const password = body['password'];
-        if(typeof username !== "string" || typeof password !== "string"){
-            return sendError(400,"invalid form : username or password is not string")
-        }
-        if(setting.forbid_remote_admin_login && username === "admin"){
-            return sendError(403,"forbid remote admin login");
-        }
-        const user = await userController.findUser(username);
-        if(user === undefined){
-            return sendError(401,"not authorized");
-        }
-        if(!user.password.check_password(password)){
-            return sendError(401,"not authorized");
-        }
-        const userPermission = await user.get_permissions();
-        const payload = sign({
-            username: user.username,
-            permission: userPermission
-        },secretKey,{expiresIn:'3h'});
-        ctx.cookies.set(loginTokenName,payload,{httpOnly:true, secure: !setting.localmode,sameSite:'strict'});
-        ctx.body = {ok:true, username: user.username, permission: userPermission}
-        console.log(`${username} logined`);
-        return;
-    };
-};
-export const LogoutMiddleware = (ctx:Koa.Context,next:Koa.Next)=>{
-    ctx.cookies.set(loginTokenName,undefined);
-    ctx.body = {ok:true};
-    return;
-}
-export const UserMiddleWare = async (ctx:Koa.ParameterizedContext<UserState>,next:Koa.Next)=>{
-    const secretKey = get_setting().jwt_secretkey;
-    const payload = ctx.cookies.get(loginTokenName);
-    const setting = get_setting();
-    if(payload == undefined){
-        ctx.state['user'] = {username:"",
-        permission:setting.guest};
-        return await next();
     }
-    const o = verify(payload,secretKey);
+    const username = body["username"];
-    if(isUserState(o)){
+    const password = body["password"];
+    //check type
+    if (typeof username !== "string" || typeof password !== "string") {
+      return sendError(
+        400,
+        "invalid form : username or password is not string",
+      );
+    }
+    //if admin login is forbidden?
+    if (username === "admin" && setting.forbid_remote_admin_login) {
+      return sendError(403, "forbiden remote admin login");
+    }
+    const user = await userController.findUser(username);
+    //username not exist
+    if (user === undefined) return sendError(401, "not authorized");
+    //password not matched
+    if (!user.password.check_password(password)) {
+      return sendError(401, "not authorized");
+    }
+    //create token
+    const userPermission = await user.get_permissions();
+    const payload = publishAccessToken(
+      secretKey,
+      user.username,
+      userPermission,
+      accessExpiredTime,
+    );
+    const payload2 = publishRefreshToken(
+      secretKey,
+      user.username,
+      refreshExpiredTime,
+    );
+    setToken(ctx, accessTokenName, payload, accessExpiredTime);
+    setToken(ctx, refreshTokenName, payload2, refreshExpiredTime);
+    ctx.body = {
+      username: user.username,
+      permission: userPermission,
+      refreshExpired: (Math.floor(Date.now() / 1000) + refreshExpiredTime),
+    };
+    console.log(`${username} logined`);
+    return;
+  };
+};
+export const LogoutMiddleware = (ctx: Koa.Context, next: Koa.Next) => {
+  ctx.cookies.set(accessTokenName, null);
+  ctx.cookies.set(refreshTokenName,null);
+  ctx.body = { ok: true };
+  return;
+};
+export const createUserMiddleWare = (knex: Knex) =>
+async (ctx: Koa.ParameterizedContext<UserState>, next: Koa.Next) => {
+    const userController = createKnexUserController(knex);
+    const refreshToken = refreshTokenHandler(userController);
+    const setting = get_setting();
+    const setGuest = async () => {
+      setToken(ctx, accessTokenName, null, 0);
+      setToken(ctx, refreshTokenName, null, 0);
+      ctx.state["user"] = { username: "", permission: setting.guest };
+      return await next();
+    };
+    return await refreshToken(ctx,setGuest,next);
+  };
+const refreshTokenHandler = (cntr:UserAccessor) => async (ctx:Koa.Context,fail: Koa.Next,next: Koa.Next)=>{
+    const payload = ctx.cookies.get(accessTokenName);
+    const setting = get_setting();
+    const secretKey = setting.jwt_secretkey;
+    const checkRefreshAndUpdate = async () => {
+      const payload2 = ctx.cookies.get(refreshTokenName);
+      if (payload2 === undefined) return await fail(); // refresh token doesn't exist
+      else {
+        try {
+          const o = verify(payload2, secretKey);
+          if (isRefreshToken(o)) {
+            const user = await cntr.findUser(o.username);
+            if (user === undefined) return await fail(); //already non-existence user
+            const perm = await user.get_permissions();
+            const payload = publishAccessToken(
+              secretKey,
+              user.username,
+              perm,
+              accessExpiredTime,
+            );
+            setToken(ctx, accessTokenName, payload, accessExpiredTime);
+            ctx.state.user = { username: o.username, permission: perm };
+          } else {
+            console.error("invalid token detected");
+            throw new Error("token form invalid");
+            return;
+          }
+        } catch (e) {
+          if (e instanceof TokenExpiredError) { // refresh token is expired.
+            return await fail();
+          } else throw e;
+        }
+      }
+      return await next();
+    };
+    if (payload == undefined) {
+      return await checkRefreshAndUpdate();
+    }
+    try {
+      const o = verify(payload, secretKey);
+      if (isUserState(o)) {
         ctx.state.user = o;
         return await next();
-    }
+      } else {
-    else{
         console.error("invalid token detected");
+        throw new Error("token form invalid");
+      }
+    } catch (e) {
+      if (e instanceof TokenExpiredError) {
+        return await checkRefreshAndUpdate();
+      } else throw e;
     }
 }
-
+export const createRefreshTokenMiddleware = (knex:Knex)=> async (ctx:Koa.Context,next:Koa.Next)=>{
-export const getAdmin = async(knex : Knex)=>{
+    const cntr= createKnexUserController(knex);
-    const cntr = createKnexUserController(knex);
+    const handler = refreshTokenHandler(cntr);
-    const admin = await cntr.findUser("admin");
+    const fail = async ()=>{
-    if(admin === undefined){
+        const user = ctx.state.user as PayloadInfo;
-        throw new Error("initial process failed!");//???
+        ctx.body = {
+            refresh:false,
+            ...user
+        }
+        ctx.type = 'json';
+    };
+    const success = async ()=>{
+        const user = ctx.state.user as PayloadInfo;
+        ctx.body = {
+            ...user,
+            refresh:true,
+            refreshExpired: Math.floor(Date.now()/1000 + accessExpiredTime),
+        }
+        ctx.type = 'json';
     }
-    return admin;
+    await handler(ctx,fail,success);
 }
+export const getAdmin = async (knex: Knex) => {
+  const cntr = createKnexUserController(knex);
+  const admin = await cntr.findUser("admin");
+  if (admin === undefined) {
+    throw new Error("initial process failed!"); //???
+  }
+  return admin;
+};
 
-export const isAdminFirst = (admin: IUser)=>{
+export const isAdminFirst = (admin: IUser) => {
-    return admin.password.hash === "unchecked" && admin.password.salt === "unchecked";
+  return admin.password.hash === "unchecked" &&
-}
+    admin.password.salt === "unchecked";
+};

src/server.ts

@@ -11,7 +11,7 @@ import { createKnexContentsAccessor } from './db/contents';
 import bodyparser from 'koa-bodyparser';
 import {error_handler} from './route/error_handler';
 
-import {UserMiddleWare,createLoginMiddleware, isAdminFirst, getAdmin, LogoutMiddleware} from './login';
+import {createUserMiddleWare,createLoginMiddleware, isAdminFirst, getAdmin, LogoutMiddleware, createRefreshTokenMiddleware} from './login';
 
 import {createInterface as createReadlineInterface} from 'readline';
 
@@ -35,7 +35,7 @@ export async function create_server(){
     let app = new Koa();
     app.use(bodyparser());
     app.use(error_handler);
-    app.use(UserMiddleWare);
+    app.use(createUserMiddleWare(db));
     //app.use(ctx=>{ctx.state['setting'] = settings});
 
     const index_html = readFileSync("index.html");
@@ -67,12 +67,13 @@ export async function create_server(){
 
     router.post('/user/login',createLoginMiddleware(db));
     router.post('/user/logout',LogoutMiddleware);
+    router.post('/user/refresh',createRefreshTokenMiddleware(db));
     
     if(setting.mode == "development"){
     let mm_count = 0;
     app.use(async (ctx,next)=>{
         console.log(`==========================${mm_count++}`);
-        const fromClient = ctx.state['user'] === undefined ? ctx.ip : ctx.state['user'].username;
+        const fromClient = ctx.state['user'].username === "" ? ctx.ip : ctx.state['user'].username;
         console.log(`${fromClient} : ${ctx.method} ${ctx.url}`);
         await next();
         //console.log(`404`);