o c:@s\dZddlmZddlmZddlmZddlmZddlmZddlmZddlmZdd lm Z dd lm Z dd lm Z dd l m Z dd l mZddl mZddl mZe rdddlmZddlmZe dddZe dddZe dddZGdddeeefZGdddeeje jfZGdddeeje jfZGdddeeje j fZ!d S)!z Dual ParserNode implementation )Any)Generic)Iterable)List)Optional)Set)Tuple)Type) TYPE_CHECKING)TypeVar) apacheparser) assertions) augeasparser) interfaces)ApacheParserNode)AugeasParserNodeGenericAugeasParserNoder)boundGenericApacheParserNoderGenericDualNode DualNodeBasec @seZdZdZdededdfddZdeddfd d Zd ede fd d Z dede dfddZ de ededede de ef ddZdS)rz Dual parser interface for in development testing. This is used as the base class for dual parser interface classes. This class handles runtime attribute value assertions.primary secondaryreturnNcCs||_||_dS)Nrr)selfrrrP/opt/certbot/lib/python3.10/site-packages/certbot_apache/_internal/dualparser.py__init__ s zDualNodeBase.__init__msgcCs|j||j|dS)z Call save for both parsers N)rsaver)rrrrrr %s zDualNodeBase.saveanamecCs@t|j|}t|j|}|dkt|g}t|st|||S)z Attribute value assertion metadata)getattrrrcallableanyr assertEqualSimple)rr!firstval secondval exclusionsrrr __getattr__*s   zDualNodeBase.__getattr__namecC|td|S)zA Traverses the ancestor tree and returns ancestors matching name find_ancestors _find_helper DualBlockNode)rr+rrrr-8szDualNodeBase.find_ancestors nodeclassfindfuncsearchkwargscKst|j||fi|}t|j||fi|}t|}t|}g} |r7|r7| ||d|dd| S|rK|D] } | ||d| dq;| S|r_|D] } | || |ddqO| St|t|ksiJ|||} | D] \} } | || | dqq| S)a.A helper for find_* functions. The function specific attributes should be passed as keyword arguments. :param interfaces.ParserNode nodeclass: The node class for results. :param str findfunc: Name of the find function to call :param str search: The search term rr)r#rrr isPassNodeListappendlen_create_matching_list)rr1r2r3r4 primary_res secondary_res pass_primarypass_secondary new_nodescmatchespsrrrr/<s6          zDualNodeBase._find_helper)__name__ __module__ __qualname____doc__rrrstrr rr*rr-r rr/rrrrrs cs*eZdZdZdeddffdd ZZS)DualCommentNodez5 Dual parser implementation of CommentNode interface r4rNc |dd|dd|d}|d}|s|r(|r|s Jt||nttjdi|tjdi|t |j |j dS)a< This initialization implementation allows ordinary initialization of CommentNode objects as well as creating a DualCommentNode object using precreated or fetched CommentNode objects if provided as optional arguments primary and secondary. Parameters other than the following are from interfaces.CommentNode: :param CommentNode primary: Primary pre-created CommentNode, mainly used when creating new DualParser nodes using add_* methods. :param CommentNode secondary: Secondary pre-created CommentNode rNrr) setdefaultpopsuperrrAugeasCommentNoder ApacheCommentNoder assertEqualrrrr4rr __class__rrri    zDualCommentNode.__init__)rBrCrDrErr __classcell__rrrPrrGesrGcsJeZdZUdZeed<deddffdd Zdeeddfdd Z Z S) DualDirectiveNodez7 Dual parser implementation of DirectiveNode interface parametersr4rNc rH)aL This initialization implementation allows ordinary initialization of DirectiveNode objects as well as creating a DualDirectiveNode object using precreated or fetched DirectiveNode objects if provided as optional arguments primary and secondary. Parameters other than the following are from interfaces.DirectiveNode: :param DirectiveNode primary: Primary pre-created DirectiveNode, mainly used when creating new DualParser nodes using add_* methods. :param DirectiveNode secondary: Secondary pre-created DirectiveNode rNrr) rIrJrKrrAugeasDirectiveNoder ApacheDirectiveNoder rNrrrOrPrrrrRzDualDirectiveNode.__init__cCs,|j||j|t|j|jdS)zf Sets parameters and asserts that both implementation successfully set the parameter sequence N)rset_parametersrr rN)rrUrrrrXs  z DualDirectiveNode.set_parameters) rBrCrDrErF__annotations__rrrrXrSrrrPrrTs rTc sPeZdZdZdeddffdd Z  d%dedeeed ee ddfd d Z  d%dedeeed ee de fd d Z  d&ded ee de fddZdeejdeejdeeejejffddZd'dedededfddZd'dededee fddZdedee fddZd(dd Zdeefd!d"Zdeefd#d$ZZS))r0z3 Dual parser implementation of BlockNode interface r4rNc rH)a, This initialization implementation allows ordinary initialization of BlockNode objects as well as creating a DualBlockNode object using precreated or fetched BlockNode objects if provided as optional arguments primary and secondary. Parameters other than the following are from interfaces.BlockNode: :param BlockNode primary: Primary pre-created BlockNode, mainly used when creating new DualParser nodes using add_* methods. :param BlockNode secondary: Secondary pre-created BlockNode rNrr) rIrJrKrrAugeasBlockNoder ApacheBlockNoder rNrrrOrPrrrrRzDualBlockNode.__init__r+rUpositioncC8|j|||}|j|||}t||t||dS)z Creates a new child BlockNode, asserts that both implementations did it in a similar way, and returns a newly created DualBlockNode object encapsulating both of the newly created objects r)radd_child_blockrr rNr0rr+rUr\ primary_new secondary_newrrrr^  zDualBlockNode.add_child_blockcCr])z Creates a new child DirectiveNode, asserts that both implementations did it in a similar way, and returns a newly created DualDirectiveNode object encapsulating both of the newly created objects r)radd_child_directiverr rNrTr_rrrrcrbz!DualBlockNode.add_child_directivecommentcCs8|jj||d}|jj||d}t||t||dS)z Creates a new child CommentNode, asserts that both implementations did it in a similar way, and returns a newly created DualCommentNode object encapsulating both of the newly created objects )rer\)r+r\r)radd_child_commentrr rNrG)rrer\r`rarrrrfrbzDualBlockNode.add_child_comment primary_listsecondary_listc Csbg}|D]*}d}|D]}z t|||}Wn ty Yq w|r+|||fqtd|S)a Matches the list of primary_list to a list of secondary_list and returns a list of tuples. This is used to create results for find_ methods. This helper function exists, because we cannot ensure that the list of search results returned by primary.find_* and secondary.find_* are ordered in a same way. The function pairs the same search results from both implementations to a list of tuples. NzCould not find a matching node.)r rNAssertionErrorr6)rrgrhmatchedr@matchrArrrr8s   z#DualBlockNode._create_matching_listTexcludecC|jtd||dS)a Performs a search for BlockNodes using both implementations and does simple checks for results. This is built upon the assumption that unimplemented find_* methods return a list with a single assertion passing object. After the assertion, it creates a list of newly created DualBlockNode instances that encapsulate the pairs of returned BlockNode objects. find_blocksrlr.rr+rlrrrrn  zDualBlockNode.find_blockscCrm)a Performs a search for DirectiveNodes using both implementations and checks the results. This is built upon the assumption that unimplemented find_* methods return a list with a single assertion passing object. After the assertion, it creates a list of newly created DualDirectiveNode instances that encapsulate the pairs of returned DirectiveNode objects. find_directivesro)r/rTrprrrrrrqzDualBlockNode.find_directivescCr,)a Performs a search for CommentNodes using both implementations and checks the results. This is built upon the assumption that unimplemented find_* methods return a list with a single assertion passing object. After the assertion, it creates a list of newly created DualCommentNode instances that encapsulate the pairs of returned CommentNode objects. find_comments)r/rG)rrerrrrs$s zDualBlockNode.find_commentschildcCs |j|j|j|jdS)zDeletes a child from the ParserNode implementations. The actual ParserNode implementations are used here directly in order to be able to match a child to the list of children.N)r delete_childr)rrtrrrru/szDualBlockNode.delete_childcC$|j}|j}t|||S)zQ Fetches the list of unsaved file paths and asserts that the lists match )r unsaved_filesrr r&)r primary_filessecondary_filesrrrrw7s   zDualBlockNode.unsaved_filescCrv)an Returns a list of file paths that have currently been parsed into the parser tree. The returned list may include paths with wildcard characters, for example: ['/etc/apache2/conf.d/*.load'] This is typically called on the root node of the ParserNode tree. :returns: list of file paths of files that have been parsed )r parsed_pathsrr assertEqualPathsList)r primary_pathssecondary_pathsrrrrz@s  zDualBlockNode.parsed_paths)NN)rdN)T)rtr0rN)rBrCrDrErrrFrrintr^rTrcrGrfrr ParserNoderr8boolrnrrrsrurrwrzrSrrrPrr0sF         r0N)"rEtypingrrrrrrrr r r certbot_apache._internalr r rr%certbot_apache._internal.apacheparserr%certbot_apache._internal.augeasparserrrrrrrLrMrGrVrWrTrZr[r0rrrrs>                   J *