auth.ts

#!/usr/bin/env node
import { Argument, Option, program } from 'commander'
const pkg = require('../../package.json')

import { Xal, Msal, TokenRefreshError } from '../lib'
import TokenStore from '../tokenstore'

class Auth {
    _options = {
        file: {
            default: '.xbox.tokens.json',
            description: 'Load a different token file',
            name: '-F, --file <path>',
        },
        auth: {
            default: 'xal',
            description: 'Choose authentication method. (choices: xal, msal)',
            name: '-a, --auth <xal|msal>',
        },

        // output: {
        //     default: 'text',
        //     description: 'Sets the output format',
        //     name: '-O, --output <path>',
        //     choices: ['text', 'json'],
        // },
    }

    _commander:typeof program
    _tokenStore:TokenStore
    _xal:Xal
    _msal:Msal

    _state
    _deviceToken

    constructor(){
        this._commander = program
            .version(pkg.version)
            .addArgument(new Argument('command', 'Command to run').choices(['auth', 'show', 'refresh', 'tokens', 'logout']).default('auth'))

        for(const arg in this._options){
            const argData = this._options[arg]
            const option = new Option(argData.name, argData.description)

            if(argData.default)
                option.default(argData.default)

            if(argData.choices)
                option.choices(argData.choices)

            this._commander.addOption(option)
        }

        program.addHelpText('after', `

Example commands:
  $ xbox-xal-auth auth
  $ xbox-xal-auth show
  $ xbox-xal-auth refresh
  $ xbox-xal-auth logout`);

        this._commander.parse();

        // Load tokenstore
        this._tokenStore = new TokenStore()
        this._tokenStore.load(this._commander.opts().file, true)

        // Load XAL
        this._xal = new Xal(this._tokenStore)
        this._msal = new Msal(this._tokenStore)
    }

    run(){
        if(this._commander.args[0] == 'auth'){

            if(this._tokenStore.hasValidAuthTokens()){
                console.log('You are already authenticated. To show the tokens, run `xbox-xal-auth show`. To re-authenticate, run `xbox-xal-auth logout` first.')
                return
            }

            if(this._commander.opts().auth === 'xal'){
                console.log('Starting authentiction using XAL method...')

                this._xal.getRedirectUri().then((redirect) => {
                    if(redirect){
                        console.log('Please authenticate using the following url:', redirect.sisuAuth.MsaOauthRedirect)

                        const readline = require('readline').createInterface({
                            input: process.stdin,
                            output: process.stdout
                        });
            
                        readline.question('Enter redirect uri: ', async redirectUri => {
                            readline.close()
                            // await this.loadTokensUsingCode(redirectUri, result.SessionId)
                            const loggedIn = await this._xal.authenticateUser(redirect, redirectUri)
                            if(loggedIn === true){
                                this._xal.refreshTokens()
                                console.log('Authentication succeeded!')
                            } else {
                                console.log('Authentication failed!')
                            }
                        })
                    }
                }).catch((error) => {
                    console.log('error', error)
                })

            } else if(this._commander.opts().auth === 'msal'){
                console.log('Starting authentiction using MSAL method...')

                this._msal.doDeviceCodeAuth().then((deviceCodeDetails:any) => {
                    if(deviceCodeDetails){
                        console.log('Please follow the instructions below:')
                        console.log(deviceCodeDetails.message)

                        this._msal.doPollForDeviceCodeAuth(deviceCodeDetails.device_code).then((tokens:any) => {
                            console.log('Authentication succeeded!')

                            this._msal.refreshUserToken().then((tokens:any) => {
                                // console.log('Tokens:', tokens)
                            }).catch((error) => {
                                console.log('Failed to refresh token:', error)
                            })
                        })
                    }
                }).catch((error) => {
                    console.log('error', error)
                })

            } else {
                console.log('Unknown authentication method:', this._commander.opts().auth)
            }

        } else if(this._commander.args[0] == 'show'){
            this.actionShow()

        } else if(this._commander.args[0] == 'refresh'){
            this.actionRefresh()

        } else if(this._commander.args[0] == 'logout'){
            this.actionLogout()

        } else if(this._commander.args[0] == 'tokens'){
            this.actionTokens()
        }
    }

    actionShow(){
        
        console.log('Current authentication status:')
        console.log(' Authentication method:', this._tokenStore.getAuthenticationMethod().toUpperCase())
        console.log(' UserToken: isAuthenticated('+this._tokenStore._userToken?.isValid()+') Seconds remaining:', this._tokenStore._userToken?.getSecondsValid())

        if(this._tokenStore.getAuthenticationMethod() === 'xal'){
            console.log(' SisuToken: isAuthenticated('+this._tokenStore._sisuToken?.isValid()+') Seconds remaining:', this._tokenStore._sisuToken?.getSecondsValid())
            console.log(' ')
            console.log(' User Hash:', this._tokenStore._sisuToken?.getUserHash())
            console.log(' Gamertag:', this._tokenStore._sisuToken?.getGamertag())
        }
    }

    actionRefresh(){
        if(this._tokenStore.getUserToken() === undefined){
            console.log('Please authenticate first using `xbox-auth auth`.')
            return
        }

        let refreshMethod:Promise<any>
        if(this._tokenStore.getAuthenticationMethod() === 'msal'){
            refreshMethod = this._msal.refreshUserToken()
        } else {
            refreshMethod = this._xal.refreshTokens()
        }

        refreshMethod.then((token) => {
            console.log('Tokens have been refreshed')

        }).catch((error) => {
            console.log('Failed to refresh token:', error)
        })

    }

    actionLogout(){
        this._tokenStore.removeAll()
        console.log('Login data has been removed. You are now logged out.')
    }

    actionTokens(){
        this.retrieveTokens().then((tokens) => {

            console.log('Tokens:\n- MSAL Token:', JSON.stringify(tokens.msalToken, null, 4), '\n- Web Token:', JSON.stringify(tokens.webToken, null, 4))
            console.log('Offering tokens:\n- xHome:', JSON.stringify(tokens.xhomeToken, null, 4), '\n- xCloud:', JSON.stringify(tokens.gpuToken, null, 4))
        }).catch((error) => {
            if(error instanceof TokenRefreshError){
                console.log('Failed to refresh user token. Please re-authenticate again using `xbox-auth auth`.\n Details:', error)
            } else {
                console.log('Failed to retrieve tokens. Please try again later.\n Details:', error)
            }
        })
        
    }

    async retrieveTokens(){
        let msalToken
        let webToken
        let streamingTokens
        
        if(this._tokenStore.getAuthenticationMethod() === 'msal'){
            msalToken = await this._msal.getMsalToken()
            webToken = await this._msal.getWebToken()
            streamingTokens = await this._msal.getStreamingTokens()
        } else {
            msalToken = await this._xal.getMsalToken()
            webToken = await this._xal.getWebToken()
            streamingTokens = await this._xal.getStreamingTokens()
        }

        const gpuToken = streamingTokens.xCloudToken
        const xhomeToken = streamingTokens.xHomeToken

        return { msalToken, webToken, xhomeToken, gpuToken }
    }
}

new Auth().run()