Hook up generating from source

Author: robertbrignull

extensions/ql-vscode/src/view/data-extensions-editor/DataExtensionsEditor.tsx

@@ -179,7 +179,7 @@ export function DataExtensionsEditor({
     [],
   );
 
-  const onGenerateClick = useCallback(() => {
+  const onGenerateFromSourceClick = useCallback(() => {
     vscode.postMessage({
       t: "generateExternalApi",
     });
@@ -279,7 +279,7 @@ export function DataExtensionsEditor({
                   Refresh
                 </VSCodeButton>
               )}
-              <VSCodeButton onClick={onGenerateClick}>
+              <VSCodeButton onClick={onGenerateFromSourceClick}>
                 {viewState?.mode === Mode.Framework
                   ? "Generate"
                   : "Download and generate"}
@@ -301,6 +301,7 @@ export function DataExtensionsEditor({
               onChange={onChange}
               onSaveModelClick={onSaveModelClick}
               onGenerateFromLlmClick={onGenerateFromLlmClick}
+              onGenerateFromSourceClick={onGenerateFromSourceClick}
             />
           </EditorContainer>
         </>

extensions/ql-vscode/src/view/data-extensions-editor/LibraryRow.tsx

@@ -87,6 +87,7 @@ type Props = {
     externalApiUsages: ExternalApiUsage[],
     modeledMethods: Record<string, ModeledMethod>,
   ) => void;
+  onGenerateFromSourceClick: () => void;
 };
 
 export const LibraryRow = ({
@@ -99,6 +100,7 @@ export const LibraryRow = ({
   onChange,
   onSaveModelClick,
   onGenerateFromLlmClick,
+  onGenerateFromSourceClick,
 }: Props) => {
   const modeledPercentage = useMemo(() => {
     return calculateModeledPercentage(externalApiUsages);
@@ -119,10 +121,14 @@ export const LibraryRow = ({
     [externalApiUsages, modeledMethods, onGenerateFromLlmClick],
   );
 
-  const handleModelFromSource = useCallback(async (e: React.MouseEvent) => {
-    e.stopPropagation();
-    e.preventDefault();
-  }, []);
+  const handleModelFromSource = useCallback(
+    async (e: React.MouseEvent) => {
+      onGenerateFromSourceClick();
+      e.stopPropagation();
+      e.preventDefault();
+    },
+    [onGenerateFromSourceClick],
+  );
 
   const handleModelDependency = useCallback(async (e: React.MouseEvent) => {
     e.stopPropagation();

extensions/ql-vscode/src/view/data-extensions-editor/ModeledMethodsList.tsx

@@ -30,6 +30,7 @@ type Props = {
     externalApiUsages: ExternalApiUsage[],
     modeledMethods: Record<string, ModeledMethod>,
   ) => void;
+  onGenerateFromSourceClick: () => void;
 };
 
 export const ModeledMethodsList = ({
@@ -41,6 +42,7 @@ export const ModeledMethodsList = ({
   onChange,
   onSaveModelClick,
   onGenerateFromLlmClick,
+  onGenerateFromSourceClick,
 }: Props) => {
   const grouped = useMemo(
     () => groupMethods(externalApiUsages, mode),
@@ -63,6 +65,7 @@ export const ModeledMethodsList = ({
           onChange={onChange}
           onSaveModelClick={onSaveModelClick}
           onGenerateFromLlmClick={onGenerateFromLlmClick}
+          onGenerateFromSourceClick={onGenerateFromSourceClick}
         />
       ))}
     </>