|
| 1 | +{ |
| 2 | + pkgs, |
| 3 | + lib, |
| 4 | + stdenv, |
| 5 | + toolchain ? { |
| 6 | + cc = "/no-toolchain-defined"; |
| 7 | + cflags = []; |
| 8 | + ld = "/no-toolchain-defined"; |
| 9 | + ldflags = []; |
| 10 | + asm = "/no-toolchain-defined"; |
| 11 | + asmflags = []; |
| 12 | + }, |
| 13 | +}: let |
| 14 | + include = { |
| 15 | + src, |
| 16 | + deps ? [], |
| 17 | + }: |
| 18 | + pkgs.runCommandLocal (builtins.baseNameOf src) {} '' |
| 19 | + mkdir -p $out/include |
| 20 | +
|
| 21 | + ARGS=() |
| 22 | + for dep in $deps; do |
| 23 | + if [ -d $dep/include ]; then |
| 24 | + ln -s $src/include/* $out/include/ |
| 25 | + fi |
| 26 | + done |
| 27 | +
|
| 28 | + ln -s ${src} $out/include/${builtins.baseNameOf src} |
| 29 | + ''; |
| 30 | + |
| 31 | + # If a raw file path is specified, turn that into a header derivation. |
| 32 | + autowrap_header = f: |
| 33 | + if !(builtins.isPath f) |
| 34 | + then f |
| 35 | + else if lib.hasSuffix ".h" f |
| 36 | + then include {src = f;} |
| 37 | + else throw "unknown file type: ${f}"; |
| 38 | + |
| 39 | + # Turn all raw file paths into derivations. |
| 40 | + # Files will be grouped so that source files can depend on header files. |
| 41 | + autowrap_deps = deps: let |
| 42 | + src = builtins.filter (f: builtins.isPath f && !(lib.hasSuffix ".h" f)) deps; |
| 43 | + other = builtins.filter (f: !(builtins.isPath f) || lib.hasSuffix ".h" f) deps; |
| 44 | + wrapped = map autowrap_header other; |
| 45 | + src_drv = |
| 46 | + map ( |
| 47 | + f: |
| 48 | + if lib.hasSuffix ".S" f |
| 49 | + then |
| 50 | + asm { |
| 51 | + src = f; |
| 52 | + deps = wrapped; |
| 53 | + } |
| 54 | + else if lib.hasSuffix ".c" f |
| 55 | + then |
| 56 | + object { |
| 57 | + src = f; |
| 58 | + deps = wrapped; |
| 59 | + } |
| 60 | + else throw "unknown file type: ${f}" |
| 61 | + ) |
| 62 | + src; |
| 63 | + in |
| 64 | + src_drv ++ wrapped; |
| 65 | + |
| 66 | + asm = { |
| 67 | + src, |
| 68 | + deps ? [], |
| 69 | + extra-asmflags ? [], |
| 70 | + asmflags ? toolchain.asmflags ++ extra-asmflags, |
| 71 | + }: |
| 72 | + stdenv.mkDerivation { |
| 73 | + name = (lib.removeSuffix ".S" (builtins.baseNameOf src)) + ".o"; |
| 74 | + inherit src asmflags; |
| 75 | + inherit (toolchain) asm; |
| 76 | + deps = autowrap_deps deps; |
| 77 | + dontUnpack = true; |
| 78 | + buildPhase = '' |
| 79 | + ARGS=() |
| 80 | + for dep in $deps; do |
| 81 | + if [ -d $dep/include ]; then |
| 82 | + ARGS+=(-I$dep/include) |
| 83 | + fi |
| 84 | + done |
| 85 | +
|
| 86 | + mkdir -p $out/obj |
| 87 | + $asm $asmflags -c -o $out/obj/$name $src ''${ARGS[@]} |
| 88 | + ''; |
| 89 | + }; |
| 90 | + |
| 91 | + object = { |
| 92 | + src, |
| 93 | + deps ? [], |
| 94 | + extra-cflags ? [], |
| 95 | + cflags ? toolchain.cflags ++ extra-cflags, |
| 96 | + }: |
| 97 | + stdenv.mkDerivation { |
| 98 | + name = (lib.removeSuffix ".c" (builtins.baseNameOf src)) + ".o"; |
| 99 | + inherit src cflags; |
| 100 | + inherit (toolchain) cc; |
| 101 | + deps = autowrap_deps deps; |
| 102 | + dontUnpack = true; |
| 103 | + buildPhase = '' |
| 104 | + ARGS=() |
| 105 | + for dep in $deps; do |
| 106 | + if [ -d $dep/include ]; then |
| 107 | + ARGS+=(-I$dep/include) |
| 108 | + fi |
| 109 | + done |
| 110 | +
|
| 111 | + mkdir -p $out/obj |
| 112 | + $cc $cflags -c -o $out/obj/$name $src ''${ARGS[@]} |
| 113 | + ''; |
| 114 | + }; |
| 115 | + |
| 116 | + static = { |
| 117 | + name, |
| 118 | + deps, |
| 119 | + }: |
| 120 | + stdenv.mkDerivation { |
| 121 | + inherit name; |
| 122 | + srcs = autowrap_deps deps; |
| 123 | + dontUnpack = true; |
| 124 | + buildPhase = '' |
| 125 | + ARGS=() |
| 126 | + for src in $srcs; do |
| 127 | + if [ -d $src/obj ]; then |
| 128 | + ARGS+=($src/obj/*) |
| 129 | + fi |
| 130 | + if [ -d $src/include ]; then |
| 131 | + mkdir -p $out/include |
| 132 | + ln -s $src/include/* $out/include/ |
| 133 | + fi |
| 134 | + done |
| 135 | +
|
| 136 | + mkdir -p $out/lib |
| 137 | + ar rcs $out/lib/$name ''${ARGS[@]} |
| 138 | + ''; |
| 139 | + }; |
| 140 | + |
| 141 | + binary = { |
| 142 | + name, |
| 143 | + deps ? [], |
| 144 | + extra-ldflags ? [], |
| 145 | + ldflags ? toolchain.ldflags ++ extra-ldflags, |
| 146 | + }: |
| 147 | + stdenv.mkDerivation { |
| 148 | + inherit name ldflags; |
| 149 | + inherit (toolchain) ld; |
| 150 | + srcs = autowrap_deps deps; |
| 151 | + dontUnpack = true; |
| 152 | + buildPhase = '' |
| 153 | + ARGS=() |
| 154 | + for src in $srcs; do |
| 155 | + if [ -d $src/obj ]; then |
| 156 | + ARGS+=($src/obj/*) |
| 157 | + fi |
| 158 | + if [ -d $src/lib ]; then |
| 159 | + ARGS+=($src/lib/*) |
| 160 | + fi |
| 161 | + done |
| 162 | +
|
| 163 | + mkdir -p $out/bin |
| 164 | + $ld $ldflags -o $out/bin/$name ''${ARGS[@]} |
| 165 | + ''; |
| 166 | + }; |
| 167 | + |
| 168 | + set = deps: |
| 169 | + pkgs.symlinkJoin { |
| 170 | + name = ""; |
| 171 | + paths = autowrap_deps deps; |
| 172 | + }; |
| 173 | +in { |
| 174 | + inherit asm include object static binary set; |
| 175 | +} |
0 commit comments