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 52 53 54 55 56 57 58 59 60 61 62 63 64 65 |
import std/[ base64, os, strutils, pegs, hashes, ] import consts template style_tag*(css: string): string = "<style>" & css & "</style>" 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.multireplace([ ("\"", "'"), ("%", "%25"), ("#", "%23"), ("{", "%7B"), ("}", "%7D"), ("<", "%3C"), (">", "%3E"), (" ", "%20"), ]) "data:image/svg+xml,$#" % [encoded_svg] else: "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") proc makeFNameUnique*(baseName, dir: string): string = ## Uses file placement (`dir`) as a unique name identifier ## Files in relative root (`dir` is empty) are returned unchanged. if dir notin ["", ".", "./"]: let dir = when dosLikeFileSystem: dir.replace('\\', '/') else: dir hashBytes = cast[array[sizeof(Hash), byte]](hash(dir)) uniquePrefix = encode(hashBytes, safe=true) baseName & '_' & uniquePrefix else: baseName |