Added support for non-beta search users (#1)

* add support for non-beta search users

Without this change, users who haven't opted into the beta-search are met with a syntax error
when launching their browser. I added an additional flag to modify what goes into the query.

Tested this by trying every combination locally and it worked as expected.

* format

* set default betaSearch value to true

---------

Co-authored-by: nbe <[email protected]>

Author: KitchenTable99

README.md

@@ -23,6 +23,7 @@ https://user-images.githubusercontent.com/33713262/226383032-113b4db8-27a3-4b8f-
         local url = csgithub.search({
           includeFilename = false,
           includeExtension = true,
+          betaSearch = true, -- set to false if you haven't opted in to GitHub Code Search (beta)
         })
 
         csgithub.open(url)

lua/csgithub/init.lua

@@ -6,6 +6,7 @@ M.search = function(args)
 	local default_args = {
 		includeFilename = false,
 		includeExtension = true,
+		betaSearch = true,
 	}
 
 	local merged_args = vim.tbl_extend("force", default_args, args or {})

lua/csgithub/query.lua

@@ -3,13 +3,27 @@ local M = {}
 M.construct_query_path = function(args)
 	local ext = vim.fn.expand("%:e")
 
-	if args.includeExtension and not args.includeFilename then
+	local onlyExtension = args.includeExtension and not args.includeFilename
+
+	if onlyExtension and args.betaSearch then
 		return "*." .. ext
+	elseif onlyExtension then
+		return "." .. ext
 	else
 		return vim.fn.expand("%:t")
 	end
 end
 
+M.construct_search_field = function(args)
+	if not args.betaSearch and args.includeFilename then
+		return "filename:"
+	elseif not args.betaSearch and args.includeExtension then
+		return "extension:"
+	else
+		return "path:"
+	end
+end
+
 M.construct_query_text = function()
 	local utils = require("csgithub.utils")
 	local text = ""
@@ -32,7 +46,8 @@ M.construct_query = function(args)
 	-- path:
 	if args.includeFilename or args.includeExtension then
 		local path = M.construct_query_path(args)
-		table.insert(query_parts, "path:" .. path)
+		local search_field = M.construct_search_field(args)
+		table.insert(query_parts, search_field .. path)
 	end
 
 	-- text