util.fb.cmdsqlite/utilmht.bas at main · thrive4/util.fb.cmdsqlite · GitHub

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
119
120
121
122
123
124
125
126
127
128
129
130
131

Function replaceimageanchor(haystack As String, needle As String) As Integer
' based on https://rosettacode.org/wiki/Count_occurrences_of_a_substring#FreeBASIC
  If haystack = "" OrElse needle = "" Then Return 0
  Dim As Integer count = 0, length = Len(needle)
  dim dummy as string
  For i As Integer = 1 To Len(haystack)
    If Mid(haystack, i, length) = needle Then
    dummy = Mid(haystack, i, (instr(i, haystack, ".") + 4) - i)
    haystack = replace(haystack, Mid(haystack, i, (instr(i, haystack, ".") + 4) - i), chr$(34) + mid(dummy, instrrev(dummy, "/") + 1))
      count += 1
      i += length - 1
    End If
  Next
  Return count
End Function

' decode a base64 encoded file
function mhtconvert(filename as string) as boolean

    ' init mht image or file input
    dim itemnr          as integer = 1
    dim listitem        as string
    dim i               as integer = 1
    ' init mht text
    dim chkcontenttype  as boolean = false
    dim tempfolder      as string
    dim orgname         as string
    dim textfile        as string
    Dim msg64           As String
    dim textitem        as string
    dim chkhtml         as boolean = false
    dim linelength      as integer = 72

    tempfolder = mid(filename, instrrev(filename, pathchar))
    tempfolder = exepath + mid(tempfolder, 1, instrrev(tempfolder, ".") - 1)

    if mkdir(tempfolder) < 0  then
        logentry("fatal", "error: could not create folder " + tempfolder)
    else
        print "exporting " + filename + " as html and text to " + tempfolder
    end if

    msg64 = ""
    textitem = ""
    orgname = mid(filename, instrrev(filename, pathchar) + 1)
    orgname = left(orgname, len(orgname) - 4) + ".html"
    textfile = tempfolder + pathchar + orgname

    Open filename For input As 1
    open textfile for output as 3

    Do Until EOF(1)
        ' stop decoding
        Line Input #1, listitem
        ' special case remove %2520 used in filenames images
        listitem = Replace(listitem, "%2520", "")
        ' filter out mht header for html
        if instr(listitem, "<html") = 0 and chkhtml = false then
            listitem = ""
        else
            chkhtml = true
        end if
        if instr(listitem, "------=_NextPart") > 0 then
            Print #2, base64decode(msg64)
            chkcontenttype = false
            msg64 = ""
            close (2)
        end if
        ' start decoding
        select case true
            case instr(listitem, "Content-Type: image") > 0
                chkcontenttype = true
            case instr(listitem, "Content-Type: text/javascript") > 0
                chkcontenttype = true
            case instr(listitem, "Content-Type: text/css") > 0
                chkcontenttype = true
            case instr(listitem, "Content-Type: font") > 0
                chkcontenttype = true
        end select
        if chkcontenttype then
            if instr(listitem, "Content-Location:") > 0 then
                ' output decoded images to a temp dir
                open tempfolder + pathchar + mid(listitem, instrrev(listitem, pathchar) + 1) for output as 2
            end if
            ' ghetto validation base64
            select case true
                case instr(listitem, " ") > 0
                    'nop
                case instr(listitem, "-") > 0
                    'nop
                case instr(listitem, ":") > 0
                    'nop
                case instr(listitem, "%") > 0
                    'nop
                case len(listitem) = 0
                    'nop
                case else
                    msg64 = msg64 + listitem
            end select
        end if
        if chkcontenttype = false then
            select case true
                case instr(listitem, "------=_NextPart") > 0
                    listitem = ""
                case instr(listitem, "Content-Type:") > 0
                    listitem = ""
                case instr(listitem, "Content-Transfer-Encoding:") > 0
                    listitem = ""
                case instr(listitem, "Content-Location:") > 0
                    listitem = ""
            end select
            ' special cases mht
            ' remove frontpage thing sticks = to end of line
            if mid(listitem, len(listitem)) = "=" then
                listitem = mid(listitem, 1, len(listitem) - 1)
            end if
            textitem = textitem + listitem
        end if
        itemnr += 1
    Loop

    ' generic replace for text and html
    textitem = htmlcleanup(textitem)
    print "nr image anchors changed: " & replaceimageanchor(textitem, chr$(34) + "file:///")
    print #3, textitem
    close

    return true

end function