codeql/java/ql/test-kotlin2/library-tests/exprs/unaryOp.ql

import java

newtype TMaybeElement =
  TElement(Element e) or
  TNoElement()

class MaybeElement extends TMaybeElement {
  abstract string toString();

  abstract Location getLocation();
}

class YesMaybeElement extends MaybeElement {
  Element e;

  YesMaybeElement() { this = TElement(e) }

  override string toString() { result = e.toString() }

  override Location getLocation() { result = e.getLocation() }
}

class NoMaybeElement extends MaybeElement {
  NoMaybeElement() { this = TNoElement() }

  override string toString() { result = "<none>" }

  override Location getLocation() { none() }
}

MaybeElement op(UnaryExpr e) {
  if exists(e.getExpr()) then result = TElement(e.getExpr()) else result = TNoElement()
}

from Expr e
select e, op(e)