aboutsummaryrefslogtreecommitdiff
path: root/prop_html2md.go
blob: 2435f231decdf59f2a6f268f90b180c4e25be838 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
////////////////////////////////////////////////////////////////////////////
// Program: html2md
// Purpose: HTML to Markdown
// Authors: Tong Sun (c) 2020, All rights reserved
////////////////////////////////////////////////////////////////////////////

package main

import (
	"fmt"
	"regexp"

	md "github.com/JohannesKaufmann/html-to-markdown"
	"github.com/JohannesKaufmann/html-to-markdown/plugin"
	"github.com/PuerkitoBio/goquery"
	"github.com/mkideal/cli"
	"github.com/mkideal/cli/clis"
)

//==========================================================================
// root handler

func html2md(ctx *cli.Context) error {
	rootArgv = ctx.RootArgv().(*rootT)
	// https://pkg.go.dev/github.com/mkideal/[email protected]/clis?tab=doc
	clis.Setup(progname, rootArgv.Verbose.Value())
	clis.Verbose(1, "%#v\n", rootArgv)
	clis.Verbose(2, "%s\n", rootArgv.Filei.Name())

	// Options handling
	opt := &md.Options{}
	opt = handleOptions(opt, rootArgv)
	clis.Verbose(1, "%#v\n", opt)

	doc, err := goquery.NewDocumentFromReader(rootArgv.Filei)
	clis.AbortOn("Reading file with goquery", err)
	content := doc.Find(rootArgv.Sel)

	domain, url := rootArgv.Domain, rootArgv.Filei.Name()
	if domain == "" && regexp.MustCompile(`(?i)^http`).MatchString(url) {
		domain = md.DomainFromURL(url)
	}
	clis.Verbose(2, "domain='%s'\n", domain)
	conv := md.NewConverter(domain, true, opt)

	// Plugin handling
	conv = handlePlugins(conv, rootArgv)
	markdown := conv.Convert(content)

	fmt.Println(markdown)

	return nil
}

func handleOptions(opt *md.Options, rootArgv *rootT) *md.Options {
	if rootArgv.OptHeadingStyle != "" {
		opt.HeadingStyle = rootArgv.OptHeadingStyle
	}
	if rootArgv.OptHorizontalRule != "" {
		opt.HorizontalRule = rootArgv.OptHorizontalRule
	}
	if rootArgv.OptBulletListMarker != "" {
		opt.BulletListMarker = rootArgv.OptBulletListMarker
	}
	if rootArgv.OptCodeBlockStyle != "" {
		opt.CodeBlockStyle = rootArgv.OptCodeBlockStyle
	}
	if rootArgv.OptFence != "" {
		opt.Fence = rootArgv.OptFence
	}
	if rootArgv.OptEmDelimiter != "" {
		opt.EmDelimiter = rootArgv.OptEmDelimiter
	}
	if rootArgv.OptStrongDelimiter != "" {
		opt.StrongDelimiter = rootArgv.OptStrongDelimiter
	}
	if rootArgv.OptLinkStyle != "" {
		opt.LinkStyle = rootArgv.OptLinkStyle
	}
	if rootArgv.OptLinkReferenceStyle != "" {
		opt.LinkReferenceStyle = rootArgv.OptLinkReferenceStyle
	}
	return opt
}

func handlePlugins(conv *md.Converter, rootArgv *rootT) *md.Converter {
	if rootArgv.PluginConfluenceAttachments {
		conv.Use(plugin.ConfluenceAttachments())
	}
	if rootArgv.PluginConfluenceCodeBlock {
		conv.Use(plugin.ConfluenceCodeBlock())
	}
	// if rootArgv.PluginFrontMatter {
	// 	conv.Use(plugin.FrontMatter())
	// }
	if rootArgv.PluginGitHubFlavored {
		conv.Use(plugin.GitHubFlavored())
	}
	if rootArgv.PluginStrikethrough {
		conv.Use(plugin.Strikethrough(""))
	}
	if rootArgv.PluginTable {
		conv.Use(plugin.Table())
	}
	if rootArgv.PluginTableCompat {
		conv.Use(plugin.TableCompat())
	}
	if rootArgv.PluginTaskListItems {
		conv.Use(plugin.TaskListItems())
	}
	// if rootArgv.PluginVimeoEmbed {
	// 	conv.Use(plugin.VimeoEmbed())
	// }
	// if rootArgv.PluginYoutubeEmbed {
	// 	conv.Use(plugin.YoutubeEmbed())
	// }
	return conv
}