path: root/client/src/app/shared/auth/auth-http.service.ts

import { Injectable } from '@angular/core'
import {
  ConnectionBackend,
  Headers,
  Http,
  Request,
  RequestMethod,
  RequestOptions,
  RequestOptionsArgs,
  Response,
  XHRBackend
} from '@angular/http'
import { Observable } from 'rxjs/Observable'

import { AuthService } from '../../core'

@Injectable()
export class AuthHttp extends Http {
  constructor (backend: ConnectionBackend, defaultOptions: RequestOptions, private authService: AuthService) {
    super(backend, defaultOptions)
  }

  request (url: string | Request, options?: RequestOptionsArgs): Observable<Response> {
    if (!options) options = {}

    options.headers = new Headers()
    this.setAuthorizationHeader(options.headers)

    return super.request(url, options)
                .catch((err) => {
                  if (err.status === 401) {
                    return this.handleTokenExpired(url, options)
                  }

                  return Observable.throw(err)
                })
  }

  delete (url: string, options?: RequestOptionsArgs): Observable<Response> {
    if (!options) options = {}
    options.method = RequestMethod.Delete

    return this.request(url, options)
  }

  get (url: string, options?: RequestOptionsArgs): Observable<Response> {
    if (!options) options = {}
    options.method = RequestMethod.Get

    return this.request(url, options)
  }

  post (url: string, body: any, options?: RequestOptionsArgs): Observable<Response> {
    if (!options) options = {}
    options.method = RequestMethod.Post
    options.body = body

    return this.request(url, options)
  }

  put (url: string, body: any, options?: RequestOptionsArgs): Observable<Response> {
    if (!options) options = {}
    options.method = RequestMethod.Put
    options.body = body

    return this.request(url, options)
  }

  private handleTokenExpired (url: string | Request, options: RequestOptionsArgs) {
    return this.authService.refreshAccessToken()
                           .flatMap(() => {
                             this.setAuthorizationHeader(options.headers)

                             return super.request(url, options)
                           })
  }

  private setAuthorizationHeader (headers: Headers) {
    headers.set('Authorization', this.authService.getRequestHeaderValue())
  }
}

export function useFactory (backend: XHRBackend, defaultOptions: RequestOptions, authService: AuthService) {
  return new AuthHttp(backend, defaultOptions, authService)
}

export const AUTH_HTTP_PROVIDERS = [
  {
    provide: AuthHttp,
    useFactory,
    deps: [ XHRBackend, RequestOptions, AuthService ]
  }
]