ast_utils.nim

# Copyright 2017 the Arraymancer contributors
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

# Tools to manipulate Nim Abstract Syntax Tree

import macros


proc hasType*(x: NimNode, t: static[string]): bool {. compileTime .} =
  ## Compile-time type checking
  sameType(x, bindSym(t))

proc isInt*(x: NimNode): bool {. compileTime .} =
  ## Compile-time type checking
  hasType(x, "int")

proc isAllInt*(slice_args: NimNode): bool {. compileTime .} =
  ## Compile-time type checking
  result = true
  for child in slice_args:
    # We don't use early return here as everything is evaluated at compile-time,
    # has no run-time impact and there are very few slice_args
    result = result and isInt(child)

proc pop*(tree: var NimNode): NimNode {. compileTime .}=
  ## varargs[untyped] consumes all arguments so the actual value should be popped
  ## https://github.com/nim-lang/Nim/issues/5855
  result = tree[tree.len-1]
  tree.del(tree.len-1)

macro getSubType*(TT: typedesc): untyped =
  # Get the subtype T of an AnyTensor[T] input
  getTypeInst(TT)[1][1]