123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398 |
- #
- #
- # The Nim Compiler
- # (c) Copyright 2017 Andreas Rumpf
- #
- # See the file "copying.txt", included in this
- # distribution, for details about the copyright.
- #
- import compiler/renderer
- import compiler/types
- import compiler/trees
- import compiler/wordrecg
- import compiler/sempass2
- import strformat
- import algorithm
- import tables
- import times
- import procmonitor
- template tryImport(module) = import module
- when compiles tryImport ../dist/checksums/src/checksums/sha1:
- import ../dist/checksums/src/checksums/sha1
- else:
- import checksums/sha1
- ## Nimsuggest is a tool that helps to give editors IDE like capabilities.
- when not defined(nimcore):
- {.error: "nimcore MUST be defined for Nim's core tooling".}
- import strutils, os, parseopt, parseutils, sequtils, net, rdstdin, sexp
- # Do NOT import suggest. It will lead to weird bugs with
- # suggestionResultHook, because suggest.nim is included by sigmatch.
- # So we import that one instead.
- import compiler / [options, commands, modules,
- passes, passaux, msgs,
- sigmatch, ast,
- idents, modulegraphs, prefixmatches, lineinfos, cmdlinehelper,
- pathutils, condsyms, syntaxes, suggestsymdb]
- when defined(nimPreviewSlimSystem):
- import std/typedthreads
- when defined(windows):
- import winlean
- else:
- import posix
- const HighestSuggestProtocolVersion = 4
- const DummyEof = "!EOF!"
- const Usage = """
- Nimsuggest - Tool to give every editor IDE like capabilities for Nim
- Usage:
- nimsuggest [options] projectfile.nim
- Options:
- --autobind automatically binds into a free port
- --port:PORT port, by default 6000
- --address:HOST binds to that address, by default ""
- --stdin read commands from stdin and write results to
- stdout instead of using sockets
- --clientProcessId:PID shutdown nimsuggest in case this process dies
- --epc use emacs epc mode
- --debug enable debug output
- --log enable verbose logging to nimsuggest.log file
- --v1 use version 1 of the protocol; for backwards compatibility
- --v2 use version 2(default) of the protocol
- --v3 use version 3 of the protocol
- --v4 use version 4 of the protocol
- --info:X information
- --info:nimVer return the Nim compiler version that nimsuggest uses internally
- --info:protocolVer return the newest protocol version that is supported
- --info:capabilities return the capabilities supported by nimsuggest
- --refresh perform automatic refreshes to keep the analysis precise
- --maxresults:N limit the number of suggestions to N
- --tester implies --stdin and outputs a line
- '""" & DummyEof & """' for the tester
- --find attempts to find the project file of the current project
- --exceptionInlayHints:on|off
- globally turn exception inlay hints on|off
- The server then listens to the connection and takes line-based commands.
- If --autobind is used, the binded port number will be printed to stdout.
- In addition, all command line options of Nim that do not affect code generation
- are supported.
- """
- type
- Mode = enum mstdin, mtcp, mepc, mcmdsug, mcmdcon
- CachedMsg = object
- info: TLineInfo
- msg: string
- sev: Severity
- CachedMsgs = seq[CachedMsg]
- var
- gPort = 6000.Port
- gAddress = "127.0.0.1"
- gMode: Mode
- gEmitEof: bool # whether we write '!EOF!' dummy lines
- gLogging = defined(logging)
- gRefresh: bool
- gAutoBind = false
- requests: Channel[string]
- results: Channel[Suggest]
- proc executeNoHooksV3(cmd: IdeCmd, file: AbsoluteFile, dirtyfile: AbsoluteFile, line, col: int; tag: string,
- graph: ModuleGraph);
- proc writelnToChannel(line: string) =
- results.send(Suggest(section: ideMsg, doc: line))
- proc sugResultHook(s: Suggest) =
- results.send(s)
- proc errorHook(conf: ConfigRef; info: TLineInfo; msg: string; sev: Severity) =
- results.send(Suggest(section: ideChk, filePath: toFullPath(conf, info),
- line: toLinenumber(info), column: toColumn(info), doc: msg,
- forth: $sev))
- proc myLog(s: string) =
- if gLogging: log(s)
- const
- seps = {':', ';', ' ', '\t'}
- Help = "usage: sug|con|def|use|dus|chk|mod|highlight|outline|known|project file.nim[;dirtyfile.nim]:line:col\n" &
- "type 'quit' to quit\n" &
- "type 'debug' to toggle debug mode on/off\n" &
- "type 'terse' to toggle terse mode on/off"
- #List of currently supported capabilities. So lang servers/ides can iterate over and check for what's enabled
- Capabilities = [
- "con", #current NimSuggest supports the `con` commmand
- "exceptionInlayHints"
- ]
- proc parseQuoted(cmd: string; outp: var string; start: int): int =
- var i = start
- i += skipWhitespace(cmd, i)
- if i < cmd.len and cmd[i] == '"':
- i += parseUntil(cmd, outp, '"', i+1)+2
- else:
- i += parseUntil(cmd, outp, seps, i)
- result = i
- proc sexp(s: IdeCmd|TSymKind|PrefixMatch): SexpNode = sexp($s)
- proc sexp(s: Suggest): SexpNode =
- # If you change the order here, make sure to change it over in
- # nim-mode.el too.
- let qp = if s.qualifiedPath.len == 0: @[] else: s.qualifiedPath
- result = convertSexp([
- s.section,
- TSymKind s.symkind,
- qp.map(newSString),
- s.filePath,
- s.forth,
- s.line,
- s.column,
- s.doc,
- s.quality
- ])
- if s.section == ideSug:
- result.add convertSexp(s.prefix)
- if s.section in {ideOutline, ideExpand} and s.version == 3:
- result.add convertSexp(s.endLine.int)
- result.add convertSexp(s.endCol)
- proc sexp(s: seq[Suggest]): SexpNode =
- result = newSList()
- for sug in s:
- result.add(sexp(sug))
- proc listEpc(): SexpNode =
- # This function is called from Emacs to show available options.
- let
- argspecs = sexp("file line column dirtyfile".split(" ").map(newSSymbol))
- docstring = sexp("line starts at 1, column at 0, dirtyfile is optional")
- result = newSList()
- for command in ["sug", "con", "def", "use", "dus", "chk", "mod", "globalSymbols", "recompile", "saved", "chkFile", "declaration", "inlayHints"]:
- let
- cmd = sexp(command)
- methodDesc = newSList()
- methodDesc.add(cmd)
- methodDesc.add(argspecs)
- methodDesc.add(docstring)
- result.add(methodDesc)
- proc findNode(n: PNode; trackPos: TLineInfo): PSym =
- #echo "checking node ", n.info
- if n.kind == nkSym:
- if isTracked(n.info, trackPos, n.sym.name.s.len): return n.sym
- else:
- for i in 0 ..< safeLen(n):
- let res = findNode(n[i], trackPos)
- if res != nil: return res
- proc symFromInfo(graph: ModuleGraph; trackPos: TLineInfo): PSym =
- let m = graph.getModule(trackPos.fileIndex)
- if m != nil and m.ast != nil:
- result = findNode(m.ast, trackPos)
- template benchmark(benchmarkName: untyped, code: untyped) =
- block:
- myLog "Started [" & benchmarkName & "]..."
- let t0 = epochTime()
- code
- let elapsed = epochTime() - t0
- let elapsedStr = elapsed.formatFloat(format = ffDecimal, precision = 3)
- myLog "CPU Time [" & benchmarkName & "] " & elapsedStr & "s"
- proc clearInstCache(graph: ModuleGraph, projectFileIdx: FileIndex) =
- if projectFileIdx == InvalidFileIdx:
- graph.typeInstCache.clear()
- graph.procInstCache.clear()
- return
- var typeIdsToDelete = newSeq[ItemId]()
- for id in graph.typeInstCache.keys:
- if id.module == projectFileIdx.int:
- typeIdsToDelete.add id
- for id in typeIdsToDelete:
- graph.typeInstCache.del id
- var procIdsToDelete = newSeq[ItemId]()
- for id in graph.procInstCache.keys:
- if id.module == projectFileIdx.int:
- procIdsToDelete.add id
- for id in procIdsToDelete:
- graph.procInstCache.del id
- proc executeNoHooks(cmd: IdeCmd, file, dirtyfile: AbsoluteFile, line, col: int, tag: string,
- graph: ModuleGraph) =
- let conf = graph.config
- if conf.suggestVersion >= 3:
- let command = fmt "cmd = {cmd} {file}:{line}:{col}"
- benchmark command:
- executeNoHooksV3(cmd, file, dirtyfile, line, col, tag, graph)
- return
- myLog("cmd: " & $cmd & ", file: " & file.string &
- ", dirtyFile: " & dirtyfile.string &
- "[" & $line & ":" & $col & "]")
- conf.ideCmd = cmd
- if cmd == ideUse and conf.suggestVersion != 0:
- graph.resetAllModules()
- var isKnownFile = true
- let dirtyIdx = fileInfoIdx(conf, file, isKnownFile)
- if not dirtyfile.isEmpty: msgs.setDirtyFile(conf, dirtyIdx, dirtyfile)
- else: msgs.setDirtyFile(conf, dirtyIdx, AbsoluteFile"")
- conf.m.trackPos = newLineInfo(dirtyIdx, line, col)
- conf.m.trackPosAttached = false
- conf.errorCounter = 0
- if conf.suggestVersion == 1:
- graph.usageSym = nil
- if not isKnownFile:
- graph.clearInstCache(dirtyIdx)
- graph.compileProject(dirtyIdx)
- if conf.suggestVersion == 0 and conf.ideCmd in {ideUse, ideDus} and
- dirtyfile.isEmpty:
- discard "no need to recompile anything"
- else:
- let modIdx = graph.parentModule(dirtyIdx)
- graph.markDirty dirtyIdx
- graph.markClientsDirty dirtyIdx
- if conf.ideCmd != ideMod:
- if isKnownFile:
- graph.clearInstCache(modIdx)
- graph.compileProject(modIdx)
- if conf.ideCmd in {ideUse, ideDus}:
- let u = if conf.suggestVersion != 1: graph.symFromInfo(conf.m.trackPos) else: graph.usageSym
- if u != nil:
- listUsages(graph, u)
- else:
- localError(conf, conf.m.trackPos, "found no symbol at this position " & (conf $ conf.m.trackPos))
- proc executeNoHooks(cmd: IdeCmd, file, dirtyfile: AbsoluteFile, line, col: int, graph: ModuleGraph) =
- executeNoHooks(cmd, file, dirtyfile, line, col, "", graph)
- proc execute(cmd: IdeCmd, file, dirtyfile: AbsoluteFile, line, col: int; tag: string,
- graph: ModuleGraph) =
- if cmd == ideChk:
- graph.config.structuredErrorHook = errorHook
- graph.config.writelnHook = myLog
- else:
- graph.config.structuredErrorHook = nil
- graph.config.writelnHook = myLog
- executeNoHooks(cmd, file, dirtyfile, line, col, tag, graph)
- proc executeEpc(cmd: IdeCmd, args: SexpNode;
- graph: ModuleGraph) =
- let
- file = AbsoluteFile args[0].getStr
- line = args[1].getNum
- column = args[2].getNum
- var dirtyfile = AbsoluteFile""
- if len(args) > 3:
- dirtyfile = AbsoluteFile args[3].getStr("")
- execute(cmd, file, dirtyfile, int(line), int(column), args[3].getStr, graph)
- proc returnEpc(socket: Socket, uid: BiggestInt, s: SexpNode|string,
- returnSymbol = "return") =
- let response = $convertSexp([newSSymbol(returnSymbol), uid, s])
- socket.send(toHex(len(response), 6))
- socket.send(response)
- template checkSanity(client, sizeHex, size, messageBuffer: typed) =
- if client.recv(sizeHex, 6) != 6:
- raise newException(ValueError, "didn't get all the hexbytes")
- if parseHex(sizeHex, size) == 0:
- raise newException(ValueError, "invalid size hex: " & $sizeHex)
- if client.recv(messageBuffer, size) != size:
- raise newException(ValueError, "didn't get all the bytes")
- proc toStdout() {.gcsafe.} =
- while true:
- let res = results.recv()
- case res.section
- of ideNone: break
- of ideMsg: echo res.doc
- of ideKnown: echo res.quality == 1
- of ideProject: echo res.filePath
- else: echo res
- proc toSocket(stdoutSocket: Socket) {.gcsafe.} =
- while true:
- let res = results.recv()
- case res.section
- of ideNone: break
- of ideMsg: stdoutSocket.send(res.doc & "\c\L")
- of ideKnown: stdoutSocket.send($(res.quality == 1) & "\c\L")
- of ideProject: stdoutSocket.send(res.filePath & "\c\L")
- else: stdoutSocket.send($res & "\c\L")
- proc toEpc(client: Socket; uid: BiggestInt) {.gcsafe.} =
- var list = newSList()
- while true:
- let res = results.recv()
- case res.section
- of ideNone: break
- of ideMsg:
- list.add sexp(res.doc)
- of ideKnown:
- list.add sexp(res.quality == 1)
- of ideProject:
- list.add sexp(res.filePath)
- else:
- list.add sexp(res)
- returnEpc(client, uid, list)
- template setVerbosity(level: typed) =
- gVerbosity = level
- conf.notes = NotesVerbosity[gVerbosity]
- proc connectToNextFreePort(server: Socket, host: string): Port =
- server.bindAddr(Port(0), host)
- let (_, port) = server.getLocalAddr
- result = port
- type
- ThreadParams = tuple[port: Port; address: string]
- proc replStdinSingleCmd(line: string) =
- requests.send line
- toStdout()
- echo ""
- flushFile(stdout)
- proc replStdin(x: ThreadParams) {.thread.} =
- if gEmitEof:
- echo DummyEof
- while true:
- let line = readLine(stdin)
- requests.send line
- if line == "quit": break
- toStdout()
- echo DummyEof
- flushFile(stdout)
- else:
- echo Help
- var line = ""
- while readLineFromStdin("> ", line):
- replStdinSingleCmd(line)
- requests.send "quit"
- proc replCmdline(x: ThreadParams) {.thread.} =
- replStdinSingleCmd(x.address)
- requests.send "quit"
- proc replTcp(x: ThreadParams) {.thread.} =
- var server = newSocket()
- if gAutoBind:
- let port = server.connectToNextFreePort(x.address)
- server.listen()
- echo port
- stdout.flushFile()
- else:
- server.bindAddr(x.port, x.address)
- server.listen()
- var inp = ""
- var stdoutSocket: Socket
- while true:
- accept(server, stdoutSocket)
- stdoutSocket.readLine(inp)
- requests.send inp
- toSocket(stdoutSocket)
- stdoutSocket.send("\c\L")
- stdoutSocket.close()
- proc argsToStr(x: SexpNode): string =
- if x.kind != SList: return x.getStr
- doAssert x.kind == SList
- doAssert x.len >= 4
- let file = x[0].getStr
- let line = x[1].getNum
- let col = x[2].getNum
- let dirty = x[3].getStr
- result = x[0].getStr.escape
- if dirty.len > 0:
- result.add ';'
- result.add dirty.escape
- result.add ':'
- result.addInt line
- result.add ':'
- result.addInt col
- proc replEpc(x: ThreadParams) {.thread.} =
- var server = newSocket()
- let port = connectToNextFreePort(server, "localhost")
- server.listen()
- echo port
- stdout.flushFile()
- var client: Socket
- # Wait for connection
- accept(server, client)
- while true:
- var
- sizeHex = ""
- size = 0
- messageBuffer = ""
- checkSanity(client, sizeHex, size, messageBuffer)
- let
- message = parseSexp($messageBuffer)
- epcApi = message[0].getSymbol
- case epcApi
- of "call":
- let
- uid = message[1].getNum
- cmd = message[2].getSymbol
- args = message[3]
- when false:
- x.ideCmd[] = parseIdeCmd(message[2].getSymbol)
- case x.ideCmd[]
- of ideSug, ideCon, ideDef, ideUse, ideDus, ideOutline, ideHighlight:
- setVerbosity(0)
- else: discard
- let fullCmd = cmd & " " & args.argsToStr
- myLog "MSG CMD: " & fullCmd
- requests.send(fullCmd)
- toEpc(client, uid)
- of "methods":
- returnEpc(client, message[1].getNum, listEpc())
- of "epc-error":
- # an unhandled exception forces down the whole process anyway, so we
- # use 'quit' here instead of 'raise'
- quit("received epc error: " & $messageBuffer)
- else:
- let errMessage = case epcApi
- of "return", "return-error":
- "no return expected"
- else:
- "unexpected call: " & epcApi
- quit errMessage
- proc execCmd(cmd: string; graph: ModuleGraph; cachedMsgs: CachedMsgs) =
- let conf = graph.config
- template sentinel() =
- # send sentinel for the input reading thread:
- results.send(Suggest(section: ideNone))
- template toggle(sw) =
- if sw in conf.globalOptions:
- excl(conf.globalOptions, sw)
- else:
- incl(conf.globalOptions, sw)
- sentinel()
- return
- template err() =
- echo Help
- sentinel()
- return
- var opc = ""
- var i = parseIdent(cmd, opc, 0)
- case opc.normalize
- of "sug": conf.ideCmd = ideSug
- of "con": conf.ideCmd = ideCon
- of "def": conf.ideCmd = ideDef
- of "use": conf.ideCmd = ideUse
- of "dus": conf.ideCmd = ideDus
- of "mod": conf.ideCmd = ideMod
- of "chk": conf.ideCmd = ideChk
- of "highlight": conf.ideCmd = ideHighlight
- of "outline": conf.ideCmd = ideOutline
- of "quit":
- sentinel()
- quit()
- of "debug": toggle optIdeDebug
- of "terse": toggle optIdeTerse
- of "known": conf.ideCmd = ideKnown
- of "project": conf.ideCmd = ideProject
- of "changed": conf.ideCmd = ideChanged
- of "globalsymbols": conf.ideCmd = ideGlobalSymbols
- of "declaration": conf.ideCmd = ideDeclaration
- of "expand": conf.ideCmd = ideExpand
- of "chkfile": conf.ideCmd = ideChkFile
- of "recompile": conf.ideCmd = ideRecompile
- of "type": conf.ideCmd = ideType
- of "inlayhints":
- if conf.suggestVersion >= 4:
- conf.ideCmd = ideInlayHints
- else:
- err()
- else: err()
- var dirtyfile = ""
- var orig = ""
- i += skipWhitespace(cmd, i)
- if i < cmd.len and cmd[i] in {'0'..'9'}:
- orig = string conf.projectFull
- else:
- i = parseQuoted(cmd, orig, i)
- if i < cmd.len and cmd[i] == ';':
- i = parseQuoted(cmd, dirtyfile, i+1)
- i += skipWhile(cmd, seps, i)
- var line = 0
- var col = -1
- i += parseInt(cmd, line, i)
- i += skipWhile(cmd, seps, i)
- i += parseInt(cmd, col, i)
- let tag = substr(cmd, i)
- if conf.ideCmd == ideKnown:
- results.send(Suggest(section: ideKnown, quality: ord(fileInfoKnown(conf, AbsoluteFile orig))))
- elif conf.ideCmd == ideProject:
- results.send(Suggest(section: ideProject, filePath: string conf.projectFull))
- else:
- if conf.ideCmd == ideChk:
- for cm in cachedMsgs: errorHook(conf, cm.info, cm.msg, cm.sev)
- execute(conf.ideCmd, AbsoluteFile orig, AbsoluteFile dirtyfile, line, col, tag, graph)
- sentinel()
- proc recompileFullProject(graph: ModuleGraph) =
- benchmark "Recompilation(clean)":
- graph.resetForBackend()
- graph.resetSystemArtifacts()
- graph.vm = nil
- graph.resetAllModules()
- GC_fullCollect()
- graph.compileProject()
- proc mainThread(graph: ModuleGraph) =
- let conf = graph.config
- myLog "searchPaths: "
- for it in conf.searchPaths:
- myLog(" " & it.string)
- proc wrHook(line: string) {.closure.} =
- if gMode == mepc:
- if gLogging: log(line)
- else:
- writelnToChannel(line)
- conf.writelnHook = wrHook
- conf.suggestionResultHook = sugResultHook
- graph.doStopCompile = proc (): bool = requests.peek() > 0
- var idle = 0
- var cachedMsgs: CachedMsgs = @[]
- while true:
- let (hasData, req) = requests.tryRecv()
- if hasData:
- conf.writelnHook = wrHook
- conf.suggestionResultHook = sugResultHook
- execCmd(req, graph, cachedMsgs)
- idle = 0
- else:
- os.sleep 250
- idle += 1
- if idle == 20 and gRefresh and conf.suggestVersion < 3:
- # we use some nimsuggest activity to enable a lazy recompile:
- conf.ideCmd = ideChk
- conf.writelnHook = proc (s: string) = discard
- cachedMsgs.setLen 0
- conf.structuredErrorHook = proc (conf: ConfigRef; info: TLineInfo; msg: string; sev: Severity) =
- cachedMsgs.add(CachedMsg(info: info, msg: msg, sev: sev))
- conf.suggestionResultHook = proc (s: Suggest) = discard
- recompileFullProject(graph)
- var
- inputThread: Thread[ThreadParams]
- proc mainCommand(graph: ModuleGraph) =
- let conf = graph.config
- clearPasses(graph)
- registerPass graph, verbosePass
- registerPass graph, semPass
- conf.setCmd cmdIdeTools
- defineSymbol(conf.symbols, $conf.backend)
- wantMainModule(conf)
- if not fileExists(conf.projectFull):
- quit "cannot find file: " & conf.projectFull.string
- add(conf.searchPaths, conf.libpath)
- conf.setErrorMaxHighMaybe # honor --errorMax even if it may not make sense here
- # do not print errors, but log them
- conf.writelnHook = proc (msg: string) = discard
- if graph.config.suggestVersion >= 3:
- graph.config.structuredErrorHook = proc (conf: ConfigRef; info: TLineInfo; msg: string; sev: Severity) =
- let suggest = Suggest(section: ideChk, filePath: toFullPath(conf, info),
- line: toLinenumber(info), column: toColumn(info), doc: msg, forth: $sev)
- graph.suggestErrors.mgetOrPut(info.fileIndex, @[]).add suggest
- # compile the project before showing any input so that we already
- # can answer questions right away:
- benchmark "Initial compilation":
- compileProject(graph)
- open(requests)
- open(results)
- if graph.config.clientProcessId != 0:
- hookProcMonitor(graph.config.clientProcessId)
- case gMode
- of mstdin: createThread(inputThread, replStdin, (gPort, gAddress))
- of mtcp: createThread(inputThread, replTcp, (gPort, gAddress))
- of mepc: createThread(inputThread, replEpc, (gPort, gAddress))
- of mcmdsug: createThread(inputThread, replCmdline,
- (gPort, "sug \"" & conf.projectFull.string & "\":" & gAddress))
- of mcmdcon: createThread(inputThread, replCmdline,
- (gPort, "con \"" & conf.projectFull.string & "\":" & gAddress))
- mainThread(graph)
- joinThread(inputThread)
- close(requests)
- close(results)
- proc processCmdLine*(pass: TCmdLinePass, cmd: string; conf: ConfigRef) =
- var p = parseopt.initOptParser(cmd)
- var findProject = false
- while true:
- parseopt.next(p)
- case p.kind
- of cmdEnd: break
- of cmdLongOption, cmdShortOption:
- case p.key.normalize
- of "help", "h":
- stdout.writeLine(Usage)
- quit()
- of "autobind":
- gMode = mtcp
- gAutoBind = true
- of "port":
- gPort = parseInt(p.val).Port
- gMode = mtcp
- of "address":
- gAddress = p.val
- gMode = mtcp
- of "stdin": gMode = mstdin
- of "cmdsug":
- gMode = mcmdsug
- gAddress = p.val
- incl(conf.globalOptions, optIdeDebug)
- of "cmdcon":
- gMode = mcmdcon
- gAddress = p.val
- incl(conf.globalOptions, optIdeDebug)
- of "epc":
- gMode = mepc
- conf.verbosity = 0 # Port number gotta be first.
- of "debug": incl(conf.globalOptions, optIdeDebug)
- of "v1": conf.suggestVersion = 1
- of "v2": conf.suggestVersion = 0
- of "v3": conf.suggestVersion = 3
- of "v4": conf.suggestVersion = 4
- of "info":
- case p.val.normalize
- of "protocolver":
- stdout.writeLine(HighestSuggestProtocolVersion)
- quit 0
- of "nimver":
- stdout.writeLine(system.NimVersion)
- quit 0
- of "capabilities":
- stdout.writeLine(Capabilities.toSeq.mapIt($it).join(" "))
- quit 0
- else:
- processSwitch(pass, p, conf)
- of "exceptioninlayhints":
- case p.val.normalize
- of "", "on": incl(conf.globalOptions, optIdeExceptionInlayHints)
- of "off": excl(conf.globalOptions, optIdeExceptionInlayHints)
- else: processSwitch(pass, p, conf)
- of "tester":
- gMode = mstdin
- gEmitEof = true
- gRefresh = false
- of "log": gLogging = true
- of "refresh":
- if p.val.len > 0:
- gRefresh = parseBool(p.val)
- else:
- gRefresh = true
- of "maxresults":
- conf.suggestMaxResults = parseInt(p.val)
- of "find":
- findProject = true
- of "clientprocessid":
- conf.clientProcessId = parseInt(p.val)
- else: processSwitch(pass, p, conf)
- of cmdArgument:
- let a = unixToNativePath(p.key)
- if dirExists(a) and not fileExists(a.addFileExt("nim")):
- conf.projectName = findProjectNimFile(conf, a)
- # don't make it worse, report the error the old way:
- if conf.projectName.len == 0: conf.projectName = a
- else:
- if findProject:
- conf.projectName = findProjectNimFile(conf, a.parentDir())
- if conf.projectName.len == 0:
- conf.projectName = a
- else:
- conf.projectName = a
- # if processArgument(pass, p, argsCount): break
- proc handleCmdLine(cache: IdentCache; conf: ConfigRef) =
- let self = NimProg(
- suggestMode: true,
- processCmdLine: processCmdLine
- )
- self.initDefinesProg(conf, "nimsuggest")
- if paramCount() == 0:
- stdout.writeLine(Usage)
- return
- self.processCmdLineAndProjectPath(conf)
- if gMode != mstdin:
- conf.writelnHook = proc (msg: string) = discard
- # Find Nim's prefix dir.
- let binaryPath = findExe("nim")
- if binaryPath == "":
- raise newException(IOError,
- "Cannot find Nim standard library: Nim compiler not in PATH")
- conf.prefixDir = AbsoluteDir binaryPath.splitPath().head.parentDir()
- if not dirExists(conf.prefixDir / RelativeDir"lib"):
- conf.prefixDir = AbsoluteDir""
- #msgs.writelnHook = proc (line: string) = log(line)
- myLog("START " & conf.projectFull.string)
- var graph = newModuleGraph(cache, conf)
- if self.loadConfigsAndProcessCmdLine(cache, conf, graph):
- mainCommand(graph)
- # v3 start
- proc recompilePartially(graph: ModuleGraph, projectFileIdx = InvalidFileIdx) =
- if projectFileIdx == InvalidFileIdx:
- myLog "Recompiling partially from root"
- else:
- myLog fmt "Recompiling partially starting from {graph.getModule(projectFileIdx)}"
- # inst caches are breaking incremental compilation when the cache caches stuff
- # from dirty buffer
- graph.clearInstCache(projectFileIdx)
- GC_fullCollect()
- try:
- benchmark "Recompilation":
- graph.compileProject(projectFileIdx)
- except Exception as e:
- myLog fmt "Failed to recompile partially with the following error:\n {e.msg} \n\n {e.getStackTrace()}"
- try:
- graph.recompileFullProject()
- except Exception as e:
- myLog fmt "Failed clean recompilation:\n {e.msg} \n\n {e.getStackTrace()}"
- func deduplicateSymInfoPair[SymInfoPair](xs: seq[SymInfoPair]): seq[SymInfoPair] =
- # xs contains duplicate items and we want to filter them by range because the
- # sym may not match. This can happen when xs contains the same definition but
- # with different signature because suggestSym might be called multiple times
- # for the same symbol (e. g. including/excluding the pragma)
- result = newSeqOfCap[SymInfoPair](xs.len)
- for itm in xs.reversed:
- var found = false
- for res in result:
- if res.info.exactEquals(itm.info):
- found = true
- break
- if not found:
- result.add(itm)
- result.reverse()
- func deduplicateSymInfoPair(xs: SuggestFileSymbolDatabase): SuggestFileSymbolDatabase =
- # xs contains duplicate items and we want to filter them by range because the
- # sym may not match. This can happen when xs contains the same definition but
- # with different signature because suggestSym might be called multiple times
- # for the same symbol (e. g. including/excluding the pragma)
- result = SuggestFileSymbolDatabase(
- lineInfo: newSeqOfCap[TinyLineInfo](xs.lineInfo.len),
- sym: newSeqOfCap[PSym](xs.sym.len),
- isDecl: newPackedBoolArray(),
- caughtExceptions: newSeqOfCap[seq[PType]](xs.caughtExceptions.len),
- caughtExceptionsSet: newPackedBoolArray(),
- fileIndex: xs.fileIndex,
- trackCaughtExceptions: xs.trackCaughtExceptions,
- isSorted: false
- )
- var i = xs.lineInfo.high
- while i >= 0:
- let itm = xs.lineInfo[i]
- var found = false
- for res in result.lineInfo:
- if res.exactEquals(itm):
- found = true
- break
- if not found:
- result.add(xs.getSymInfoPair(i))
- dec i
- result.reverse()
- proc findSymData(graph: ModuleGraph, trackPos: TLineInfo):
- ref SymInfoPair =
- let db = graph.fileSymbols(trackPos.fileIndex).deduplicateSymInfoPair
- doAssert(db.fileIndex == trackPos.fileIndex)
- for i in db.lineInfo.low..db.lineInfo.high:
- if isTracked(db.lineInfo[i], TinyLineInfo(line: trackPos.line, col: trackPos.col), db.sym[i].name.s.len):
- var res = db.getSymInfoPair(i)
- new(result)
- result[] = res
- break
- func isInRange*(current, startPos, endPos: TinyLineInfo, tokenLen: int): bool =
- result =
- (current.line > startPos.line or (current.line == startPos.line and current.col>=startPos.col)) and
- (current.line < endPos.line or (current.line == endPos.line and current.col <= endPos.col))
- proc findSymDataInRange(graph: ModuleGraph, startPos, endPos: TLineInfo):
- seq[SymInfoPair] =
- result = newSeq[SymInfoPair]()
- let db = graph.fileSymbols(startPos.fileIndex).deduplicateSymInfoPair
- for i in db.lineInfo.low..db.lineInfo.high:
- if isInRange(db.lineInfo[i], TinyLineInfo(line: startPos.line, col: startPos.col), TinyLineInfo(line: endPos.line, col: endPos.col), db.sym[i].name.s.len):
- result.add(db.getSymInfoPair(i))
- proc findSymData(graph: ModuleGraph, file: AbsoluteFile; line, col: int):
- ref SymInfoPair =
- let
- fileIdx = fileInfoIdx(graph.config, file)
- trackPos = newLineInfo(fileIdx, line, col)
- result = findSymData(graph, trackPos)
- proc findSymDataInRange(graph: ModuleGraph, file: AbsoluteFile; startLine, startCol, endLine, endCol: int):
- seq[SymInfoPair] =
- let
- fileIdx = fileInfoIdx(graph.config, file)
- startPos = newLineInfo(fileIdx, startLine, startCol)
- endPos = newLineInfo(fileIdx, endLine, endCol)
- result = findSymDataInRange(graph, startPos, endPos)
- proc markDirtyIfNeeded(graph: ModuleGraph, file: string, originalFileIdx: FileIndex) =
- let sha = $sha1.secureHashFile(file)
- if graph.config.m.fileInfos[originalFileIdx.int32].hash != sha or graph.config.ideCmd in {ideSug, ideCon}:
- myLog fmt "{file} changed compared to last compilation"
- graph.markDirty originalFileIdx
- graph.markClientsDirty originalFileIdx
- else:
- myLog fmt "No changes in file {file} compared to last compilation"
- proc suggestResult(graph: ModuleGraph, sym: PSym, info: TLineInfo,
- defaultSection = ideNone, endLine: uint16 = 0, endCol = 0) =
- let section = if defaultSection != ideNone:
- defaultSection
- elif sym.info.exactEquals(info):
- ideDef
- else:
- ideUse
- let suggest = symToSuggest(graph, sym, isLocal=false, section,
- info, 100, PrefixMatch.None, false, 0,
- endLine = endLine, endCol = endCol)
- suggestResult(graph.config, suggest)
- proc suggestInlayHintResultType(graph: ModuleGraph, sym: PSym, info: TLineInfo,
- defaultSection = ideNone, endLine: uint16 = 0, endCol = 0) =
- let section = if defaultSection != ideNone:
- defaultSection
- elif sym.info.exactEquals(info):
- ideDef
- else:
- ideUse
- var suggestDef = symToSuggest(graph, sym, isLocal=false, section,
- info, 100, PrefixMatch.None, false, 0, true,
- endLine = endLine, endCol = endCol)
- suggestDef.inlayHintInfo = suggestToSuggestInlayTypeHint(suggestDef)
- suggestDef.section = ideInlayHints
- if sym.kind == skForVar:
- suggestDef.inlayHintInfo.allowInsert = false
- suggestResult(graph.config, suggestDef)
- proc suggestInlayHintResultException(graph: ModuleGraph, sym: PSym, info: TLineInfo,
- defaultSection = ideNone, caughtExceptions: seq[PType], caughtExceptionsSet: bool, endLine: uint16 = 0, endCol = 0) =
- if not caughtExceptionsSet:
- return
- if sym.kind == skParam and sfEffectsDelayed in sym.flags:
- return
- var raisesList: seq[PType] = @[getEbase(graph, info)]
- let t = sym.typ
- if not isNil(t) and not isNil(t.n) and t.n.len > 0 and t.n[0].len > exceptionEffects:
- let effects = t.n[0]
- if effects.kind == nkEffectList and effects.len == effectListLen:
- let effs = effects[exceptionEffects]
- if not isNil(effs):
- raisesList = @[]
- for eff in items(effs):
- if not isNil(eff):
- raisesList.add(eff.typ)
- var propagatedExceptionList: seq[PType] = @[]
- for re in raisesList:
- var exceptionIsPropagated = true
- for ce in caughtExceptions:
- if isNil(ce) or safeInheritanceDiff(re, ce) <= 0:
- exceptionIsPropagated = false
- break
- if exceptionIsPropagated:
- propagatedExceptionList.add(re)
- if propagatedExceptionList.len == 0:
- return
- let section = if defaultSection != ideNone:
- defaultSection
- elif sym.info.exactEquals(info):
- ideDef
- else:
- ideUse
- var suggestDef = symToSuggest(graph, sym, isLocal=false, section,
- info, 100, PrefixMatch.None, false, 0, true,
- endLine = endLine, endCol = endCol)
- suggestDef.inlayHintInfo = suggestToSuggestInlayExceptionHintLeft(suggestDef, propagatedExceptionList)
- suggestDef.section = ideInlayHints
- suggestResult(graph.config, suggestDef)
- suggestDef.inlayHintInfo = suggestToSuggestInlayExceptionHintRight(suggestDef, propagatedExceptionList)
- suggestResult(graph.config, suggestDef)
- const
- # kinds for ideOutline and ideGlobalSymbols
- searchableSymKinds = {skField, skEnumField, skIterator, skMethod, skFunc, skProc, skConverter, skTemplate}
- proc symbolEqual(left, right: PSym): bool =
- # More relaxed symbol comparison
- return left.info.exactEquals(right.info) and left.name == right.name
- proc findDef(n: PNode, line: uint16, col: int16): PNode =
- if n.kind in {nkProcDef, nkIteratorDef, nkTemplateDef, nkMethodDef, nkMacroDef}:
- if n.info.line == line:
- return n
- else:
- for i in 0 ..< safeLen(n):
- let res = findDef(n[i], line, col)
- if res != nil: return res
- proc findByTLineInfo(trackPos: TLineInfo, infoPairs: SuggestFileSymbolDatabase):
- ref SymInfoPair =
- result = nil
- if infoPairs.fileIndex == trackPos.fileIndex:
- for i in infoPairs.lineInfo.low..infoPairs.lineInfo.high:
- let s = infoPairs.getSymInfoPair(i)
- if s.info.exactEquals trackPos:
- new(result)
- result[] = s
- break
- proc outlineNode(graph: ModuleGraph, n: PNode, endInfo: TLineInfo, infoPairs: SuggestFileSymbolDatabase): bool =
- proc checkSymbol(sym: PSym, info: TLineInfo): bool =
- result = (sym.owner.kind in {skModule, skType} or sym.kind in {skProc, skMethod, skIterator, skTemplate, skType})
- if n.kind == nkSym and n.sym.checkSymbol(n.info):
- graph.suggestResult(n.sym, n.sym.info, ideOutline, endInfo.line, endInfo.col)
- return true
- elif n.kind == nkIdent:
- let symData = findByTLineInfo(n.info, infoPairs)
- if symData != nil and symData.sym.checkSymbol(symData.info):
- let sym = symData.sym
- graph.suggestResult(sym, sym.info, ideOutline, endInfo.line, endInfo.col)
- return true
- proc handleIdentOrSym(graph: ModuleGraph, n: PNode, endInfo: TLineInfo, infoPairs: SuggestFileSymbolDatabase): bool =
- for child in n:
- if child.kind in {nkIdent, nkSym}:
- if graph.outlineNode(child, endInfo, infoPairs):
- return true
- elif child.kind == nkPostfix:
- if graph.handleIdentOrSym(child, endInfo, infoPairs):
- return true
- proc iterateOutlineNodes(graph: ModuleGraph, n: PNode, infoPairs: SuggestFileSymbolDatabase) =
- var matched = true
- if n.kind == nkIdent:
- let symData = findByTLineInfo(n.info, infoPairs)
- if symData != nil and symData.sym.kind == skEnumField and symData.info.exactEquals(symData.sym.info):
- let sym = symData.sym
- graph.suggestResult(sym, sym.info, ideOutline, n.endInfo.line, n.endInfo.col)
- elif (n.kind in {nkFuncDef, nkProcDef, nkTypeDef, nkMacroDef, nkTemplateDef, nkConverterDef, nkEnumFieldDef, nkConstDef}):
- matched = handleIdentOrSym(graph, n, n.endInfo, infoPairs)
- else:
- matched = false
- if n.kind != nkFormalParams:
- for child in n:
- graph.iterateOutlineNodes(child, infoPairs)
- proc calculateExpandRange(n: PNode, info: TLineInfo): TLineInfo =
- if ((n.kind in {nkFuncDef, nkProcDef, nkIteratorDef, nkTemplateDef, nkMethodDef, nkConverterDef} and
- n.info.exactEquals(info)) or
- (n.kind in {nkCall, nkCommand} and n[0].info.exactEquals(info))):
- result = n.endInfo
- else:
- for child in n:
- result = child.calculateExpandRange(info)
- if result != unknownLineInfo:
- return result
- result = unknownLineInfo
- proc executeNoHooksV3(cmd: IdeCmd, file: AbsoluteFile, dirtyfile: AbsoluteFile, line, col: int; tag: string,
- graph: ModuleGraph) =
- let conf = graph.config
- conf.writelnHook = proc (s: string) = discard
- conf.structuredErrorHook = proc (conf: ConfigRef; info: TLineInfo;
- msg: string; sev: Severity) =
- let suggest = Suggest(section: ideChk, filePath: toFullPath(conf, info),
- line: toLinenumber(info), column: toColumn(info), doc: msg, forth: $sev)
- graph.suggestErrors.mgetOrPut(info.fileIndex, @[]).add suggest
- conf.ideCmd = cmd
- myLog fmt "cmd: {cmd}, file: {file}[{line}:{col}], dirtyFile: {dirtyfile}, tag: {tag}"
- var fileIndex: FileIndex
- if not (cmd in {ideRecompile, ideGlobalSymbols}):
- if not fileInfoKnown(conf, file):
- myLog fmt "{file} is unknown, returning no results"
- return
- fileIndex = fileInfoIdx(conf, file)
- msgs.setDirtyFile(
- conf,
- fileIndex,
- if dirtyfile.isEmpty: AbsoluteFile"" else: dirtyfile)
- if not dirtyfile.isEmpty:
- graph.markDirtyIfNeeded(dirtyFile.string, fileInfoIdx(conf, file))
- # these commands require fully compiled project
- if cmd in {ideUse, ideDus, ideGlobalSymbols, ideChk, ideInlayHints} and graph.needsCompilation():
- graph.recompilePartially()
- # when doing incremental build for the project root we should make sure that
- # everything is unmarked as no longer beeing dirty in case there is no
- # longer reference to a particular module. E. g. A depends on B, B is marked
- # as dirty and A loses B import.
- graph.unmarkAllDirty()
- # these commands require partially compiled project
- elif cmd in {ideSug, ideCon, ideOutline, ideHighlight, ideDef, ideChkFile, ideType, ideDeclaration, ideExpand} and
- (graph.needsCompilation(fileIndex) or cmd in {ideSug, ideCon}):
- # for ideSug use v2 implementation
- if cmd in {ideSug, ideCon}:
- conf.m.trackPos = newLineInfo(fileIndex, line, col)
- conf.m.trackPosAttached = false
- else:
- conf.m.trackPos = default(TLineInfo)
- graph.recompilePartially(fileIndex)
- case cmd
- of ideDef:
- let s = graph.findSymData(file, line, col)
- if not s.isNil:
- graph.suggestResult(s.sym, s.sym.info)
- of ideType:
- let s = graph.findSymData(file, line, col)
- if not s.isNil:
- let typeSym = s.sym.typ.sym
- if typeSym != nil:
- graph.suggestResult(typeSym, typeSym.info, ideType)
- elif s.sym.typ.len != 0:
- let genericType = s.sym.typ[0].sym
- graph.suggestResult(genericType, genericType.info, ideType)
- of ideUse, ideDus:
- let symbol = graph.findSymData(file, line, col)
- if not symbol.isNil:
- var res: seq[SymInfoPair] = @[]
- for s in graph.suggestSymbolsIter:
- if s.sym.symbolEqual(symbol.sym):
- res.add(s)
- for s in res.deduplicateSymInfoPair():
- graph.suggestResult(s.sym, s.info)
- of ideHighlight:
- let sym = graph.findSymData(file, line, col)
- if not sym.isNil:
- let fs = graph.fileSymbols(fileIndex)
- var usages: seq[SymInfoPair] = @[]
- for i in fs.lineInfo.low..fs.lineInfo.high:
- if fs.sym[i] == sym.sym:
- usages.add(fs.getSymInfoPair(i))
- myLog fmt "Found {usages.len} usages in {file.string}"
- for s in usages:
- graph.suggestResult(s.sym, s.info)
- of ideRecompile:
- graph.recompileFullProject()
- of ideChanged:
- graph.markDirtyIfNeeded(file.string, fileIndex)
- of ideSug, ideCon:
- # ideSug/ideCon performs partial build of the file, thus mark it dirty for the
- # future calls.
- graph.markDirtyIfNeeded(file.string, fileIndex)
- graph.recompilePartially(fileIndex)
- let m = graph.getModule fileIndex
- incl m.flags, sfDirty
- of ideOutline:
- let n = parseFile(fileIndex, graph.cache, graph.config)
- graph.iterateOutlineNodes(n, graph.fileSymbols(fileIndex).deduplicateSymInfoPair)
- of ideChk:
- myLog fmt "Reporting errors for {graph.suggestErrors.len} file(s)"
- for sug in graph.suggestErrorsIter:
- suggestResult(graph.config, sug)
- of ideChkFile:
- let errors = graph.suggestErrors.getOrDefault(fileIndex, @[])
- myLog fmt "Reporting {errors.len} error(s) for {file.string}"
- for error in errors:
- suggestResult(graph.config, error)
- of ideGlobalSymbols:
- var
- counter = 0
- res: seq[SymInfoPair] = @[]
- for s in graph.suggestSymbolsIter:
- if (sfGlobal in s.sym.flags or s.sym.kind in searchableSymKinds) and
- s.sym.info == s.info:
- if contains(s.sym.name.s, file.string):
- inc counter
- res = res.filterIt(not it.info.exactEquals(s.info))
- res.add s
- # stop after first 1000 matches...
- if counter > 1000:
- break
- # ... then sort them by weight ...
- res.sort() do (left, right: SymInfoPair) -> int:
- let
- leftString = left.sym.name.s
- rightString = right.sym.name.s
- leftIndex = leftString.find(file.string)
- rightIndex = rightString.find(file.string)
- if leftIndex == rightIndex:
- result = cmp(toLowerAscii(leftString),
- toLowerAscii(rightString))
- else:
- result = cmp(leftIndex, rightIndex)
- # ... and send first 100 results
- if res.len > 0:
- for i in 0 .. min(100, res.len - 1):
- let s = res[i]
- graph.suggestResult(s.sym, s.info)
- of ideDeclaration:
- let s = graph.findSymData(file, line, col)
- if not s.isNil:
- # find first mention of the symbol in the file containing the definition.
- # It is either the definition or the declaration.
- var first: SymInfoPair
- let db = graph.fileSymbols(s.sym.info.fileIndex).deduplicateSymInfoPair
- for i in db.lineInfo.low..db.lineInfo.high:
- if s.sym.symbolEqual(db.sym[i]):
- first = db.getSymInfoPair(i)
- break
- if s.info.exactEquals(first.info):
- # we are on declaration, go to definition
- graph.suggestResult(first.sym, first.sym.info, ideDeclaration)
- else:
- # we are on definition or usage, look for declaration
- graph.suggestResult(first.sym, first.info, ideDeclaration)
- of ideExpand:
- var level: int = high(int)
- let index = skipWhitespace(tag, 0);
- let trimmed = substr(tag, index)
- if not (trimmed == "" or trimmed == "all"):
- discard parseInt(trimmed, level, 0)
- conf.expandPosition = newLineInfo(fileIndex, line, col)
- conf.expandLevels = level
- conf.expandProgress = false
- conf.expandNodeResult = ""
- graph.markDirty fileIndex
- graph.markClientsDirty fileIndex
- graph.recompilePartially()
- var suggest = Suggest()
- suggest.section = ideExpand
- suggest.version = 3
- suggest.line = line
- suggest.column = col
- suggest.doc = graph.config.expandNodeResult
- if suggest.doc != "":
- let
- n = parseFile(fileIndex, graph.cache, graph.config)
- endInfo = n.calculateExpandRange(conf.expandPosition)
- suggest.endLine = endInfo.line
- suggest.endCol = endInfo.col
- suggestResult(graph.config, suggest)
- graph.markDirty fileIndex
- graph.markClientsDirty fileIndex
- of ideInlayHints:
- myLog fmt "Executing inlayHints"
- var endLine = 0
- var endCol = -1
- var i = 0
- i += skipWhile(tag, seps, i)
- i += parseInt(tag, endLine, i)
- i += skipWhile(tag, seps, i)
- i += parseInt(tag, endCol, i)
- i += skipWhile(tag, seps, i)
- var typeHints = true
- var exceptionHints = false
- while i <= tag.high:
- var token: string
- i += parseUntil(tag, token, seps, i)
- i += skipWhile(tag, seps, i)
- case token:
- of "+typeHints":
- typeHints = true
- of "-typeHints":
- typeHints = false
- of "+exceptionHints":
- exceptionHints = true
- of "-exceptionHints":
- exceptionHints = false
- else:
- myLog fmt "Discarding unknown inlay hint parameter {token}"
- let s = graph.findSymDataInRange(file, line, col, endLine, endCol)
- for q in s:
- if typeHints and q.sym.kind in {skLet, skVar, skForVar, skConst} and q.isDecl and not q.sym.hasUserSpecifiedType:
- graph.suggestInlayHintResultType(q.sym, q.info, ideInlayHints)
- if exceptionHints and q.sym.kind in {skProc, skFunc, skMethod, skVar, skLet, skParam} and not q.isDecl:
- graph.suggestInlayHintResultException(q.sym, q.info, ideInlayHints, caughtExceptions = q.caughtExceptions, caughtExceptionsSet = q.caughtExceptionsSet)
- else:
- myLog fmt "Discarding {cmd}"
- # v3 end
- when isMainModule:
- handleCmdLine(newIdentCache(), newConfigRef())
- else:
- export Suggest
- export IdeCmd
- export AbsoluteFile
- type NimSuggest* = ref object
- graph: ModuleGraph
- idle: int
- cachedMsgs: CachedMsgs
- proc initNimSuggest*(project: string, nimPath: string = ""): NimSuggest =
- var retval: ModuleGraph
- proc mockCommand(graph: ModuleGraph) =
- retval = graph
- let conf = graph.config
- conf.setCmd cmdIdeTools
- defineSymbol(conf.symbols, $conf.backend)
- clearPasses(graph)
- registerPass graph, verbosePass
- registerPass graph, semPass
- wantMainModule(conf)
- if not fileExists(conf.projectFull):
- quit "cannot find file: " & conf.projectFull.string
- add(conf.searchPaths, conf.libpath)
- conf.setErrorMaxHighMaybe
- # do not print errors, but log them
- conf.writelnHook = myLog
- conf.structuredErrorHook = nil
- # compile the project before showing any input so that we already
- # can answer questions right away:
- compileProject(graph)
- proc mockCmdLine(pass: TCmdLinePass, cmd: string; conf: ConfigRef) =
- conf.suggestVersion = 0
- let a = unixToNativePath(project)
- if dirExists(a) and not fileExists(a.addFileExt("nim")):
- conf.projectName = findProjectNimFile(conf, a)
- # don't make it worse, report the error the old way:
- if conf.projectName.len == 0: conf.projectName = a
- else:
- conf.projectName = a
- # if processArgument(pass, p, argsCount): break
- let
- cache = newIdentCache()
- conf = newConfigRef()
- self = NimProg(
- suggestMode: true,
- processCmdLine: mockCmdLine
- )
- self.initDefinesProg(conf, "nimsuggest")
- self.processCmdLineAndProjectPath(conf)
- if gMode != mstdin:
- conf.writelnHook = proc (msg: string) = discard
- # Find Nim's prefix dir.
- if nimPath == "":
- let binaryPath = findExe("nim")
- if binaryPath == "":
- raise newException(IOError,
- "Cannot find Nim standard library: Nim compiler not in PATH")
- conf.prefixDir = AbsoluteDir binaryPath.splitPath().head.parentDir()
- if not dirExists(conf.prefixDir / RelativeDir"lib"):
- conf.prefixDir = AbsoluteDir""
- else:
- conf.prefixDir = AbsoluteDir nimPath
- #msgs.writelnHook = proc (line: string) = log(line)
- myLog("START " & conf.projectFull.string)
- var graph = newModuleGraph(cache, conf)
- if self.loadConfigsAndProcessCmdLine(cache, conf, graph):
- mockCommand(graph)
- if gLogging:
- myLog("Search paths:")
- for it in conf.searchPaths:
- myLog(" " & it.string)
- retval.doStopCompile = proc (): bool = false
- return NimSuggest(graph: retval, idle: 0, cachedMsgs: @[])
- proc runCmd*(nimsuggest: NimSuggest, cmd: IdeCmd, file, dirtyfile: AbsoluteFile, line, col: int): seq[Suggest] =
- var retval: seq[Suggest] = @[]
- let conf = nimsuggest.graph.config
- conf.ideCmd = cmd
- conf.writelnHook = proc (line: string) =
- retval.add(Suggest(section: ideMsg, doc: line))
- conf.suggestionResultHook = proc (s: Suggest) =
- retval.add(s)
- conf.writelnHook = proc (s: string) =
- stderr.write s & "\n"
- if conf.ideCmd == ideKnown:
- retval.add(Suggest(section: ideKnown, quality: ord(fileInfoKnown(conf, file))))
- elif conf.ideCmd == ideProject:
- retval.add(Suggest(section: ideProject, filePath: string conf.projectFull))
- else:
- if conf.ideCmd == ideChk:
- for cm in nimsuggest.cachedMsgs: errorHook(conf, cm.info, cm.msg, cm.sev)
- if conf.ideCmd == ideChk:
- conf.structuredErrorHook = proc (conf: ConfigRef; info: TLineInfo; msg: string; sev: Severity) =
- retval.add(Suggest(section: ideChk, filePath: toFullPath(conf, info),
- line: toLinenumber(info), column: toColumn(info), doc: msg,
- forth: $sev))
- else:
- conf.structuredErrorHook = nil
- executeNoHooks(conf.ideCmd, file, dirtyfile, line, col, nimsuggest.graph)
- return retval
|