import cors from '@koa/cors'
import { Context, Middleware } from 'koa'
export default function (): Middleware {
  const conf = {
    origin: (ctx: Context) => {
      const requestOrigin = ctx.get('Origin')
      if (/ksyun\.com/.test(requestOrigin)) {
        return requestOrigin
      } else {
        return 'ksyun.com'
      }
    },
    methods: ['GET', 'PUT', 'POST'],
    credentials: true
  }
  return cors(conf)
}