1 files changed, 31 insertions, 0 deletions
diff --git a/gopls/internal/lsp/format.go b/gopls/internal/lsp/format.go
new file mode 100644
index 000000000..773a4690e
--- /dev/null
+++ b/gopls/internal/lsp/format.go
@@ -0,0 +1,31 @@
+// Copyright 2018 The Go Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style
+// license that can be found in the LICENSE file.
+
+package lsp
+
+import (
+	"context"
+
+	"golang.org/x/tools/gopls/internal/lsp/mod"
+	"golang.org/x/tools/gopls/internal/lsp/protocol"
+	"golang.org/x/tools/gopls/internal/lsp/source"
+	"golang.org/x/tools/gopls/internal/lsp/work"
+)
+
+func (s *Server) formatting(ctx context.Context, params *protocol.DocumentFormattingParams) ([]protocol.TextEdit, error) {
+	snapshot, fh, ok, release, err := s.beginFileRequest(ctx, params.TextDocument.URI, source.UnknownKind)
+	defer release()
+	if !ok {
+		return nil, err
+	}
+	switch snapshot.View().FileKind(fh) {
+	case source.Mod:
+		return mod.Format(ctx, snapshot, fh)
+	case source.Go:
+		return source.Format(ctx, snapshot, fh)
+	case source.Work:
+		return work.Format(ctx, snapshot, fh)
+	}
+	return nil, nil
+}