Implement logic for export cmd

internal/cmd/export.go

@@ -1,6 +1,14 @@
 package cmd
 
 import (
+	"fmt"
+	"os"
+	fp "path/filepath"
+	"strings"
+	"time"
+
+	"github.com/go-shiori/shiori/internal/database"
+
 	"github.com/spf13/cobra"
 )
 
@@ -9,7 +17,74 @@ func exportCmd() *cobra.Command {
 		Use:   "export target-file",
 		Short: "Export bookmarks into HTML file in Netscape Bookmark format",
 		Args:  cobra.ExactArgs(1),
+		Run:   exportHandler,
 	}
 
 	return cmd
 }
+
+func exportHandler(cmd *cobra.Command, args []string) {
+	// Fetch bookmarks from database
+	bookmarks, err := DB.GetBookmarks(database.GetBookmarksOptions{})
+	if err != nil {
+		cError.Printf("Failed to get bookmarks: %v\n", err)
+		return
+	}
+
+	if len(bookmarks) == 0 {
+		cError.Println("No saved bookmarks yet")
+		return
+	}
+
+	// Make sure destination directory exist
+	dstDir := fp.Dir(args[0])
+	os.MkdirAll(dstDir, os.ModePerm)
+
+	// Create destination file
+	dstFile, err := os.Create(args[0])
+	if err != nil {
+		cError.Printf("Failed to create destination file: %v\n", err)
+		return
+	}
+	defer dstFile.Close()
+
+	// Write exported bookmark to file
+	fmt.Fprintln(dstFile, ``+
+		`<!DOCTYPE NETSCAPE-Bookmark-file-1>`+
+		`<META HTTP-EQUIV="Content-Type" CONTENT="text/html; charset=UTF-8">`+
+		`<TITLE>Bookmarks</TITLE>`+
+		`<H1>Bookmarks</H1>`+
+		`<DL>`)
+
+	for _, book := range bookmarks {
+		// Create Unix timestamp for bookmark
+		modifiedTime, err := time.Parse("2006-01-02 15:04:05", book.Modified)
+		if err != nil {
+			modifiedTime = time.Now()
+		}
+		unixTimestamp := modifiedTime.Unix()
+
+		// Create tags for bookmarks
+		tags := []string{}
+		for _, tag := range book.Tags {
+			tags = append(tags, tag.Name)
+		}
+		strTags := strings.Join(tags, ",")
+
+		// Write to file
+		exportLine := fmt.Sprintf(`<DT><A HREF="%s" ADD_DATE="%d" LAST_MODIFIED="%d" TAGS="%s">%s</A>`,
+			book.URL, unixTimestamp, unixTimestamp, strTags, book.Title)
+		fmt.Fprintln(dstFile, exportLine)
+	}
+
+	fmt.Fprintln(dstFile, "</DL>")
+
+	// Flush data to storage
+	err = dstFile.Sync()
+	if err != nil {
+		cError.Printf("Failed to export the bookmarks: %v\n", err)
+		return
+	}
+
+	fmt.Println("Export finished")
+}