File size: 4,490 Bytes
08c964e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 | from dataclasses import dataclass
from typing import Union
from .. import *
# 用于按条件查找块内的指令的索引，并支持从某个索引位置开始插入指令
class _IndexBlockTools:
@dataclass
class OpL: # 查找第一个指令
directive: str = ""
offset: int = 0
parameter: str = ""
comment: str = ""
remove_comment: bool = False
class OpR(OpL): # 查找最后一个指令
pass
def __init__(self):
self._block: Union[Block, Http, Config, Upstream] = Block()
def find_index(self, *ops: Union[OpL, OpR], default: int = -1) -> int:
directives = self._block.get_directives()
if self._block.__class__ is Http:
directives = self._block.directives
ops = [op for op in ops if op.directive != "server"]
elif self._block.__class__ is Upstream:
directives = self._block.directives
ops = [op for op in ops if op.directive != "server"]
for op in ops:
target_idx = -1
for i, directive in enumerate(directives):
if op.directive and op.directive == directive.get_name() and (
op.parameter == "" or any(op.parameter in p for p in directive.get_parameters())
):
target_idx = i
if type(op) is self.OpL:
return target_idx + op.offset
elif op.comment:
comments = directive.get_comment()
for c_dix, comment in enumerate(comments[::-1]):
if op.comment in comment:
if op.remove_comment:
comments.pop(0 - c_dix - 1)
directive.set_comment(comments)
target_idx = i
if type(op) is self.OpL:
return target_idx + op.offset
if target_idx >= 0:
return target_idx + op.offset
return default
def insert(self, idx: int, *directives: IDirective):
if idx < 0:
raise ValueError("索引不能小于0")
cls = type(self._block)
prep = [] if idx == 0 else self._block.directives[:idx]
after = [] if idx >= (len(self._block.directives)-1 ) else self._block.directives[idx:]
if cls in (Block, Config):
self._block.directives = prep + list(directives) + after
elif cls is Http:
srv_list, dir_list = [], []
for directive in directives:
if directive.__class__ is Server:
srv_list.append(directive)
else:
dir_list.append(directive)
self._block.servers.extend(srv_list)
self._block.directives = prep + list(dir_list) + after
elif cls is Upstream:
srv_list, dir_list = [], []
for directive in directives:
if directive.__class__ is UpstreamServer:
srv_list.append(directive)
else:
dir_list.append(directive)
self._block.servers.extend(srv_list)
self._block.directives = prep + dir_list + after
def insert_after(self, idx: int, *directives: IDirective):
if idx < 0:
raise ValueError("索引不能小于0")
cls = type(self._block)
if cls in (Block, Config):
self._block.directives = self._block.directives[:idx] + list(directives) + self._block.directives[idx:]
elif cls is Http:
srv_list, dir_list = [], []
for directive in directives:
if directive.__class__ is Server:
srv_list.append(directive)
else:
dir_list.append(directive)
self._block.servers.extend(srv_list)
self._block.directives = self._block.directives[:idx] + list(dir_list) + self._block.directives[idx:]
elif cls is Upstream:
srv_list, dir_list = [], []
for directive in directives:
if directive.__class__ is UpstreamServer:
srv_list.append(directive)
else:
dir_list.append(directive)
self._block.servers.extend(srv_list)
self._block.directives = self._block.directives[:idx] + dir_list + self._block.directives[idx:]
|