all repos — hastyscribe @ b4eb8d20cebd6ab7e28041c761364471d617a4a7

A professional markdown compiler.

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