Add PRQL (#69)

* Add PRQL * Update README.md * Update analyzer_test.go * Update scanner_test.go * Update README.md Co-authored-by: João Daniel Rufino <joaodaniel0405@gmail.com> --------- Co-authored-by: João Daniel Rufino <joaodaniel0405@gmail.com>
JoaoDanielRufino · Oct 6, 2023 · 09ccbee · 09ccbee
1 parent 8b12162
commit 09ccbee
Show file tree

Hide file tree

Showing 5 changed files with 38 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -191,6 +191,7 @@ $ gcloc languages
   PowerShell       |  .ps1          |  #              |
   Processing       |  .pde          |  //             |  /* */
   Protocol Buffers |  .proto        |  //             |
+  PRQL             |  .prql         |  #              |
   Python           |  .py           |  #              |  """ """
   R                |  .r .R         |  #              |
   Rego             |  .rego         |  #              |

diff --git a/internal/constants/languages.go b/internal/constants/languages.go
@@ -183,6 +183,11 @@ var Languages = language.Languages{
 		MultiLineComments: [][]string{{"\"\"\"", "\"\"\""}},
 		Extensions:        []string{".py"},
 	},
+	"PRQL": {
+		LineComments:      []string{"#"},
+		MultiLineComments: [][]string{},
+		Extensions:        []string{".prql"},
+	},
 	"R": {
 		LineComments:      []string{"#"},
 		MultiLineComments: [][]string{},

diff --git a/pkg/analyzer/analyzer_test.go b/pkg/analyzer/analyzer_test.go
@@ -92,6 +92,11 @@ func TestMatchingFiles(t *testing.T) {
 					Extension: ".py",
 					Language:  "Python",
 				},
+				{
+					FilePath:  filepath.Join(codeSamplesDir, "example.prql"),
+					Extension: ".prql",
+					Language:  "PRQL",
+				},
 				{
 					FilePath:  filepath.Join(codeSamplesDir, "main.rb"),
 					Extension: ".rb",
@@ -189,6 +194,11 @@ func TestMatchingFiles(t *testing.T) {
 					Extension: ".py",
 					Language:  "Python",
 				},
+				{
+					FilePath:  filepath.Join(codeSamplesDir, "example.prql"),
+					Extension: ".prql",
+					Language:  "PRQL",
+				},
 				{
 					FilePath:  filepath.Join(codeSamplesDir, "main.rb"),
 					Extension: ".rb",
@@ -287,6 +297,11 @@ func TestMatchingFiles(t *testing.T) {
 					Extension: ".py",
 					Language:  "Python",
 				},
+				{
+					FilePath:  filepath.Join(codeSamplesDir, "example.prql"),
+					Extension: ".prql",
+					Language:  "PRQL",
+				},
 				{
 					FilePath:  filepath.Join(codeSamplesDir, "main.rb"),
 					Extension: ".rb",

diff --git a/pkg/scanner/scanner_test.go b/pkg/scanner/scanner_test.go
@@ -113,6 +113,11 @@ func TestScan(t *testing.T) {
 			Extension: ".txt",
 			Language:  "Plain Text",
 		},
+		{
+			FilePath:  filepath.Join(codeSamplesDir, "example.prql"),
+			Extension: ".prql",
+			Language:  "PRQL",
+		},
 	}
 
 	expected := []scanResult{
@@ -242,6 +247,13 @@ func TestScan(t *testing.T) {
 			BlankLines: 13,
 			Comments:   0,
 		},
+		{
+			Metadata:   files[18],
+			Lines:      5,
+			CodeLines:  4,
+			BlankLines: 0,
+			Comments:   1,
+		},
 	}
 
 	result, err := scanner.Scan(files)

diff --git a/test/fixtures/code_samples/example.prql b/test/fixtures/code_samples/example.prql
@@ -0,0 +1,5 @@
+# This is a comment
+from track_plays
+filter plays > 10_000
+sort length
+take 10