From 2f5704b114d033e81725f18459f9293a9adfee1e Mon Sep 17 00:00:00 2001
From: "charles.deng" <cilendeng@gmail.com>
Date: Mon, 10 Oct 2022 00:11:18 +0800
Subject: Stream writer support to set inline rich text cell (#1121)

Co-authored-by: zhengchao.deng <zhengchao.deng@meican.com>
---
 cell.go | 42 +++++++++++++++++++++++++-----------------
 1 file changed, 25 insertions(+), 17 deletions(-)

(limited to 'cell.go')

diff --git a/cell.go b/cell.go
index 6beb3b2..80eb035 100644
--- a/cell.go
+++ b/cell.go
@@ -885,6 +885,28 @@ func newRpr(fnt *Font) *xlsxRPr {
 	return &rpr
 }
 
+// setRichText provides a function to set rich text of a cell.
+func setRichText(runs []RichTextRun) ([]xlsxR, error) {
+	var (
+		textRuns       []xlsxR
+		totalCellChars int
+	)
+	for _, textRun := range runs {
+		totalCellChars += len(textRun.Text)
+		if totalCellChars > TotalCellChars {
+			return textRuns, ErrCellCharsLength
+		}
+		run := xlsxR{T: &xlsxT{}}
+		_, run.T.Val, run.T.Space = setCellStr(textRun.Text)
+		fnt := textRun.Font
+		if fnt != nil {
+			run.RPr = newRpr(fnt)
+		}
+		textRuns = append(textRuns, run)
+	}
+	return textRuns, nil
+}
+
 // SetCellRichText provides a function to set cell with rich text by given
 // worksheet. For example, set rich text on the A1 cell of the worksheet named
 // Sheet1:
@@ -1016,24 +1038,10 @@ func (f *File) SetCellRichText(sheet, cell string, runs []RichTextRun) error {
 		return err
 	}
 	c.S = f.prepareCellStyle(ws, col, row, c.S)
-	si := xlsxSI{}
-	sst := f.sharedStringsReader()
-	var textRuns []xlsxR
-	totalCellChars := 0
-	for _, textRun := range runs {
-		totalCellChars += len(textRun.Text)
-		if totalCellChars > TotalCellChars {
-			return ErrCellCharsLength
-		}
-		run := xlsxR{T: &xlsxT{}}
-		_, run.T.Val, run.T.Space = setCellStr(textRun.Text)
-		fnt := textRun.Font
-		if fnt != nil {
-			run.RPr = newRpr(fnt)
-		}
-		textRuns = append(textRuns, run)
+	si, sst := xlsxSI{}, f.sharedStringsReader()
+	if si.R, err = setRichText(runs); err != nil {
+		return err
 	}
-	si.R = textRuns
 	for idx, strItem := range sst.SI {
 		if reflect.DeepEqual(strItem, si) {
 			c.T, c.V = "s", strconv.Itoa(idx)
-- 
cgit v1.2.1