codeql/javascript/ql/test/library-tests/CustomAbstractValueDefinitions/CustomAbstractValueDefinitions.ql

import javascript
import semmle.javascript.dataflow.InferredTypes
import semmle.javascript.dataflow.internal.FlowSteps as FlowSteps
import semmle.javascript.dataflow.internal.AbstractPropertiesImpl as AbstractPropertiesImpl
import semmle.javascript.dataflow.CustomAbstractValueDefinitions

class MyCustomAbstractValueDefinition extends CustomAbstractValueDefinition {

  DataFlow::ValueNode node;

  MyCustomAbstractValueDefinition() {
    DataFlow::valueNode(this) = node and
    node instanceof DataFlow::ObjectLiteralNode and
    exists (DataFlow::PropWrite pwn |
      pwn.writes(node, "custom", any(BooleanLiteral l | l.getValue() = "true").flow())
    )
  }

  override boolean getBooleanValue() {
    result = true
  }

  override predicate isCoercibleToNumber() {
    none()
  }

  override PrimitiveAbstractValue toPrimitive() {
    result = TAbstractOtherString()
  }

  override InferredType getType() { result = TTObject() }

  override predicate shouldTrackProperties() {
    exists (DataFlow::PropWrite pwn |
      pwn.writes(node, "trackProps", any(BooleanLiteral l | l.getValue() = "true").flow())
    )
  }

}

boolean flowProps(AbstractValue val) {
  if FlowSteps::shouldTrackProperties(val) then
    result = true
  else
    result = false
}

boolean typeProps(AbstractValue val) {
  if AbstractPropertiesImpl::shouldTrackProperties(val) then
    result = true
  else
    result = false
}

from MyCustomAbstractValueDefinition def, AbstractValue val
where def.getAbstractValue() = val
select def, val, flowProps(val), typeProps(val)