From c70a50e76b5b905b593b18a7d15776d02bbbf4f3 Mon Sep 17 00:00:00 2001
From: aarondill <aaronsacks2006@gmail.com>
Date: Sun, 18 Jun 2023 04:30:20 -0500
Subject: [PATCH] Handle errors in finding binary path

This avoids constant nil errors (fixes #77), and quits early if the binary can't be found.
Notifications can easily be added in future commits if so desired to warn the user about what is wrong.
---
 lua/tabnine/binary.lua | 24 ++++++++++++++++++++++--
 1 file changed, 22 insertions(+), 2 deletions(-)

diff --git a/lua/tabnine/binary.lua b/lua/tabnine/binary.lua
index bd6e65c..e704dca 100644
--- a/lua/tabnine/binary.lua
+++ b/lua/tabnine/binary.lua
@@ -23,6 +23,8 @@ local function arch_and_platform()
 		return "x86_64-pc-windows-gnu"
 	elseif os_uname.sysname == "Windows_NT" then
 		return "i686-pc-windows-gnu"
+	else
+		return nil
 	end
 end
 
@@ -46,7 +48,21 @@ local function binary_path()
 
 	table.sort(paths)
 
-	return binaries_path .. "/" .. tostring(paths[#paths]) .. "/" .. arch_and_platform() .. "/" .. binary_name()
+	local version = paths[#paths]
+	if not version then
+		return nil -- Is it installed?
+	end
+	local machine = arch_and_platform()
+	if not machine then
+		return nil -- Is this machine supported?
+	end
+	local binary = binaries_path .. "/" .. tostring(version) .. "/" .. machine .. "/" .. binary_name()
+
+	if vim.fn.filereadable(binary) then -- Double check that it's installed
+		return binary
+	else
+		return nil -- File doesn't exist or isn't readable. Is it installed?
+	end
 end
 
 local function optional_args()
@@ -62,7 +78,7 @@ local function optional_args()
 end
 
 function TabnineBinary:start()
-	self.handle, self.pid = uv.spawn(binary_path(), {
+	self.handle, self.pid = uv.spawn(self.binary_path, {
 		args = vim.list_extend({
 			"--client",
 			"nvim",
@@ -103,12 +119,16 @@ function TabnineBinary:new(o)
 	self.restart_counter = 0
 	self.handle = nil
 	self.pid = nil
+	self.binary_path = binary_path()
 	self.callbacks = {}
 
 	return o
 end
 
 function TabnineBinary:request(request, on_response)
+	if not self.binary_path then
+		return function() end -- To avoid breaking user configurations
+	end
 	if not self.pid then
 		self.restart_counter = self.restart_counter + 1
 		self:start()