nwaku/waku/common/utils/parse_size_units.nim
Ivan FB fd6a71cdd7
chore: Bump dependencies for v0.31.0 (#2885)
* bump_dependencies.md: add nim-results dependency
* change imports stew/results to results
* switching to Nim 2.0.8
* waku.nimble: reflect the requirement nim 1.6.0 to 2.0.8
  Adding --mm:refc as nim 2.0 enables a new garbage collector that we're
  not yet ready to support
* adapt waku code to Nim 2.0
* gcsafe adaptations because Nim 2.0 is more strict
2024-07-09 13:14:28 +02:00

52 lines
1.5 KiB
Nim

import std/strutils, results, regex
proc parseMsgSize*(input: string): Result[uint64, string] =
## Parses size strings such as "1.2 KiB" or "3Kb" and returns the equivalent number of bytes
## if the parse task goes well. If not, it returns an error describing the problem.
const RegexDef = """\s*(\d+([\,\.]\d*)?)\s*([Kk]{0,1}[i]?[Bb]{1})"""
const RegexParseSize = re2(RegexDef)
var m: RegexMatch2
if input.match(RegexParseSize, m) == false:
return err("error in parseSize. regex is not matching: " & RegexDef)
var value: float
try:
value = parseFloat(input[m.captures[0]].replace(",", "."))
except ValueError:
return err(
"invalid size in parseSize: " & getCurrentExceptionMsg() & " error parsing: " &
input[m.captures[0]] & " KKK : " & $m
)
let units = input[m.captures[2]].toLowerAscii() # units is "kib", or "kb", or "b".
var multiplier: float
case units
of "kb":
multiplier = 1000
of "kib":
multiplier = 1024
of "ib":
return err("wrong units. ib or iB aren't allowed.")
else: ## bytes
multiplier = 1
value = value * multiplier
return ok(uint64(value))
proc parseCorrectMsgSize*(input: string): uint64 =
## This proc always returns an int and wraps the following proc:
##
## proc parseMsgSize*(input: string): Result[int, string] = ...
##
## in case of error, it just returns 0, and this is expected to
## be called only from a controlled and well-known inputs
let ret = parseMsgSize(input).valueOr:
return 0
return ret