All files / src/abap/nodes structure_node.ts

94.83% Statements 110/116
91.67% Branches 55/60
100% Functions 20/20
94.83% Lines 110/116

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 2331x         1x     1x       20559x 20559x       156763x         289x 621x 61x 560x 332x   228x 228x 208x       20x       48x 48x 66x   48x       57x 57x 57x             1999x 1999x 2890x 345x     1999x       4996x 4996x 6817x 403x     4996x       3826x 3987x 3694x 293x 183x   110x 110x 43x       89x       5398x 9751x 9751x 1496x     3902x       58x 58x 56x   2x           212x   212x 212x             789x   789x 789x             9536x 9536x 19090x   9536x       1839x 1839x 3625x   1839x       20744x 20744x 40253x 13491x 26762x 1126x     20744x       1313x 1313x 2620x 1639x   981x     1313x       4629x   4629x 9192x 5560x 3632x 66x   3632x   4629x       59236x 59236x     59236x 97374x 55146x 42228x 2056x   40172x     59236x       6463x 1578x   4885x 12026x 1064x     3821x       18965x 840x   18125x 37431x 26208x 11223x 2913x   8310x 8310x         15212x      
import {AbstractNode} from "./_abstract_node";
import {Token} from "../1_lexer/tokens/_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()) {
      Eif (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(): Token {
    const child = this.getFirstChild();
 
    Eif (child !== undefined) {
      return child.getFirstToken();
    }
 
    throw new Error("StructureNode, getFirstToken, unexpected type");
  }
 
  public getLastToken(): Token {
    const child = this.getLastChild();
 
    Eif (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 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 findAllStructures(type: new () => IStructure): StructureNode[] {
    const ret: StructureNode[] = [];
    Iif (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);
        Iif (res) {
          return res;
        }
      }
    }
    return undefined;
  }
 
}