All files / src/abap/nodes structure_node.ts

94.48% Statements 257/272
94% Branches 94/100
100% Functions 22/22
94.48% Lines 257/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 40365x 40365x 40365x 1x 1x 355229x 355229x 1x 1x 1x 363x 728x 74x 728x 365x 654x 289x 289x 253x 253x 289x 728x 36x 36x 1x 1x 45x 45x 63x 63x 45x 45x 1x 1x 215x 311x 181x 181x 311x 34x 34x 1x 1x 3763x 3763x 5720x 634x 634x 5720x 3763x 3763x 1x 1x 8360x 8360x 14486x 727x 727x 14486x 8360x 8360x 1x 1x 6390x 7024x 5666x 7024x 769x 1358x 589x 589x 75x 75x 589x 7024x 649x 649x 1x 1x 11285x 21185x 21185x 3076x 3076x 21185x 8209x 8209x 1x 1x 2043x 2043x 1579x 1579x 464x 464x     1x 1x 418x 418x 418x 418x 418x       1x 1x 1456x 1456x 1456x 1456x 1456x       1x 1x 23624x 23624x 45559x 45559x 23624x 23624x 1x 1x 8513x 8513x 8513x 16128x 9408x 16128x 6720x 6720x 16128x 8513x 8513x 1x 1x 5740x 5740x 10920x 10920x 5740x 5740x 1x 1x 82312x 82312x 159931x 60708x 159931x 2800x 2800x 159931x 82312x 82312x 1x 1x 2402x 2402x 4701x 2909x 4701x 1792x 1792x 4701x 2402x 2402x 1x 1x 31579x 31579x 31579x 60767x 35326x 60767x 245x 245x 25441x 25441x 31579x 31579x 1x 1x 2383x 2383x 5767x 195x 195x 5767x 2383x 4322x 2533x 2533x 1789x 1789x 4322x 4539x       4539x 1789x 1789x 1789x 4322x 2188x 2188x 1x 1x 121603x 121603x     121603x 209401x 116420x 209401x 4026x 92981x 88955x 88955x 209401x 121603x 121603x 1x 1x 11778x 2868x 2868x 11778x 22226x 2104x 2104x 22226x 6806x 6806x 1x 1x 30659x     30659x 65683x 45375x 65683x 5500x 20308x 14808x 14808x 14x 14x 14808x 65683x 25145x 25145x 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;
  }
 
}