jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

fix(shareModal): import removeFileNameFromComponents function to fix compilation error

Browse source 
feat(shareModal): add logic to remove file names from flows before sharing to improve privacy and security
feat(reactflowUtils): add removeFileNameFromComponents function to recursively remove file names from flow components
This commit is contained in:
anovazzi1 2023-11-30 16:43:28 -03:00
commit 67e520cfd6
2 changed files with 19 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/frontend/src/modals/shareModal/index.tsx
View file

@ -13,7 +13,10 @@ import {
saveFlowStore, saveFlowStore,
} from "../../controllers/API"; } from "../../controllers/API";
import { FlowType } from "../../types/flow"; import { FlowType } from "../../types/flow";
import { removeApiKeys } from "../../utils/reactflowUtils"; import {
removeApiKeys,
removeFileNameFromComponents,
} from "../../utils/reactflowUtils";
import { getTagsIds } from "../../utils/storeUtils"; import { getTagsIds } from "../../utils/storeUtils";
import BaseModal from "../baseModal"; import BaseModal from "../baseModal";
@ -81,6 +84,8 @@ export default function ShareModal({
}, [component, open, internalOpen]); }, [component, open, internalOpen]);
const handleShareComponent = () => { const handleShareComponent = () => {
//remove file names from flows before sharing
removeFileNameFromComponents(component);
const flow: FlowType = checked const flow: FlowType = checked
? { ? {
id: component!.id, id: component!.id,

13
src/frontend/src/utils/reactflowUtils.ts
View file

@ -1096,3 +1096,16 @@ export function updateComponentNameAndType(
data: any, data: any,
component: NodeDataType component: NodeDataType
) {} ) {}
export function removeFileNameFromComponents(flow: FlowType) {
flow.data!.nodes.forEach((node: NodeType) => {
Object.keys(node.data.node!.template).forEach((field) => {
if (node.data.node?.template[field].type === "file") {
node.data.node!.template[field].value = "";
}
});
if (node.data.node?.flow) {
removeFileNameFromComponents(node.data.node.flow);
}
});
}