sample-ui-component-library 0.0.46-dev → 0.0.48-dev

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "sample-ui-component-library",
-  "version": "0.0.46-dev",
+  "version": "0.0.48-dev",
   "description": "A library which contains sample UI elements that can be used for populating layouts.",
   "main": "dist/cjs/index.js",
   "module": "dist/esm/index.js",
@@ -46,11 +46,13 @@
     "@storybook/test": "^8.6.11",
     "@storybook/theming": "^8.6.11",
     "css-loader": "^7.1.2",
+    "dal-engine-core-js-lib-dev": "^0.0.4",
     "gh-pages": "^6.3.0",
     "prop-types": "^15.8.1",
     "raw-loader": "^4.0.2",
     "react": "^18.2.0",
     "react-dom": "^18.2.0",
+    "reaflow": "^5.4.1",
     "rollup": "^4.37.0",
     "rollup-plugin-peer-deps-external": "^2.2.4",
     "rollup-plugin-postcss": "^4.0.2",
@@ -69,7 +71,6 @@
     "@dagrejs/dagre": "^1.1.4",
     "@monaco-editor/react": "^4.7.0",
     "@xyflow/react": "^12.6.0",
-    "dal-engine-core-js-lib-dev": "^0.0.4",
     "react-bootstrap-icons": "^1.11.5"
   }
 }

package/rollup.config.mjs

@@ -1,13 +1,19 @@
 import resolve from '@rollup/plugin-node-resolve';
 import commonjs from '@rollup/plugin-commonjs';
 import terser from '@rollup/plugin-terser';
-import external from 'rollup-plugin-peer-deps-external';
+import peerDepsExternal  from 'rollup-plugin-peer-deps-external';
 import postcss from 'rollup-plugin-postcss';
 import json from '@rollup/plugin-json';
 import { babel } from '@rollup/plugin-babel';
 
 export default {
     input: 'src/index.js',
+    external: [
+        'react',
+        'react-dom',
+        '@dnd-kit/core',
+        "reaflow"
+    ],
     output: [
         {
             file: 'dist/cjs/index.js',
@@ -21,7 +27,7 @@ export default {
         },
     ],
     plugins: [
-        external(['react', 'react-dom', '@dnd-kit/core',"reaflow"]),
+        peerDepsExternal(),
         resolve({
             extensions: ['.js', '.jsx'],
         }),

package/src/components/BehavioralGraphBuilder/BehavioralGraphBuilder.jsx

@@ -53,7 +53,9 @@ export const BehavioralGraphBuilder = forwardRef(({connectBehaviors, deleteTrans
 
     // Callbacks
     const handleWheel = useCallback((e) => {
-        (e.deltaY < 0)?canvasRef.current.zoomIn():canvasRef.current.zoomOut();
+        if (canvasRef.current) {
+            (e.deltaY < 0)?canvasRef.current.zoomIn():canvasRef.current.zoomOut();
+        }
     }, [canvasRef]);
 
     const nodeClick = useCallback((e, node) => {

package/src/components/BehavioralGraphBuilder/BehavioralGraphBuilder.scss

@@ -7,6 +7,6 @@
     overflow:hidden;
 }
 
-canvas {
+.canvas-wrapper > canvas {
     overflow: hidden;
 }

package/src/components/BehavioralGraphBuilder/helper.js

@@ -1,19 +1,18 @@
 
 /**
- * Converts a DAL design specification object into React Flow elements (nodes and edges).
- * @param {Object} design 
+ * Converts a DAL design specification object into nodes and edges for reaflow.
+ * @param {Object} engine - The engine instance containing the design specification.
  * @returns {Object} An object containing nodes and edges for React Flow
  */
 export const designToNodes = (engine) => {
-
     let edges = [];
     let nodes = [];
 
     for (let i = 0; i < engine.graph.nodes.length; i++) {
         const node = engine.graph.nodes[i];
         nodes.push({
-            id: node.getBehavior().name,
-            text: node.getBehavior().name,
+            id: node.getBehavior().getName(),
+            text: node.getBehavior().getName(),
         });
     }
     
@@ -25,8 +24,8 @@ export const designToNodes = (engine) => {
 
         node.getGoToBehaviors().forEach((goTo) => {
             edges.push({
-                id: `${node.getBehavior().name}->${goTo}`,
-                from: node.getBehavior().name,
+                id: `${node.getBehavior().getName()}->${goTo}`,
+                from: node.getBehavior().getName(),
                 to: goTo,
             });
         });

package/src/components/Editor/Editor.jsx

@@ -17,6 +17,11 @@ import { EditorContext } from "./EditorContext";
 
 import { editorReducer, initialState } from "./EditorReducer";
 
+const MODES = {
+    MAPPING: 1,
+    IMPLEMENTATION: 2
+}
+
 /**
  * Renders the editor component with support for tabs.
  * 
@@ -45,6 +50,10 @@ export const Editor = forwardRef(({ }, ref) => {
         dispatch({ type: "RESET_STATE"});
         dispatch({ type: "SET_PARENT_TAB_GROUP_ID", payload: id });
     }, []);
+    
+    const setMapping = useCallback((fileName, mapping) => {
+        dispatch({ type: "SET_MAPPING", payload: { fileName, mapping } });
+    }, []);
 
     const api = useMemo(() => {
         return {
@@ -53,9 +62,10 @@ export const Editor = forwardRef(({ }, ref) => {
             setTabGroupId,
             selectTab,
             closeTab,
-            moveTab
+            moveTab,
+            setMapping
         };
-    }, [state, addTab, selectTab, closeTab, moveTab, setTabGroupId]);
+    }, [state, addTab, selectTab, closeTab, moveTab, setTabGroupId, setMapping]);
 
     useImperativeHandle(ref, () => api, [api]);
 

package/src/components/Editor/EditorReducer.js

@@ -99,6 +99,25 @@ export const editorReducer = (state, action) => {
                 tabs: prevTabs
             };
         }
+
+        case "SET_MAPPING": {
+            const { mapping, fileName } = action.payload;
+            console.log(state);
+            for (let i = 0; i < state.tabs.length; i++) {
+                const tab = state.tabs[i];
+                if (tab.name === fileName) {
+                    const updatedTab = { ...tab, mapping: mapping };
+                    const newTabs = [...state.tabs];
+                    newTabs[i] = updatedTab;
+                    return {
+                        ...state,
+                        tabs: newTabs,
+                        activeTab: updatedTab
+                    };
+                }
+            }
+            return state;
+        }
         
         case "RESET_STATE": {
             return initialState;

package/src/components/Editor/MonacoInstance/MonacoInstance.jsx

@@ -1,4 +1,4 @@
-import React, { useEffect, useRef, useState } from 'react';
+import React, { useCallback, useLayoutEffect, useEffect, useRef, useState } from 'react';
 import PropTypes from 'prop-types';
 
 import Editor from '@monaco-editor/react';
@@ -17,14 +17,22 @@ export const MonacoInstance = ({ }) => {
     const containerRef = useRef(null);
     const frameRef = useRef(0);
 
-    useEffect(() => {
+    const [overlayDivs, setOverlayDivs] = useState();
+    const [overlayRanges, setOverlayRanges] = useState();
+
+    useLayoutEffect(() => {
         if (state.activeTab) {
             setEditorContent(state.activeTab.content);
             setShowEditor(true);
+            if (state.activeTab.mapping) {
+                addOverlays(state.activeTab.mapping);
+            } else {
+                setOverlayDivs([]);
+            }
         } else {
             setShowEditor(false);
         }
-    }, [state.activeTab]);
+    }, [state.activeTab, editorRef]);
 
     useEffect(() => {
         content.current = editorContent;
@@ -38,13 +46,53 @@ export const MonacoInstance = ({ }) => {
      * @param {Object} editor 
      * @param {Object} monaco 
      */
-    const handleEditorDidMount = (editor, monaco) => {
+    const handleEditorDidMount = useCallback((editor, monaco) => {
         editorRef.current = editor;
         if (content?.current) {
             editorRef.current.setValue(content.current);
         }
         editorRef.current.layout();
-    }
+        if (state.activeTab?.mapping) {
+            addOverlays(state.activeTab.mapping);
+        } else {
+            setOverlayDivs([]);
+        }
+    }, [state.activeTab]);
+
+    // Add overlays to editor for the given line ranges.
+    const addOverlays = useCallback((ranges) => {
+        if (!editorRef.current) return;
+
+        const lineCount = editorRef.current.getModel().getLineCount();
+        const lineHeight = editorRef.current.getOption(monaco.editor.EditorOption.lineHeight);
+        
+        const divs = [];
+        ranges.forEach((entry) => {
+             const top = editorRef.current.getTopForLineNumber(entry.start_line) - editorRef.current.getScrollTop();
+             let bottom = editorRef.current.getTopForLineNumber(entry.end_line + 1) - editorRef.current.getScrollTop();
+            if (entry.end_line >= lineCount) {
+                bottom = bottom + lineHeight;
+            }
+             const overlayDiv = <div className="line-block-overlay" style={{ top: top + "px", height: (bottom - top) + "px" }}></div>;
+             divs.push(overlayDiv);
+        });
+        setOverlayDivs(divs);
+    }, [editorRef?.current]);
+
+    // Scroll the editor and update overlays on wheel event.
+    const handleWheel = useCallback((e) => {
+        if (!editorRef.current) return;
+        const deltaY = e.deltaY;
+        const currentScrollTop = editorRef.current.getScrollTop();
+        editorRef.current.setScrollTop(currentScrollTop + deltaY);
+        if (state.activeTab?.mapping) {
+            addOverlays(state.activeTab.mapping);
+        } else {
+            setOverlayDivs([]);
+        }
+    }, [overlayRanges, state.activeTab]);
+
+
 
     // Editor options for Monaco Editor.
     const editorOptions = {
@@ -70,6 +118,7 @@ export const MonacoInstance = ({ }) => {
             cancelAnimationFrame(frameRef.current);
             frameRef.current = requestAnimationFrame(() => {
                 editorRef.current?.layout();
+                addOverlays(state.activeTab.mapping); 
             });
         });
 
@@ -79,7 +128,7 @@ export const MonacoInstance = ({ }) => {
             cancelAnimationFrame(frameRef.current);
             ro.disconnect();
         };
-    }, []);
+    }, [state.activeTab, addOverlays]);
 
     /**
      * Render the editor if there is an active tab, otherwise render a placeholder message.
@@ -104,6 +153,9 @@ export const MonacoInstance = ({ }) => {
     return (
         <div className="editor-container" ref={containerRef}>
             {renderEditor()}
+            <div className="overlay-layer" onWheel={handleWheel}>
+                {overlayDivs}
+            </div>
         </div>
     )
 }

package/src/components/Editor/MonacoInstance/MonacoInstance.scss

@@ -11,4 +11,24 @@
     position:relative;
     width:100%;
     height: 100%;
+}
+
+.overlay-layer {
+    position: absolute; 
+    left: 0;
+    right: 0;
+    inset: 0;
+    overflow:hidden;
+}
+
+.line-block-overlay {
+    z-index: 999;
+    position: absolute;
+    left: 0;
+    right: 0;
+}
+
+.line-block-overlay:hover {
+    background-color: rgba(255, 166, 0, 0.3);
+    cursor: pointer;
 }

package/src/stories/BehavioralGraphBuilder.stories.js

@@ -46,19 +46,19 @@ const Template = (args) => {
             activeTool: activeTool,
             design: design,
         });
-    }, []);
+    }, [activeTool, design]);
 
     useEffect(() => {
         if (editorRef.current) {
             const engine = new DALEngine({ name: "testEngine" });
-            // engine.deserialize(JSON.stringify(design));
             setEngine(engine);
             editorRef.current.updateEngine(engine);
-            setTimeout(() => {
+            const timerId = setTimeout(() => {
                 engine.addNode("testBehavior", []);
                 engine.addNode("testBehavior2", []);
                 editorRef.current.updateEngine(engine);
             }, 4000);
+            return () => clearTimeout(timerId);
         }
     }, [design, editorRef]);
 

package/src/stories/Editor.stories.js

@@ -11,6 +11,8 @@ import {
 } from "@dnd-kit/core";
 
 import WorkspaceSampleTree from "./data/FileBrowser/workspace_sample.json"
+import transactiondb_mapping from "./data/Mapping/TransactionDB_mapping.json"
+import mapping from "./data/Mapping/mapping.json"
 
 import "./EditorStories.scss"
 
@@ -52,15 +54,12 @@ const Template = (args) => {
             }
         });
 
-
-        const node = {
-            "name": "SAMPLE",
-            "type": "file",
-            "uid": "dissr-f6459410-1634-4dbc-8d76-35896822158d",
-            "content": "1234"
-        }
-
-        editorRef.current.addTab(node,2);
+        const result = flattenTree(WorkspaceSampleTree.tree).find(
+            (obj) => obj.name === "TransactionDB.py"
+        );
+        editorRef.current.addTab(result);
+        // editorRef.current.setMapping("FrenchTranslator.py", mapping);
+        editorRef.current.setMapping("TransactionDB.py", transactiondb_mapping);
     }, []);
 
 const [dragging, setDragging] = useState(false);

package/src/stories/components/ToolBar/ToolBar.js

@@ -30,12 +30,13 @@ export function ToolBar({ onSelectTool }) {
           className="icon"
         />
       </div>
-      <div className="toolbarContainer bottom"></div>
+      <div className="toolbarContainer bottom">
         <Floppy
           onClick={(e) => selectTool("save")}
           title="Save Design"
           className="icon"
         />
+      </div>
     </div>
   );
 }

package/src/stories/data/FileBrowser/workspace_sample.json

@@ -1 +1,66 @@
-{"tree":[{"name":"library_manager_single_threaded","type":"folder","uid":"dir-ba56de84-31dd-4ccf-ae98-a98b0581f803","children":[{"name":"library_manager.py","type":"file","uid":"dir-3d7aaf8e-6224-4eb7-8301-936447fef430","content":"import sys\n\ndef place_book_on_shelf(book_shelf, name, genre):\n    '''\n    This function places the given book on the shelf \n    in a slot determined by the first letter of its name.\n    \n    :param book_shelf: Bookshelf to place the book on.\n    :param name: Name of the book.\n    :param genre: Genre of the book.\n    '''\n    print(f\"Accepted book: {name} (Genre: {genre})\")\n    \n    firstLetter = name[0].upper()\n\n    if (firstLetter not in book_shelf):\n        book_shelf[firstLetter] = []\n    else:\n        print(\"Slot for book already exists.\")\n\n    book_shelf[firstLetter].append(name)\n\n    return book_shelf\n\ndef accept_book():\n    '''\n    Accepts a book and genre from the user and returns it.\n    '''\n    print(\"\\nEnter book details:\")\n        \n    name = input(\"Book name: \")\n\n    genre = input(\"Genre: \")\n\n    return {\"name\":name, \"genre\":genre}\n\n\ndef library_manager():\n    '''\n    Runs the library manager to accept books until the user decides to stop.\n    '''\n    book_shelf = {}\n\n    while True:\n        book_details = accept_book()\n        book_shelf = place_book_on_shelf(\n            book_shelf, \n            book_details[\"name\"], \n            book_details[\"genre\"]\n        )\n\n        more = input(\"Add another book? (y): \").lower()\n        \n        if more != \"y\":\n            break\n\n    print(\"Exiting library manager, goodbye.\")\n\nif __name__ == \"__main__\":\n    sys.exit(library_manager())\n    "},{"name":"readme","type":"file","uid":"dir-5aa507ee-e43d-4d77-b49a-b6e0b1da36f5","content":"This is version 1 of the library manager and it contains a simple implementation that accepts a book name and genre before placing it on the shelf. It places it on the shelf using the first letter of the book name. It continues to accept books until the user decides to stop.\n\nThis version of the library manager will also use instrumentation to identify abstractions by line to overcome the limitations of existing techniques."}]},{"name":"text_translator","type":"folder","uid":"dir-cbb97582-e3da-403e-aa7f-a524a5a7cff7","children":[{"name":"FrenchTranslator.py","type":"file","uid":"dir-516008b2-e80c-4a3a-bb00-d651885ac15d","content":"import threading\nimport queue\nfrom deep_translator import GoogleTranslator # type: ignore\n\nclass FrenchTranslator(threading.Thread):\n    def __init__(self, id, queue, packerQueue):\n        super().__init__(daemon=True)\n        self.queue = queue\n        self.id = id\n        self.packerQueue = packerQueue\n        self.start()\n\n    def run(self):\n        while True:\n            try:\n                job = self.queue.get(timeout=10)\n            except queue.Empty:\n                continue\n\n            translatedMsg = GoogleTranslator(source=\"auto\", target=\"french\").translate(job[\"value\"][\"data\"])\n            self.packerQueue.put({\n                \"uid\": job[\"uid\"],\n                \"value\": {\n                    \"language\": \"french\",\n                    \"original\": job[\"value\"][\"data\"],\n                    \"translated\": translatedMsg\n                }\n            })"},{"name":"Packer.py","type":"file","uid":"dir-e5085241-ff4f-434d-85ab-7f50e673ea4d","content":"import threading\nimport queue\nfrom TransactionDB import TransactionDB\n\nclass PackerThread(threading.Thread):\n    def __init__(self, queue):\n        super().__init__(daemon=True)\n        self.queue = queue\n        self.items = {}\n        self.start()\n\n    def run(self):\n        self.db = TransactionDB()\n        while True:\n            try:\n                translatedJobPart = self.queue.get(timeout=10)\n            except queue.Empty:\n                continue\n\n            self.db.addTranslation(translatedJobPart)\n\n            if self.db.isDone(translatedJobPart[\"uid\"]):\n                self.db.setDone(translatedJobPart[\"uid\"])"},{"name":"SpanishTranslator.py","type":"file","uid":"dir-b2a78dfd-aa94-4068-a5f7-24c396d88825","content":"import threading\nimport queue\nfrom deep_translator import GoogleTranslator # type: ignore\n\nclass SpanishTranslator(threading.Thread):\n    def __init__(self, id, queue, packerQueue):\n        super().__init__(daemon=True)\n        self.queue = queue\n        self.id = id\n        self.packerQueue = packerQueue\n        self.start()\n\n    def run(self):\n        while True:\n            try:\n                job = self.queue.get(timeout=10)\n            except queue.Empty:\n                continue\n\n            translatedMsg = GoogleTranslator(source=\"auto\", target=\"spanish\").translate(job[\"value\"][\"data\"])\n            self.packerQueue.put({\n                \"uid\": job[\"uid\"],\n                \"value\": {\n                    \"language\": \"spanish\",\n                    \"original\": job[\"value\"][\"data\"],\n                    \"translated\": translatedMsg\n                }\n            })"},{"name":"TamilTranslator.py","type":"file","uid":"dir-594ff411-b916-4a88-9077-3d5b8ff7f871","content":"import threading\nimport queue\nfrom deep_translator import GoogleTranslator # type: ignore\n\nclass TamilTranslator(threading.Thread):\n    def __init__(self, id, queue, packerQueue):\n        super().__init__(daemon=True)\n        self.queue = queue\n        self.id = id\n        self.packerQueue = packerQueue\n        self.start()\n\n    def run(self):\n        while True:\n            try:\n                job = self.queue.get(timeout=10)\n            except queue.Empty:\n                continue\n\n            translatedMsg = GoogleTranslator(source=\"auto\", target=\"tamil\").translate(job[\"value\"][\"data\"])\n            self.packerQueue.put({\n                \"uid\": job[\"uid\"],\n                \"value\": {\n                    \"language\": \"tamil\",\n                    \"original\": job[\"value\"][\"data\"],\n                    \"translated\": translatedMsg\n                }\n            })"},{"name":"readme","type":"file","uid":"dir-f6459410-1634-4dbc-8d76-35896822158d","content":"This folder contains a text translator that uses three workers to translate the text and uses a packer to add the results to a cabinet."}]}]}
+{
+    "tree": [
+        {
+            "name": "library_manager_single_threaded",
+            "type": "folder",
+            "uid": "dir-ba56de84-31dd-4ccf-ae98-a98b0581f803",
+            "children": [
+                {
+                    "name": "library_manager.py",
+                    "type": "file",
+                    "uid": "dir-3d7aaf8e-6224-4eb7-8301-936447fef430",
+                    "content": "import sys\n\ndef place_book_on_shelf(book_shelf, name, genre):\n    '''\n    This function places the given book on the shelf \n    in a slot determined by the first letter of its name.\n    \n    :param book_shelf: Bookshelf to place the book on.\n    :param name: Name of the book.\n    :param genre: Genre of the book.\n    '''\n    print(f\"Accepted book: {name} (Genre: {genre})\")\n    \n    firstLetter = name[0].upper()\n\n    if (firstLetter not in book_shelf):\n        book_shelf[firstLetter] = []\n    else:\n        print(\"Slot for book already exists.\")\n\n    book_shelf[firstLetter].append(name)\n\n    return book_shelf\n\ndef accept_book():\n    '''\n    Accepts a book and genre from the user and returns it.\n    '''\n    print(\"\\nEnter book details:\")\n        \n    name = input(\"Book name: \")\n\n    genre = input(\"Genre: \")\n\n    return {\"name\":name, \"genre\":genre}\n\n\ndef library_manager():\n    '''\n    Runs the library manager to accept books until the user decides to stop.\n    '''\n    book_shelf = {}\n\n    while True:\n        book_details = accept_book()\n        book_shelf = place_book_on_shelf(\n            book_shelf, \n            book_details[\"name\"], \n            book_details[\"genre\"]\n        )\n\n        more = input(\"Add another book? (y): \").lower()\n        \n        if more != \"y\":\n            break\n\n    print(\"Exiting library manager, goodbye.\")\n\nif __name__ == \"__main__\":\n    sys.exit(library_manager())\n    "
+                },
+                {
+                    "name": "readme",
+                    "type": "file",
+                    "uid": "dir-5aa507ee-e43d-4d77-b49a-b6e0b1da36f5",
+                    "content": "This is version 1 of the library manager and it contains a simple implementation that accepts a book name and genre before placing it on the shelf. It places it on the shelf using the first letter of the book name. It continues to accept books until the user decides to stop.\n\nThis version of the library manager will also use instrumentation to identify abstractions by line to overcome the limitations of existing techniques."
+                }
+            ]
+        },
+        {
+            "name": "text_translator",
+            "type": "folder",
+            "uid": "dir-cbb97582-e3da-403e-aa7f-a524a5a7cff7",
+            "children": [
+                {
+                    "name": "FrenchTranslator.py",
+                    "type": "file",
+                    "uid": "dir-516008b2-e80c-4a3a-bb00-d651885ac15d",
+                    "content": "import threading\nimport queue\nfrom deep_translator import GoogleTranslator # type: ignore\n\nclass FrenchTranslator(threading.Thread):\n    def __init__(self, id, queue, packerQueue):\n        super().__init__(daemon=True)\n        self.queue = queue\n        self.id = id\n        self.packerQueue = packerQueue\n        self.start()\n\n    def run(self):\n        while True:\n            try:\n                job = self.queue.get(timeout=10)\n            except queue.Empty:\n                continue\n\n            translatedMsg = GoogleTranslator(source=\"auto\", target=\"french\").translate(job[\"value\"][\"data\"])\n            self.packerQueue.put({\n                \"uid\": job[\"uid\"],\n                \"value\": {\n                    \"language\": \"french\",\n                    \"original\": job[\"value\"][\"data\"],\n                    \"translated\": translatedMsg\n                }\n            })"
+                },
+                {
+                    "name": "Packer.py",
+                    "type": "file",
+                    "uid": "dir-e5085241-ff4f-434d-85ab-7f50e673ea4d",
+                    "content": "import threading\nimport queue\nfrom TransactionDB import TransactionDB\n\nclass PackerThread(threading.Thread):\n    def __init__(self, queue):\n        super().__init__(daemon=True)\n        self.queue = queue\n        self.items = {}\n        self.start()\n\n    def run(self):\n        self.db = TransactionDB()\n        while True:\n            try:\n                translatedJobPart = self.queue.get(timeout=10)\n            except queue.Empty:\n                continue\n\n            self.db.addTranslation(translatedJobPart)\n\n            if self.db.isDone(translatedJobPart[\"uid\"]):\n                self.db.setDone(translatedJobPart[\"uid\"])"
+                },
+                {
+                    "name": "SpanishTranslator.py",
+                    "type": "file",
+                    "uid": "dir-b2a78dfd-aa94-4068-a5f7-24c396d88825",
+                    "content": "import threading\nimport queue\nfrom deep_translator import GoogleTranslator # type: ignore\n\nclass SpanishTranslator(threading.Thread):\n    def __init__(self, id, queue, packerQueue):\n        super().__init__(daemon=True)\n        self.queue = queue\n        self.id = id\n        self.packerQueue = packerQueue\n        self.start()\n\n    def run(self):\n        while True:\n            try:\n                job = self.queue.get(timeout=10)\n            except queue.Empty:\n                continue\n\n            translatedMsg = GoogleTranslator(source=\"auto\", target=\"spanish\").translate(job[\"value\"][\"data\"])\n            self.packerQueue.put({\n                \"uid\": job[\"uid\"],\n                \"value\": {\n                    \"language\": \"spanish\",\n                    \"original\": job[\"value\"][\"data\"],\n                    \"translated\": translatedMsg\n                }\n            })"
+                },
+                {
+                    "name": "TamilTranslator.py",
+                    "type": "file",
+                    "uid": "dir-594ff411-b916-4a88-9077-3d5b8ff7f871",
+                    "content": "import threading\nimport queue\nfrom deep_translator import GoogleTranslator # type: ignore\n\nclass TamilTranslator(threading.Thread):\n    def __init__(self, id, queue, packerQueue):\n        super().__init__(daemon=True)\n        self.queue = queue\n        self.id = id\n        self.packerQueue = packerQueue\n        self.start()\n\n    def run(self):\n        while True:\n            try:\n                job = self.queue.get(timeout=10)\n            except queue.Empty:\n                continue\n\n            translatedMsg = GoogleTranslator(source=\"auto\", target=\"tamil\").translate(job[\"value\"][\"data\"])\n            self.packerQueue.put({\n                \"uid\": job[\"uid\"],\n                \"value\": {\n                    \"language\": \"tamil\",\n                    \"original\": job[\"value\"][\"data\"],\n                    \"translated\": translatedMsg\n                }\n            })"
+                },
+                {
+                    "name": "readme",
+                    "type": "file",
+                    "uid": "dir-f6459410-1634-4dbc-8d76-35896822158d",
+                    "content": "This folder contains a text translator that uses three workers to translate the text and uses a packer to add the results to a cabinet."
+                },
+                {
+                    "name": "TransactionDB.py",
+                    "type": "file",
+                    "uid": "dir-f6459410-1634-4dbc-8d76-35sfdvsd6822158d",
+                    "content": "import sqlite3\nimport atexit\nimport uuid\n\nclass TransactionDB():\n\n    def __init__(self):\n        atexit.register(self.cleanup_function)\n        self.initializeDB()\n\n    def cleanup_function(self):\n        self.conn.close()\n\n    def initializeDB(self):\n\n        self.conn = sqlite3.connect(\"translations.db\")\n        self.cursor = self.conn.cursor()\n\n        self.cursor.execute(\"\"\"\n        CREATE TABLE IF NOT EXISTS translation_jobs (\n            job_id   INTEGER PRIMARY KEY AUTOINCREMENT,\n            uid      TEXT NOT NULL UNIQUE,\n            english  TEXT NOT NULL,\n            french   TEXT,\n            spanish  TEXT,\n            tamil    TEXT,\n            done     INTEGER NOT NULL DEFAULT 0\n        );\n        \"\"\")\n        self.conn.commit()\n\n    def addJob(self, data):\n        uid = str(uuid.uuid4())\n        self.conn.execute(\n            \"\"\"\n            INSERT INTO translation_jobs (uid, english)\n            VALUES (?, ?)\n            \"\"\",\n            (uid, data)\n        )\n        self.conn.commit()\n\n        return {\n            \"uid\": uid,\n            \"value\": {\"data\": data}\n        }\n    \n    def addTranslation(self, msg):\n        self.conn.execute(\n            f\"\"\"\n            UPDATE translation_jobs\n            SET {msg[\"value\"][\"language\"]} = ?\n            WHERE uid = ?\n            \"\"\",\n            (msg[\"value\"][\"translated\"], msg[\"uid\"])\n        )\n        self.conn.commit()\n\n\n    def isDone(self, uid):\n        cur = self.conn.execute(\n            \"\"\"\n            SELECT\n                french  IS NOT NULL\n            AND spanish IS NOT NULL\n            AND tamil   IS NOT NULL\n            FROM translation_jobs\n            WHERE uid = ?\n            \"\"\",\n            (uid,)\n        )\n        row = cur.fetchone()\n        return bool(row[0]) if row else False\n    \n    def setDone(self, uid):\n        self.conn.execute(\n            \"\"\"\n            UPDATE translation_jobs\n            SET done = 1\n            WHERE uid = ?\n            \"\"\",\n            (uid,)\n        )\n        self.conn.commit()\n\n\n    "
+                }
+            ]
+        }
+    ]
+}

package/src/stories/data/Mapping/TransactionDB_mapping.json

@@ -0,0 +1,149 @@
+[
+    {
+        "type": "Import",
+        "uid": "41375952-c2b0-4d81-9833-f7e38afe5130",
+        "start_line": 1,
+        "end_line": 1,
+        "source": "import sqlite3"
+    },
+    {
+        "type": "Import",
+        "uid": "a8ceb80a-390d-4d0e-a3b4-db9f299a067b",
+        "start_line": 2,
+        "end_line": 2,
+        "source": "import atexit"
+    },
+    {
+        "type": "Import",
+        "uid": "eccfad0c-1487-4d39-84e2-24107925be66",
+        "start_line": 3,
+        "end_line": 3,
+        "source": "import uuid"
+    },
+    {
+        "type": "Expr",
+        "uid": "b13d053d-b448-42cf-adce-2720d7026c28",
+        "start_line": 8,
+        "end_line": 8,
+        "source": "atexit.register(self.cleanup_function)"
+    },
+    {
+        "type": "Expr",
+        "uid": "d735f8a9-b2d9-4af7-b8d0-021bbb8a8945",
+        "start_line": 9,
+        "end_line": 9,
+        "source": "self.initializeDB()"
+    },
+    {
+        "type": "Expr",
+        "uid": "f96705c3-5afb-4ad2-ad3b-66e0f8c1fa44",
+        "start_line": 12,
+        "end_line": 12,
+        "source": "self.conn.close()"
+    },
+    {
+        "type": "Assign",
+        "uid": "3597725d-8fba-486f-9788-4f346ba91be9",
+        "start_line": 16,
+        "end_line": 16,
+        "source": "self.conn = sqlite3.connect(\"translations.db\")"
+    },
+    {
+        "type": "Assign",
+        "uid": "8e34543e-d1a8-4f0b-9945-9fc61b192e88",
+        "start_line": 17,
+        "end_line": 17,
+        "source": "self.cursor = self.conn.cursor()"
+    },
+    {
+        "type": "Expr",
+        "uid": "3071e179-638f-428c-9afc-e7c92997cec8",
+        "start_line": 19,
+        "end_line": 29,
+        "source": "self.cursor.execute(\"\"\"\n        CREATE TABLE IF NOT EXISTS translation_jobs (\n            job_id   INTEGER PRIMARY KEY AUTOINCREMENT,\n            uid      TEXT NOT NULL UNIQUE,\n            english  TEXT NOT NULL,\n            french   TEXT,\n            spanish  TEXT,\n            tamil    TEXT,\n            done     INTEGER NOT NULL DEFAULT 0\n        );\n        \"\"\")"
+    },
+    {
+        "type": "Expr",
+        "uid": "053a69cc-9959-4011-83b9-9cd01948d560",
+        "start_line": 30,
+        "end_line": 30,
+        "source": "self.conn.commit()"
+    },
+    {
+        "type": "Assign",
+        "uid": "c0c88646-99fc-45de-b441-1efbfb617467",
+        "start_line": 33,
+        "end_line": 33,
+        "source": "uid = str(uuid.uuid4())"
+    },
+    {
+        "type": "Expr",
+        "uid": "a0dc4483-bf8d-4790-b6ee-11d4c3ff6747",
+        "start_line": 34,
+        "end_line": 40,
+        "source": "self.conn.execute(\n            \"\"\"\n            INSERT INTO translation_jobs (uid, english)\n            VALUES (?, ?)\n            \"\"\",\n            (uid, data)\n        )"
+    },
+    {
+        "type": "Expr",
+        "uid": "5daf0842-9da2-473f-906f-177b2297e42a",
+        "start_line": 41,
+        "end_line": 41,
+        "source": "self.conn.commit()"
+    },
+    {
+        "type": "Return",
+        "uid": "e5507376-e78c-49a8-953f-a50f7d534e91",
+        "start_line": 43,
+        "end_line": 46,
+        "source": "return {\n            \"uid\": uid,\n            \"value\": {\"data\": data}\n        }"
+    },
+    {
+        "type": "Expr",
+        "uid": "5c91cc00-0bab-45f4-8c44-52b9cbba0033",
+        "start_line": 49,
+        "end_line": 56,
+        "source": "self.conn.execute(\n            f\"\"\"\n            UPDATE translation_jobs\n            SET {msg[\"value\"][\"language\"]} = ?\n            WHERE uid = ?\n            \"\"\",\n            (msg[\"value\"][\"translated\"], msg[\"uid\"])\n        )"
+    },
+    {
+        "type": "Expr",
+        "uid": "50dbde63-e2af-4788-bba3-72ce9aa6cd4b",
+        "start_line": 57,
+        "end_line": 57,
+        "source": "self.conn.commit()"
+    },
+    {
+        "type": "Assign",
+        "uid": "709389e7-985b-433e-b2cb-e77d5a4b8b54",
+        "start_line": 61,
+        "end_line": 71,
+        "source": "cur = self.conn.execute(\n            \"\"\"\n            SELECT\n                french  IS NOT NULL\n            AND spanish IS NOT NULL\n            AND tamil   IS NOT NULL\n            FROM translation_jobs\n            WHERE uid = ?\n            \"\"\",\n            (uid,)\n        )"
+    },
+    {
+        "type": "Assign",
+        "uid": "31c90fb9-89a3-45d9-a2cf-ab0941b26668",
+        "start_line": 72,
+        "end_line": 72,
+        "source": "row = cur.fetchone()"
+    },
+    {
+        "type": "Return",
+        "uid": "2273d6fc-aa36-4abf-9f98-8259a9d3fd0b",
+        "start_line": 73,
+        "end_line": 73,
+        "source": "return bool(row[0]) if row else False"
+    },
+    {
+        "type": "Expr",
+        "uid": "695d4481-ba24-458d-80db-0e600296cf2b",
+        "start_line": 76,
+        "end_line": 83,
+        "source": "self.conn.execute(\n            \"\"\"\n            UPDATE translation_jobs\n            SET done = 1\n            WHERE uid = ?\n            \"\"\",\n            (uid,)\n        )"
+    },
+    {
+        "type": "Expr",
+        "uid": "7d6571df-d51f-461a-8a2f-59ce793a3444",
+        "start_line": 84,
+        "end_line": 84,
+        "source": "self.conn.commit()"
+    }
+]