From 08795920159e960ab5ec1f67258cff5c071bb8bb Mon Sep 17 00:00:00 2001
From: sloumdrone <brian.michael.evans@gmail.com>
Date: Thu, 19 Sep 2019 15:32:26 -0700
Subject: [PATCH] Adds search with terms inline, also gemini file rendering

---
 client.go        | 48 +++++++++++++++++++++++++++---------------------
 gemini/gemini.go | 27 +++++++++++++++++----------
 2 files changed, 44 insertions(+), 31 deletions(-)

diff --git a/client.go b/client.go
index 4a49108..6d3aa28 100644
--- a/client.go
+++ b/client.go
@@ -277,7 +277,7 @@ func (c *client) simpleCommand(action string) {
 		c.BookMarks.ToggleOpen()
 		c.Draw()
 	case "SEARCH":
-		c.search()
+		c.search("")
 	case "HELP", "?":
 		go c.Visit(helplocation)
 	default:
@@ -296,6 +296,8 @@ func (c *client) doCommand(action string, values []string) {
 	switch action {
 	case "CHECK", "C":
 		c.displayConfigValue(values[0])
+	case "SEARCH":
+		c.search(strings.Join(values, " "))
 	default:
 	  c.SetMessage(fmt.Sprintf("Unknown action %q", action), true)
 		c.DrawMessage()
@@ -505,21 +507,27 @@ func (c *client) doLinkCommand(action, target string) {
 
 }
 
-func (c *client) search() {
-	c.ClearMessage()
-	c.ClearMessageLine()
-	// TODO handle keeping the full command bar here
-	// like was done for regular command entry
-	// maybe split into separate function
-	fmt.Print("?")
-	entry, err := cui.GetLine()
-	c.ClearMessageLine()
-	if err != nil {
-		c.SetMessage(err.Error(), true)
-		c.DrawMessage()
-		return
-	} else if strings.TrimSpace(entry) == "" {
-		return
+func (c *client) search(q string) {
+	var entry string
+	var err error
+	if q == "" {
+		c.ClearMessage()
+		c.ClearMessageLine()
+		if c.Options["theme"] == "normal" {
+			fmt.Printf("\033[7m%*.*s\r", c.Width, c.Width, "")
+		}
+		fmt.Print("?")
+		entry, err = cui.GetLine()
+		c.ClearMessageLine()
+		if err != nil {
+			c.SetMessage(err.Error(), true)
+			c.DrawMessage()
+			return
+		} else if strings.TrimSpace(entry) == "" {
+			return
+		}
+	} else {
+		entry = q
 	}
 	u, err := MakeUrl(c.Options["searchengine"])
 	if err != nil {
@@ -614,7 +622,7 @@ func (c *client) displayConfigValue(setting string) {
 
 func (c *client) SetMessage(msg string, isError bool) {
 	c.MessageIsErr = isError
-	c.Message = msg
+	c.Message = strings.ReplaceAll(msg, "\t", "%09")
 }
 
 func (c *client) DrawMessage() {
@@ -682,7 +690,7 @@ func (c *client) goToLink(l string) {
 func (c *client) SetHeaderUrl() {
 	if c.PageState.Length > 0 {
 		u := c.PageState.History[c.PageState.Position].Location.Full
-		c.TopBar.url = u
+		c.TopBar.url = strings.ReplaceAll(u, "\t", "%09")
 	} else {
 		c.TopBar.url = ""
 	}
@@ -692,6 +700,7 @@ func (c *client) Visit(url string) {
 	c.SetMessage("Loading...", false)
 	c.DrawMessage()
 
+	url = strings.ReplaceAll(url, "%09", "\t")
 	u, err := MakeUrl(url)
 	if err != nil {
 		c.SetMessage(err.Error(), true)
@@ -741,9 +750,6 @@ func (c *client) Visit(url string) {
 				c.DrawMessage()
 			}
 		}
-
-		// c.SetMessage("Bombadillo has not mastered Gemini yet, check back soon", false)
-		// c.DrawMessage()
 	case "telnet":
 		c.SetMessage("Attempting to start telnet session", false)
 		c.DrawMessage()
diff --git a/gemini/gemini.go b/gemini/gemini.go
index e8edee0..f2fd88c 100644
--- a/gemini/gemini.go
+++ b/gemini/gemini.go
@@ -131,19 +131,26 @@ func parseGemini(b, rootUrl string) (string, []string) {
 	for i, ln := range splitContent {
 		splitContent[i] = strings.Trim(ln, "\r\n")
 		if len([]rune(ln)) > 3 && ln[:2] == "=>" {
-			trimmedSubLn := strings.Trim(ln[2:], "\r\n\t \a")
-			lineSplit := strings.SplitN(trimmedSubLn, " ", 2)
-			if len(lineSplit) != 2 {
-				lineSplit = append(lineSplit, lineSplit[0])
+			var link, decorator string
+			subLn := strings.Trim(ln[2:], "\r\n\t \a")
+			splitPoint := strings.IndexAny(subLn, " \t")
+
+			if splitPoint < 0 || len([]rune(subLn)) - 1 <= splitPoint {
+				link = subLn
+				decorator = subLn
+			} else {
+				link = strings.Trim(subLn[:splitPoint], "\t\n\r \a")
+				decorator = strings.Trim(subLn[splitPoint:], "\t\n\r \a")
 			}
-			lineSplit[0] = strings.Trim(lineSplit[0], "\t\n\r \a")
-			lineSplit[1] = strings.Trim(lineSplit[1], "\t\n\r \a")
-			if len(lineSplit[0]) > 0 && lineSplit[0][0] == '/' {
-				lineSplit[0] = fmt.Sprintf("%s%s", rootUrl, lineSplit[0])
+
+			if len(link) > 0 && link[0] == '/' {
+				link = fmt.Sprintf("%s%s", rootUrl, link)
+			} else if len(link) > 0 && strings.Index(link, "://") < 0 {
+				link = fmt.Sprintf("%s/%s", rootUrl, link)
 			}
-			links = append(links, lineSplit[0])
+			links = append(links, link)
 			linknum := fmt.Sprintf("[%d]", len(links))
-			splitContent[i] = fmt.Sprintf("%-5s %s", linknum, lineSplit[1])
+			splitContent[i] = fmt.Sprintf("%-5s %s", linknum, decorator)
 		}
 	}
 	return strings.Join(splitContent, "\n"), links