Simple implementation for podman-desktop/api/aunthentication API

Signed-off-by: Denis Golovin <[email protected]>

Author: dgolovin

src/authentication-service.ts

@@ -19,7 +19,7 @@
  *  IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
  *  WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  *--------------------------------------------------------------------------------------------*/
-import { window } from '@podman-desktop/api';
+import { AuthenticationSession, window, EventEmitter, AuthenticationProviderSessionChangeEvent } from '@podman-desktop/api';
 import { ServerResponse } from 'node:http';
 import { Client, generators, Issuer, TokenSet } from 'openid-client';
 import { createServer, startServer } from './authentication-server';
@@ -60,18 +60,18 @@ interface IStoredSession {
   };
 }
 
-// export const onDidChangeSessions = new vscode.EventEmitter<vscode.AuthenticationProviderAuthenticationSessionsChangeEvent>();
+export const onDidChangeSessions = new EventEmitter<AuthenticationProviderSessionChangeEvent>();
 
 export const REFRESH_NETWORK_FAILURE = 'Network failure';
 
-export interface RedHatAuthenticationSession {
+export interface RedHatAuthenticationSession extends AuthenticationSession{
   idToken: string | undefined;
   readonly id: string;
   readonly accessToken: string;
   readonly scopes: ReadonlyArray<string>;
-  account?: {
+  account: {
     label: string;
-    id: string;
+    id?: string;
   };
 }
 
@@ -227,7 +227,7 @@ export class RedHatAuthenticationService {
     }
 
     if (added.length || removed.length) {
-      // onDidChangeSessions.fire({ added: added, removed: removed, changed: [] });
+      onDidChangeSessions.fire({ added: added, removed: removed, changed: [] });
     }
   }
 
@@ -240,7 +240,7 @@ export class RedHatAuthenticationService {
       id: token.sessionId,
       accessToken: token.accessToken!,
       idToken: token.idToken,
-      // account: token.account,
+      account: token.account,
       scopes: token.scope.split(' '),
     };
   }
@@ -409,7 +409,7 @@ export class RedHatAuthenticationService {
         setTimeout(async () => {
           try {
             const refreshedToken = await this.refreshToken(token.refreshToken, scope, token.sessionId);
-            // onDidChangeSessions.fire({ added: [], removed: [], changed: [this.convertToSessionSync(refreshedToken)] });
+            onDidChangeSessions.fire({ added: [], removed: [], changed: [this.convertToSessionSync(refreshedToken)] });
           } catch (e: any) {
             if (e.message === REFRESH_NETWORK_FAILURE) {
               const didSucceedOnRetry = await this.handleRefreshNetworkError(
@@ -422,7 +422,7 @@ export class RedHatAuthenticationService {
               }
             } else {
               await this.removeSession(token.sessionId);
-              // onDidChangeSessions.fire({ added: [], removed: [this.convertToSessionSync(token)], changed: [] });
+              onDidChangeSessions.fire({ added: [], removed: [this.convertToSessionSync(token)], changed: [] });
             }
           }
         }, 1000 * (token.expiresIn - 30)),
@@ -515,7 +515,7 @@ export class RedHatAuthenticationService {
         const token = this._tokens.find(token => token.sessionId === sessionId);
         if (token) {
           token.accessToken = undefined;
-          // onDidChangeSessions.fire({ added: [], removed: [], changed: [this.convertToSessionSync(token)] });
+          onDidChangeSessions.fire({ added: [], removed: [], changed: [this.convertToSessionSync(token)] });
         }
       }
 

src/extension.ts

@@ -18,7 +18,7 @@
 
 import * as extensionApi from '@podman-desktop/api';
 import { getAuthConfig } from './configuration';
-import { RedHatAuthenticationService } from './authentication-service';
+import { onDidChangeSessions, RedHatAuthenticationService, RedHatAuthenticationSession } from './authentication-service';
 import { shell } from 'electron';
 const menuItemsRegistered: extensionApi.Disposable[] = [];
 
@@ -59,6 +59,7 @@ async function initMenu(extensionContext: extensionApi.ExtensionContext): Promis
 }
 
 let loginService:RedHatAuthenticationService;
+let currentSession: RedHatAuthenticationSession;
 
 async function getAutenticatonService() {
   if (!loginService) {
@@ -68,41 +69,56 @@ async function getAutenticatonService() {
   return loginService;
 }
 
-const _onDidChangeSessions = new extensionApi.EventEmitter<extensionApi.AuthenticationProviderSessionChangeEvent>();
+let authService: RedHatAuthenticationService;
+
+async function getAuthService() {
+  if (!authService) {
+    authService = await getAutenticatonService();
+  }
+  return authService;
+}
 
 export async function activate(extensionContext: extensionApi.ExtensionContext): Promise<void> {
   console.log('starting extension redhat-authentication');
 
   await initMenu(extensionContext);
 
-  extensionApi.authentication.registerAuthenticationProvider({
-    onDidChangeSessions: _onDidChangeSessions.event,
-    createSession: function (): Promise<extensionApi.AuthenticationSession> {
-      throw new Error('Function not implemented.');
-    },
-    getSessions: function (): Promise<extensionApi.AuthenticationSession[]> {
-      throw new Error('Function not implemented.');
-    },
-    removeSession: function (id: string): Promise<void> {
-      throw new Error('Function not implemented.');
-    },
-    id: 'redhat.autentication-provider',
-    displayName: 'Red Hat',
-  });
+  extensionApi.authentication.registerAuthenticationProvider(
+    'redhat.autentication-provider',
+    'Red Hat', {
+      onDidChangeSessions: onDidChangeSessions.event,
+      createSession: async function (scopes: string[]): Promise<extensionApi.AuthenticationSession> {
+        const service = await getAuthService();
+        const session = await service.createSession(scopes.sort().join(' '));
+        onDidChangeSessions.fire({added: [session]});
+        return session;
+      },
+      getSessions: async function (scopes: string[]): Promise<extensionApi.AuthenticationSession[]> {
+        const service = await getAuthService();
+        return service.getSessions(scopes)
+      },
+      removeSession: async function (sessionId: string): Promise<void> {
+        const service = await getAuthService();
+        const session = await service.removeSession(sessionId);
+        onDidChangeSessions.fire({removed: [session]});
+      }
+    });
 
   const SignInCommand = extensionApi.commands.registerCommand('redhat.authentication.signin', async () => {
-    const service = await getAutenticatonService();
-    const session = await service.createSession('openid');
-
-    AuthMenuItem.label = `Red Hat (${session.account.label})`;
+    loginService = await getAutenticatonService();
+    currentSession = await loginService.createSession('openid');
+    onDidChangeSessions.fire({added: [currentSession], removed:[], changed:[]});
+    AuthMenuItem.label = `Red Hat (${currentSession.account.label})`;
     AuthMenuItem.
       submenu = [SignInMenuItem(false), SignOutMenuItem(true), Separator, SignUpMenuItem()];
     const subscription = extensionApi.tray.registerMenuItem(AuthMenuItem);
     extensionContext.subscriptions.push(subscription);
   });
 
   const SignOutCommand = extensionApi.commands.registerCommand('redhat.authentication.signout', async () => {
-    loginService = undefined;
+    loginService.removeSession(currentSession.id);
+    onDidChangeSessions.fire({added: [], removed:[currentSession], changed:[]});
+    currentSession = undefined;
     AuthMenuItem.label = `Red Hat`;
     AuthMenuItem.
       submenu = [SignInMenuItem(true), SignOutMenuItem(false), Separator, SignUpMenuItem()];