more typesafety

webapp/src/ai_integration.ts

@@ -4,6 +4,8 @@
 import {GlobalState} from 'mattermost-webapp/packages/types/src/store';
 import {useSelector} from 'react-redux';
 
+import {BotSelector, Bot, BotsLoaderHook} from './types/ai';
+
 export const aiPluginID = 'mattermost-ai';
 
 export const useAIAvailable = () => {
@@ -12,21 +14,21 @@ export const useAIAvailable = () => {
 };
 
 export const useAIAvailableBots = () => {
-    return useSelector<GlobalState, any[]>((state) => {
+    return useSelector<GlobalState, Bot[]>((state) => {
         //@ts-ignore plugins state is a thing
         return state['plugins-' + aiPluginID]?.bots || [];
     });
 };
 
 export const useBotSelector = () => {
-    return useSelector<GlobalState, any[]>((state) => {
+    return useSelector<GlobalState, BotSelector>((state) => {
         //@ts-ignore plugins state is a thing
         return state['plugins-' + aiPluginID]?.botSelector;
     });
 };
 
 export const useBotsLoaderHook = () => {
-    return useSelector<GlobalState, any[]>((state) => {
+    return useSelector<GlobalState, BotsLoaderHook>((state) => {
         //@ts-ignore plugins state is a thing
         return state['plugins-' + aiPluginID]?.botsLoaderHook || (() => null);
     });

webapp/src/components/modals/ai_modal.tsx

@@ -15,6 +15,8 @@ import {Textbox} from 'src/webapp_globals';
 import {generateStatusUpdate} from 'src/client';
 import {useBotSelector, useBotsLoaderHook} from 'src/ai_integration';
 
+import {Bot} from 'src/types/ai';
+
 import postEventListener, {PostUpdateWebsocketMessage} from 'src/websocket';
 
 type Version = {
@@ -53,12 +55,12 @@ const AIModal = ({playbookRunId, onAccept, onClose, isOpen}: Props) => {
     const [copied, setCopied] = useState(false);
     const [instruction, setInstruction] = useState('');
     const suggestionBox = useRef<HTMLDivElement|null>(null);
-    const BotSelector = useBotSelector() as any;
-    const useBotlist = useBotsLoaderHook() as any;
+    const BotSelector = useBotSelector();
+    const useBotlist = useBotsLoaderHook();
     const {bots, activeBot, setActiveBot} = useBotlist();
     const [currentVersion, setCurrentVersion] = useState<number>(0);
     const [versions, setVersions] = useState<Version[]>([]);
-    const [generating, setGenerating] = useState<any>(null);
+    const [generating, setGenerating] = useState(false);
 
     useEffect(() => {
         if (activeBot?.id && isOpen) {
@@ -89,7 +91,7 @@ const AIModal = ({playbookRunId, onAccept, onClose, isOpen}: Props) => {
         };
     }, [generating]);
 
-    const onBotChange = useCallback((bot: any) => {
+    const onBotChange = useCallback((bot: Bot) => {
         setActiveBot(bot);
         setCurrentVersion(versions.length + 1);
         setVersions([...versions, {instruction: '', value: '', prevValue: ''}]);

webapp/src/types/ai.tsx

@@ -0,0 +1,14 @@
+import React from 'react'
+
+export type Bot = {
+  id: string
+}
+
+type BotSelectorProps = {
+  bots: Bot[]
+  activeBot: Bot
+  setActiveBot: (bot: Bot) => void
+}
+
+export type BotsLoaderHook = () => BotSelectorProps
+export type BotSelector = React.FunctionComponent<BotSelectorProps>