dww-rog-test/src/index.js

dww-rog-test

import React, { Component, createContext } from 'react'
import OneGraphAuth from 'onegraph-auth'

const AuthContext = createContext()

class AuthProvider extends Component {
  state = {
    auth: null,
    status: {},
    headers: {},
  }

  componentDidMount() {
    const auth = new OneGraphAuth({
      appId: this.props.appId,
    })

    window.auth = auth

    auth.servicesStatus().then(status =>
      this.setState({
        headers: auth.authHeaders(),
        status: Object.keys(status).reduce((out, service) => {
          out[service] = status[service].isLoggedIn
          return out
        }, {}),
        auth,
      })
    )
  }

  login = (service, callback) => {
    const { auth, status, headers } = this.state

    if (auth) {
      auth.login(service).then(() => {
        auth.isLoggedIn(service).then(isLoggedIn => {
          callback && callback(isLoggedIn)

          this.setState({
            status: {
              ...status,
              [service]: isLoggedIn,
            },
            headers: auth.authHeaders(),
          })
        })
      })
    }
  }

  logout = (service, callback) => {
    const { auth, status, headers } = this.state

    auth.logout(service).then(() => {
      auth.isLoggedIn(service).then(isLoggedIn => {
        callback && callback(isLoggedIn)

        this.setState({
          status: {
            ...status,
            [service]: isLoggedIn,
          },
          headers: auth.authHeaders(),
        })
      })
    })
  }

  render() {
    const { appId, children } = this.props
    const { auth, status, headers } = this.state

    const authInterface = {
      status,
      headers,
      login: this.login,
      logout: this.logout,
      appId,
    }

    return (
      <AuthContext.Provider value={authInterface}>
        {this.props.children}
      </AuthContext.Provider>
    )
  }
}

const AuthConsumer = AuthContext.Consumer

export { AuthContext, AuthConsumer, AuthProvider }