register.controller.js

import angular from 'angular'
import _ from 'lodash'
import { BUSY_PROGRESS_MESSAGE, DOMAIN, WIPRO_SSO_PROVIDER, V3_JWT, V2_JWT, V2_SSO, AUTH0_REFRESH, AUTH0_JWT, ZENDESK_JWT } from '../../../core/constants.js'
import { registerUser, socialRegistration, identifySSOProvider } from '../../../core/auth.js'
import { npad } from '../../../core/utils.js'
import { generateReturnUrl, redirectTo } from '../../../core/url.js'
import { getToken, decodeToken, setToken } from '../../../core/token.js'
import { getNewJWT } from '../../../core/auth.js'

(function() {
  'use strict'

  const SKILL_PICKER_URL = 'https://www.' + DOMAIN + '/settings/profile'

  angular.module('accounts').controller('TCRegistrationController', TCRegistrationController)

  TCRegistrationController.$inject = ['$log', '$scope', '$state', '$stateParams', 'UserService', 'ISO3166', '$timeout']

  function TCRegistrationController($log, $scope, $state, $stateParams, UserService, ISO3166, $timeout) {
    var vm = this
    vm.registering = false
    // auth0 login data, passed from another states as state param
    vm.auth0Data = $stateParams.auth0Data
    // SSO user data extracted from auth0 login data
    vm.ssoUser = vm.auth0Data && vm.auth0Data.ssoUserData ? vm.auth0Data.ssoUserData : null
    // regForm is used to pre-populate form items
    vm.regForm = $stateParams.regForm
    if(vm.regForm) {
      vm.username = vm.regForm.handle
      vm.firstname = vm.regForm.firstName
      vm.lastname = vm.regForm.lastName
      vm.countryObj = ISO3166.getCountryObjFromCountryCode(vm.regForm.country)
    }
    
    // prepares utm params, if available
    var utm = {
      source : $stateParams && $stateParams.utm_source ? $stateParams.utm_source : '',
      medium : $stateParams && $stateParams.utm_medium ? $stateParams.utm_medium : '',
      campaign : $stateParams && $stateParams.utm_campaign ? $stateParams.utm_campaign : ''
    }

    // Set default for toggle password directive
    vm.defaultPlaceholder = 'Create Password'
    vm.busyMessage = BUSY_PROGRESS_MESSAGE
    vm.retUrl = $stateParams && $stateParams.retUrl ? $stateParams.retUrl : SKILL_PICKER_URL
    vm.countries = ISO3166.getAllCountryObjects()

    vm.$stateParams = $stateParams

    // watch form to detect particular changes in it.
    // https://stackoverflow.com/questions/22436501/simple-angularjs-form-is-undefined-in-scope
    $scope.$watch('registerForm', function(registerForm) {
      if (vm.ssoUser) {
        loadSSOUser(vm.ssoUser)
      }
    })
    $scope.$watch('vm.email', function(email) {
      vm.ssoForced = !!(identifySSOProvider(email))     
    })

    $scope.usernameFocusLoss = function () {
      $timeout(function () {
        vm.usernameTips = false
      }, 100);
    }

    $scope.emailFocusLoss = function () {
      $timeout(function () {
        vm.emailTips = false
      }, 100);
    }

    vm.updateCountry = function (angucompleteCountryObj) {
      var countryCode = _.get(angucompleteCountryObj, 'originalObject.code', undefined)

      var isValidCountry = _.isUndefined(countryCode) ? false : true
      vm.registerForm.country.$setValidity('required', isValidCountry)
      vm.isValidCountry = isValidCountry
      if (isValidCountry) {
        vm.country = angucompleteCountryObj.originalObject
      }
    }

    function setV3Tokens({token, zendeskJwt}) {
      $log.debug('Received v3 tokens')
      setToken(V3_JWT, token || '')
      setToken(ZENDESK_JWT, zendeskJwt || '')
      $log.debug('Redirecting to ' + vm.retUrl)
      var error = redirectTo(generateReturnUrl(vm.retUrl))
      $scope.$apply(function() {
        vm.registering = false
        if (error) {
          vm.error = 'Invalid URL is assigned to the return-URL.'
        }
      })
    }

    function startSSO() {
      $state.go ('SSO_LOGIN', 
        {
          app: 'member',
          email   : vm.email,
          regForm : {
            handle  : vm.username,
            firstName: vm.firstname,
            lastName : vm.lastname,
            country  : vm.country.code
          },
          retUrl  : vm.retUrl
        }
      )
    }

    vm.register = function() {
      if (!vm.ssoUser) {
        const provider = identifySSOProvider(vm.email)
        if (provider) {
          startSSO()
          return
        }
      }
      
      vm.registering = true
      var userInfo = {
        handle: vm.username,
        firstName: vm.firstname,
        lastName: vm.lastname,
        email: vm.email,
        country: {
          code: npad(vm.country.code, 3),
          isoAlpha3Code: vm.country.alpha3,
          isoAlpha2Code: vm.country.alpha2
        },
        utmSource: utm.source,
        utmMedium: utm.medium,
        utmCampaign: utm.campaign
      }

      if (!vm.isSocialRegistration && !vm.ssoUser) {// if not social or sso registration
        userInfo.credential = { password: vm.password }
      } else if (vm.ssoUser) {//SSO user
        userInfo.active = true, // activate in registration
        userInfo.profile = {
          name: vm.ssoUser.name,
          email: vm.ssoUser.email,
          providerType: 'samlp',
          provider: vm.ssoUser.ssoProvider,
          userId: vm.ssoUser.ssoUserId
        }
      } else {
        userInfo.profile = {
          userId: vm.socialUserId,
          name: vm.firstname + ' ' + vm.lastname,
          email: vm.socialProfile.email,
          emailVerified: vm.socialProfile.email_verified,
          providerType: vm.socialProvider,
          context: {
            handle: vm.username,
            accessToken: vm.socialContext.accessToken,
            auth0UserId: vm.socialProfile.user_id
          }
        }
      }
      var redirectURL = vm.retUrl ? vm.retUrl : SKILL_PICKER_URL;
      var body = {
        param: userInfo,
        options: {
          afterActivationURL: redirectURL
        }
      }

      registerUser(body)
      .then(function(data) {
        vm.registering = false
        $log.debug('Registered successfully')

        if (vm.ssoUser && vm.auth0Data) {
          // LOG IN user
          setToken(AUTH0_JWT, vm.auth0Data.idToken)
          setToken(AUTH0_REFRESH, vm.auth0Data.refreshToken)
          $log.debug('Getting v3jwt')
          getNewJWT()
            .then(setV3Tokens)
            .catch(function(err) {
              vm.registering = false
              vm.errMsg = err && err.message ? err.message : 'Error in logging in new user'
              $scope.$apply()
              $log.error('Error in logging in new user', err)
            })
        } else {
          // In the future, go to dashboard
          $state.go('MEMBER_REGISTRATION_SUCCESS', {
            ssoUser : !!vm.ssoUser,
            retUrl  : redirectURL
          })
        }
      })
      .catch(function(err) {
        vm.registering = false
        vm.errMsg = err && err.message ? err.message : 'Error in registering new user'
        $scope.$apply()
        $log.error('Error in registering new user', err)
      })
    }

    vm.socialRegister = function(provider) {
      vm.errMsg = null
      socialRegistration(provider, null)
      .then(function(resp) {
        if (resp.status === 'SUCCESS') {
          var socialData = resp.data
          vm.socialUserId = socialData.socialUserId
          vm.username = socialData.username
          if (socialData.username) {
            vm.registerForm.username.$setDirty()
          }
          vm.firstname = socialData.firstname
          if (socialData.firstname) {
            vm.registerForm.firstname.$setDirty()
          }
          vm.lastname = socialData.lastname
          if (socialData.lastname) {
            vm.registerForm.lastname.$setDirty()
          }
          if (socialData.email) {
            vm.registerForm.email.$setDirty()
          }
          vm.email = socialData.email
          vm.socialProfile = socialData.socialProfile
          vm.socialProvider = socialData.socialProvider
          vm.socialContext= {'accessToken':  socialData.accessToken}
          vm.isSocialRegistration = true
        } else {
          vm.isSocialRegistration = false
        }
        $scope.$apply()
      })
      .catch(function(err) {
        switch (err.status) {
        case 'SOCIAL_PROFILE_ALREADY_EXISTS':
          vm.errMsg = 'An account with that profile already exists. Please login to access your account.'

          $log.error('Error registering user with social account', err)

          break

        default:
          vm.errMsg = 'Whoops! Something went wrong. Please try again later.'

          $log.error('Error registering user with social account', err)
        }
        vm.isSocialRegistration = false
        $scope.$apply()
      })
    }

    function loadSSOUser(ssoUser) {
      vm.ssoUser = ssoUser
      
      if (ssoUser && ssoUser.firstName) {
        vm.firstname = ssoUser.firstName
        vm.registerForm.firstname.$setDirty()
      }
      if (ssoUser && ssoUser.lastName) {
        vm.lastname = ssoUser.lastName
        vm.registerForm.lastname.$setDirty()
      }
      if (ssoUser && ssoUser.email) {
        vm.email = ssoUser.email
        vm.registerForm.email.$setDirty()
      }
    }

    vm.showRegistrationPage = function(){
      $state.go('MEMBER_REGISTRATION', $stateParams)
    }

  }
})()