Handle complete_request in gojupyterscaffold

cmd/lgo-internal/kernel.go

@@ -118,6 +118,11 @@ func (h *handlers) HandleExecuteRequest(ctx context.Context, r *scaffold.Execute
 	}
 }
 
+func (*handlers) HandleComplete(req *scaffold.CompleteRequest) *scaffold.CompleteReply {
+	// Not implemented
+	return nil
+}
+
 func (h *handlers) HandleInspect(r *scaffold.InspectRequest) *scaffold.InspectReply {
 	index := -1
 	count := 0

jupyter/gojupyterscaffold-example/main.go

@@ -64,6 +64,15 @@ loop:
 	return res
 }
 
+func (*handlers) HandleComplete(req *scaffold.CompleteRequest) *scaffold.CompleteReply {
+	return &scaffold.CompleteReply{
+		Status:      "ok",
+		Matches:     []string{"abc", "xyz", "123", "こんにちは"},
+		CursorStart: req.CursorPos,
+		CursorEnd:   req.CursorPos,
+	}
+}
+
 func (*handlers) HandleInspect(req *scaffold.InspectRequest) *scaffold.InspectReply {
 	return &scaffold.InspectReply{
 		Status: "ok",

jupyter/gojupyterscaffold/api.go

@@ -8,6 +8,7 @@ type RequestHandlers interface {
 		req *ExecuteRequest,
 		writeStream func(name, text string),
 		writeDisplayData func(data *DisplayData)) *ExecuteResult
+	HandleComplete(req *CompleteRequest) *CompleteReply
 	HandleInspect(req *InspectRequest) *InspectReply
 }
 
@@ -55,6 +56,34 @@ type InspectReply struct {
 	Data map[string]interface{} `json:"data,omitempty"`
 }
 
+// http://jupyter-client.readthedocs.io/en/latest/messaging.html#completion
+type CompleteRequest struct {
+	// The code context in which completion is requested
+	// this may be up to an entire multiline cell, such as
+	// 'foo = a.isal'
+	Code string `json:"code"`
+	// The cursor position within 'code' (in unicode characters) where completion is requested
+	CursorPos int `json:"cursor_pos"`
+}
+
+type CompleteReply struct {
+	// The list of all matches to the completion request, such as
+	// ['a.isalnum', 'a.isalpha'] for the above example.
+	Matches []string `json:"matches"`
+
+	// The range of text that should be replaced by the above matches when a completion is accepted.
+	// typically cursor_end is the same as cursor_pos in the request.
+	CursorStart int `json:"cursor_start"`
+	CursorEnd   int `json:"cursor_end"`
+
+	// 'metadata' is omitted
+
+	// status should be 'ok' unless an exception was raised during the request,
+	// in which case it should be 'error', along with the usual error message content
+	// in other messages.
+	Status string `json:"status"`
+}
+
 // http://jupyter-client.readthedocs.io/en/latest/messaging.html#execution-results
 type ExecuteResult struct {
 	Status         string `json:"status"`

jupyter/gojupyterscaffold/message.go

@@ -39,6 +39,8 @@ func newContentForMsgType(header *messageHeader) interface{} {
 	switch header.MsgType {
 	case "execute_request":
 		return &ExecuteRequest{}
+	case "complete_request":
+		return &CompleteRequest{}
 	case "inspect_request":
 		return &InspectRequest{}
 	}

jupyter/gojupyterscaffold/shelsocket.go

@@ -312,6 +312,20 @@ func (s *shellSocket) handleMessages() error {
 		// TODO: Send shutdown_reply
 	case "execute_request":
 		s.execQueue.push(&msg, s)
+	case "complete_request":
+		go func() {
+			reply := s.handlers.HandleComplete(msg.Content.(*CompleteRequest))
+			res := newMessageWithParent(&msg)
+			res.Header.MsgType = "complete_reply"
+			if reply != nil {
+				res.Content = reply
+			} else {
+				res.Content = &CompleteReply{
+					Status: "ok",
+				}
+			}
+			s.pushResult(res)
+		}()
 	case "inspect_request":
 		go func() {
 			reply := s.handlers.HandleInspect(msg.Content.(*InspectRequest))