Christophe Weblog Wiki Code Publications Music
sort out printing functions
[swankr.git] / swank.R
1 ### This program is free software; you can redistribute it and/or
2 ### modify it under the terms of the GNU General Public Licence as
3 ### published by the Free Software Foundation; either version 2 of the
4 ### Licence, or (at your option) any later version.
5 ###
6 ### This program is distributed in the hope that it will be useful,
7 ### but WITHOUT ANY WARRANTY; without even the implied warranty of
8 ### MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
9 ### GNU General Public Licence for more details.
10 ###
11 ### A copy of version 2 of the GNU General Public Licence is available
12 ### at <http://www.gnu.org/licenses/old-licenses/gpl-2.0.txt>; the
13 ### latest version of the GNU General Public Licence is available at
14 ### <http://www.gnu.org/licenses/gpl.txt>.
15
16 swank <- function(port=4005) {
17   acceptConnections(port, FALSE)
18 }
19
20 startSwank <- function(portFile) {
21   acceptConnections(FALSE, portFile)
22 }
23
24 acceptConnections <- function(port, portFile) {
25   s <- socketConnection(host="localhost", server=TRUE, port=port, open="r+b")
26   on.exit(close(s))
27   serve(s)
28 }
29
30 serve <- function(io) {
31   mainLoop(io)
32 }
33
34 mainLoop <- function(io) {
35   slimeConnection <- new.env()
36   slimeConnection$io <- io
37   while(TRUE) {
38     withRestarts(tryCatch(dispatch(slimeConnection, readPacket(io)),
39                           swankTopLevel=function(c) NULL),
40                  abort="return to SLIME's toplevel")
41   }
42 }
43
44 dispatch <- function(slimeConnection, event, sldbState=NULL) {
45   str(event)
46   kind <- event[[1]]
47   if(kind == quote(`:emacs-rex`)) {
48     do.call("emacsRex", c(list(slimeConnection), list(sldbState), event[-1]))
49   }
50 }
51
52 sendToEmacs <- function(slimeConnection, obj) {
53   io <- slimeConnection$io
54   str(obj)
55   payload <- writeSexpToString(obj)
56   writeChar(sprintf("%06x", nchar(payload)), io, eos=NULL)
57   writeChar(payload, io, eos=NULL)
58   flush(io)
59   cat(sprintf("%06x", nchar(payload)), payload, sep="")
60 }
61
62 callify <- function(form) {
63   ## we implement here the conversion from Lisp S-expression (or list)
64   ## expressions of code into our own, swankr, calling convention,
65   ## with slimeConnection and sldbState as first and second arguments.
66   ## as.call() gets us part of the way, but we need to walk the list
67   ## recursively to mimic CL:EVAL; we need to avoid converting R
68   ## special operators which we are punning (only `quote`, for now)
69   ## into this calling convention.
70   if(is.list(form)) {
71     if(form[[1]] == quote(quote)) {
72       as.call(form)
73     } else {
74       as.call(c(list(form[[1]], quote(slimeConnection), quote(sldbState)), lapply(form[-1], callify)))
75     }
76   } else {
77     form
78   }
79 }
80
81 emacsRex <- function(slimeConnection, sldbState, form, pkg, thread, id, level=0) {
82   ok <- FALSE
83   value <- NULL
84   tryCatch({
85     withCallingHandlers({
86       call <- callify(form)
87       value <- eval(call)
88       ok <- TRUE
89     }, error=function(c) {
90       newSldbState <- makeSldbState(c, if(is.null(sldbState)) 0 else sldbState$level+1, id)
91       withRestarts(sldbLoop(slimeConnection, newSldbState, id), abort=paste("return to sldb level", newSldbState$level)) })},
92     finally=sendToEmacs(slimeConnection, list(quote(`:return`), if(ok) list(quote(`:ok`), value) else list(quote(`:abort`)), id)))
93 }
94
95 makeSldbState <- function(condition, level, id) {
96   calls <- rev(sys.calls())[-1]
97   frames <- rev(sys.frames())[-1]
98   restarts <- rev(computeRestarts(condition))[-1]
99   ret <- list(condition=condition, level=level, id=id, restarts=restarts, calls=calls, frames=frames)
100   class(ret) <- c("sldbState", class(ret))
101   ret
102 }
103
104 sldbLoop <- function(slimeConnection, sldbState, id) {
105   tryCatch({
106     io <- slimeConnection$io
107     sendToEmacs(slimeConnection, c(list(quote(`:debug`), id, sldbState$level), `swank:debugger-info-for-emacs`(slimeConnection, sldbState)))
108     sendToEmacs(slimeConnection, list(quote(`:debug-activate`), id, sldbState$level, FALSE))
109     while(TRUE) {
110       dispatch(slimeConnection, readPacket(io), sldbState)
111     }
112   }, finally=sendToEmacs(slimeConnection, c(list(quote(`:debug-return`), id, sldbState$level, FALSE))))
113 }
114
115 readPacket <- function(io) {
116   socketSelect(list(io))
117   header <- readChunk(io, 6)
118   len <- strtoi(header, base=16)
119   payload <- readChunk(io, len)
120   readSexpFromString(payload)
121 }
122
123 readChunk <- function(io, len) {
124   buffer <- readChar(io, len)
125   if(nchar(buffer) != len) {
126     stop("short read in readChunk")
127   }
128   buffer
129 }
130
131 readSexpFromString <- function(string) {
132   pos <- 1
133   read <- function() {
134     skipWhitespace()
135     char <- substr(string, pos, pos)
136     switch(char,
137            "("=readList(),
138            "\""=readString(),
139            "'"=readQuote(),
140            {
141              if(pos > nchar(string))
142                stop("EOF during read")
143              obj <- readNumberOrSymbol()
144              if(obj == quote(`.`)) {
145                stop("Consing dot not implemented")
146              }
147              obj
148            })
149   }
150   skipWhitespace <- function() {
151     while(substr(string, pos, pos) %in% c(" ", "\t", "\n")) {
152       pos <<- pos + 1
153     }
154   }
155   readList <- function() {
156     ret <- list()
157     pos <<- pos + 1
158     while(TRUE) {
159       skipWhitespace()
160       char <- substr(string, pos, pos)
161       if(char == ")") {
162         pos <<- pos + 1
163         break
164       } else {
165         obj <- read()
166         if(length(obj) == 1 && obj == quote(`.`)) {
167           stop("Consing dot not implemented")
168         }
169         ret <- c(ret, list(obj))
170       }
171     }
172     ret
173   }
174   readString <- function() {
175     ret <- ""
176     addChar <- function(c) { ret <<- paste(ret, c, sep="") }
177     while(TRUE) {
178       pos <<- pos + 1
179       char <- substr(string, pos, pos)
180       switch(char,
181              "\""={ pos <<- pos + 1; break },
182              "\\"={ pos <<- pos + 1
183                     char2 <- substr(string, pos, pos)
184                     switch(char2,
185                            "\""=addChar(char2),
186                            "\\"=addChar(char2),
187                            stop("Unrecognized escape character")) },
188              addChar(char))
189     }
190     ret
191   }
192   readNumberOrSymbol <- function() {
193     token <- readToken()
194     if(nchar(token)==0) {
195       stop("End of file reading token")
196     } else if(grepl("^[0-9]+$", token)) {
197       strtoi(token)
198     } else if(grepl("^[0-9]+\\.[0-9]+$", token)) {
199       as.double(token)
200     } else {
201       as.name(token)
202     }
203   }
204   readToken <- function() {
205     token <- ""
206     while(TRUE) {
207       char <- substr(string, pos, pos)
208       if(char == "") {
209         break;
210       } else if(char %in% c(" ", "\n", "\t", "(", ")", "\"", "'")) {
211         break;
212       } else {
213         token <- paste(token, char, sep="")
214         pos <<- pos + 1
215       }
216     }
217     token
218   }
219   read()
220 }
221
222 writeSexpToString <- function(obj) {
223   writeSexpToStringLoop <- function(obj) {
224     switch(typeof(obj),
225            "character"={ string <- paste(string, "\"", gsub("([\"\\])", "\\\\\\1", obj), "\"", sep="") },
226            "list"={ string <- paste(string, "(", sep="")
227                     max <- length(obj)
228                     if(max > 0) {
229                       for(i in 1:max) {
230                         string <- paste(string, writeSexpToString(obj[[i]]), sep="")
231                         if(i != max) {
232                           string <- paste(string, " ", sep="")
233                         }
234                       }
235                     }
236                     string <- paste(string, ")", sep="") },
237            "symbol"={ string <- paste(string, as.character(obj), sep="") },
238            "logical"={ string <- if(obj) { paste(string, "t", sep="") } else { paste(string, "nil", sep="") }},
239            "double"={ string <- paste(string, as.character(obj), sep="") },
240            "integer"={ string <- paste(string, as.character(obj), sep="") },
241            stop(paste("can't write object ", obj, sep="")))
242     string
243   }
244   string <- ""
245   writeSexpToStringLoop(obj)
246 }
247
248 prin1ToString <- function(val) {
249   paste(deparse(val, backtick=TRUE, control=c("delayPromises", "keepNA")),
250         sep="", collapse="\n")
251 }
252
253 printToString <- function(val) {
254   paste(capture.output(print(val)), sep="", collapse="\n")
255 }
256
257 `swank:connection-info` <- function (slimeConnection, sldbState) {
258   list(quote(`:pid`), Sys.getpid(),
259        quote(`:package`), list(quote(`:name`), "R", quote(`:prompt`), "R> "),
260        quote(`:lisp-implementation`), list(quote(`:type`), "R",
261                                            quote(`:name`), "R",
262                                            quote(`:version`), paste(R.version$major, R.version$minor, sep=".")))
263 }
264
265 `swank:swank-require` <- function (slimeConnection, sldbState, contribs) {
266   for(contrib in contribs) {
267     filename <- sprintf("%s.R", as.character(contrib))
268     if(file.exists(filename)) {
269       source(filename, verbose=TRUE)
270     }
271   }
272   list()
273 }
274
275 `swank:create-repl` <- function(slimeConnection, sldbState, env, ...) {
276   list("R", "R")
277 }
278
279 makeReplResult <- function(value) {
280   string <- printToString(value)
281   list(quote(`:write-string`), string,
282        quote(`:repl-result`))
283 }
284
285 makeReplResultFunction <- makeReplResult
286
287 sendReplResult <- function(slimeConnection, value) {
288   result <- makeReplResultFunction(value)
289   sendToEmacs(slimeConnection, result)
290 }
291
292 sendReplResultFunction <- sendReplResult
293
294 `swank:listener-eval` <- function(slimeConnection, sldbState, string) {
295   string <- gsub("#\\.\\(swank:lookup-presented-object-or-lose([^)]*)\\)", ".(`swank:lookup-presented-object-or-lose`(slimeConnection, sldbState,\\1))", string)
296   expr <- parse(text=string)[[1]]
297   lookedup <- do.call("bquote", list(expr))
298   value <- eval(lookedup, envir = globalenv())
299   sendReplResultFunction(slimeConnection, value)
300   list()
301 }
302
303 `swank:autodoc` <- function(slimeConnection, sldbState, rawForm, ...) {
304   "No Arglist Information"
305 }
306
307 `swank:operator-arglist` <- function(slimeConnection, sldbState, op, package) {
308   list()
309 }
310
311 `swank:throw-to-toplevel` <- function(slimeConnection, sldbState) {
312   condition <- simpleCondition("Throw to toplevel")
313   class(condition) <- c("swankTopLevel", class(condition))
314   signalCondition(condition)
315 }
316
317 `swank:backtrace` <- function(slimeConnection, sldbState, from=0, to=NULL) {
318   calls <- sldbState$calls
319   if(is.null(to)) to <- length(calls)
320   from <- from+1
321   calls <- lapply(calls[from:to],
322                   { frameNumber <- from-1;
323                     function (x) {
324                       ret <- list(frameNumber, paste(format(x), sep="", collapse=" "))
325                       frameNumber <<- 1+frameNumber
326                       ret
327                     }
328                   })
329 }
330
331 computeRestartsForEmacs <- function (sldbState) {
332   lapply(sldbState$restarts,
333          function(x) {
334            ## this is all a little bit internalsy
335            restartName <- x[[1]][[1]]
336            description <- restartDescription(x)
337            list(restartName, if(is.null(description)) restartName else description)
338          })
339 }
340
341 `swank:debugger-info-for-emacs` <- function(slimeConnection, sldbState, from=0, to=NULL) {
342   list(list(as.character(sldbState$condition), sprintf("  [%s]", class(sldbState$condition)[[1]]), FALSE),
343        computeRestartsForEmacs(sldbState),
344        `swank:backtrace`(slimeConnection, sldbState, from, to),
345        list(sldbState$id))
346 }
347
348 `swank:invoke-nth-restart-for-emacs` <- function(slimeConnection, sldbState, level, n) {
349   if(sldbState$level == level) {
350     invokeRestart(sldbState$restarts[[n+1]])
351   }
352 }
353
354 `swank:frame-source-location` <- function(slimeConnection, sldbState, n) {
355   call <- sldbState$calls[[n+1]]
356   srcref <- attr(call, "srcref")
357   srcfile <- attr(srcref, "srcfile")
358   if(is.null(srcfile)) {
359     list(quote(`:error`), "no srcfile")
360   } else {
361     list(quote(`:location`),
362          list(quote(`:file`), sprintf("%s/%s", srcfile$wd, srcfile$filename)),
363          list(quote(`:line`), srcref[[1]], srcref[[2]]-1),
364          FALSE)
365   }
366 }
367
368 `swank:buffer-first-change` <- function(slimeConnection, sldbState, filename) {
369   FALSE
370 }
371
372 `swank:eval-string-in-frame` <- function(slimeConnection, sldbState, string, index) {
373   frame <- sldbState$frames[[1+index]]
374   withRetryRestart("retry SLIME interactive evaluation request",
375                    value <- eval(parse(text=string), envir=frame))
376   printToString(value)
377 }
378
379 `swank:frame-locals-and-catch-tags` <- function(slimeConnection, sldbState, index) {
380   str(sldbState$frames)
381   frame <- sldbState$frames[[1+index]]
382   objs <- ls(envir=frame)
383   list(lapply(objs, function(name) { list(quote(`:name`), name,
384                                           quote(`:id`), 0,
385                                           quote(`:value`), printToString(eval(parse(text=name), envir=frame))) }),
386        list())
387 }
388
389 `swank:simple-completions` <- function(slimeConnection, sldbState, prefix, package) {
390   ## fails multiply if prefix contains regexp metacharacters
391   matches <- apropos(sprintf("^%s", prefix), ignore.case=FALSE)
392   nmatches <- length(matches)
393   if(nmatches == 0) {
394     list(list(), "")
395   } else {
396     longest <- matches[order(nchar(matches))][1]
397     while(length(grep(sprintf("^%s", longest), matches)) < nmatches) {
398       longest <- substr(longest, 1, nchar(longest)-1)
399     }
400     list(as.list(matches), longest)
401   }
402 }
403
404 `swank:compile-string-for-emacs` <- function(slimeConnection, sldbState, string, buffer, position, filename, policy) {
405   # FIXME: I think in parse() here we can use srcref to associate
406   # buffer/filename/position to the objects.  Or something.
407   withRestarts({ times <- system.time(eval(parse(text=string), envir = globalenv())) },
408                abort="abort compilation")
409   list(quote(`:compilation-result`), list(), TRUE, times[3])
410 }
411
412 withRetryRestart <- function(description, expr) {
413   call <- substitute(expr)
414   retry <- TRUE
415   while(retry) {
416     retry <- FALSE
417     withRestarts(eval.parent(call),
418                  retry=list(description=description,
419                    handler=function() retry <<- TRUE))
420   }
421 }
422
423 `swank:interactive-eval` <-  function(slimeConnection, sldbState, string) {
424   withRetryRestart("retry SLIME interactive evaluation request",
425                    value <- eval(parse(text=string), envir=globalenv()))
426   prin1ToString(value)
427 }
428
429 `swank:eval-and-grab-output` <- function(slimeConnection, sldbState, string) {
430   withRetryRestart("retry SLIME interactive evaluation request",
431                    { output <-
432                        capture.output(value <- eval(parse(text=string),
433                                                     envir=globalenv())) })
434   output <- paste(output, sep="", collapse="\n")
435   list(output, prin1ToString(value))
436 }
437
438 `swank:find-definitions-for-emacs` <- function(slimeConnection, sldbState, string) {
439   if(exists(string, envir = globalenv())) {
440     thing <- get(string, envir = globalenv())
441     if(inherits(thing, "function")) {
442       body <- body(thing)
443       srcref <- attr(body, "srcref")
444       srcfile <- attr(body, "srcfile")
445       if(is.null(srcfile)) {
446         list()
447       } else {
448         filename <- get("filename", srcfile)
449         list(list(sprintf("function %s", string),
450                   list(quote(`:location`),
451                        list(quote(`:file`), sprintf("%s/%s", srcfile$wd, srcfile$filename)),
452                        list(quote(`:line`), srcref[[2]][[1]], srcref[[2]][[2]]-1),
453                        list())))
454       }
455     } else {
456       list()
457     }
458   } else {
459     list()
460   }
461 }
462
463 `swank:value-for-editing` <- function(slimeConnection, sldbState, string) {
464   paste(deparse(eval(parse(text=string), envir = globalenv()), control="all"),
465         collapse="\n", sep="")
466 }
467
468 `swank:commit-edited-value` <- function(slimeConnection, sldbState, string, value) {
469   eval(parse(text=sprintf("%s <- %s", string, value)), envir = globalenv())
470   TRUE
471 }
472
473 resetInspector <- function(slimeConnection) {
474   assign("istate", list(), envir=slimeConnection)
475   assign("inspectorHistory", NULL, envir=slimeConnection)
476 }
477
478 `swank:init-inspector` <- function(slimeConnection, sldbState, string) {
479   withRetryRestart("retry SLIME inspection request",
480                    { resetInspector(slimeConnection)
481                      value <- inspectObject(slimeConnection, eval(parse(text=string), envir=globalenv()))
482                    })
483   value
484 }
485
486 inspectObject <- function(slimeConnection, object) {
487   previous <- slimeConnection$istate
488   slimeConnection$istate <- new.env()
489   slimeConnection$istate$object <- object
490   slimeConnection$istate$previous <- previous
491   slimeConnection$istate$content <- emacsInspect(object)
492   if(!(object %in% slimeConnection$inspectorHistory)) {
493     slimeConnection$inspectorHistory <- c(slimeConnection$inspectorHistory, object)
494   }
495   if(!is.null(slimeConnection$istate$previous)) {
496     slimeConnection$istate$previous$`next` <- slimeConnection$istate
497   }
498   istateToElisp(slimeConnection$istate)
499 }
500
501 valuePart <- function(istate, object, string) {
502   list(quote(`:value`),
503        if(is.null(string)) printToString(object) else string,
504        assignIndexInParts(object, istate))
505 }
506
507 preparePart <- function(istate, part) {
508   if(is.character(part)) {
509     list(part)
510   } else {
511     switch(as.character(part[[1]]),
512            `:newline` = list("\n"),
513            `:value` = valuePart(istate, part[[2]], part[[3]]),
514            `:line` = list(printToString(part[[2]]), ": ",
515              valuePart(istate, part[[3]], NULL), "\n"))
516   }
517 }
518
519 prepareRange <- function(istate, start, end) {
520   range <- istate$content[start+1:min(end+1, length(istate$content))]
521   ps <- NULL
522   for(part in range) {
523     ps <- c(ps, preparePart(istate, part))
524   }
525   list(ps, if(length(ps)<end-start) { start+length(ps) } else { end+1000 },
526        start, end)
527 }
528
529 assignIndexInParts <- function(object, istate) {
530   ret <- 1+length(istate$parts)
531   istate$parts <- c(istate$parts, list(object))
532   ret
533 }
534
535 istateToElisp <- function(istate) {
536   list(quote(`:title`), deparse(istate$object, control="all", nlines=1),
537        quote(`:id`), assignIndexInParts(istate$object, istate),
538        quote(`:content`), prepareRange(istate, 0, 500))
539 }
540
541 emacsInspect <- function(object) {
542   UseMethod("emacsInspect")
543 }
544
545 emacsInspect.default <- function(thing) {
546   c(list(paste("a ", class(thing)[[1]], sep=""), list(quote(`:newline`))))
547 }
548
549 emacsInspect.list <- function(list) {
550   c(list("a list", list(quote(`:newline`))),
551     mapply(function(name, value) { list(list(quote(`:line`), name, value)) },
552            names(list), list))
553 }
554
555 emacsInspect.numeric <- function(numeric) {
556   c(list("a numeric", list(quote(`:newline`))),
557     mapply(function(name, value) { list(list(quote(`:line`), name, value)) },
558            (1:length(numeric)), numeric))
559 }
560
561 `swank:quit-inspector` <- function(slimeConnection, sldbState) {
562   resetInspector(slimeConnection)
563   FALSE
564 }
565
566 `swank:inspector-nth-part` <- function(slimeConnection, sldbState, index) {
567   slimeConnection$istate$parts[[index]]
568 }
569
570 `swank:inspect-nth-part` <- function(slimeConnection, sldbState, index) {
571   object <- `swank:inspector-nth-part`(slimeConnection, sldbState, index)
572   inspectObject(slimeConnection, object)
573 }
574
575 `swank:inspector-pop` <- function(slimeConnection, sldbState) {
576   if(!is.null(slimeConnection$istate$previous)) {
577     slimeConnection$istate <- slimeConnection$istate$previous
578     istateToElisp(slimeConnection$istate)
579   } else {
580     FALSE
581   }
582 }
583
584 `swank:inspector-next` <- function(slimeConnection, sldbState) {
585   if(!is.null(slimeConnection$istate$`next`)) {
586     slimeConnection$istate <- slimeConnection$istate$`next`
587     istateToElisp(slimeConnection$istate)
588   } else {
589     FALSE
590   }
591 }
592
593 `swank:inspector-eval` <- function(slimeConnection, sldbState, string) {
594   expr <- parse(text=string)[[1]]
595   object <- slimeConnection$istate$object
596   if(inherits(object, "list")|inherits(object, "environment")) {
597     substituted <- substituteDirect(expr, object)
598     eval(substituted, envir=globalenv())
599   } else {
600     eval(expr, envir=globalenv())
601   }
602 }
603
604 `swank:inspect-current-condition` <- function(slimeConnection, sldbState) {
605   resetInspector(slimeConnection)
606   inspectObject(slimeConnection, sldbState$condition)
607 }
608
609 `swank:inspect-frame-var` <- function(slimeConnection, sldbState, frame, var) {
610   resetInspector(slimeConnection)
611   frame <- sldbState$frames[[1+frame]]
612   name <- ls(envir=frame)[[1+var]]
613   object <- get(name, envir=frame)
614   inspectObject(slimeConnection, object)
615 }