You can close this window now.
', + errorTemplate: 'Check the console for more information.
', + }); + + return authResponse; + } catch (error) { + throw error; + } + } + + /** + * Handles the response from a popup or redirect. If response is null, will check if we have any accounts and attempt to sign in. + * @param response + */ + async handleResponse(response) { + if (response !== null) { + this.account = response.account; + } else { + this.account = await this.getAccount(); + } + + return this.account; + } + + /** + * Calls getAllAccounts and determines the correct account to sign into, currently defaults to first account found in cache. + * https://github.com/AzureAD/microsoft-authentication-library-for-js/blob/dev/lib/msal-common/docs/Accounts.md + */ + async getAccount() { + const currentAccounts = await this.cache.getAllAccounts(); + + if (!currentAccounts) { + console.log('No accounts detected'); + return null; + } + + if (currentAccounts.length > 1) { + // Add choose account code here + console.log('Multiple accounts detected, need to add choose account code.'); + return currentAccounts[0]; + } else if (currentAccounts.length === 1) { + return currentAccounts[0]; + } else { + return null; + } + } +} + +module.exports = AuthProvider; diff --git a/renderer/graph/authConfig.js b/renderer/graph/authConfig.js new file mode 100644 index 0000000..860b214 --- /dev/null +++ b/renderer/graph/authConfig.js @@ -0,0 +1,48 @@ +/* + * Copyright (c) Microsoft Corporation. All rights reserved. + * Licensed under the MIT License. + */ + +const { LogLevel } = require("@azure/msal-node"); + +/** + * Configuration object to be passed to MSAL instance on creation. + * For a full list of MSAL.js configuration parameters, visit: + * https://github.com/AzureAD/microsoft-authentication-library-for-js/blob/dev/lib/msal-node/docs/configuration.md + */ +const AAD_ENDPOINT_HOST = "https://login.microsoftonline.com/"; // include the trailing slash + +const msalConfig = { + auth: { + clientId: "d581ab07-3a21-44d3-84c4-16b06bef6266", + authority: `${AAD_ENDPOINT_HOST}common`, + }, + system: { + loggerOptions: { + loggerCallback(loglevel, message, containsPii) { + console.log(message); + }, + piiLoggingEnabled: false, + logLevel: LogLevel.Verbose, + }, + }, +}; + +/** + * Add here the endpoints and scopes when obtaining an access token for protected web APIs. For more information, see: + * https://github.com/AzureAD/microsoft-authentication-library-for-js/blob/dev/lib/msal-browser/docs/resources-and-scopes.md + */ +const GRAPH_ENDPOINT_HOST = "https://graph.microsoft.com/"; // include the trailing slash + +const protectedResources = { + graphMe: { + endpoint: `${GRAPH_ENDPOINT_HOST}v1.0/me`, + scopes: ["User.Read"], + } +}; + + +module.exports = { + msalConfig: msalConfig, + protectedResources: protectedResources, +}; diff --git a/renderer/graph/constants.js b/renderer/graph/constants.js new file mode 100644 index 0000000..c27ba43 --- /dev/null +++ b/renderer/graph/constants.js @@ -0,0 +1,16 @@ +/* + * Copyright (c) Microsoft Corporation. All rights reserved. + * Licensed under the MIT License. + */ + +const IPC_MESSAGES = { + SHOW_WELCOME_MESSAGE: 'SHOW_WELCOME_MESSAGE', + LOGIN: 'LOGIN', + LOGOUT: 'LOGOUT', + GET_PROFILE: 'GET_PROFILE', + SET_PROFILE: 'SET_PROFILE', +} + +module.exports = { + IPC_MESSAGES: IPC_MESSAGES, +} diff --git a/renderer/graph/graph.js b/renderer/graph/graph.js new file mode 100644 index 0000000..4e0f105 --- /dev/null +++ b/renderer/graph/graph.js @@ -0,0 +1,61 @@ +const { Client } = require('@microsoft/microsoft-graph-client'); +const { msalConfig } = require('./authConfig'); +const AuthProvider = require('./AuthProvider'); +require('isomorphic-fetch'); + +class GraphService { + constructor() { + this.authProvider = new AuthProvider(msalConfig); + this.graphClient = null; + } + + async initialize() { + try { + // Get account + await this.authProvider.login(); + + // Get token with OneDrive scopes + const tokenRequest = { + scopes: ['Files.Read', 'Files.Read.All', 'Sites.Read.All'] + }; + + const authResponse = await this.authProvider.getToken(tokenRequest); + + // Initialize Graph client + this.graphClient = Client.init({ + authProvider: (done) => { + done(null, authResponse.accessToken); + } + }); + + return authResponse.accessToken; + } catch (error) { + console.error('Error initializing Graph service:', error); + throw error; + } + } + + async listFolderContents(folderPath) { + if (!this.graphClient) { + await this.initialize(); + } + + try { + // Clean up the path + const cleanPath = folderPath.replace(/^\/+|\/+$/g, ''); + const endpoint = `/me/drive/root:/${cleanPath}:/children`; + + // Get items from the folder + const response = await this.graphClient.api(endpoint) + .select('id,name,size,file,@microsoft.graph.downloadUrl') + .get(); + + return response.value; + } catch (error) { + console.error('Error listing folder contents:', error); + throw error; + } + } +} + +module.exports = new GraphService(); \ No newline at end of file diff --git a/renderer/graphApi.js b/renderer/graph/graphApi.js similarity index 100% rename from renderer/graphApi.js rename to renderer/graph/graphApi.js diff --git a/renderer/renderer.js b/renderer/renderer.js index 86b41a7..0b4dd3e 100644 --- a/renderer/renderer.js +++ b/renderer/renderer.js @@ -1,5 +1,8 @@ const { ipcRenderer, session } = require('electron'); -const graphApi = require('./graphApi'); +const graphApi = require('./graph/graphApi'); + + + // Wait for webview to load const webview = document.getElementById('main-content'); @@ -82,17 +85,23 @@ document.getElementById('sync-button').addEventListener('click', async () => { syncButton.textContent = 'Syncing...'; syncButton.disabled = true; - // Get folder contents using GraphApiClient - console.log('renderer: Getting folder contents:', onedriveSource); - const items = await graphApi.listFolderContents(onedriveSource); - console.log('renderer: Found items:', items.length); + // Get folder contents using GraphService + console.log('Getting folder contents:', onedriveSource); + const graphService = require('./graph/graph'); + + // This will handle authentication and get folder contents + const items = await graphService.listFolderContents(onedriveSource); + console.log('Found items:', items.length); - // Process the items - // TODO: Handle the items as needed showStatus(`Found ${items.length} items`); // Send the items to the main process - ipcRenderer.send('start-sync', { destFolder, onedriveSource , items}); + ipcRenderer.send('start-sync', { + destFolder, + onedriveSource, + items, + accessToken: await graphService.initialize() // Get fresh token for download + }); } catch (error) { console.error('Sync error:', error);