setta 0.0.1.dev0__py3-none-any.whl → 0.0.2__py3-none-any.whl

setta/static/frontend/assets/logo/logo.svg

@@ -0,0 +1,8 @@
+<svg width="237" height="99" viewBox="0 0 237 99" fill="none" xmlns="http://www.w3.org/2000/svg">
+<path d="M177.36 41.3772C177.36 35.647 178.426 30.6165 180.558 26.2855C182.757 21.9546 185.722 18.6231 189.453 16.291C193.184 13.959 197.349 12.793 201.946 12.793C205.877 12.793 209.309 13.5925 212.241 15.1916C215.239 16.7908 217.538 18.8896 219.137 21.4882V13.5925H236.227V69.3618H219.137V61.4661C217.471 64.0647 215.139 66.1636 212.141 67.7627C209.209 69.3618 205.778 70.1613 201.846 70.1613C197.316 70.1613 193.184 68.9953 189.453 66.6633C185.722 64.2646 182.757 60.8998 180.558 56.5688C178.426 52.1713 177.36 47.1074 177.36 41.3772ZM219.137 41.4772C219.137 37.2128 217.937 33.848 215.539 31.3827C213.207 28.9174 210.342 27.6848 206.944 27.6848C203.545 27.6848 200.647 28.9174 198.248 31.3827C195.916 33.7814 194.75 37.1129 194.75 41.3772C194.75 45.6415 195.916 49.0397 198.248 51.5716C200.647 54.0369 203.545 55.2696 206.944 55.2696C210.342 55.2696 213.207 54.0369 215.539 51.5716C217.937 49.1063 219.137 45.7415 219.137 41.4772Z" fill="#4CE2FF"/>
+<path d="M176.827 54.8697V69.3618H168.131C161.935 69.3618 157.104 67.8626 153.639 64.8642C150.175 61.7993 148.442 56.8353 148.442 49.9724V27.7847H141.646V13.5925H148.442V0H165.533V13.5925H176.727V27.7847H165.533V50.1723C165.533 51.8381 165.933 53.0374 166.732 53.7704C167.532 54.5033 168.864 54.8697 170.73 54.8697H176.827Z" fill="#4CE2FF"/>
+<path d="M140.222 54.8697V69.3618H131.527C125.33 69.3618 120.499 67.8626 117.035 64.8642C113.57 61.7993 111.838 56.8353 111.838 49.9724V27.7847H105.041V13.5925H111.838V0H128.928V13.5925H140.122V27.7847H128.928V50.1723C128.928 51.8381 129.328 53.0374 130.127 53.7704C130.927 54.5033 132.26 54.8697 134.125 54.8697H140.222Z" fill="#4CE2FF"/>
+<path d="M104.916 40.5777C104.916 42.1768 104.816 43.8425 104.616 45.5749H65.9377C66.2042 49.0397 67.3036 51.7049 69.2358 53.5705C71.2347 55.3695 73.6667 56.269 76.5318 56.269C80.7961 56.269 83.7612 54.47 85.4269 50.872H103.617C102.684 54.5366 100.985 57.8348 98.5197 60.7665C96.121 63.6982 93.0894 65.997 89.4247 67.6627C85.7601 69.3285 81.6623 70.1613 77.1315 70.1613C71.6678 70.1613 66.8038 68.9953 62.5395 66.6633C58.2752 64.3312 54.9437 60.9997 52.545 56.6688C50.1464 52.3378 48.947 47.274 48.947 41.4772C48.947 35.6804 50.113 30.6165 52.4451 26.2855C54.8438 21.9546 58.1753 18.6231 62.4396 16.291C66.7039 13.959 71.6012 12.793 77.1315 12.793C82.5285 12.793 87.3259 13.9257 91.5235 16.1911C95.7212 18.4565 98.9861 21.6881 101.318 25.8858C103.717 30.0834 104.916 34.9807 104.916 40.5777ZM87.4258 36.0801C87.4258 33.1484 86.4264 30.8164 84.4275 29.084C82.4286 27.3516 79.9299 26.4854 76.9316 26.4854C74.0665 26.4854 71.6345 27.3183 69.6356 28.984C67.7033 30.6498 66.504 33.0152 66.0376 36.0801H87.4258Z" fill="#4CE2FF"/>
+<path d="M25.6858 70.1613C20.8219 70.1613 16.4909 69.3285 12.693 67.6627C8.8951 65.997 5.89675 63.7316 3.69796 60.8665C1.49917 57.9348 0.26652 54.6699 0 51.0719H16.8907C17.0906 53.0041 17.9901 54.5699 19.5892 55.7693C21.1883 56.9686 23.1539 57.5683 25.4859 57.5683C27.6181 57.5683 29.2505 57.1685 30.3832 56.3689C31.5826 55.5028 32.1823 54.4034 32.1823 53.0708C32.1823 51.4716 31.3494 50.3056 29.6836 49.5727C28.0179 48.7731 25.3194 47.9069 21.5881 46.9741C17.5903 46.0413 14.2588 45.0752 11.5936 44.0757C8.92841 43.0096 6.62968 41.3772 4.69741 39.1784C2.76514 36.913 1.79901 33.8813 1.79901 30.0834C1.79901 26.8852 2.6652 23.9868 4.39758 21.3882C6.19658 18.723 8.79515 16.6242 12.1933 15.0917C15.658 13.5592 19.7558 12.793 24.4865 12.793C31.4826 12.793 36.9796 14.5253 40.9774 17.9901C45.0418 21.4549 47.3739 26.0523 47.9735 31.7825H32.1823C31.9157 29.8502 31.0495 28.3177 29.5837 27.185C28.1845 26.0523 26.3188 25.486 23.9868 25.486C21.9879 25.486 20.4554 25.8858 19.3893 26.6853C18.3232 27.4182 17.7902 28.451 17.7902 29.7836C17.7902 31.3827 18.6231 32.5821 20.2888 33.3816C22.0212 34.1812 24.6864 34.9807 28.2844 35.7803C32.4155 36.8464 35.7803 37.9125 38.3788 38.9785C40.9774 39.978 43.2428 41.6437 45.1751 43.9758C47.174 46.2412 48.2068 49.3062 48.2734 53.1707C48.2734 56.4356 47.3406 59.3673 45.4749 61.9659C43.6759 64.4978 41.044 66.4967 37.5793 67.9626C34.1812 69.4284 30.2167 70.1613 25.6858 70.1613Z" fill="#4CE2FF"/>
+<rect y="86" width="48" height="13" fill="#4CE2FF"/>
+</svg>

setta/static/frontend/assets/pen-455d7d8a.svg

@@ -0,0 +1,19 @@
+<svg
+  width="24"
+  height="24"
+  viewBox="0 0 24 24"
+  fill="white"
+  style="fill:white; filter: drop-shadow(0px 3px 3px rgba(0, 0, 0, 0.4));"
+  xmlns="http://www.w3.org/2000/svg"
+>
+  <path
+    fill-rule="evenodd"
+    clip-rule="evenodd"
+    d="M21.2635 2.29289C20.873 1.90237 20.2398 1.90237 19.8493 2.29289L18.9769 3.16525C17.8618 2.63254 16.4857 2.82801 15.5621 3.75165L4.95549 14.3582L10.6123 20.0151L21.2189 9.4085C22.1426 8.48486 22.338 7.1088 21.8053 5.99367L22.6777 5.12132C23.0682 4.7308 23.0682 4.09763 22.6777 3.70711L21.2635 2.29289ZM16.9955 10.8035L10.6123 17.1867L7.78392 14.3582L14.1671 7.9751L16.9955 10.8035ZM18.8138 8.98525L19.8047 7.99429C20.1953 7.60376 20.1953 6.9706 19.8047 6.58007L18.3905 5.16586C18 4.77534 17.3668 4.77534 16.9763 5.16586L15.9853 6.15683L18.8138 8.98525Z"
+    fill="white"
+  />
+  <path
+    d="M2 22.9502L4.12171 15.1717L9.77817 20.8289L2 22.9502Z"
+    fill="white"
+  />
+</svg>

setta/static/frontend/assets/web-vitals-44a8e082.js

@@ -0,0 +1 @@
+var m,l,C,h,f=function(n,e){return{name:n,value:e===void 0?-1:e,delta:0,entries:[],id:"v2-".concat(Date.now(),"-").concat(Math.floor(8999999999999*Math.random())+1e12)}},T=function(n,e){try{if(PerformanceObserver.supportedEntryTypes.includes(n)){if(n==="first-input"&&!("PerformanceEventTiming"in self))return;var i=new PerformanceObserver(function(r){return r.getEntries().map(e)});return i.observe({type:n,buffered:!0}),i}}catch{}},y=function(n,e){var i=function r(t){t.type!=="pagehide"&&document.visibilityState!=="hidden"||(n(t),e&&(removeEventListener("visibilitychange",r,!0),removeEventListener("pagehide",r,!0)))};addEventListener("visibilitychange",i,!0),addEventListener("pagehide",i,!0)},g=function(n){addEventListener("pageshow",function(e){e.persisted&&n(e)},!0)},v=function(n,e,i){var r;return function(t){e.value>=0&&(t||i)&&(e.delta=e.value-(r||0),(e.delta||r===void 0)&&(r=e.value,n(e)))}},d=-1,w=function(){return document.visibilityState==="hidden"?0:1/0},F=function(){y(function(n){var e=n.timeStamp;d=e},!0)},L=function(){return d<0&&(d=w(),F(),g(function(){setTimeout(function(){d=w(),F()},0)})),{get firstHiddenTime(){return d}}},A=function(n,e){var i,r=L(),t=f("FCP"),o=function(c){c.name==="first-contentful-paint"&&(u&&u.disconnect(),c.startTime<r.firstHiddenTime&&(t.value=c.startTime,t.entries.push(c),i(!0)))},a=performance.getEntriesByName&&performance.getEntriesByName("first-contentful-paint")[0],u=a?null:T("paint",o);(a||u)&&(i=v(n,t,e),a&&o(a),g(function(c){t=f("FCP"),i=v(n,t,e),requestAnimationFrame(function(){requestAnimationFrame(function(){t.value=performance.now()-c.timeStamp,i(!0)})})}))},b=!1,E=-1,R=function(n,e){b||(A(function(s){E=s.value}),b=!0);var i,r=function(s){E>-1&&n(s)},t=f("CLS",0),o=0,a=[],u=function(s){if(!s.hadRecentInput){var B=a[0],q=a[a.length-1];o&&s.startTime-q.startTime<1e3&&s.startTime-B.startTime<5e3?(o+=s.value,a.push(s)):(o=s.value,a=[s]),o>t.value&&(t.value=o,t.entries=a,i())}},c=T("layout-shift",u);c&&(i=v(r,t,e),y(function(){c.takeRecords().map(u),i(!0)}),g(function(){o=0,E=-1,t=f("CLS",0),i=v(r,t,e)}))},p={passive:!0,capture:!0},H=new Date,P=function(n,e){m||(m=e,l=n,C=new Date,k(removeEventListener),D())},D=function(){if(l>=0&&l<C-H){var n={entryType:"first-input",name:m.type,target:m.target,cancelable:m.cancelable,startTime:m.timeStamp,processingStart:m.timeStamp+l};h.forEach(function(e){e(n)}),h=[]}},I=function(n){if(n.cancelable){var e=(n.timeStamp>1e12?new Date:performance.now())-n.timeStamp;n.type=="pointerdown"?function(i,r){var t=function(){P(i,r),a()},o=function(){a()},a=function(){removeEventListener("pointerup",t,p),removeEventListener("pointercancel",o,p)};addEventListener("pointerup",t,p),addEventListener("pointercancel",o,p)}(e,n):P(e,n)}},k=function(n){["mousedown","keydown","touchstart","pointerdown"].forEach(function(e){return n(e,I,p)})},M=function(n,e){var i,r=L(),t=f("FID"),o=function(u){u.startTime<r.firstHiddenTime&&(t.value=u.processingStart-u.startTime,t.entries.push(u),i(!0))},a=T("first-input",o);i=v(n,t,e),a&&y(function(){a.takeRecords().map(o),a.disconnect()},!0),a&&g(function(){var u;t=f("FID"),i=v(n,t,e),h=[],l=-1,m=null,k(addEventListener),u=o,h.push(u),D()})},S=new Set,N=function(n,e){var i,r=L(),t=f("LCP"),o=function(c){var s=c.startTime;s<r.firstHiddenTime&&(t.value=s,t.entries.push(c)),i()},a=T("largest-contentful-paint",o);if(a){i=v(n,t,e);var u=function(){S.has(t.id)||(a.takeRecords().map(o),a.disconnect(),S.add(t.id),i(!0))};["keydown","click"].forEach(function(c){addEventListener(c,u,{once:!0,capture:!0})}),y(u,!0),g(function(c){t=f("LCP"),i=v(n,t,e),requestAnimationFrame(function(){requestAnimationFrame(function(){t.value=performance.now()-c.timeStamp,S.add(t.id),i(!0)})})})}},O=function(n){var e,i=f("TTFB");e=function(){try{var r=performance.getEntriesByType("navigation")[0]||function(){var t=performance.timing,o={entryType:"navigation",startTime:0};for(var a in t)a!=="navigationStart"&&a!=="toJSON"&&(o[a]=Math.max(t[a]-t.navigationStart,0));return o}();if(i.value=i.delta=r.responseStart,i.value<0)return;i.entries=[r],n(i)}catch{}},document.readyState==="complete"?setTimeout(e,0):addEventListener("pageshow",e)};export{R as getCLS,A as getFCP,M as getFID,N as getLCP,O as getTTFB};

setta/static/frontend/browserconfig.xml

@@ -0,0 +1,9 @@
+<?xml version="1.0" encoding="utf-8"?>
+<browserconfig>
+    <msapplication>
+        <tile>
+            <square150x150logo src="/mstile-150x150.png"/>
+            <TileColor>#da532c</TileColor>
+        </tile>
+    </msapplication>
+</browserconfig>

setta/static/frontend/index.html

@@ -0,0 +1,30 @@
+<!doctype html>
+<!-- Need overscroll-x-none: https://github.com/xyflow/xyflow/discussions/3379#discussioncomment-10171429 -->
+<html lang="en" class="overscroll-x-none">
+  <head>
+    <meta charset="utf-8" />
+    <link rel="apple-touch-icon" sizes="180x180" href="/static/apple-touch-icon.png" />
+    <link rel="icon" type="image/png" sizes="32x32" href="/static/favicon-32x32.png" />
+    <link rel="icon" type="image/png" sizes="16x16" href="/static/favicon-16x16.png" />
+    <link rel="manifest" href="/static/site.webmanifest" />
+    <link rel="mask-icon" href="/static/safari-pinned-tab.svg" color="#00b5d8" />
+    <meta name="msapplication-TileColor" content="#da532c" />
+    <meta name="theme-color" content="#00b5d8" />
+    <meta name="viewport" content="width=device-width, initial-scale=1" />
+    <meta name="description" content="setta" />
+    
+
+    <title>setta.dev</title>
+    <script type="module" crossorigin src="/static/assets/index-ee99dc72.js"></script>
+    <link rel="stylesheet" href="/static/assets/index-1d4b4ecf.css">
+  </head>
+  <body>
+    <noscript>You need to enable JavaScript to run this app.</noscript>
+    <main
+      id="root"
+      class="flex h-screen w-screen flex-col items-center bg-setta-light font-sans dark:bg-setta-dark"
+    ></main>
+
+    
+  </body>
+</html>

setta/static/frontend/manifest.json

@@ -0,0 +1,25 @@
+{
+  "short_name": "React App",
+  "name": "Create React App Sample",
+  "icons": [
+    {
+      "src": "favicon.ico",
+      "sizes": "64x64 32x32 24x24 16x16",
+      "type": "image/x-icon"
+    },
+    {
+      "src": "logo192.png",
+      "type": "image/png",
+      "sizes": "192x192"
+    },
+    {
+      "src": "logo512.png",
+      "type": "image/png",
+      "sizes": "512x512"
+    }
+  ],
+  "start_url": ".",
+  "display": "standalone",
+  "theme_color": "#000000",
+  "background_color": "#ffffff"
+}

setta/static/frontend/robots.txt

@@ -0,0 +1,3 @@
+# https://www.robotstxt.org/robotstxt.html
+User-agent: *
+Disallow:

setta/static/frontend/safari-pinned-tab.svg

@@ -0,0 +1,18 @@
+<?xml version="1.0" standalone="no"?>
+<!DOCTYPE svg PUBLIC "-//W3C//DTD SVG 20010904//EN"
+ "http://www.w3.org/TR/2001/REC-SVG-20010904/DTD/svg10.dtd">
+<svg version="1.0" xmlns="http://www.w3.org/2000/svg"
+ width="700.000000pt" height="700.000000pt" viewBox="0 0 700.000000 700.000000"
+ preserveAspectRatio="xMidYMid meet">
+<metadata>
+Created by potrace 1.14, written by Peter Selinger 2001-2017
+</metadata>
+<g transform="translate(0.000000,700.000000) scale(0.100000,-0.100000)"
+fill="#000000" stroke="none">
+<path d="M690 6986 c-344 -64 -607 -325 -676 -671 -11 -59 -13 -515 -11 -2840
+1 -1524 5 -2779 9 -2790 3 -11 9 -33 13 -50 58 -251 256 -477 506 -576 59 -23
+56 -22 149 -43 57 -13 427 -15 2840 -13 1526 1 2787 6 2801 10 204 56 303 109
+426 228 107 103 182 232 225 385 l23 79 2 2770 c2 2963 5 2823 -45 2965 -92
+263 -334 481 -593 536 -13 2 -44 9 -69 14 -72 15 -5519 11 -5600 -4z"/>
+</g>
+</svg>

setta/static/frontend/site.webmanifest

@@ -0,0 +1,19 @@
+{
+    "name": "",
+    "short_name": "",
+    "icons": [
+        {
+            "src": "/android-chrome-192x192.png",
+            "sizes": "192x192",
+            "type": "image/png"
+        },
+        {
+            "src": "/android-chrome-512x512.png",
+            "sizes": "512x512",
+            "type": "image/png"
+        }
+    ],
+    "theme_color": "#ffffff",
+    "background_color": "#ffffff",
+    "display": "standalone"
+}

setta/tasks/fns/__init__.py

@@ -0,0 +1,9 @@
+from .codeAreaAutocomplete import codeAreaAutocomplete
+from .codeAreaFindTemplateVars import codeAreaFindTemplateVars
+from .codeAreaInitializeCode import codeAreaInitializeCode
+from .findEVRefs import findEVRefs
+from .parametersRequest import parametersRequest
+from .textFieldAutocomplete import textFieldCompletion
+from .textFieldInitializeCode import textFieldInitializeCode
+from .typeCheck import typeCheck
+from .utils import SettaInMemoryFn, TaskDefinition

setta/tasks/fns/codeAreaAutocomplete.py

@@ -0,0 +1,209 @@
+from typing import Any, Optional
+
+from pydantic import BaseModel
+
+from setta.code_gen.create_runnable_scripts import generate_final_code_for_section
+from setta.code_gen.export_selected import get_gen_code_template_var
+from setta.code_gen.find_placeholders import tp
+from setta.code_gen.python.check_scope import are_positions_in_scope_with_variable
+from setta.code_gen.python.generate_code import get_chars_in_line_before_position
+from setta.code_gen.python.position_line_col import position_to_line_col
+from setta.tasks.fns.codeAreaFindTemplateVars import findTemplateVarsLocally
+from setta.tasks.fns.findEVRefs import findEVRefsLocally
+from setta.tasks.task_runner import RunType
+from setta.utils.constants import C
+
+from .codeAreaInitializeCode import get_cache_key
+from .utils import (
+    TaskDefinition,
+    prepare_code_for_scope_checking,
+    send_document_open_request_with_diagnostics,
+)
+
+
+class CodeAreaAutocompleteContent(BaseModel):
+    projectConfigId: str
+    projectConfigName: str
+    codeLanguage: str
+    sectionId: str
+    sectionName: str
+    fullText: str
+    position: int
+    candidateEVRefs: list
+    candidateTemplateVars: list
+    cached: Optional[Any] = None
+
+
+async def _codeAreaAutocomplete(message, lsp_writers):
+    request_id = message.id
+    message = CodeAreaAutocompleteContent.parse_obj(message.content)
+
+    (
+        template_var_replacement_values,
+        generated_var_names,
+        referencable_var_names,
+        exporter_obj,
+        fullNameToInfo,
+        fullNameToSectionId,
+    ) = message.cached
+
+    templateVars = await findTemplateVarsLocally(
+        message.sectionId,
+        message.candidateTemplateVars,
+        message.fullText,
+        fullNameToSectionId,
+        lsp_writers,
+    )
+
+    gen_code_template_var = get_gen_code_template_var(templateVars)
+    if (
+        gen_code_template_var
+        and exporter_obj.code_gen_template_var_section_details["section"].get("id")
+        != message.sectionId
+    ):
+        return {"content": {"data": C.CODE_AREA_GEN_CODE_NOT_INITIALIZED}}
+
+    evRefs = await findEVRefsLocally(
+        message.sectionId,
+        None,
+        message.candidateEVRefs,
+        templateVars,
+        True,
+        message.fullText,
+        fullNameToInfo,
+        lsp_writers,
+    )
+
+    (
+        code,
+        _,
+        _,
+        ref_template_var_positions,
+    ) = await generate_final_code_for_section(
+        sectionId=message.sectionId,
+        code=message.fullText,
+        codeLanguage=message.codeLanguage,
+        evRefs=evRefs,
+        templateVars=templateVars,
+        exporter_obj=exporter_obj,
+        lsp_writers=lsp_writers,
+        folder_path=lsp_writers["full"].project_code_folder(message.projectConfigName),
+        cursor_position=message.position,
+        template_var_replacement_values=template_var_replacement_values,
+    )
+
+    cursor_position = ref_template_var_positions["cursor"]
+    line, column = position_to_line_col(code, cursor_position)
+
+    generated_var_names, referencable_var_names = calculate_referencable_var_names(
+        message.fullText,  # use original code
+        templateVars,  # use original template vars
+        message.position,  # use original cursor position
+        generated_var_names,
+        referencable_var_names,
+    )
+
+    code_id = await send_document_open_request_with_diagnostics(
+        lsp_writers["full"],
+        code,
+        exporter_obj,
+        ref_template_var_positions,
+        message.projectConfigName,
+        message.sectionName,
+        message.sectionId,
+    )
+
+    await lsp_writers["full"].send_autocomplete_request(
+        line,
+        column,
+        code_id=code_id,
+        request_id=request_id,
+        referencable_var_names=referencable_var_names,
+        generated_var_names=generated_var_names,
+        cursor_offset=cursor_position - message.position,
+    )
+
+    return None
+
+
+def calculate_referencable_var_names(
+    code,
+    original_template_vars,
+    original_cursor_position,
+    generated_var_names,
+    referencable_var_names,
+):
+    gen_code_template_var = get_gen_code_template_var(original_template_vars)
+    gen_code_position = (
+        gen_code_template_var["startPos"] if gen_code_template_var else None
+    )
+
+    if gen_code_position is None or original_cursor_position <= gen_code_position:
+        return [], special_keywords_for_autocomplete(gen_code_template_var)
+
+    code, the_special_var_name = prepare_code_for_scope_checking(
+        code, original_template_vars, gen_code_template_var
+    )
+
+    code = add_placeholder_to_empty_line_for_scope_checking(
+        code, original_cursor_position
+    )
+
+    response = are_positions_in_scope_with_variable(
+        code, [original_cursor_position], the_special_var_name
+    )
+
+    if response[0]:
+        referencable_var_names = [
+            *special_keywords_for_autocomplete(gen_code_template_var),
+            *referencable_var_names,
+        ]
+        return generated_var_names, referencable_var_names
+
+    return [], []
+
+
+# The are_positions_in_scope_with_variable function doesn't work with empty lines
+# So if the cursor is on an empty line, we add a dummy _
+def add_placeholder_to_empty_line_for_scope_checking(code, cursor_position):
+    is_empty_line = False
+    before_chars = get_chars_in_line_before_position(code, cursor_position)
+
+    if len(before_chars.strip()) == 0:
+        line_end_index = code.find("\n", cursor_position)
+        if line_end_index == -1:
+            # Check remaining characters if we're at the last line
+            after_chars = code[cursor_position:]
+        else:
+            after_chars = code[cursor_position:line_end_index]
+
+        if len(after_chars.strip()) == 0:
+            is_empty_line = True
+
+    if is_empty_line:
+        code = code[:cursor_position] + "_" + code[cursor_position:]
+
+    return code
+
+
+def special_keywords_for_autocomplete(gen_code_template_var):
+    if not gen_code_template_var:
+        return [tp(C.SETTA_GENERATED_PYTHON)]
+    return []
+
+
+def read_cache(cache, message):
+    message = message.content
+    message["cached"] = cache.get(
+        get_cache_key(message["projectConfigId"], message["sectionId"])
+    )
+
+
+codeAreaAutocomplete = TaskDefinition(
+    name="codeAreaAutocomplete",
+    return_message_type="codeAreaAutocompleteReturn",
+    fn=_codeAreaAutocomplete,
+    read_cache=read_cache,
+    run_as=RunType.NONE,
+    server_dependencies=["lsp_writers"],
+)

setta/tasks/fns/codeAreaFindTemplateVars.py

@@ -0,0 +1,128 @@
+import asyncio
+from typing import List
+
+from pydantic import BaseModel
+
+from setta.code_gen.python.position_line_col import position_to_line_col
+from setta.database.utils import create_new_id
+from setta.tasks.task_runner import RunType
+
+from .utils import TaskDefinition, replace_template_vars_with_random_names
+
+
+class CodeAreaFindTemplateVarsContent(BaseModel):
+    sectionId: str
+    candidateTemplateVars: List[list]
+    fullText: str
+
+
+async def _codeAreaFindTemplateVars(message, lsp_writers):
+    message_id = message.id
+    message = CodeAreaFindTemplateVarsContent.parse_obj(message.content)
+    request_ids = await sendTemplateVarRequests(
+        message_id,
+        message.sectionId,
+        message.candidateTemplateVars,
+        message.fullText,
+        lsp_writers,
+        local=False,
+    )
+    if len(request_ids) == 0:
+        return {"content": {"data": {"positions": [], "numMessages": 1}}}
+
+    return None
+
+
+async def sendTemplateVarRequests(
+    message_id, sectionId, candidateTemplateVars, fullText, lsp_writers, local
+):
+    if len(candidateTemplateVars) == 0:
+        return []
+
+    code, positions_to_test, _ = replace_template_vars_with_random_names(
+        candidateTemplateVars, fullText
+    )
+    code_id = f"{sectionId}-find-template-vars"
+
+    await lsp_writers["basic"].send_document_open_request(code_id, code)
+    num_messages = len(positions_to_test)
+    request_ids = []
+    for idx in range(num_messages):
+        startPos = positions_to_test[idx]
+        line, col = position_to_line_col(code, startPos)
+        request_id = f"{message_id}-{idx}"
+        request_ids.append(request_id)
+        lsp_writers["basic"].create_asyncio_task(
+            lsp_writers["basic"].send_document_highlight_request(
+                line=line,
+                character=col,
+                code_id=code_id,
+                request_id=request_id,
+                return_id=message_id,
+                num_messages=num_messages,
+            ),
+            request_id,
+            local=local,
+        )
+
+    return request_ids
+
+
+async def findTemplateVarsLocally(
+    sectionId,
+    candidateTemplateVars,
+    fullText,
+    fullNameToSectionId,
+    lsp_writers,
+):
+    templateVarRequestId = create_new_id()
+    request_ids = await sendTemplateVarRequests(
+        templateVarRequestId,
+        sectionId,
+        candidateTemplateVars,
+        fullText,
+        lsp_writers,
+        local=True,
+    )
+
+    if len(request_ids) == 0:
+        return []
+
+    responses = await waitForTemplateVarResponses(request_ids, lsp_writers)
+    positions = []
+    for r in responses:
+        positions.extend(r["data"]["positions"])
+    return construct_template_vars(positions, fullNameToSectionId, fullText)
+
+
+def construct_template_vars(keyword_positions, fullNameToSectionId, fullText):
+    output = []
+    for k in keyword_positions:
+        keyword = fullText[k["startPos"] : k["endPos"]]
+        if keyword in fullNameToSectionId:
+            output.append(
+                {
+                    "sectionId": fullNameToSectionId[keyword],
+                    "startPos": k["startPos"],
+                    "keyword": keyword,
+                }
+            )
+    return output
+
+
+async def waitForTemplateVarResponses(request_ids, lsp_writers):
+    request_futures = []
+    for r in request_ids:
+        future = lsp_writers["basic"].wait_for_response(r)
+        request_futures.append(future)
+
+    return await asyncio.gather(*request_futures)
+
+
+codeAreaFindTemplateVars = TaskDefinition(
+    name="codeAreaFindTemplateVars",
+    return_message_type="codeAreaFindTemplateVarsReturn",
+    fn=_codeAreaFindTemplateVars,
+    run_as=RunType.NONE,
+    server_dependencies=["lsp_writers"],
+)