summary refs log tree commit diff
path: root/pkgs/development/tools/poetry2nix/poetry2nix/pep508.nix
blob: 93a395326eb76c04a8625cf3b7e734aa29e66235 (plain) (blame)
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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
{ lib, stdenv, poetryLib }: python:

let
  inherit (poetryLib) ireplace;

  # Like builtins.substring but with stop being offset instead of length
  substr = start: stop: s: builtins.substring start (stop - start) s;

  # Strip leading/trailing whitespace from string
  stripStr = s: lib.elemAt (builtins.split "^ *" (lib.elemAt (builtins.split " *$" s) 0)) 2;

  findSubExpressionsFun = acc: c: (
    if c == "(" then (
      let
        posNew = acc.pos + 1;
        isOpen = acc.openP == 0;
        startPos = if isOpen then posNew else acc.startPos;
      in
        acc // {
          inherit startPos;
          exprs = acc.exprs ++ [ (substr acc.exprPos (acc.pos - 1) acc.expr) ];
          pos = posNew;
          openP = acc.openP + 1;
        }
    ) else if c == ")" then (
      let
        openP = acc.openP - 1;
        exprs = findSubExpressions (substr acc.startPos acc.pos acc.expr);
      in
        acc // {
          inherit openP;
          pos = acc.pos + 1;
          exprs = if openP == 0 then acc.exprs ++ [ exprs ] else acc.exprs;
          exprPos = if openP == 0 then acc.pos + 1 else acc.exprPos;
        }
    ) else acc // { pos = acc.pos + 1; }
  );

  # Make a tree out of expression groups (parens)
  findSubExpressions = expr: let
    acc = builtins.foldl' findSubExpressionsFun {
      exprs = [];
      expr = expr;
      pos = 0;
      openP = 0;
      exprPos = 0;
      startPos = 0;
    } (lib.stringToCharacters expr);
    tailExpr = (substr acc.exprPos acc.pos expr);
    tailExprs = if tailExpr != "" then [ tailExpr ] else [];
  in
    acc.exprs ++ tailExprs;

  parseExpressions = exprs: let
    splitCond = (
      s: builtins.map
        (x: stripStr (if builtins.typeOf x == "list" then (builtins.elemAt x 0) else x))
        (builtins.split " (and|or) " (s + " "))
    );

    mapfn = expr: (
      if (builtins.match "^ ?$" expr != null) then null  # Filter empty
      else if (builtins.elem expr [ "and" "or" ]) then {
        type = "bool";
        value = expr;
      }
      else {
        type = "expr";
        value = expr;
      }
    );

    parse = expr: builtins.filter (x: x != null) (builtins.map mapfn (splitCond expr));

  in
    builtins.foldl' (
      acc: v: acc ++ (
        if builtins.typeOf v == "string" then parse v else [ (parseExpressions v) ]
      )
    ) [] exprs;

  # Transform individual expressions to structured expressions
  # This function also performs variable substitution, replacing environment markers with their explicit values
  transformExpressions = exprs: let
    variables = {
      os_name = "posix"; # TODO: Check other platforms
      sys_platform = (
        if stdenv.isLinux then "linux"
        else if stdenv.isDarwin then "darwin"
        else throw "Unsupported platform"
      );
      platform_machine = stdenv.platform.kernelArch;
      platform_python_implementation = "CPython"; # Only CPython supported for now
      platform_release = ""; # Field not reproducible
      platform_system = (
        if stdenv.isLinux then "Linux"
        else if stdenv.isDarwin then "Darwin"
        else throw "Unsupported platform"
      );
      platform_version = ""; # Field not reproducible
      python_version = python.passthru.pythonVersion;
      python_full_version = python.version;
      implementation_name = "cpython"; # Only cpython supported for now
      implementation_version = python.version;
      extra = "";
    };

    substituteVar = value: if builtins.hasAttr value variables then (builtins.toJSON variables."${value}") else value;

    processVar = value: builtins.foldl' (acc: v: v acc) value [
      stripStr
      substituteVar
    ];

  in
    if builtins.typeOf exprs == "set" then (
      if exprs.type == "expr" then (
        let
          mVal = ''[a-zA-Z0-9\'"_\. ]+'';
          mOp = "in|[!=<>]+";
          e = stripStr exprs.value;
          m = builtins.map stripStr (builtins.match ''^(${mVal}) *(${mOp}) *(${mVal})$'' e);
        in
          {
            type = "expr";
            value = {
              op = builtins.elemAt m 1;
              values = [
                (processVar (builtins.elemAt m 0))
                (processVar (builtins.elemAt m 2))
              ];
            };
          }
      ) else exprs
    ) else builtins.map transformExpressions exprs;

  # Recursively eval all expressions
  evalExpressions = exprs: let
    unmarshal = v: (
      # TODO: Handle single quoted values
      if v == "True" then true
      else if v == "False" then false
      else builtins.fromJSON v
    );
    hasElem = needle: haystack: builtins.elem needle (builtins.filter (x: builtins.typeOf x == "string") (builtins.split " " haystack));
    op = {
      "<=" = x: y: (unmarshal x) <= (unmarshal y);
      "<" = x: y: (unmarshal x) < (unmarshal y);
      "!=" = x: y: x != y;
      "==" = x: y: x == y;
      ">=" = x: y: (unmarshal x) >= (unmarshal y);
      ">" = x: y: (unmarshal x) > (unmarshal y);
      "~=" = v: c: let
        parts = builtins.splitVersion c;
        pruned = lib.take ((builtins.length parts) - 1) parts;
        upper = builtins.toString (
          (lib.toInt (builtins.elemAt pruned (builtins.length pruned - 1))) + 1
        );
        upperConstraint = builtins.concatStringsSep "." (ireplace (builtins.length pruned - 1) upper pruned);
      in
        op.">=" v c && op."<" v upperConstraint;
      "===" = x: y: x == y;
      "in" = x: y: let
        values = builtins.filter (x: builtins.typeOf x == "string") (builtins.split " " (unmarshal y));
      in
        builtins.elem (unmarshal x) values;
    };
  in
    if builtins.typeOf exprs == "set" then (
      if exprs.type == "expr" then (
        let
          expr = exprs;
          result = (op."${expr.value.op}") (builtins.elemAt expr.value.values 0) (builtins.elemAt expr.value.values 1);
        in
          {
            type = "value";
            value = result;
          }
      ) else exprs
    ) else builtins.map evalExpressions exprs;

  # Now that we have performed an eval all that's left to do is to concat the graph into a single bool
  reduceExpressions = exprs: let
    cond = {
      "and" = x: y: x && y;
      "or" = x: y: x || y;
    };
    reduceExpressionsFun = acc: v: (
      if builtins.typeOf v == "set" then (
        if v.type == "value" then (
          acc // {
            value = cond."${acc.cond}" acc.value v.value;
          }
        ) else if v.type == "bool" then (
          acc // {
            cond = v.value;
          }
        ) else throw "Unsupported type"
      ) else if builtins.typeOf v == "list" then (
        let
          ret = builtins.foldl' reduceExpressionsFun {
            value = true;
            cond = "and";
          } v;
        in
          acc // {
            value = cond."${acc.cond}" acc.value ret.value;
          }
      ) else throw "Unsupported type"
    );
  in
    (
      builtins.foldl' reduceExpressionsFun {
        value = true;
        cond = "and";
      } exprs
    ).value;

in
e: builtins.foldl' (acc: v: v acc) e [
  findSubExpressions
  parseExpressions
  transformExpressions
  evalExpressions
  reduceExpressions
]