import axios, { AxiosRequestHeaders } from 'axios';
import Config                         from '../config/Config';
import SessionManager                 from './SessionManager';
import FormData                       from 'form-data';
import logger                         from '../shared/logging/WinstonLogger';
import GitlabManager                  from './GitlabManager';
import { StatusCodes }                from 'http-status-codes';


class HttpManager {
    public handleCommandErrors: boolean = true;

    registerAxiosInterceptor() {
        this.registerRequestInterceptor();
        this.registerResponseInterceptor();
    }

    private registerRequestInterceptor() {
        axios.interceptors.request.use((config) => {
            if ( config.data instanceof FormData ) {
                config.headers = { ...config.headers, ...(config.data as FormData).getHeaders() } as AxiosRequestHeaders;
            }

            if ( config.url && (config.url.indexOf(Config.apiURL) !== -1) ) {
                if ( config.data && Object.keys(config.data).length > 0 ) {
                    config.headers['Content-Type'] = 'multipart/form-data';
                }

                if ( SessionManager.isLogged ) {
                    config.headers.Authorization = 'Bearer ' + SessionManager.token;
                }
            }

            if ( GitlabManager.isLogged && config.url && config.url.indexOf(Config.gitlab.apiURL) !== -1 ) {
                config.headers['PRIVATE-TOKEN'] = GitlabManager.token;
            }

            return config;
        });
    }

    private registerResponseInterceptor() {
        axios.interceptors.response.use((response) => {
            if ( response.data && response.data.sessionToken ) {
                SessionManager.token = response.data.sessionToken;
            }

            return response;
        }, (error) => {
            if ( error.response ) {
                if ( this.handleCommandErrors ) {
                    if ( error.response.url && error.response.url.indexOf(Config.apiURL) !== -1 ) {
                        switch ( error.response.status ) {
                            case StatusCodes.UNAUTHORIZED:   // Unauthorized
                                logger.error('Session expired or inexistent. Please login again.');
                                process.exit(1);
                                break;
                            case StatusCodes.FORBIDDEN:   // Forbidden
                                logger.error('Forbidden access.');
                                process.exit(1);
                                break;
                        }
                    }
                } else {
                    this.handleCommandErrors = true;
                }
            } else {
                logger.error('Error connecting to the server.');
                process.exit(1);
            }


            return Promise.reject(error);
        });
    }
}


export default new HttpManager();