122 lines
		
	
	
		
			2.0 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			122 lines
		
	
	
		
			2.0 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| package main
 | |
| 
 | |
| // #cgo darwin CFLAGS: -I/usr/local/opt/readline/include
 | |
| // #cgo darwin LDFLAGS: -L/usr/local/opt/readline/lib
 | |
| // #cgo LDFLAGS: -lreadline
 | |
| // #include <stdio.h>
 | |
| // #include <stdlib.h>
 | |
| // #include <readline/readline.h>
 | |
| // #include <readline/history.h>
 | |
| import "C"
 | |
| import (
 | |
| 	"os"
 | |
| 	"os/signal"
 | |
| 	"strings"
 | |
| 	"syscall"
 | |
| 	"unsafe"
 | |
| )
 | |
| 
 | |
| func initReadLine() {
 | |
| 	C.rl_catch_sigwinch = 0
 | |
| 	C.rl_catch_signals = 0
 | |
| 	c := make(chan os.Signal, 1)
 | |
| 	signal.Notify(c, syscall.SIGWINCH)
 | |
| 	signal.Notify(c, os.Interrupt)
 | |
| 	go func() {
 | |
| 		for sig := range c {
 | |
| 			switch sig {
 | |
| 			case syscall.SIGWINCH:
 | |
| 				C.rl_resize_terminal()
 | |
| 
 | |
| 			case os.Interrupt:
 | |
| 				C.rl_cleanup_after_signal()
 | |
| 			default:
 | |
| 
 | |
| 			}
 | |
| 		}
 | |
| 	}()
 | |
| }
 | |
| 
 | |
| func readLine(prompt *string) *string {
 | |
| 	var p *C.char
 | |
| 
 | |
| 	//readline allows an empty prompt(NULL)
 | |
| 	if prompt != nil {
 | |
| 		p = C.CString(*prompt)
 | |
| 	}
 | |
| 
 | |
| 	ret := C.readline(p)
 | |
| 
 | |
| 	if p != nil {
 | |
| 		C.free(unsafe.Pointer(p))
 | |
| 	}
 | |
| 
 | |
| 	if ret == nil {
 | |
| 		return nil
 | |
| 	} //EOF
 | |
| 
 | |
| 	s := C.GoString(ret)
 | |
| 	C.free(unsafe.Pointer(ret))
 | |
| 	return &s
 | |
| }
 | |
| 
 | |
| func addHistory(s string) {
 | |
| 	p := C.CString(s)
 | |
| 	C.add_history(p)
 | |
| 	C.free(unsafe.Pointer(p))
 | |
| }
 | |
| 
 | |
| var indentCount = 0
 | |
| var str = ""
 | |
| 
 | |
| func (self *JSRepl) setIndent() {
 | |
| 	open := strings.Count(str, "{")
 | |
| 	open += strings.Count(str, "(")
 | |
| 	closed := strings.Count(str, "}")
 | |
| 	closed += strings.Count(str, ")")
 | |
| 	indentCount = open - closed
 | |
| 	if indentCount <= 0 {
 | |
| 		self.prompt = "> "
 | |
| 	} else {
 | |
| 		self.prompt = strings.Join(make([]string, indentCount*2), "..")
 | |
| 		self.prompt += " "
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func (self *JSRepl) read() {
 | |
| 	initReadLine()
 | |
| L:
 | |
| 	for {
 | |
| 		switch result := readLine(&self.prompt); true {
 | |
| 		case result == nil:
 | |
| 			break L
 | |
| 
 | |
| 		case *result != "":
 | |
| 			str += *result + "\n"
 | |
| 
 | |
| 			self.setIndent()
 | |
| 
 | |
| 			if indentCount <= 0 {
 | |
| 				if *result == "exit" {
 | |
| 					self.Stop()
 | |
| 					break L
 | |
| 				}
 | |
| 
 | |
| 				addHistory(str[:len(str)-1]) //allow user to recall this line
 | |
| 
 | |
| 				self.parseInput(str)
 | |
| 
 | |
| 				str = ""
 | |
| 			}
 | |
| 		}
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func (self *JSRepl) PrintValue(v interface{}) {
 | |
| 	method, _ := self.re.vm.Get("prettyPrint")
 | |
| 	v, err := self.re.vm.ToValue(v)
 | |
| 	if err == nil {
 | |
| 		method.Call(method, v)
 | |
| 	}
 | |
| }
 |