src/hastyscribepkg/utils.nim
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 |
import
std/base64,
std/os,
std/strutils,
std/pegs
import
consts
proc style_tag*(css: string): string =
result = "<style>$1</style>" % [css]
proc style_link_tag*(css: string): string =
result = "<link rel=\"stylesheet\" href=\"$1\"/>" % [css]
proc encode_image*(contents, format: string): string =
if format == "svg":
let encoded_svg = contents
.replace("\"", "'")
.replace("%", "%25")
.replace("#", "%23")
.replace("{", "%7B")
.replace("}", "%7D")
.replace("<", "%3C")
.replace(">", "%3E")
.replace(" ", "%20")
return "data:image/svg+xml,$#" % [encoded_svg]
else:
return "data:image/$format;base64,$enc_contents" % ["format", format, "enc_contents", contents.encode]
proc encode_image_file*(file, format: string): string =
if (file.fileExists):
let contents = file.readFile
return encode_image(contents, format)
else:
stderr.writeLine("Warning: image '" & file & "' not found.")
return file
proc image_format*(imgfile: string): string =
let peg_imgformat = peg"i'.png' / i'.jpg' / i'.jpeg' / i'.gif' / i'.svg' / i'.bmp' / i'.webp' @$"
return imgfile.substr(imgfile.find(peg_imgformat)+1, imgfile.len-1)
proc watermark_css*(imgfile: string): string =
if imgfile == "":
result = ""
else:
let img = imgfile.encode_image_file(imgfile.image_format)
result = (watermark_style % [img]).style_tag
proc add_jump_to_top_links*(document: string): string =
result = document.replacef(peg"{'</h' [23456] '>'}", "<a href=\"#document-top\" title=\"Go to top\"></a>$1")
|