o DDi@sddlmZddlmZmZddlmZmZmZm Z ddl m Z m Z ddl mZddlmZeddGd d d Zed dGd d d eZdedefddZGdddeeZdS))contextmanager) dataclassfield)CallableIteratorListOptional)CSTNodeModule) CodegenState)BaseMetadataProviderT)frozenc@s"eZdZUdZeed<eed<dS)CodeSpanz Represents the position of a piece of code by its starting position and length. Note: This class does not specify the unit of distance - it can be bytes, Unicode characters, or something else entirely. startlengthN)__name__ __module__ __qualname____doc__int__annotations__rrV/var/www/Datamplify/venv/lib/python3.10/site-packages/libcst/metadata/span_provider.pyrs  rFc seZdZUeeed<dZeee ge fed<dZ e ed<e e dZee ed<dfd d Zd e ddffd d Zd e ddfddZdeddfddZdeddfddZeddddedeedeededfddZZS)SpanProvidingCodegenStateproviderN get_lengthrposition)default_factory_stackreturncs$t|jD]}||qdSN)superadd_indent_tokens indent_tokens_update_position)selftoken __class__rrr"&s   z+SpanProvidingCodegenState.add_indent_tokensvaluecst|||dSr )r! add_tokenr$)r%r)r'rrr*+s z#SpanProvidingCodegenState.add_tokencCs |jpt}|j||7_dSr )rlenr)r%r)rrrrr$/s z*SpanProvidingCodegenState._update_positionnodecCs|j|jdSr )rappendr)r%r,rrrbefore_codegen3sz(SpanProvidingCodegenState.before_codegencCs<|j}||jjvr|j}t|||d|jj|<dSdSN)r)rpopr _computedrr)r%r,rendrrr after_codegen6s  z'SpanProvidingCodegenState.after_codegen) start_nodeend_noder4r5c cs|j}z7dVW|j}|dur|jj|jn|}|dur,|jj|}|j|j|}n||}t||d|jj|<dS|j}|durJ|jj|jn|}|dur_|jj|}|j|j|}n||}t||d|jj|<wr/)rrr1rrr)r%r,r4r5rr2end_spanrrrrrecord_syntactic_position=s0  z3SpanProvidingCodegenState.record_syntactic_position)rN)rrrr rrrrrstrrrrlistrrr"r*r$r r.r3rrr7 __classcell__rrr'rrs,  rr)rcCst|dS)Nutf8)r+encode)r)rrrbyte_length_in_utf8Wsr=c@s"eZdZdZdeddfddZdS)ByteSpanPositionProvidera Generates offset and length metadata for nodes' positions. For each :class:`CSTNode` this provider generates a :class:`CodeSpan` that contains the byte-offset of the node from the start of the file, and its length (also in bytes). The whitespace owned by the node is not included in this length. Note: offset and length measure bytes, not characters (which is significant for example in the case of Unicode characters encoded in more than one byte) modulerNcCs"t|j|j|td}||dS)N)default_indentdefault_newlinerr)rr@rAr=_codegen)r%r?staterrr _gen_implhsz"ByteSpanPositionProvider._gen_impl)rrrrr rDrrrrr>[s r>N) contextlibr dataclassesrrtypingrrrrlibcstr r libcst._nodes.internalr libcst.metadata.base_providerr rrr8rr=r>rrrrs   7