Skip to content

Commit

Permalink
AI App Ramp up + Presence Integration
Browse files Browse the repository at this point in the history
  • Loading branch information
chentong7 committed Oct 22, 2024
1 parent b8e887e commit 8a65277
Show file tree
Hide file tree
Showing 7 changed files with 148 additions and 4 deletions.
2 changes: 1 addition & 1 deletion examples/apps/ai-collab/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@ You can run this example using the following steps:
- For an even faster build, you can add the package name to the build command, like this:
`pnpm run build:fast --nolint @fluid-example/ai-collab`
1. Start a Tinylicious server by running `pnpm start:server` from this directory.
1. In a separate terminal also from this directory, run `pnpm next:dev` and open http://localhost:3000/ in a
1. In a separate terminal also from this directory, run `pnpm start` and open http://localhost:3000/ in a
web browser to see the app running.

### Using SharePoint embedded instead of tinylicious
Expand Down
5 changes: 4 additions & 1 deletion examples/apps/ai-collab/package.json
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,9 @@
"start": "next dev",
"start:server": "tinylicious"
},
"dependencies": {},
"dependencies": {
"@fluid-experimental/presence": "workspace:~"
},
"devDependencies": {
"@azure/identity": "^4.4.1",
"@azure/msal-browser": "^3.25.0",
Expand Down Expand Up @@ -62,6 +64,7 @@
"react": "^18.3.1",
"react-dom": "^18.3.1",
"rimraf": "^4.4.0",
"source-map-loader": "^5.0.0",
"tinylicious": "^5.0.0",
"typechat": "^0.1.1",
"typescript": "~5.4.5",
Expand Down
11 changes: 11 additions & 0 deletions examples/apps/ai-collab/src/app/page.tsx
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@

"use client";

import { acquirePresenceViaDataObject } from "@fluid-experimental/presence";
import {
Box,
Button,
Expand All @@ -18,7 +19,10 @@ import {
import type { IFluidContainer, TreeView } from "fluid-framework";
import React, { useEffect, useState } from "react";

import { buildUserPresence, type UserPresence } from "./presence";

import { TaskGroup } from "@/components/TaskGroup";
import UserProfilePhoto from "@/components/UserProfilePhoto";
import {
CONTAINER_SCHEMA,
INITIAL_APP_STATE,
Expand Down Expand Up @@ -47,6 +51,7 @@ export async function createAndInitializeContainer(): Promise<
export default function TasksListPage(): JSX.Element {
const [selectedTaskGroup, setSelectedTaskGroup] = useState<SharedTreeTaskGroup>();
const [treeView, setTreeView] = useState<TreeView<typeof SharedTreeAppState>>();
const [onlineUsers, setOnlineUsers] = useState<UserPresence>();

const { container, isFluidInitialized, data } = useFluidContainerNextJs(
containerIdFromUrl(),
Expand All @@ -57,6 +62,11 @@ export default function TasksListPage(): JSX.Element {
(fluidContainer) => {
const _treeView = fluidContainer.initialObjects.appState.viewWith(TREE_CONFIGURATION);
setTreeView(_treeView);

const presence = acquirePresenceViaDataObject(fluidContainer.initialObjects.presence);
const _onlineUsers = buildUserPresence(presence);
setOnlineUsers(_onlineUsers);

return { sharedTree: _treeView };
},
);
Expand All @@ -79,6 +89,7 @@ export default function TasksListPage(): JSX.Element {
sx={{ display: "flex", flexDirection: "column", alignItems: "center" }}
maxWidth={"lg"}
>
{onlineUsers && <UserProfilePhoto onlineUsers={onlineUsers} />}
<Typography variant="h2" sx={{ my: 3 }}>
My Work Items
</Typography>
Expand Down
26 changes: 26 additions & 0 deletions examples/apps/ai-collab/src/app/presence.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,26 @@
/*!
* Copyright (c) Microsoft Corporation and contributors. All rights reserved.
* Licensed under the MIT License.
*/

import {
IPresence,
LatestMap,
type PresenceStates,
type PresenceStatesSchema,
} from "@fluid-experimental/presence";

export interface User {
photo: string;
}

const statesSchema = {
onlineUsers: LatestMap<{ value: User }, `id-${string}`>(),
} satisfies PresenceStatesSchema;

export type UserPresence = PresenceStates<typeof statesSchema>;

export function buildUserPresence(presence: IPresence): UserPresence {
const states = presence.getStates("name:app-client-states", statesSchema);
return states;
}
92 changes: 92 additions & 0 deletions examples/apps/ai-collab/src/components/UserProfilePhoto.tsx
Original file line number Diff line number Diff line change
@@ -0,0 +1,92 @@
/*!
* Copyright (c) Microsoft Corporation and contributors. All rights reserved.
* Licensed under the MIT License.
*/

import { InteractiveBrowserCredential } from "@azure/identity";
import { Client } from "@microsoft/microsoft-graph-client";
import { TokenCredentialAuthenticationProvider } from "@microsoft/microsoft-graph-client/authProviders/azureTokenCredentials";
import { Avatar, Badge } from "@mui/material";
import React, { useEffect, useState } from "react";
import { v4 as uuid } from "uuid";

import type { UserPresence } from "@/app/presence";

interface UserProfilePhotoProps {
onlineUsers: UserPresence;
}

const UserProfilePhoto: React.FC<UserProfilePhotoProps> = ({ onlineUsers }) => {
const [photos, setPhotos] = useState<string[]>([]);
const [isPhotoFetched, setIsPhotoFetched] = useState<boolean>(false);

useEffect(() => {
const allPhotos: string[] = [];
for (const element of Array.from(onlineUsers.onlineUsers.clientValues())) {
for (const user of Array.from(element.items.values())) {
allPhotos.push(user.value.value.photo);
}
}
setPhotos(allPhotos);
}, [onlineUsers]);

useEffect(() => {
const fetchPhoto = async () => {
const clientId = process.env.NEXT_PUBLIC_SPE_CLIENT_ID;
const tenantId = process.env.NEXT_PUBLIC_SPE_ENTRA_TENANT_ID;
if (tenantId === undefined || clientId === undefined) {
return;
}

const credential = new InteractiveBrowserCredential({
clientId,
tenantId,
});

const authProvider = new TokenCredentialAuthenticationProvider(credential, {
scopes: ["User.Read"],
});

const client = Client.initWithMiddleware({ authProvider });
try {
const photoBlob = await client.api("/me/photo/$value").get();
const photoUrl = URL.createObjectURL(photoBlob);
setPhotos((prevPhotos) => {
if (!prevPhotos.includes(photoUrl)) {
return [...prevPhotos, photoUrl];
}
return prevPhotos;
});
onlineUsers.onlineUsers.local.set(`id-${uuid()}`, { value: { photo: photoUrl } });
setIsPhotoFetched(true);
} catch (error) {
console.error(error);
}
};

if (!isPhotoFetched) {
fetchPhoto();
}
}, [isPhotoFetched, onlineUsers]);

return (
<div>
{photos.length === 0 ? (
<Avatar alt="User Photo" sx={{ width: 56, height: 56 }} />
) : (
photos.map((photo, index) => (
<Badge
key={index}
overlap="circular"
anchorOrigin={{ vertical: "bottom", horizontal: "right" }}
variant="dot"
>
<Avatar alt="User Photo" src={photo} sx={{ width: 56, height: 56 }} />
</Badge>
))
)}
</div>
);
};

export default UserProfilePhoto;
9 changes: 7 additions & 2 deletions examples/apps/ai-collab/src/types/sharedTreeAppSchema.ts
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
*/

import { SharedTree, SchemaFactory, TreeViewConfiguration } from "fluid-framework";

import { ExperimentalPresenceManager } from "@fluid-experimental/presence";
import type { Engineer, Task, TaskGroup } from "./task";

// The string passed to the SchemaFactory should be unique
Expand Down Expand Up @@ -155,7 +155,12 @@ export const INITIAL_APP_STATE = {
} as const;

export const CONTAINER_SCHEMA = {
initialObjects: { appState: SharedTree },
initialObjects: {
appState: SharedTree,
// A Presence Manager object temporarily needs to be placed within container schema
// https://github.com/microsoft/FluidFramework/blob/main/packages/framework/presence/README.md#onboarding
presence: ExperimentalPresenceManager,
},
};

export const TREE_CONFIGURATION = new TreeViewConfiguration({
Expand Down
7 changes: 7 additions & 0 deletions pnpm-lock.yaml

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

0 comments on commit 8a65277

Please sign in to comment.