# wrapper over vscode to control extensions per project (extensions folder will be created in execution path)
{ lib                             
, writeShellScriptBin             
, extensionsFromVscodeMarketplace 
, vscodeDefault
}:
## User input
{ vscode ? vscodeDefault                          
# extensions to be symlinked into the project's extensions folder
, nixExtensions        ? []                   
# extensions to be copied into the project's extensions folder
, mutableExtensions    ? []        
, vscodeExtsFolderName ? ".vscode-exts"        
}:
let 
  nixExtsDrvs = extensionsFromVscodeMarketplace nixExtensions;
  mutExtsDrvs = extensionsFromVscodeMarketplace mutableExtensions;

  #removed not defined extensions
  rmExtensions =  lib.optionalString (nixExtensions++mutableExtensions != []) ''
    find ${vscodeExtsFolderName} -mindepth 1 -maxdepth 1 ${lib.concatMapStringsSep " " (e : ''! -iname ${e.publisher}.${e.name}'') (nixExtensions++mutableExtensions)} -exec sudo rm -rf {} \;
  '';
  #copy mutable extension out of the nix store
  cpExtensions = ''
    ${lib.concatMapStringsSep "\n" (e : ''ln -sfn ${e}/share/vscode/extensions/* ${vscodeExtsFolderName}/'') nixExtsDrvs}
    ${lib.concatMapStringsSep "\n" (e : ''
      cp -a ${e}/share/vscode/extensions/${e.vscodeExtUniqueId} ${vscodeExtsFolderName}/${e.vscodeExtUniqueId}-${(lib.findSingle (ext: ''${ext.publisher}.${ext.name}'' == e.vscodeExtUniqueId) "" "m" mutableExtensions ).version}
      '') mutExtsDrvs} 
  '';
in
  writeShellScriptBin "code" ''
    if ! [[ "$@" =~ "--list-extension" ]]; then 
      mkdir -p ${vscodeExtsFolderName} 
      ${rmExtensions}
      ${cpExtensions}
    fi
    ${vscode}/bin/code --extensions-dir ${vscodeExtsFolderName} "$@"
  ''