B64EmbedImages init

Author: sommerf-lf

custom/conf/app.example.ini

@@ -1704,6 +1704,9 @@ LEVEL = Info
 ;;
 ;; convert \r\n to \n for Sendmail
 ;SENDMAIL_CONVERT_CRLF = true
+;;
+;; convert links of attached images to inline images
+;B64_EMBED_IMAGES = true
 
 ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
 ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;

modules/setting/mailer.go

@@ -28,6 +28,7 @@ type Mailer struct {
 	SendAsPlainText      bool                `ini:"SEND_AS_PLAIN_TEXT"`
 	SubjectPrefix        string              `ini:"SUBJECT_PREFIX"`
 	OverrideHeader       map[string][]string `ini:"-"`
+	B64EmbedImages       bool                `ini:"-"`
 
 	// SMTP sender
 	Protocol             string `ini:"PROTOCOL"`

services/mailer/mail.go

@@ -6,10 +6,14 @@ package mailer
 
 import (
 	"bytes"
+	"code.gitea.io/gitea/modules/storage"
 	"context"
+	"encoding/base64"
 	"fmt"
+	"golang.org/x/net/html"
 	"html/template"
 	"mime"
+	"net/http"
 	"regexp"
 	"strconv"
 	"strings"
@@ -232,6 +236,15 @@ func composeIssueCommentMessages(ctx *mailCommentContext, lang string, recipient
 		return nil, err
 	}
 
+	if setting.MailService.B64EmbedImages {
+		bodyStr := string(body)
+		bodyStr, err = inlineImages(bodyStr, ctx)
+		if err != nil {
+			return nil, err
+		}
+		body = template.HTML(bodyStr)
+	}
+
 	actType, actName, tplName := actionToTemplate(ctx.Issue, ctx.ActionType, commentType, reviewType)
 
 	if actName != "new" {
@@ -363,6 +376,106 @@ func composeIssueCommentMessages(ctx *mailCommentContext, lang string, recipient
 	return msgs, nil
 }
 
+func inlineImages(body string, ctx *mailCommentContext) (string, error) {
+	doc, err := html.Parse(strings.NewReader(body))
+	if err != nil {
+		log.Error("Failed to parse HTML body: %v", err)
+		return "", err
+	}
+
+	var processNode func(*html.Node)
+	processNode = func(n *html.Node) {
+		if n.Type == html.ElementNode {
+			if n.Data == "img" {
+				// Process <img> tags
+				for i, attr := range n.Attr {
+					if attr.Key == "src" {
+						attachmentPath := attr.Val
+						// Read the attachment file and encode it as base64
+						dataURI, err := attachmentSrcToDataURI(attachmentPath, ctx)
+						if err != nil {
+							log.Error("attachmentSrcToDataURI failed: %v", err)
+							continue
+						}
+						log.Trace("Old value of src attribute: %s, new value (first 100 characters): %s", attr.Val, dataURI[:100])
+						n.Attr[i].Val = dataURI
+					}
+				}
+			}
+		}
+
+		if n.FirstChild != nil {
+			log.Trace("Processing child nodes of <%s>", n.Data)
+		}
+		for c := n.FirstChild; c != nil; c = c.NextSibling {
+			processNode(c)
+		}
+	}
+
+	processNode(doc)
+
+	var buf bytes.Buffer
+	err = html.Render(&buf, doc)
+	if err != nil {
+		log.Error("Failed to render modified HTML: %v", err)
+		return "", err
+	}
+	return buf.String(), nil
+}
+
+// Helper function to convert attachment source to data URI
+
+func attachmentSrcToDataURI(attachmentPath string, ctx *mailCommentContext) (string, error) {
+	// Extract the UUID from the attachment path
+	parts := strings.Split(attachmentPath, "/attachments/")
+	attachmentUUID := ""
+	if len(parts) > 1 {
+		attachmentUUID = parts[1]
+	} else {
+		return "", fmt.Errorf("Invalid attachment path: %s", attachmentPath)
+	}
+	// Retrieve the attachment using the UUID
+	attachment, err := repo_model.GetAttachmentByUUID(ctx, attachmentUUID)
+	log.Trace("attachmentUUID: %s", attachmentUUID)
+	if err != nil {
+		return "", err
+	}
+
+	//log attachment.DownloadPath
+	log.Trace("attachment.DownloadURL(): %s", attachment.DownloadURL())
+
+	fr, err := storage.Attachments.Open(attachment.RelativePath())
+	if err != nil {
+		return "", err
+	}
+	defer func(fr storage.Object) {
+		err := fr.Close()
+		if err != nil {
+		}
+	}(fr)
+
+	//fr is io.ReadSeeker
+	//get this as bytes:
+	content := make([]byte, attachment.Size)
+	_, err = fr.Read(content)
+	if err != nil {
+		return "", err
+	}
+
+	// Detect the MIME type
+	mimeType := http.DetectContentType(content)
+
+	log.Trace("MIME type: %s", mimeType)
+
+	// Encode the content to base64
+	encoded := base64.StdEncoding.EncodeToString(content)
+	log.Trace("First 100 characters of encoded content: %s", encoded[:100])
+
+	// Construct the data URI
+	dataURI := fmt.Sprintf("data:%s;base64,%s", mimeType, encoded)
+	return dataURI, nil
+}
+
 func generateMessageIDForIssue(issue *issues_model.Issue, comment *issues_model.Comment, actionType activities_model.ActionType) string {
 	var path string
 	if issue.IsPull {