Show similar library names in lib search

Author: 2qar

cli/lib/search.go

@@ -88,8 +88,8 @@ func (res result) Data() interface{} {
 
 		names := []LibName{}
 		results := res.results.GetLibraries()
-		for _, lsr := range results {
-			names = append(names, LibName{lsr.Name})
+		for _, lib := range results {
+			names = append(names, LibName{lib.Name})
 		}
 
 		return NamesOnly{
@@ -113,9 +113,18 @@ func (res result) String() string {
 
 	var out strings.Builder
 
+	if res.results.GetStatus() == rpc.LibrarySearchStatus_failed {
+		out.WriteString("No libraries matching your search.\nDid you mean...\n")
+	}
+
 	for _, lib := range results {
-		out.WriteString(fmt.Sprintf("Name: \"%s\"\n", lib.Name))
-		if res.namesOnly {
+		if res.results.GetStatus() == rpc.LibrarySearchStatus_success {
+			out.WriteString(fmt.Sprintf("Name: \"%s\"\n", lib.Name))
+			if res.namesOnly {
+				continue
+			}
+		} else {
+			out.WriteString(fmt.Sprintf("%s\n", lib.Name))
 			continue
 		}
 

commands/lib/search.go

@@ -23,6 +23,7 @@ import (
 	"github.com/arduino/arduino-cli/arduino/libraries/librariesindex"
 	"github.com/arduino/arduino-cli/commands"
 	rpc "github.com/arduino/arduino-cli/rpc/commands"
+	"github.com/imjasonmiller/godice"
 	semver "go.bug.st/relaxed-semver"
 )
 
@@ -34,6 +35,7 @@ func LibrarySearch(ctx context.Context, req *rpc.LibrarySearchReq) (*rpc.Library
 	}
 
 	res := []*rpc.SearchedLibrary{}
+	status := rpc.LibrarySearchStatus_success
 
 	for _, lib := range lm.Index.Libraries {
 		qry := strings.ToLower(req.GetQuery())
@@ -46,16 +48,27 @@ func LibrarySearch(ctx context.Context, req *rpc.LibrarySearchReq) (*rpc.Library
 			}
 			latest := GetLibraryParameters(lib.Latest)
 
-			searchedlib := &rpc.SearchedLibrary{
+			searchedLib := &rpc.SearchedLibrary{
 				Name:     lib.Name,
 				Releases: releases,
 				Latest:   latest,
 			}
-			res = append(res, searchedlib)
+			res = append(res, searchedLib)
 		}
 	}
 
-	return &rpc.LibrarySearchResp{Libraries: res}, nil
+	if len(res) == 0 {
+		status = rpc.LibrarySearchStatus_failed
+		for _, lib := range lm.Index.Libraries {
+			if godice.CompareString(req.GetQuery(), lib.Name) > 0.7 {
+				res = append(res, &rpc.SearchedLibrary{
+					Name: lib.Name,
+				})
+			}
+		}
+	}
+
+	return &rpc.LibrarySearchResp{Libraries: res, Status: status}, nil
 }
 
 // GetLibraryParameters FIXMEDOC

go.mod

@@ -19,6 +19,7 @@ require (
 	github.com/gofrs/uuid v3.2.0+incompatible
 	github.com/golang/protobuf v1.3.3
 	github.com/h2non/filetype v1.0.8 // indirect
+	github.com/imjasonmiller/godice v0.1.2
 	github.com/juju/errors v0.0.0-20181118221551-089d3ea4e4d5 // indirect
 	github.com/juju/loggo v0.0.0-20190526231331-6e530bcce5d8 // indirect
 	github.com/juju/testing v0.0.0-20190429233213-dfc56b8c09fc // indirect

go.sum

@@ -85,6 +85,8 @@ github.com/h2non/filetype v1.0.8 h1:le8gpf+FQA0/DlDABbtisA1KiTS0Xi+YSC/E8yY3Y14=
 github.com/h2non/filetype v1.0.8/go.mod h1:isekKqOuhMj+s/7r3rIeTErIRy4Rub5uBWHfvMusLMU=
 github.com/hashicorp/hcl v1.0.0 h1:0Anlzjpi4vEasTeNFn2mLJgTSwt0+6sfsiTG8qcWGx4=
 github.com/hashicorp/hcl v1.0.0/go.mod h1:E5yfLk+7swimpb2L/Alb/PJmXilQ/rhwaUYs4T20WEQ=
+github.com/imjasonmiller/godice v0.1.2 h1:T1/sW/HoDzFeuwzOOuQjmeMELz9CzZ53I2CnD+08zD4=
+github.com/imjasonmiller/godice v0.1.2/go.mod h1:8cTkdnVI+NglU2d6sv+ilYcNaJ5VSTBwvMbFULJd/QQ=
 github.com/inconshreveable/mousetrap v1.0.0 h1:Z8tu5sraLXCXIcARxBp/8cbvlwVa7Z1NHg9XEKhtSvM=
 github.com/inconshreveable/mousetrap v1.0.0/go.mod h1:PxqpIevigyE2G7u3NXJIT2ANytuPF1OarO4DADm73n8=
 github.com/jonboulle/clockwork v0.1.0/go.mod h1:Ii8DK3G1RaLaWxj9trq07+26W01tbo22gdxWY5EU2bo=

rpc/commands/lib.pb.go

@@ -82,8 +82,14 @@ message LibrarySearchReq {
     string query = 2;
 }
 
+enum LibrarySearchStatus {
+    failed = 0;
+    success = 1;
+}
+
 message LibrarySearchResp {
     repeated SearchedLibrary libraries = 1;
+    LibrarySearchStatus status = 2;
 }
 
 message SearchedLibrary {