Fix stdout reopen issue

This commit is contained in:
Ganesh Viswanathan 2020-01-14 14:23:05 -06:00
commit 359bc6e38a
5 changed files with 48 additions and 51 deletions

View file

@ -83,7 +83,7 @@ proc searchAstForNode(ast: ref Ast, node: TSNode, nimState: NimState): bool =
if nimState.gState.debug: if nimState.gState.debug:
nimState.nodeBranch.add $node.tsNodeType() nimState.nodeBranch.add $node.tsNodeType()
echo "#" & spaces(nimState.nodeBranch.len * 2) & nimState.nodeBranch[^1] necho "#" & spaces(nimState.nodeBranch.len * 2) & nimState.nodeBranch[^1]
if ast.children.nBl: if ast.children.nBl:
if childNames.contains(ast.regex) or if childNames.contains(ast.regex) or
@ -111,14 +111,14 @@ proc searchAstForNode(ast: ref Ast, node: TSNode, nimState: NimState): bool =
result = node.saveNodeData(nimState) result = node.saveNodeData(nimState)
else: else:
if nimState.gState.debug: if nimState.gState.debug:
echo "#" & spaces(nimState.nodeBranch.len * 2) & &" {ast.getRegexForAstChildren()} !=~ {childNames}" necho "#" & spaces(nimState.nodeBranch.len * 2) & &" {ast.getRegexForAstChildren()} !=~ {childNames}"
elif node.getTSNodeNamedChildCountSansComments() == 0: elif node.getTSNodeNamedChildCountSansComments() == 0:
result = node.saveNodeData(nimState) result = node.saveNodeData(nimState)
if nimState.gState.debug: if nimState.gState.debug:
discard nimState.nodeBranch.pop() discard nimState.nodeBranch.pop()
if nimstate.nodeBranch.Bl: if nimstate.nodeBranch.Bl:
echo "" necho ""
proc searchAst(root: TSNode, astTable: AstTable, nimState: NimState) = proc searchAst(root: TSNode, astTable: AstTable, nimState: NimState) =
var var
@ -133,7 +133,7 @@ proc searchAst(root: TSNode, astTable: AstTable, nimState: NimState) =
if name in astTable: if name in astTable:
for ast in astTable[name]: for ast in astTable[name]:
if nimState.gState.debug: if nimState.gState.debug:
echo "\n# " & nimState.getNodeVal(node).replace("\n", "\n# ") & "\n" necho "\n# " & nimState.getNodeVal(node).replace("\n", "\n# ") & "\n"
if searchAstForNode(ast, node, nimState): if searchAstForNode(ast, node, nimState):
ast.tonim(ast, node, nimState) ast.tonim(ast, node, nimState)
if nimState.gState.debug: if nimState.gState.debug:
@ -166,8 +166,8 @@ proc searchAst(root: TSNode, astTable: AstTable, nimState: NimState) =
if node == root: if node == root:
break break
proc printNimHeader*() = proc printNimHeader*(gState: State) =
echo """# Generated at $1 gecho """# Generated at $1
# Command line: # Command line:
# $2 $3 # $2 $3
@ -194,32 +194,32 @@ proc printNim*(gState: State, fullpath: string, root: TSNode, astTable: AstTable
root.searchAst(astTable, nimState) root.searchAst(astTable, nimState)
if nimState.enumStr.nBl: if nimState.enumStr.nBl:
echo &"{nimState.enumStr}\n" necho &"{nimState.enumStr}\n"
nimState.constStr = nimState.getOverrideFinal(nskConst) & nimState.constStr nimState.constStr = nimState.getOverrideFinal(nskConst) & nimState.constStr
if nimState.constStr.nBl: if nimState.constStr.nBl:
echo &"const{nimState.constStr}\n" necho &"const{nimState.constStr}\n"
echo &""" necho &"""
{{.pragma: {nimState.impShort}, importc{nimState.getHeader()}.}} {{.pragma: {nimState.impShort}, importc{nimState.getHeader()}.}}
{{.pragma: {nimState.impShort}C, {nimState.impShort}, cdecl{nimState.getDynlib()}.}} {{.pragma: {nimState.impShort}C, {nimState.impShort}, cdecl{nimState.getDynlib()}.}}
""" """
nimState.typeStr = nimState.getOverrideFinal(nskType) & nimState.typeStr nimState.typeStr = nimState.getOverrideFinal(nskType) & nimState.typeStr
if nimState.typeStr.nBl: if nimState.typeStr.nBl:
echo &"type{nimState.typeStr}\n" necho &"type{nimState.typeStr}\n"
nimState.procStr = nimState.getOverrideFinal(nskProc) & nimState.procStr nimState.procStr = nimState.getOverrideFinal(nskProc) & nimState.procStr
if nimState.procStr.nBl: if nimState.procStr.nBl:
echo &"{nimState.procStr}\n" necho &"{nimState.procStr}\n"
if nimState.gState.debug: if nimState.gState.debug:
if nimState.debugStr.nBl: if nimState.debugStr.nBl:
echo nimState.debugStr necho nimState.debugStr
if nimState.skipStr.nBl: if nimState.skipStr.nBl:
let let
hash = nimState.skipStr.hash().abs() hash = nimState.skipStr.hash().abs()
sname = getTempDir() / &"nimterop_{$hash}.h" sname = getTempDir() / &"nimterop_{$hash}.h"
echo &"# Writing skipped definitions to {sname}\n" necho &"# Writing skipped definitions to {sname}\n"
writeFile(sname, nimState.skipStr) writeFile(sname, nimState.skipStr)

View file

@ -462,7 +462,7 @@ proc configure*(path, check: string, flags = "") =
echo "# Running autogen.sh" echo "# Running autogen.sh"
echo execAction( echo execAction(
&"cd {(path / i).parentDir().sanitizePath} && bash autogen.sh").output &"cd {(path / i).parentDir().sanitizePath} && bash ./autogen.sh").output
break break
@ -479,7 +479,7 @@ proc configure*(path, check: string, flags = "") =
echo "# Running configure " & flags echo "# Running configure " & flags
var var
cmd = &"cd {path.sanitizePath} && bash configure" cmd = &"cd {path.sanitizePath} && bash ./configure"
if flags.len != 0: if flags.len != 0:
cmd &= &" {flags}" cmd &= &" {flags}"

View file

@ -62,6 +62,8 @@ type
onSymbol*, onSymbolOverride*: OnSymbol onSymbol*, onSymbolOverride*: OnSymbol
onSymbolOverrideFinal*: OnSymbolOverrideFinal onSymbolOverrideFinal*: OnSymbolOverrideFinal
outputHandle*: File
NimState {.used.} = ref object NimState {.used.} = ref object
identifiers*: TableRef[string, string] identifiers*: TableRef[string, string]
@ -94,3 +96,14 @@ const modeDefault {.used.} = $cpp # TODO: USE this everywhere relevant
when not declared(CIMPORT): when not declared(CIMPORT):
export gAtoms, gExpressions, gEnumVals, Kind, Ast, AstTable, State, NimState, export gAtoms, gExpressions, gEnumVals, Kind, Ast, AstTable, State, NimState,
nBl, Bl, CompileMode, modeDefault nBl, Bl, CompileMode, modeDefault
# Redirect output to file when required
template gecho*(args: string) {.dirty.} =
if gState.outputHandle.isNil:
echo args
else:
gState.outputHandle.writeLine(args)
template necho*(args: string) {.dirty.} =
let gState = nimState.gState
gecho args

View file

@ -735,7 +735,7 @@ proc parseGrammar*(): AstTable =
else: else:
result[n].add(ast) result[n].add(ast)
proc printGrammar*(astTable: AstTable) = proc printGrammar*(gState: State, astTable: AstTable) =
for name in astTable.keys(): for name in astTable.keys():
for ast in astTable[name]: for ast in astTable[name]:
echo ast.printAst() gecho ast.printAst()

View file

@ -22,12 +22,12 @@ proc printLisp(gState: State, root: TSNode) =
if node.tsNodeNamedChildCount() != 0: if node.tsNodeNamedChildCount() != 0:
if gState.pretty: if gState.pretty:
echo "" gecho ""
nextnode = node.tsNodeNamedChild(0) nextnode = node.tsNodeNamedChild(0)
depth += 1 depth += 1
else: else:
if gState.pretty: if gState.pretty:
echo ")" gecho ")"
else: else:
stdout.write ")" stdout.write ")"
nextnode = node.tsNodeNextNamedSibling() nextnode = node.tsNodeNextNamedSibling()
@ -39,7 +39,7 @@ proc printLisp(gState: State, root: TSNode) =
if depth == -1: if depth == -1:
break break
if gState.pretty: if gState.pretty:
echo spaces(depth) & ")" gecho spaces(depth) & ")"
else: else:
stdout.write ")" stdout.write ")"
if node == root: if node == root:
@ -95,7 +95,7 @@ proc process(gState: State, path: string, astTable: AstTable) =
elif gState.pnim: elif gState.pnim:
gState.printNim(path, root, astTable) gState.printNim(path, root, astTable)
elif gState.preprocess: elif gState.preprocess:
echo gState.code gecho gState.code
# CLI processing with default values # CLI processing with default values
proc main( proc main(
@ -149,32 +149,23 @@ proc main(
if pluginSourcePath.nBl: if pluginSourcePath.nBl:
gState.loadPlugin(pluginSourcePath) gState.loadPlugin(pluginSourcePath)
# Backup stdout
var var
outputFile = output outputFile = output
outputHandle: File
stdoutBackup = stdout
check = check or stub check = check or stub
# Fix output file extention # Fix output file extention for Nim mode
if outputFile.len != 0: if outputFile.len != 0 and pnim:
if outputFile.splitFile().ext != ".nim": if outputFile.splitFile().ext != ".nim":
outputFile = outputFile & ".nim" outputFile = outputFile & ".nim"
# Check needs a file # Check needs a file
if check and outputFile.len == 0: if check and outputFile.len == 0:
outputFile = getTempDir() / "toast_" & ($getTime().toUnix()).addFileExt("nim") outputFile = getTempDir() / "toast_" & ($getTime().toUnix()).addFileExt("nim")
when defined(windows):
# https://github.com/nim-lang/Nim/issues/12939
echo &"Cannot print wrapper with check on Windows, review {outputFile}\n"
# Redirect output to file # Redirect output to file
if outputFile.len != 0: if outputFile.len != 0:
when defined(windows): doAssert gState.outputHandle.open(outputFile, fmWrite),
doAssert stdout.reopen(outputFile, fmWrite), &"Failed to write to {outputFile}" &"Failed to write to {outputFile}"
else:
doAssert outputHandle.open(outputFile, fmWrite), &"Failed to write to {outputFile}"
stdout = outputHandle
# Process grammar into AST # Process grammar into AST
let let
@ -182,17 +173,17 @@ proc main(
if pgrammar: if pgrammar:
# Print AST of grammar # Print AST of grammar
astTable.printGrammar() gState.printGrammar(astTable)
elif source.nBl: elif source.nBl:
# Print source after preprocess or Nim output # Print source after preprocess or Nim output
if gState.pnim: if gState.pnim:
printNimHeader() gState.printNimHeader()
for src in source: for src in source:
gState.process(src.expandSymlinkAbs(), astTable) gState.process(src.expandSymlinkAbs(), astTable)
when not defined(windows): # Close outputFile
# Restore stdout if outputFile.len != 0:
stdout = stdoutBackup gState.outputHandle.close()
# Check Nim output # Check Nim output
if gState.pnim and check: if gState.pnim and check:
@ -202,12 +193,6 @@ proc main(
if err != 0: if err != 0:
# Failed check so try stubbing # Failed check so try stubbing
if stub: if stub:
# Close output file to prepend stubs
when not defined(windows):
outputHandle.close()
else:
stdout.close()
# Find undeclared identifiers in error # Find undeclared identifiers in error
var var
data = "" data = ""
@ -234,14 +219,13 @@ proc main(
# Rerun nim check on stubbed wrapper # Rerun nim check on stubbed wrapper
(check, err) = execCmdEx(&"{gState.nim} check {outputFile}") (check, err) = execCmdEx(&"{gState.nim} check {outputFile}")
doAssert err == 0, "# Nim check with stub failed:\n\n" & check doAssert err == 0, data & "# Nim check with stub failed:\n\n" & check
else: else:
doAssert err == 0, "# Nim check failed:\n\n" & check doAssert err == 0, outputFile.readFile() & "# Nim check failed:\n\n" & check
when not defined(windows): # Print wrapper if temporarily redirected to file
# Print wrapper if temporarily redirected to file if check and output.len == 0:
if check and output.len == 0: stdout.write outputFile.readFile()
stdout.write outputFile.readFile()
when isMainModule: when isMainModule:
# Setup cligen command line help and short flags # Setup cligen command line help and short flags