如何在callsite目录中使用slurp / gorge / staticRead / staticExec?

问题描述 投票:1回答:2

编译时函数slurp / gorge / staticRead / staticExec似乎使用源文件的目录作为工作目录。在大多数情况下,这是期望的行为,因为源代码和编译时资源的关系是固定的。但是,我如何在库中使用这些函数,以便它们引用用户提供的资源?

结构示例:

.
├── client
│   ├── client.nim
│   └── resource.data
└── library
    └── library.nim

我想在库中提供bundle函数,它允许客户端调用像bundle("resource.data")这样的东西。在图书馆内部可以使用例如slurp(givenResourcePath)。然而,这将失败,因为相对于resource.data,slurp查找library.nim。有没有办法使用这些函数并引用相对于调用点的文件?

注意:我尝试使用模板/宏生成执行slurp的AST,但是甚至查找都与library.nim相关。

nim
2个回答
3
投票

处理此问题的最简单方法是依赖使用instantiationInfo的助手模板来获取宏调用者的源路径。

您可以创建一个名为bundles.nim的模块:

import os

macro bundleImpl(userPath, resource: static string): untyped =
  let resourcePath = splitFile(userPath).dir / resource 
  echo "FULL RESOURCE PATH ", resourcePath
  echo "FILE CONTENTS:"
  echo staticRead(resourcePath)

template bundle*(resource: static string) =
  bundleImpl(instantiationInfo(-1, fullPaths = true).filename, resource)

然后,您可以按预期的方式从任何模块使用它:

import
  bundles

bundle "test.txt"

我系统的结果如下:

FULL RESOURCE PATH /Users/zahary/nim/scratch/test.txt
FILE CONTENTS:
<test.txt contents>

2
投票

这可以通过使用一个小技巧的宏来解决:查找slurp的实现显示它使用slurp AST节点的lineinfo来确定其工作目录。默认情况下,使用宏构造AST会附加一个引用library.nim的lineinfo,因此,slurp使用库路径。要修改行为,我们可以从调用点读取lineinfo并将其附加到slurp节点:

macro bundle*(resource: string): untyped =
  # create slurp call node
  var slurpCall = newCall(ident "slurp", newStrLitNode resource.strVal)

  # forward callsite lineinfo to affect working directory behavior
  slurpCall.copyLineInfo(resource)

  # embed slurpCall somewhere in output AST
  # ...
© www.soinside.com 2019 - 2024. All rights reserved.