shiori/internal/cmd/export.go

91 lines
2 KiB
Go
Raw Normal View History

2019-05-21 11:31:40 +08:00
package cmd
import (
2019-05-23 18:05:46 +08:00
"fmt"
"os"
fp "path/filepath"
"strings"
"time"
"github.com/go-shiori/shiori/internal/database"
2019-05-21 11:31:40 +08:00
"github.com/spf13/cobra"
)
func exportCmd() *cobra.Command {
cmd := &cobra.Command{
Use: "export target-file",
Short: "Export bookmarks into HTML file in Netscape Bookmark format",
Args: cobra.ExactArgs(1),
2019-05-23 18:05:46 +08:00
Run: exportHandler,
2019-05-21 11:31:40 +08:00
}
return cmd
}
2019-05-23 18:05:46 +08:00
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")
}