redirect login page on nextjs

needLogin.js

/* @flow */
import React, { Component } from 'react'
import Router from 'next/router'
import { checkLogin } from '../../redux/modules/auth'

type Props = {
  location: string,
}

export default ({ location }: Props) => (ComposedComponent: any) => class needLogin extends Component<Object> {
  static async getInitialProps ({ res, reduxStore }: { res: any, reduxStore: any }) {
    const { getState, dispatch } = reduxStore

    if (res) {
      await dispatch(checkLogin()).catch((e) => console.error(e))
      if (!getState().app.auth.login) {
        res.writeHead(302, {
          Location: '/login?location=' + location
        })
        res.end()
      }
    } else {
      if (!getState().app.auth.login) {
        Router.push({ pathname: '/login', query: { location } })
      }
    }

    return {}
  }

  render () {
    return <ComposedComponent {...this.props} />
  }
}