All files / src/abap/nodes structure_node.ts

95.22% Statements 259/272
95.04% Branches 96/101
100% Functions 22/22
95.22% Lines 259/272

Press n or j to go to the next uncovered block, b, p or k for the previous block.

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 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 2721x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 38193x 38193x 38193x 1x 1x 336367x 336367x 1x 1x 1x 350x 710x 70x 710x 360x 640x 280x 280x 244x 244x 280x 710x 36x 36x 1x 1x 45x 45x 63x 63x 45x 45x 1x 1x 172x 177x 171x 171x 177x 1x 1x 1x 1x 3530x 3530x 5290x 546x 546x 5290x 3530x 3530x 1x 1x 7821x 7821x 13593x 695x 695x 13593x 7821x 7821x 1x 1x 5801x 6237x 5333x 6237x 524x 904x 380x 380x 73x 73x 380x 6237x 395x 395x 1x 1x 10738x 20245x 20245x 2891x 2891x 20245x 7847x 7847x 1x 1x 146x 146x 145x 145x 1x 1x     1x 1x 390x 390x 390x 390x 390x       1x 1x 1395x 1395x 1395x 1395x 1395x       1x 1x 21349x 21349x 41190x 41190x 21349x 21349x 1x 1x 7745x 7745x 7745x 14652x 8556x 14652x 6096x 6096x 14652x 7745x 7745x 1x 1x 5222x 5222x 9928x 9928x 5222x 5222x 1x 1x 76994x 76994x 149738x 55960x 149738x 2588x 2588x 149738x 76994x 76994x 1x 1x 2235x 2235x 4379x 2719x 4379x 1660x 1660x 4379x 2235x 2235x 1x 1x 27488x 27488x 27488x 52865x 30745x 52865x 195x 195x 22120x 22120x 27488x 27488x 1x 1x 1047x 1047x 2987x 102x 102x 2987x 1047x 1890x 1103x 1103x 787x 787x 1890x 2361x       2361x 787x 787x 787x 1890x 945x 945x 1x 1x 112927x 112927x     112927x 194114x 107730x 194114x 3707x 86384x 82677x 82677x 194114x 112927x 112927x 1x 1x 11405x 2783x 2783x 11405x 21457x 2014x 2014x 21457x 6608x 6608x 1x 1x 41210x 1475x 1475x 41210x 85587x 60147x 85587x 6083x 25440x 19357x 19357x 14x 14x 19357x 85587x 33638x 33638x 1x 1x
import {AbstractNode} from "./_abstract_node";
import {AbstractToken} from "../1_lexer/tokens/abstract_token";
import {IStatement} from "../2_statements/statements/_statement";
import {IStructure} from "../3_structures/structures/_structure";
import {IStatementRunnable} from "../2_statements/statement_runnable";
import {StatementNode} from "./statement_node";
import {ExpressionNode} from "./expression_node";
 
export class StructureNode extends AbstractNode<StructureNode | StatementNode> {
  private readonly structure: IStructure;
 
  public constructor(structure: IStructure) {
    super();
    this.structure = structure;
  }
 
  public get() {
    return this.structure;
  }
 
  // todo, remove this method, the logic should never go up in the tree
  public findParent(node: StatementNode): StructureNode | undefined {
    for (const child of this.getChildren()) {
      if (child === node) {
        return this;
      } else if (child instanceof StatementNode) {
        continue;
      } else {
        const res = child.findParent(node);
        if (res) {
          return res;
        }
      }
    }
    return undefined;
  }
 
  public concatTokens(): string {
    let concat = "";
    for (const child of this.getChildren()) {
      concat = concat + child.concatTokens();
    }
    return concat;
  }
 
  public findDirectStatement(type: new () => IStatement): StatementNode | undefined {
    for (const child of this.getChildren()) {
      if (child instanceof StatementNode && child.get() instanceof type) {
        return child;
      }
    }
    return undefined;
  }
 
  public findDirectStatements(type: new () => IStatement): StatementNode[] {
    const ret: StatementNode[] = [];
    for (const child of this.getChildren()) {
      if (child instanceof StatementNode && child.get() instanceof type) {
        ret.push(child);
      }
    }
    return ret;
  }
 
  public findDirectStructures(type: new () => IStructure): StructureNode[] {
    const ret: StructureNode[] = [];
    for (const child of this.getChildren()) {
      if (child instanceof StructureNode && child.get() instanceof type) {
        ret.push(child);
      }
    }
    return ret;
  }
 
  public findFirstStatement(type: new () => IStatement): StatementNode | undefined {
    for (const child of this.getChildren()) {
      if (child.get() instanceof type) {
        return child as StatementNode;
      } else if (child instanceof StatementNode) {
        continue;
      } else {
        const res = child.findFirstStatement(type);
        if (res) {
          return res;
        }
      }
    }
    return undefined;
  }
 
  public findFirstExpression(type: new () => IStatementRunnable): ExpressionNode | undefined {
    for (const child of this.getChildren()) {
      const res = child.findFirstExpression(type);
      if (res) {
        return res;
      }
    }
    return undefined;
  }
 
  public getFirstStatement(): StatementNode | undefined {
    for (const child of this.getChildren()) {
      if (child instanceof StatementNode) {
        return child;
      }
      return child.getFirstStatement();
    }
    return undefined;
  }
 
  public getFirstToken(): AbstractToken {
    const child = this.getFirstChild();
 
    if (child !== undefined) {
      return child.getFirstToken();
    }

    throw new Error("StructureNode, getFirstToken, unexpected type");
  }
 
  public getLastToken(): AbstractToken {
    const child = this.getLastChild();
 
    if (child !== undefined) {
      return child.getLastToken();
    }

    throw new Error("StructureNode, getLastToken, unexpected type");
  }
 
  public findAllExpressions(type: new () => IStatementRunnable): ExpressionNode[] {
    const ret: ExpressionNode[] = [];
    for (const child of this.getChildren()) {
      ret.push(...child.findAllExpressions(type));
    }
    return ret;
  }
 
  public findAllExpressionsRecursive(type: new () => IStatementRunnable): ExpressionNode[] {
    const ret: ExpressionNode[] = [];
 
    for (const child of this.getChildren()) {
      if (child instanceof StatementNode) {
        ret.push(...child.findAllExpressionsRecursive(type));
      } else {
        ret.push(...child.findAllExpressionsRecursive(type));
      }
    }
    return ret;
  }
 
  public findAllExpressionsMulti(type: (new () => IStatementRunnable)[]): ExpressionNode[] {
    const ret: ExpressionNode[] = [];
    for (const child of this.getChildren()) {
      ret.push(...child.findAllExpressionsMulti(type));
    }
    return ret;
  }
 
  public findAllStatements(type: new () => IStatement): StatementNode[] {
    const ret: StatementNode[] = [];
    for (const child of this.getChildren()) {
      if (child instanceof StructureNode) {
        ret.push(...child.findAllStatements(type));
      } else if (child.get() instanceof type) {
        ret.push(child);
      }
    }
    return ret;
  }
 
  public findAllStatementNodes(): StatementNode[] {
    const ret: StatementNode[] = [];
    for (const child of this.getChildren()) {
      if (child instanceof StatementNode) {
        ret.push(child);
      } else {
        ret.push(...child.findAllStatementNodes());
      }
    }
    return ret;
  }
 
  public findAllStructuresRecursive(type: new () => IStructure): StructureNode[] {
    const ret: StructureNode[] = [];
 
    for (const child of this.getChildren()) {
      if (child instanceof StatementNode) {
        continue;
      } else if (child.get() instanceof type) {
        ret.push(child);
      }
      ret.push(...child.findAllStructuresRecursive(type));
    }
    return ret;
  }
 
  public findAllStructuresMulti(type: (new () => IStructure)[]): StructureNode[] {
    const ret: StructureNode[] = [];
    for (const t of type) {
      if (this.get() instanceof t) {
        return [this];
      }
    }
    for (const child of this.getChildren()) {
      if (child instanceof StatementNode) {
        continue;
      }
 
      let found = false;
      for (const t of type) {
        if (this.get() instanceof t) {
          ret.push(child);
          found = true;
        }
      }
      if (found === false) {
        ret.push(...child.findAllStructuresMulti(type));
      }
    }
    return ret;
  }
 
  public findAllStructures(type: new () => IStructure): StructureNode[] {
    const ret: StructureNode[] = [];
    if (this.get() instanceof type) {
      return [this];
    }
    for (const child of this.getChildren()) {
      if (child instanceof StatementNode) {
        continue;
      } else if (child.get() instanceof type) {
        ret.push(child);
      } else {
        ret.push(...child.findAllStructures(type));
      }
    }
    return ret;
  }
 
  public findDirectStructure(type: new () => IStructure): StructureNode | undefined {
    if (this.get() instanceof type) {
      return this;
    }
    for (const child of this.getChildren()) {
      if (child.get() instanceof type) {
        return child as StructureNode;
      }
    }
    return undefined;
  }
 
  public findFirstStructure(type: new () => IStructure): StructureNode | undefined {
    if (this.get() instanceof type) {
      return this;
    }
    for (const child of this.getChildren()) {
      if (child instanceof StatementNode) {
        continue;
      } else if (child.get() instanceof type) {
        return child;
      } else {
        const res = child.findFirstStructure(type);
        if (res) {
          return res;
        }
      }
    }
    return undefined;
  }
 
}