core/lib/strings.nix

1554 lines
43 KiB
Nix
Raw Normal View History

2024-06-30 08:16:52 +00:00
# String manipulation functions.
2024-05-01 22:14:04 +00:00
{ lib }:
let
inherit (builtins) length;
inherit (lib.trivial) warnIf;
2024-06-30 08:16:52 +00:00
asciiTable = import ./ascii-table.nix;
2024-05-01 22:14:04 +00:00
in
rec {
inherit (builtins)
compareVersions
elem
elemAt
filter
fromJSON
genList
head
isInt
isList
isAttrs
isPath
isString
match
parseDrvName
readFile
replaceStrings
split
storeDir
stringLength
substring
tail
toJSON
typeOf
unsafeDiscardStringContext
;
2024-06-30 08:16:52 +00:00
/*
Concatenate a list of strings.
2024-05-01 22:14:04 +00:00
Type: concatStrings :: [string] -> string
Example:
concatStrings ["foo" "bar"]
=> "foobar"
*/
concatStrings = builtins.concatStringsSep "";
2024-06-30 08:16:52 +00:00
/*
Map a function over a list and concatenate the resulting strings.
2024-05-01 22:14:04 +00:00
Type: concatMapStrings :: (a -> string) -> [a] -> string
Example:
concatMapStrings (x: "a" + x) ["foo" "bar"]
=> "afooabar"
*/
concatMapStrings = f: list: concatStrings (map f list);
2024-06-30 08:16:52 +00:00
/*
Like `concatMapStrings` except that the f functions also gets the
position as a parameter.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Type: concatImapStrings :: (int -> a -> string) -> [a] -> string
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
concatImapStrings (pos: x: "${toString pos}-${x}") ["foo" "bar"]
=> "1-foo2-bar"
2024-05-01 22:14:04 +00:00
*/
concatImapStrings = f: list: concatStrings (lib.imap1 f list);
2024-06-30 08:16:52 +00:00
/*
Place an element between each element of a list
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Type: intersperse :: a -> [a] -> [a]
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
intersperse "/" ["usr" "local" "bin"]
=> ["usr" "/" "local" "/" "bin"].
2024-05-01 22:14:04 +00:00
*/
intersperse =
# Separator to add between elements
separator:
# Input list
list:
2024-06-30 08:16:52 +00:00
if list == [ ] || length list == 1 then
list
else
tail (
lib.concatMap (x: [
separator
x
]) list
);
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
/*
Concatenate a list of strings with a separator between each element
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Type: concatStringsSep :: string -> [string] -> string
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
concatStringsSep "/" ["usr" "local" "bin"]
=> "usr/local/bin"
2024-05-01 22:14:04 +00:00
*/
concatStringsSep = builtins.concatStringsSep;
2024-06-30 08:16:52 +00:00
/*
Maps a function over a list of strings and then concatenates the
result with the specified separator interspersed between
elements.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Type: concatMapStringsSep :: string -> (a -> string) -> [a] -> string
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
concatMapStringsSep "-" (x: toUpper x) ["foo" "bar" "baz"]
=> "FOO-BAR-BAZ"
2024-05-01 22:14:04 +00:00
*/
concatMapStringsSep =
# Separator to add between elements
sep:
# Function to map over the list
f:
# List of input strings
2024-06-30 08:16:52 +00:00
list:
concatStringsSep sep (map f list);
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
/*
Same as `concatMapStringsSep`, but the mapping function
additionally receives the position of its argument.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Type: concatIMapStringsSep :: string -> (int -> a -> string) -> [a] -> string
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
concatImapStringsSep "-" (pos: x: toString (x / pos)) [ 6 6 6 ]
=> "6-3-2"
2024-05-01 22:14:04 +00:00
*/
concatImapStringsSep =
# Separator to add between elements
sep:
# Function that receives elements and their positions
f:
# List of input strings
2024-06-30 08:16:52 +00:00
list:
concatStringsSep sep (lib.imap1 f list);
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
/*
Concatenate a list of strings, adding a newline at the end of each one.
Defined as `concatMapStrings (s: s + "\n")`.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Type: concatLines :: [string] -> string
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
concatLines [ "foo" "bar" ]
=> "foo\nbar\n"
2024-05-01 22:14:04 +00:00
*/
concatLines = concatMapStrings (s: s + "\n");
/*
Replicate a string n times,
and concatenate the parts into a new string.
Type: replicate :: int -> string -> string
Example:
replicate 3 "v"
=> "vvv"
replicate 5 "hello"
=> "hellohellohellohellohello"
*/
replicate = n: s: concatStrings (lib.lists.replicate n s);
2024-06-30 08:16:52 +00:00
/*
Construct a Unix-style, colon-separated search path consisting of
the given `subDir` appended to each of the given paths.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Type: makeSearchPath :: string -> [string] -> string
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
makeSearchPath "bin" ["/root" "/usr" "/usr/local"]
=> "/root/bin:/usr/bin:/usr/local/bin"
makeSearchPath "bin" [""]
=> "/bin"
2024-05-01 22:14:04 +00:00
*/
makeSearchPath =
# Directory name to append
subDir:
# List of base paths
paths:
concatStringsSep ":" (map (path: path + "/" + subDir) (filter (x: x != null) paths));
2024-06-30 08:16:52 +00:00
/*
Construct a Unix-style search path by appending the given
`subDir` to the specified `output` of each of the packages. If no
output by the given name is found, fallback to `.out` and then to
the default.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Type: string -> string -> [package] -> string
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
makeSearchPathOutput "dev" "bin" [ pkgs.openssl pkgs.zlib ]
=> "/nix/store/9rz8gxhzf8sw4kf2j2f1grr49w8zx5vj-openssl-1.0.1r-dev/bin:/nix/store/wwh7mhwh269sfjkm6k5665b5kgp7jrk2-zlib-1.2.8/bin"
2024-05-01 22:14:04 +00:00
*/
makeSearchPathOutput =
# Package output to use
output:
# Directory name to append
subDir:
# List of packages
2024-06-30 08:16:52 +00:00
pkgs:
makeSearchPath subDir (map (lib.getOutput output) pkgs);
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
/*
Construct a library search path (such as RPATH) containing the
libraries for a set of packages
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
makeLibraryPath [ "/usr" "/usr/local" ]
=> "/usr/lib:/usr/local/lib"
pkgs = import <nixpkgs> { }
makeLibraryPath [ pkgs.openssl pkgs.zlib ]
=> "/nix/store/9rz8gxhzf8sw4kf2j2f1grr49w8zx5vj-openssl-1.0.1r/lib:/nix/store/wwh7mhwh269sfjkm6k5665b5kgp7jrk2-zlib-1.2.8/lib"
2024-05-01 22:14:04 +00:00
*/
makeLibraryPath = makeSearchPathOutput "lib" "lib";
2024-06-30 08:16:52 +00:00
/*
Construct an include search path (such as C_INCLUDE_PATH) containing the
header files for a set of packages or paths.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
makeIncludePath [ "/usr" "/usr/local" ]
=> "/usr/include:/usr/local/include"
pkgs = import <nixpkgs> { }
makeIncludePath [ pkgs.openssl pkgs.zlib ]
=> "/nix/store/9rz8gxhzf8sw4kf2j2f1grr49w8zx5vj-openssl-1.0.1r-dev/include:/nix/store/wwh7mhwh269sfjkm6k5665b5kgp7jrk2-zlib-1.2.8-dev/include"
2024-05-01 22:14:04 +00:00
*/
makeIncludePath = makeSearchPathOutput "dev" "include";
2024-06-30 08:16:52 +00:00
/*
Construct a binary search path (such as $PATH) containing the
binaries for a set of packages.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
makeBinPath ["/root" "/usr" "/usr/local"]
=> "/root/bin:/usr/bin:/usr/local/bin"
2024-05-01 22:14:04 +00:00
*/
makeBinPath = makeSearchPathOutput "bin" "bin";
2024-06-30 08:16:52 +00:00
/*
Normalize path, removing extraneous /s
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Type: normalizePath :: string -> string
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
normalizePath "/a//b///c/"
=> "/a/b/c/"
2024-05-01 22:14:04 +00:00
*/
2024-06-30 08:16:52 +00:00
normalizePath =
s:
warnIf (isPath s)
2024-05-01 22:14:04 +00:00
''
lib.strings.normalizePath: The argument (${toString s}) is a path value, but only strings are supported.
Path values are always normalised in Nix, so there's no need to call this function on them.
This function also copies the path to the Nix store and returns the store path, the same as "''${path}" will, which may not be what you want.
This behavior is deprecated and will throw an error in the future.''
(
2024-06-30 08:16:52 +00:00
builtins.foldl' (x: y: if y == "/" && hasSuffix "/" x then x else x + y) "" (stringToCharacters s)
2024-05-01 22:14:04 +00:00
);
2024-06-30 08:16:52 +00:00
/*
Depending on the boolean `cond', return either the given string
or the empty string. Useful to concatenate against a bigger string.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Type: optionalString :: bool -> string -> string
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
optionalString true "some-string"
=> "some-string"
optionalString false "some-string"
=> ""
2024-05-01 22:14:04 +00:00
*/
optionalString =
# Condition
cond:
# String to return if condition is true
2024-06-30 08:16:52 +00:00
string:
if cond then string else "";
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
/*
Determine whether a string has given prefix.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Type: hasPrefix :: string -> string -> bool
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
hasPrefix "foo" "foobar"
=> true
hasPrefix "foo" "barfoo"
=> false
2024-05-01 22:14:04 +00:00
*/
hasPrefix =
# Prefix to check for
pref:
# Input string
str:
# Before 23.05, paths would be copied to the store before converting them
# to strings and comparing. This was surprising and confusing.
2024-06-30 08:16:52 +00:00
warnIf (isPath pref)
2024-05-01 22:14:04 +00:00
''
lib.strings.hasPrefix: The first argument (${toString pref}) is a path value, but only strings are supported.
There is almost certainly a bug in the calling code, since this function always returns `false` in such a case.
This function also copies the path to the Nix store, which may not be what you want.
This behavior is deprecated and will throw an error in the future.
You might want to use `lib.path.hasPrefix` instead, which correctly supports paths.''
(substring 0 (stringLength pref) str == pref);
2024-06-30 08:16:52 +00:00
/*
Determine whether a string has given suffix.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Type: hasSuffix :: string -> string -> bool
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
hasSuffix "foo" "foobar"
=> false
hasSuffix "foo" "barfoo"
=> true
2024-05-01 22:14:04 +00:00
*/
hasSuffix =
# Suffix to check for
suffix:
# Input string
content:
let
lenContent = stringLength content;
lenSuffix = stringLength suffix;
in
# Before 23.05, paths would be copied to the store before converting them
# to strings and comparing. This was surprising and confusing.
2024-06-30 08:16:52 +00:00
warnIf (isPath suffix)
2024-05-01 22:14:04 +00:00
''
lib.strings.hasSuffix: The first argument (${toString suffix}) is a path value, but only strings are supported.
There is almost certainly a bug in the calling code, since this function always returns `false` in such a case.
This function also copies the path to the Nix store, which may not be what you want.
This behavior is deprecated and will throw an error in the future.''
2024-06-30 08:16:52 +00:00
(lenContent >= lenSuffix && substring (lenContent - lenSuffix) lenContent content == suffix);
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
/*
Determine whether a string contains the given infix
2024-05-01 22:14:04 +00:00
Type: hasInfix :: string -> string -> bool
Example:
hasInfix "bc" "abcd"
=> true
hasInfix "ab" "abcd"
=> true
hasInfix "cd" "abcd"
=> true
hasInfix "foo" "abcd"
=> false
*/
2024-06-30 08:16:52 +00:00
hasInfix =
infix: content:
2024-05-01 22:14:04 +00:00
# Before 23.05, paths would be copied to the store before converting them
# to strings and comparing. This was surprising and confusing.
2024-06-30 08:16:52 +00:00
warnIf (isPath infix)
2024-05-01 22:14:04 +00:00
''
lib.strings.hasInfix: The first argument (${toString infix}) is a path value, but only strings are supported.
There is almost certainly a bug in the calling code, since this function always returns `false` in such a case.
This function also copies the path to the Nix store, which may not be what you want.
This behavior is deprecated and will throw an error in the future.''
(builtins.match ".*${escapeRegex infix}.*" "${content}" != null);
2024-06-30 08:16:52 +00:00
/*
Convert a string to a list of characters (i.e. singleton strings).
This allows you to, e.g., map a function over each character. However,
note that this will likely be horribly inefficient; Nix is not a
general purpose programming language. Complex string manipulations
should, if appropriate, be done in a derivation.
Also note that Nix treats strings as a list of bytes and thus doesn't
handle unicode.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Type: stringToCharacters :: string -> [string]
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
stringToCharacters ""
=> [ ]
stringToCharacters "abc"
=> [ "a" "b" "c" ]
stringToCharacters "🦄"
=> [ "<EFBFBD>" "<EFBFBD>" "<EFBFBD>" "<EFBFBD>" ]
2024-05-01 22:14:04 +00:00
*/
2024-06-30 08:16:52 +00:00
stringToCharacters = s: genList (p: substring p 1 s) (stringLength s);
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
/*
Manipulate a string character by character and replace them by
strings before concatenating the results.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Type: stringAsChars :: (string -> string) -> string -> string
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
stringAsChars (x: if x == "a" then "i" else x) "nax"
=> "nix"
2024-05-01 22:14:04 +00:00
*/
stringAsChars =
# Function to map over each individual character
f:
# Input string
2024-06-30 08:16:52 +00:00
s:
concatStrings (map f (stringToCharacters s));
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
/*
Convert char to ascii value, must be in printable range
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Type: charToInt :: string -> int
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
charToInt "A"
=> 65
charToInt "("
=> 40
2024-05-01 22:14:04 +00:00
*/
charToInt = c: builtins.getAttr c asciiTable;
2024-06-30 08:16:52 +00:00
/*
Escape occurrence of the elements of `list` in `string` by
prefixing it with a backslash.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Type: escape :: [string] -> string -> string
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
escape ["(" ")"] "(foo)"
=> "\\(foo\\)"
2024-05-01 22:14:04 +00:00
*/
escape = list: replaceStrings list (map (c: "\\${c}") list);
2024-06-30 08:16:52 +00:00
/*
Escape occurrence of the element of `list` in `string` by
converting to its ASCII value and prefixing it with \\x.
Only works for printable ascii characters.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Type: escapeC = [string] -> string -> string
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
escapeC [" "] "foo bar"
=> "foo\\x20bar"
2024-05-01 22:14:04 +00:00
*/
2024-06-30 08:16:52 +00:00
escapeC = list: replaceStrings list (map (c: "\\x${toLower (lib.toHexString (charToInt c))}") list);
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
/*
Escape the string so it can be safely placed inside a URL
query.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Type: escapeURL :: string -> string
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
escapeURL "foo/bar baz"
=> "foo%2Fbar%20baz"
2024-05-01 22:14:04 +00:00
*/
2024-06-30 08:16:52 +00:00
escapeURL =
let
unreserved = [
"A"
"B"
"C"
"D"
"E"
"F"
"G"
"H"
"I"
"J"
"K"
"L"
"M"
"N"
"O"
"P"
"Q"
"R"
"S"
"T"
"U"
"V"
"W"
"X"
"Y"
"Z"
"a"
"b"
"c"
"d"
"e"
"f"
"g"
"h"
"i"
"j"
"k"
"l"
"m"
"n"
"o"
"p"
"q"
"r"
"s"
"t"
"u"
"v"
"w"
"x"
"y"
"z"
"0"
"1"
"2"
"3"
"4"
"5"
"6"
"7"
"8"
"9"
"-"
"_"
"."
"~"
];
toEscape = builtins.removeAttrs asciiTable unreserved;
in
replaceStrings (builtins.attrNames toEscape) (
lib.mapAttrsToList (_: c: "%${fixedWidthString 2 "0" (lib.toHexString c)}") toEscape
);
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
/*
Quote string to be used safely within the Bourne shell.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Type: escapeShellArg :: string -> string
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
escapeShellArg "esc'ape\nme"
=> "'esc'\\''ape\nme'"
2024-05-01 22:14:04 +00:00
*/
2024-06-30 08:16:52 +00:00
escapeShellArg = arg: "'${replaceStrings [ "'" ] [ "'\\''" ] (toString arg)}'";
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
/*
Quote all arguments to be safely passed to the Bourne shell.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Type: escapeShellArgs :: [string] -> string
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
escapeShellArgs ["one" "two three" "four'five"]
=> "'one' 'two three' 'four'\\''five'"
2024-05-01 22:14:04 +00:00
*/
escapeShellArgs = concatMapStringsSep " " escapeShellArg;
2024-06-30 08:16:52 +00:00
/*
Test whether the given name is a valid POSIX shell variable name.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Type: string -> bool
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
isValidPosixName "foo_bar000"
=> true
isValidPosixName "0-bad.jpg"
=> false
2024-05-01 22:14:04 +00:00
*/
isValidPosixName = name: match "[a-zA-Z_][a-zA-Z0-9_]*" name != null;
2024-06-30 08:16:52 +00:00
/*
Translate a Nix value into a shell variable declaration, with proper escaping.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
The value can be a string (mapped to a regular variable), a list of strings
(mapped to a Bash-style array) or an attribute set of strings (mapped to a
Bash-style associative array). Note that "string" includes string-coercible
values like paths or derivations.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Strings are translated into POSIX sh-compatible code; lists and attribute sets
assume a shell that understands Bash syntax (e.g. Bash or ZSH).
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Type: string -> (string | listOf string | attrsOf string) -> string
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
''
${toShellVar "foo" "some string"}
[[ "$foo" == "some string" ]]
''
2024-05-01 22:14:04 +00:00
*/
2024-06-30 08:16:52 +00:00
toShellVar =
name: value:
2024-05-01 22:14:04 +00:00
lib.throwIfNot (isValidPosixName name) "toShellVar: ${name} is not a valid shell variable name" (
2024-06-30 08:16:52 +00:00
if isAttrs value && !isStringLike value then
"declare -A ${name}=(${
concatStringsSep " " (lib.mapAttrsToList (n: v: "[${escapeShellArg n}]=${escapeShellArg v}") value)
})"
else if isList value then
"declare -a ${name}=(${escapeShellArgs value})"
else
"${name}=${escapeShellArg value}"
2024-05-01 22:14:04 +00:00
);
2024-06-30 08:16:52 +00:00
/*
Translate an attribute set into corresponding shell variable declarations
using `toShellVar`.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Type: attrsOf (string | listOf string | attrsOf string) -> string
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
let
foo = "value";
bar = foo;
in ''
${toShellVars { inherit foo bar; }}
[[ "$foo" == "$bar" ]]
''
2024-05-01 22:14:04 +00:00
*/
toShellVars = vars: concatStringsSep "\n" (lib.mapAttrsToList toShellVar vars);
2024-06-30 08:16:52 +00:00
/*
Turn a string into a Nix expression representing that string
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Type: string -> string
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
escapeNixString "hello\${}\n"
=> "\"hello\\\${}\\n\""
2024-05-01 22:14:04 +00:00
*/
2024-06-30 08:16:52 +00:00
escapeNixString = s: escape [ "$" ] (toJSON s);
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
/*
Turn a string into an exact regular expression
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Type: string -> string
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
escapeRegex "[^a-z]*"
=> "\\[\\^a-z]\\*"
2024-05-01 22:14:04 +00:00
*/
escapeRegex = escape (stringToCharacters "\\[{()^$?*+|.");
2024-06-30 08:16:52 +00:00
/*
Quotes a string if it can't be used as an identifier directly.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Type: string -> string
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
escapeNixIdentifier "hello"
=> "hello"
escapeNixIdentifier "0abc"
=> "\"0abc\""
2024-05-01 22:14:04 +00:00
*/
2024-06-30 08:16:52 +00:00
escapeNixIdentifier =
s:
2024-05-01 22:14:04 +00:00
# Regex from https://github.com/NixOS/nix/blob/d048577909e383439c2549e849c5c2f2016c997e/src/libexpr/lexer.l#L91
2024-06-30 08:16:52 +00:00
if match "[a-zA-Z_][a-zA-Z0-9_'-]*" s != null then s else escapeNixString s;
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
/*
Escapes a string such that it is safe to include verbatim in an XML
document.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Type: string -> string
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
escapeXML ''"test" 'test' < & >''
=> "&quot;test&quot; &apos;test&apos; &lt; &amp; &gt;"
*/
escapeXML =
builtins.replaceStrings
[
"\""
"'"
"<"
">"
"&"
]
[
"&quot;"
"&apos;"
"&lt;"
"&gt;"
"&amp;"
];
2024-05-01 22:14:04 +00:00
# warning added 12-12-2022
replaceChars = lib.warn "lib.replaceChars is a deprecated alias of lib.replaceStrings." builtins.replaceStrings;
# Case conversion utilities.
lowerChars = stringToCharacters "abcdefghijklmnopqrstuvwxyz";
upperChars = stringToCharacters "ABCDEFGHIJKLMNOPQRSTUVWXYZ";
2024-06-30 08:16:52 +00:00
/*
Converts an ASCII string to lower-case.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Type: toLower :: string -> string
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
toLower "HOME"
=> "home"
2024-05-01 22:14:04 +00:00
*/
toLower = replaceStrings upperChars lowerChars;
2024-06-30 08:16:52 +00:00
/*
Converts an ASCII string to upper-case.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Type: toUpper :: string -> string
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
toUpper "home"
=> "HOME"
2024-05-01 22:14:04 +00:00
*/
toUpper = replaceStrings lowerChars upperChars;
2024-06-30 08:16:52 +00:00
/*
Appends string context from another string. This is an implementation
detail of Nix and should be used carefully.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Strings in Nix carry an invisible `context` which is a list of strings
representing store paths. If the string is later used in a derivation
attribute, the derivation will properly populate the inputDrvs and
inputSrcs.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
pkgs = import <nixpkgs> { };
addContextFrom pkgs.coreutils "bar"
=> "bar"
2024-05-01 22:14:04 +00:00
*/
addContextFrom = a: b: substring 0 0 a + b;
2024-06-30 08:16:52 +00:00
/*
Cut a string with a separator and produces a list of strings which
were separated by this separator.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
splitString "." "foo.bar.baz"
=> [ "foo" "bar" "baz" ]
splitString "/" "/usr/local/bin"
=> [ "" "usr" "local" "bin" ]
2024-05-01 22:14:04 +00:00
*/
2024-06-30 08:16:52 +00:00
splitString =
sep: s:
2024-05-01 22:14:04 +00:00
let
2024-06-30 08:16:52 +00:00
splits = builtins.filter builtins.isString (
builtins.split (escapeRegex (toString sep)) (toString s)
);
2024-05-01 22:14:04 +00:00
in
2024-06-30 08:16:52 +00:00
map (addContextFrom s) splits;
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
/*
Return a string without the specified prefix, if the prefix matches.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Type: string -> string -> string
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
removePrefix "foo." "foo.bar.baz"
=> "bar.baz"
removePrefix "xxx" "foo.bar.baz"
=> "foo.bar.baz"
2024-05-01 22:14:04 +00:00
*/
removePrefix =
# Prefix to remove if it matches
prefix:
# Input string
str:
# Before 23.05, paths would be copied to the store before converting them
# to strings and comparing. This was surprising and confusing.
2024-06-30 08:16:52 +00:00
warnIf (isPath prefix)
2024-05-01 22:14:04 +00:00
''
lib.strings.removePrefix: The first argument (${toString prefix}) is a path value, but only strings are supported.
There is almost certainly a bug in the calling code, since this function never removes any prefix in such a case.
This function also copies the path to the Nix store, which may not be what you want.
This behavior is deprecated and will throw an error in the future.''
2024-06-30 08:16:52 +00:00
(
let
preLen = stringLength prefix;
in
if substring 0 preLen str == prefix then
# -1 will take the string until the end
substring preLen (-1) str
else
str
);
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
/*
Return a string without the specified suffix, if the suffix matches.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Type: string -> string -> string
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
removeSuffix "front" "homefront"
=> "home"
removeSuffix "xxx" "homefront"
=> "homefront"
2024-05-01 22:14:04 +00:00
*/
removeSuffix =
# Suffix to remove if it matches
suffix:
# Input string
str:
# Before 23.05, paths would be copied to the store before converting them
# to strings and comparing. This was surprising and confusing.
2024-06-30 08:16:52 +00:00
warnIf (isPath suffix)
2024-05-01 22:14:04 +00:00
''
lib.strings.removeSuffix: The first argument (${toString suffix}) is a path value, but only strings are supported.
There is almost certainly a bug in the calling code, since this function never removes any suffix in such a case.
This function also copies the path to the Nix store, which may not be what you want.
This behavior is deprecated and will throw an error in the future.''
2024-06-30 08:16:52 +00:00
(
let
sufLen = stringLength suffix;
sLen = stringLength str;
in
if sufLen <= sLen && suffix == substring (sLen - sufLen) sufLen str then
substring 0 (sLen - sufLen) str
else
str
);
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
/*
Return true if string v1 denotes a version older than v2.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
versionOlder "1.1" "1.2"
=> true
versionOlder "1.1" "1.1"
=> false
2024-05-01 22:14:04 +00:00
*/
versionOlder = v1: v2: compareVersions v2 v1 == 1;
2024-06-30 08:16:52 +00:00
/*
Return true if string v1 denotes a version equal to or newer than v2.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
versionAtLeast "1.1" "1.0"
=> true
versionAtLeast "1.1" "1.1"
=> true
versionAtLeast "1.1" "1.2"
=> false
2024-05-01 22:14:04 +00:00
*/
versionAtLeast = v1: v2: !versionOlder v1 v2;
2024-06-30 08:16:52 +00:00
/*
This function takes an argument that's either a derivation or a
derivation's "name" attribute and extracts the name part from that
argument.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
getName "youtube-dl-2016.01.01"
=> "youtube-dl"
getName pkgs.youtube-dl
=> "youtube-dl"
*/
getName =
let
parse = drv: (parseDrvName drv).name;
in
x: if isString x then parse x else x.pname or (parse x.name);
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
/*
This function takes an argument that's either a derivation or a
derivation's "name" attribute and extracts the version part from that
argument.
Example:
getVersion "youtube-dl-2016.01.01"
=> "2016.01.01"
getVersion pkgs.youtube-dl
=> "2016.01.01"
2024-05-01 22:14:04 +00:00
*/
2024-06-30 08:16:52 +00:00
getVersion =
let
parse = drv: (parseDrvName drv).version;
in
x: if isString x then parse x else x.version or (parse x.name);
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
/*
Extract name with version from URL. Ask for separator which is
supposed to start extension.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
nameFromURL "https://nixos.org/releases/nix/nix-1.7/nix-1.7-x86_64-linux.tar.bz2" "-"
=> "nix"
nameFromURL "https://nixos.org/releases/nix/nix-1.7/nix-1.7-x86_64-linux.tar.bz2" "_"
=> "nix-1.7-x86"
2024-05-01 22:14:04 +00:00
*/
2024-06-30 08:16:52 +00:00
nameFromURL =
url: sep:
2024-05-01 22:14:04 +00:00
let
components = splitString "/" url;
filename = lib.last components;
name = head (splitString sep filename);
2024-06-30 08:16:52 +00:00
in
assert name != filename;
name;
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
/*
Create a "-D<feature>:<type>=<value>" string that can be passed to typical
2024-05-01 22:14:04 +00:00
CMake invocations.
Type: cmakeOptionType :: string -> string -> string -> string
@param feature The feature to be set
@param type The type of the feature to be set, as described in
https://cmake.org/cmake/help/latest/command/set.html
the possible values (case insensitive) are:
BOOL FILEPATH PATH STRING INTERNAL
@param value The desired value
Example:
cmakeOptionType "string" "ENGINE" "sdl2"
=> "-DENGINE:STRING=sdl2"
*/
2024-06-30 08:16:52 +00:00
cmakeOptionType =
let
types = [
"BOOL"
"FILEPATH"
"PATH"
"STRING"
"INTERNAL"
];
in
type: feature: value:
2024-05-01 22:14:04 +00:00
assert (elem (toUpper type) types);
assert (isString feature);
assert (isString value);
"-D${feature}:${toUpper type}=${value}";
2024-06-30 08:16:52 +00:00
/*
Create a -D<condition>={TRUE,FALSE} string that can be passed to typical
2024-05-01 22:14:04 +00:00
CMake invocations.
Type: cmakeBool :: string -> bool -> string
@param condition The condition to be made true or false
@param flag The controlling flag of the condition
Example:
cmakeBool "ENABLE_STATIC_LIBS" false
=> "-DENABLESTATIC_LIBS:BOOL=FALSE"
*/
2024-06-30 08:16:52 +00:00
cmakeBool =
condition: flag:
2024-05-01 22:14:04 +00:00
assert (lib.isString condition);
assert (lib.isBool flag);
cmakeOptionType "bool" condition (lib.toUpper (lib.boolToString flag));
2024-06-30 08:16:52 +00:00
/*
Create a -D<feature>:STRING=<value> string that can be passed to typical
2024-05-01 22:14:04 +00:00
CMake invocations.
This is the most typical usage, so it deserves a special case.
Type: cmakeFeature :: string -> string -> string
@param condition The condition to be made true or false
@param flag The controlling flag of the condition
Example:
cmakeFeature "MODULES" "badblock"
=> "-DMODULES:STRING=badblock"
*/
2024-06-30 08:16:52 +00:00
cmakeFeature =
feature: value:
2024-05-01 22:14:04 +00:00
assert (lib.isString feature);
assert (lib.isString value);
cmakeOptionType "string" feature value;
2024-06-30 08:16:52 +00:00
/*
Create a -D<feature>=<value> string that can be passed to typical Meson
2024-05-01 22:14:04 +00:00
invocations.
Type: mesonOption :: string -> string -> string
@param feature The feature to be set
@param value The desired value
Example:
mesonOption "engine" "opengl"
=> "-Dengine=opengl"
*/
2024-06-30 08:16:52 +00:00
mesonOption =
feature: value:
2024-05-01 22:14:04 +00:00
assert (lib.isString feature);
assert (lib.isString value);
"-D${feature}=${value}";
2024-06-30 08:16:52 +00:00
/*
Create a -D<condition>={true,false} string that can be passed to typical
2024-05-01 22:14:04 +00:00
Meson invocations.
Type: mesonBool :: string -> bool -> string
@param condition The condition to be made true or false
@param flag The controlling flag of the condition
Example:
mesonBool "hardened" true
=> "-Dhardened=true"
mesonBool "static" false
=> "-Dstatic=false"
*/
2024-06-30 08:16:52 +00:00
mesonBool =
condition: flag:
2024-05-01 22:14:04 +00:00
assert (lib.isString condition);
assert (lib.isBool flag);
mesonOption condition (lib.boolToString flag);
2024-06-30 08:16:52 +00:00
/*
Create a -D<feature>={enabled,disabled} string that can be passed to
2024-05-01 22:14:04 +00:00
typical Meson invocations.
Type: mesonEnable :: string -> bool -> string
@param feature The feature to be enabled or disabled
@param flag The controlling flag
Example:
mesonEnable "docs" true
=> "-Ddocs=enabled"
mesonEnable "savage" false
=> "-Dsavage=disabled"
*/
2024-06-30 08:16:52 +00:00
mesonEnable =
feature: flag:
2024-05-01 22:14:04 +00:00
assert (lib.isString feature);
assert (lib.isBool flag);
mesonOption feature (if flag then "enabled" else "disabled");
2024-06-30 08:16:52 +00:00
/*
Create an --{enable,disable}-<feature> string that can be passed to
standard GNU Autoconf scripts.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
enableFeature true "shared"
=> "--enable-shared"
enableFeature false "shared"
=> "--disable-shared"
2024-05-01 22:14:04 +00:00
*/
2024-06-30 08:16:52 +00:00
enableFeature =
flag: feature:
2024-05-01 22:14:04 +00:00
assert lib.isBool flag;
assert lib.isString feature; # e.g. passing openssl instead of "openssl"
"--${if flag then "enable" else "disable"}-${feature}";
2024-06-30 08:16:52 +00:00
/*
Create an --{enable-<feature>=<value>,disable-<feature>} string that can be passed to
standard GNU Autoconf scripts.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
enableFeatureAs true "shared" "foo"
=> "--enable-shared=foo"
enableFeatureAs false "shared" (throw "ignored")
=> "--disable-shared"
2024-05-01 22:14:04 +00:00
*/
2024-06-30 08:16:52 +00:00
enableFeatureAs =
flag: feature: value:
2024-05-01 22:14:04 +00:00
enableFeature flag feature + optionalString flag "=${value}";
2024-06-30 08:16:52 +00:00
/*
Create an --{with,without}-<feature> string that can be passed to
standard GNU Autoconf scripts.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
withFeature true "shared"
=> "--with-shared"
withFeature false "shared"
=> "--without-shared"
2024-05-01 22:14:04 +00:00
*/
2024-06-30 08:16:52 +00:00
withFeature =
flag: feature:
2024-05-01 22:14:04 +00:00
assert isString feature; # e.g. passing openssl instead of "openssl"
"--${if flag then "with" else "without"}-${feature}";
2024-06-30 08:16:52 +00:00
/*
Create an --{with-<feature>=<value>,without-<feature>} string that can be passed to
standard GNU Autoconf scripts.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
withFeatureAs true "shared" "foo"
=> "--with-shared=foo"
withFeatureAs false "shared" (throw "ignored")
=> "--without-shared"
2024-05-01 22:14:04 +00:00
*/
2024-06-30 08:16:52 +00:00
withFeatureAs =
flag: feature: value:
2024-05-01 22:14:04 +00:00
withFeature flag feature + optionalString flag "=${value}";
2024-06-30 08:16:52 +00:00
/*
Create a fixed width string with additional prefix to match
required width.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
This function will fail if the input string is longer than the
requested length.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Type: fixedWidthString :: int -> string -> string -> string
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
fixedWidthString 5 "0" (toString 15)
=> "00015"
2024-05-01 22:14:04 +00:00
*/
2024-06-30 08:16:52 +00:00
fixedWidthString =
width: filler: str:
2024-05-01 22:14:04 +00:00
let
strw = lib.stringLength str;
reqWidth = width - (lib.stringLength filler);
in
2024-06-30 08:16:52 +00:00
assert lib.assertMsg (strw <= width)
"fixedWidthString: requested string length (${toString width}) must not be shorter than actual length (${toString strw})";
if strw == width then str else filler + fixedWidthString reqWidth filler str;
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
/*
Format a number adding leading zeroes up to fixed width.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
fixedWidthNumber 5 15
=> "00015"
2024-05-01 22:14:04 +00:00
*/
fixedWidthNumber = width: n: fixedWidthString width "0" (toString n);
2024-06-30 08:16:52 +00:00
/*
Convert a float to a string, but emit a warning when precision is lost
during the conversion
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
floatToString 0.000001
=> "0.000001"
floatToString 0.0000001
=> trace: warning: Imprecise conversion from float to string 0.000000
"0.000000"
*/
floatToString =
float:
let
result = toString float;
precise = float == fromJSON result;
in
lib.warnIf (!precise) "Imprecise conversion from float to string ${result}" result;
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
/*
Soft-deprecated function. While the original implementation is available as
isConvertibleWithToString, consider using isStringLike instead, if suitable.
*/
isCoercibleToString =
lib.warnIf (lib.isInOldestRelease 2305)
"lib.strings.isCoercibleToString is deprecated in favor of either isStringLike or isConvertibleWithToString. Only use the latter if it needs to return true for null, numbers, booleans and list of similarly coercibles."
isConvertibleWithToString;
/*
Check whether a list or other value can be passed to toString.
Many types of value are coercible to string this way, including int, float,
null, bool, list of similarly coercible values.
*/
isConvertibleWithToString =
let
types = [
"null"
"int"
"float"
"bool"
];
in
x: isStringLike x || elem (typeOf x) types || (isList x && lib.all isConvertibleWithToString x);
/*
Check whether a value can be coerced to a string.
The value must be a string, path, or attribute set.
String-like values can be used without explicit conversion in
string interpolations and in most functions that expect a string.
*/
isStringLike = x: isString x || isPath x || x ? outPath || x ? __toString;
/*
Check whether a value is a store path.
Example:
isStorePath "/nix/store/d945ibfx9x185xf04b890y4f9g3cbb63-python-2.7.11/bin/python"
=> false
isStorePath "/nix/store/d945ibfx9x185xf04b890y4f9g3cbb63-python-2.7.11"
=> true
isStorePath pkgs.python
=> true
isStorePath [] || isStorePath 42 || isStorePath {} ||
=> false
*/
isStorePath =
x:
2024-05-01 22:14:04 +00:00
if isStringLike x then
2024-06-30 08:16:52 +00:00
let
str = toString x;
in
substring 0 1 str == "/" && dirOf str == storeDir
2024-05-01 22:14:04 +00:00
else
false;
2024-06-30 08:16:52 +00:00
/*
Parse a string as an int. Does not support parsing of integers with preceding zero due to
ambiguity between zero-padded and octal numbers. See toIntBase10.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Type: string -> int
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
toInt "1337"
=> 1337
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
toInt "-4"
=> -4
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
toInt " 123 "
=> 123
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
toInt "00024"
=> error: Ambiguity in interpretation of 00024 between octal and zero padded integer.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
toInt "3.14"
=> error: floating point JSON numbers are not supported
2024-05-01 22:14:04 +00:00
*/
toInt =
let
matchStripInput = match "[[:space:]]*(-?[[:digit:]]+)[[:space:]]*";
matchLeadingZero = match "0[[:digit:]]+";
in
str:
let
# RegEx: Match any leading whitespace, possibly a '-', one or more digits,
# and finally match any trailing whitespace.
strippedInput = matchStripInput str;
# RegEx: Match a leading '0' then one or more digits.
2024-06-30 08:16:52 +00:00
isLeadingZero = matchLeadingZero (head strippedInput) == [ ];
2024-05-01 22:14:04 +00:00
# Attempt to parse input
parsedInput = fromJSON (head strippedInput);
generalError = "toInt: Could not convert ${escapeNixString str} to int.";
in
2024-06-30 08:16:52 +00:00
# Error on presence of non digit characters.
if strippedInput == null then
throw generalError
# Error on presence of leading zero/octal ambiguity.
else if isLeadingZero then
throw "toInt: Ambiguity in interpretation of ${escapeNixString str} between octal and zero padded integer."
# Error if parse function fails.
else if !isInt parsedInput then
throw generalError
# Return result.
else
parsedInput;
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
/*
Parse a string as a base 10 int. This supports parsing of zero-padded integers.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Type: string -> int
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
toIntBase10 "1337"
=> 1337
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
toIntBase10 "-4"
=> -4
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
toIntBase10 " 123 "
=> 123
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
toIntBase10 "00024"
=> 24
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
toIntBase10 "3.14"
=> error: floating point JSON numbers are not supported
2024-05-01 22:14:04 +00:00
*/
toIntBase10 =
let
matchStripInput = match "[[:space:]]*0*(-?[[:digit:]]+)[[:space:]]*";
matchZero = match "0+";
in
str:
let
# RegEx: Match any leading whitespace, then match any zero padding,
# capture possibly a '-' followed by one or more digits,
# and finally match any trailing whitespace.
strippedInput = matchStripInput str;
# RegEx: Match at least one '0'.
2024-06-30 08:16:52 +00:00
isZero = matchZero (head strippedInput) == [ ];
2024-05-01 22:14:04 +00:00
# Attempt to parse input
parsedInput = fromJSON (head strippedInput);
generalError = "toIntBase10: Could not convert ${escapeNixString str} to int.";
in
2024-06-30 08:16:52 +00:00
# Error on presence of non digit characters.
if strippedInput == null then
throw generalError
# In the special case zero-padded zero (00000), return early.
else if isZero then
0
# Error if parse function fails.
else if !isInt parsedInput then
throw generalError
# Return result.
else
parsedInput;
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
/*
Read a list of paths from `file`, relative to the `rootPath`.
Lines beginning with `#` are treated as comments and ignored.
Whitespace is significant.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
NOTE: This function is not performant and should be avoided.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
readPathsFromFile /prefix
./pkgs/development/libraries/qt-5/5.4/qtbase/series
=> [ "/prefix/dlopen-resolv.patch" "/prefix/tzdir.patch"
"/prefix/dlopen-libXcursor.patch" "/prefix/dlopen-openssl.patch"
"/prefix/dlopen-dbus.patch" "/prefix/xdg-config-dirs.patch"
"/prefix/nix-profiles-library-paths.patch"
"/prefix/compose-search-path.patch" ]
*/
readPathsFromFile = lib.warn "lib.readPathsFromFile is deprecated, use a list instead." (
rootPath: file:
let
lines = lib.splitString "\n" (readFile file);
removeComments = lib.filter (line: line != "" && !(lib.hasPrefix "#" line));
relativePaths = removeComments lines;
absolutePaths = map (path: rootPath + "/${path}") relativePaths;
in
absolutePaths
);
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
/*
Read the contents of a file removing the trailing \n
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Type: fileContents :: path -> string
Example:
$ echo "1.0" > ./version
fileContents ./version
=> "1.0"
2024-05-01 22:14:04 +00:00
*/
fileContents = file: removeSuffix "\n" (readFile file);
2024-06-30 08:16:52 +00:00
/*
Creates a valid derivation name from a potentially invalid one.
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Type: sanitizeDerivationName :: String -> String
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
sanitizeDerivationName "../hello.bar # foo"
=> "-hello.bar-foo"
sanitizeDerivationName ""
=> "unknown"
sanitizeDerivationName pkgs.hello
=> "-nix-store-2g75chlbpxlrqn15zlby2dfh8hr9qwbk-hello-2.10"
2024-05-01 22:14:04 +00:00
*/
sanitizeDerivationName =
2024-06-30 08:16:52 +00:00
let
okRegex = match "[[:alnum:]+_?=-][[:alnum:]+._?=-]*";
in
string:
# First detect the common case of already valid strings, to speed those up
if stringLength string <= 207 && okRegex string != null then
unsafeDiscardStringContext string
else
lib.pipe string [
# Get rid of string context. This is safe under the assumption that the
# resulting string is only used as a derivation name
unsafeDiscardStringContext
# Strip all leading "."
(x: elemAt (match "\\.*(.*)" x) 0)
# Split out all invalid characters
# https://github.com/NixOS/nix/blob/2.3.2/src/libstore/store-api.cc#L85-L112
# https://github.com/NixOS/nix/blob/2242be83c61788b9c0736a92bb0b5c7bbfc40803/nix-rust/src/store/path.rs#L100-L125
(split "[^[:alnum:]+._?=-]+")
# Replace invalid character ranges with a "-"
(concatMapStrings (s: if lib.isList s then "-" else s))
# Limit to 211 characters (minus 4 chars for ".drv")
(x: substring (lib.max (stringLength x - 207) 0) (-1) x)
# If the result is empty, replace it with "unknown"
(x: if stringLength x == 0 then "unknown" else x)
];
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
/*
Computes the Levenshtein distance between two strings.
Complexity O(n*m) where n and m are the lengths of the strings.
Algorithm adjusted from https://stackoverflow.com/a/9750974/6605742
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Type: levenshtein :: string -> string -> int
Example:
levenshtein "foo" "foo"
=> 0
levenshtein "book" "hook"
=> 1
levenshtein "hello" "Heyo"
=> 3
*/
levenshtein =
a: b:
let
# Two dimensional array with dimensions (stringLength a + 1, stringLength b + 1)
arr = lib.genList (i: lib.genList (j: dist i j) (stringLength b + 1)) (stringLength a + 1);
d = x: y: lib.elemAt (lib.elemAt arr x) y;
dist =
i: j:
let
c = if substring (i - 1) 1 a == substring (j - 1) 1 b then 0 else 1;
in
if j == 0 then
i
else if i == 0 then
j
else
lib.min (lib.min (d (i - 1) j + 1) (d i (j - 1) + 1)) (d (i - 1) (j - 1) + c);
in
d (stringLength a) (stringLength b);
# Returns the length of the prefix common to both strings.
commonPrefixLength =
a: b:
2024-05-01 22:14:04 +00:00
let
m = lib.min (stringLength a) (stringLength b);
2024-06-30 08:16:52 +00:00
go =
i:
if i >= m then
m
else if substring i 1 a == substring i 1 b then
go (i + 1)
else
i;
in
go 0;
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
# Returns the length of the suffix common to both strings.
commonSuffixLength =
a: b:
2024-05-01 22:14:04 +00:00
let
m = lib.min (stringLength a) (stringLength b);
2024-06-30 08:16:52 +00:00
go =
i:
if i >= m then
m
else if substring (stringLength a - i - 1) 1 a == substring (stringLength b - i - 1) 1 b then
go (i + 1)
else
i;
in
go 0;
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
/*
Returns whether the levenshtein distance between two strings is at most some value
Complexity is O(min(n,m)) for k <= 2 and O(n*m) otherwise
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Type: levenshteinAtMost :: int -> string -> string -> bool
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
Example:
levenshteinAtMost 0 "foo" "foo"
=> true
levenshteinAtMost 1 "foo" "boa"
=> false
levenshteinAtMost 2 "foo" "boa"
=> true
levenshteinAtMost 2 "This is a sentence" "this is a sentense."
=> false
levenshteinAtMost 3 "This is a sentence" "this is a sentense."
=> true
*/
levenshteinAtMost =
let
infixDifferAtMost1 = x: y: stringLength x <= 1 && stringLength y <= 1;
# This function takes two strings stripped by their common pre and suffix,
# and returns whether they differ by at most two by Levenshtein distance.
# Because of this stripping, if they do indeed differ by at most two edits,
# we know that those edits were (if at all) done at the start or the end,
# while the middle has to have stayed the same. This fact is used in the
# implementation.
infixDifferAtMost2 =
x: y:
let
xlen = stringLength x;
ylen = stringLength y;
# This function is only called with |x| >= |y| and |x| - |y| <= 2, so
# diff is one of 0, 1 or 2
diff = xlen - ylen;
# Infix of x and y, stripped by the left and right most character
xinfix = substring 1 (xlen - 2) x;
yinfix = substring 1 (ylen - 2) y;
# x and y but a character deleted at the left or right
xdelr = substring 0 (xlen - 1) x;
xdell = substring 1 (xlen - 1) x;
ydelr = substring 0 (ylen - 1) y;
ydell = substring 1 (ylen - 1) y;
in
2024-05-01 22:14:04 +00:00
# A length difference of 2 can only be gotten with 2 delete edits,
# which have to have happened at the start and end of x
# Example: "abcdef" -> "bcde"
2024-06-30 08:16:52 +00:00
if diff == 2 then
xinfix == y
2024-05-01 22:14:04 +00:00
# A length difference of 1 can only be gotten with a deletion on the
# right and a replacement on the left or vice versa.
# Example: "abcdef" -> "bcdez" or "zbcde"
2024-06-30 08:16:52 +00:00
else if diff == 1 then
xinfix == ydelr || xinfix == ydell
2024-05-01 22:14:04 +00:00
# No length difference can either happen through replacements on both
# sides, or a deletion on the left and an insertion on the right or
# vice versa
# Example: "abcdef" -> "zbcdez" or "bcdefz" or "zabcde"
2024-06-30 08:16:52 +00:00
else
xinfix == yinfix || xdelr == ydell || xdell == ydelr;
2024-05-01 22:14:04 +00:00
2024-06-30 08:16:52 +00:00
in
k:
if k <= 0 then
a: b: a == b
else
let
f =
a: b:
let
alen = stringLength a;
blen = stringLength b;
prelen = commonPrefixLength a b;
suflen = commonSuffixLength a b;
presuflen = prelen + suflen;
ainfix = substring prelen (alen - presuflen) a;
binfix = substring prelen (blen - presuflen) b;
in
# Make a be the bigger string
if alen < blen then
f b a
# If a has over k more characters than b, even with k deletes on a, b can't be reached
else if alen - blen > k then
false
else if k == 1 then
infixDifferAtMost1 ainfix binfix
else if k == 2 then
infixDifferAtMost2 ainfix binfix
else
levenshtein ainfix binfix <= k;
in
f;
2024-05-01 22:14:04 +00:00
}