Add LanguageID to file

internal/filesystem/document.go

@@ -42,6 +42,10 @@ func (d *document) Version() int {
 	return d.meta.Version()
 }
 
+func (d *document) LanguageID() string {
+	return d.meta.dh.LanguageID()
+}
+
 func (d *document) IsOpen() bool {
 	return d.meta.IsOpen()
 }

internal/filesystem/filesystem_test.go

@@ -590,6 +590,10 @@ func (fh *testHandler) Version() int {
 	return 0
 }
 
+func (fh *testHandler) LanguageID() string {
+	return "terraform"
+}
+
 func testDocumentStorage() DocumentStorage {
 	fs := NewFilesystem()
 	fs.logger = testLogger()

internal/filesystem/types.go

@@ -19,6 +19,7 @@ type DocumentHandler interface {
 	FullPath() string
 	Dir() string
 	Filename() string
+	LanguageID() string
 }
 
 type VersionedDocumentHandler interface {

internal/langserver/handlers/did_open_test.go

@@ -2,10 +2,16 @@ package handlers
 
 import (
 	"fmt"
+	"path/filepath"
 	"testing"
 
+	"github.com/google/go-cmp/cmp"
+	"github.com/hashicorp/terraform-ls/internal/filesystem"
 	"github.com/hashicorp/terraform-ls/internal/langserver"
 	"github.com/hashicorp/terraform-ls/internal/langserver/session"
+	"github.com/hashicorp/terraform-ls/internal/lsp"
+	"github.com/hashicorp/terraform-ls/internal/terraform/exec"
+	"github.com/stretchr/testify/mock"
 )
 
 func TestLangServer_didOpenWithoutInitialization(t *testing.T) {
@@ -24,3 +30,69 @@ func TestLangServer_didOpenWithoutInitialization(t *testing.T) {
 		}
 	}`, TempDir(t).URI())}, session.SessionNotInitialized.Err())
 }
+
+func TestLangServer_didOpenLanguageIdStored(t *testing.T) {
+	tmpDir := TempDir(t)
+	fs := filesystem.NewFilesystem()
+
+	ls := langserver.NewLangServerMock(t, NewMockSession(&MockSessionInput{
+		TerraformCalls: &exec.TerraformMockCalls{
+			PerWorkDir: map[string][]*mock.Call{
+				tmpDir.Dir(): validTfMockCalls(),
+			},
+		},
+		Filesystem: fs,
+	}))
+	stop := ls.Start(t)
+	defer stop()
+
+	ls.Call(t, &langserver.CallRequest{
+		Method: "initialize",
+		ReqParams: fmt.Sprintf(`{
+	    "capabilities": {},
+	    "rootUri": %q,
+	    "processId": 12345
+	}`, tmpDir.URI())})
+	ls.Notify(t, &langserver.CallRequest{
+		Method:    "initialized",
+		ReqParams: "{}",
+	})
+
+	originalText := `variable "service_host" {
+  default = "blah"
+}
+`
+	ls.Call(t, &langserver.CallRequest{
+		Method: "textDocument/didOpen",
+		ReqParams: fmt.Sprintf(`{
+    "textDocument": {
+        "languageId": "terraform",
+        "version": 0,
+        "uri": "%s/main.tf",
+        "text": %q
+    }
+}`, TempDir(t).URI(), originalText)})
+	path := filepath.Join(TempDir(t).Dir(), "main.tf")
+	doc, err := fs.GetDocument(lsp.FileHandlerFromPath(path))
+	if err != nil {
+		t.Fatal(err)
+	}
+	languageID := doc.LanguageID()
+	if diff := cmp.Diff(languageID, string("terraform")); diff != "" {
+		t.Fatalf("unexpected languageID: %s", diff)
+	}
+	fullPath := doc.FullPath()
+	if diff := cmp.Diff(fullPath, string(path)); diff != "" {
+		t.Fatalf("unexpected fullPath: %s", diff)
+	}
+
+	version := doc.Version()
+	if diff := cmp.Diff(version, int(0)); diff != "" {
+		t.Fatalf("unexpected version: %s", diff)
+	}
+
+	uri := doc.URI()
+	if diff := cmp.Diff(uri, string(fmt.Sprintf("file://%s", path))); diff != "" {
+		t.Fatalf("unexpected URI: %s", diff)
+	}
+}

internal/lsp/file.go

@@ -11,13 +11,15 @@ type File interface {
 	Dir() string
 	Filename() string
 	Lines() source.Lines
+	LanguageID() string
 }
 
 type file struct {
-	fh      *fileHandler
-	ls      source.Lines
-	text    []byte
-	version int
+	fh         *fileHandler
+	ls         source.Lines
+	text       []byte
+	version    int
+	languageID string
 }
 
 func (f *file) URI() string {
@@ -44,6 +46,10 @@ func (f *file) Lines() source.Lines {
 	return f.lines()
 }
 
+func (f *file) LanguageID() string {
+	return f.languageID
+}
+
 func (f *file) lines() source.Lines {
 	if f.ls == nil {
 		f.ls = source.MakeSourceLines(f.fh.Filename(), f.text)
@@ -57,8 +63,9 @@ func (f *file) Version() int {
 
 func FileFromDocumentItem(doc lsp.TextDocumentItem) *file {
 	return &file{
-		fh:      FileHandlerFromDocumentURI(doc.URI),
-		text:    []byte(doc.Text),
-		version: int(doc.Version),
+		fh:         FileHandlerFromDocumentURI(doc.URI),
+		text:       []byte(doc.Text),
+		version:    int(doc.Version),
+		languageID: doc.LanguageID,
 	}
 }

internal/lsp/file_handler.go

@@ -67,6 +67,10 @@ func (fh *fileHandler) URI() string {
 	return string(fh.uri)
 }
 
+func (fh *fileHandler) LanguageID() string {
+	return ""
+}
+
 type versionedFileHandler struct {
 	fileHandler
 	v int