From 4594bb47fd59ee9236d54e1e075ecfd6601304b9 Mon Sep 17 00:00:00 2001
From: Alicia Sykes <alicia@omg.lol>
Date: Sat, 11 Nov 2023 18:33:30 +0000
Subject: [PATCH] If yarn not present, then enable Corepack

---
 config/zsh/aliases/node-js.zsh | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/config/zsh/aliases/node-js.zsh b/config/zsh/aliases/node-js.zsh
index efd54e2..3996abf 100644
--- a/config/zsh/aliases/node-js.zsh
+++ b/config/zsh/aliases/node-js.zsh
@@ -86,6 +86,7 @@ print_node_versions () {
   # Print versions of core Node things
   get_version 'node' 'Node.js'
   get_version 'npm' 'NPM'
+  get_version 'corepack' 'Corepack'
   get_version 'yarn' 'Yarn'
   get_version 'nvm' 'NVM'
   get_version 'ni' 'ni'
@@ -123,6 +124,27 @@ function source_nvm node npm yarn $NVM_LAZY_CMD {
   hash "$0" 2> /dev/null && command "$0" "$@"
 }
 
+# Helper function to enable Corepack / use Yarn
+enable_corepack () {
+  if ! hash 'yarn' 2> /dev/null && hash 'corepack' 2> /dev/null; then
+    echo -e "\033[1;93mEnabling Corepack...\033[0m"
+    corepack enable
+  else
+    echo -e "\033[1;31mCorepack already enabled, skipping...\033[0m"
+  fi
+}
+
+# Wrapper function for Yarn, which sets up Yarn if it's not yet found
+yarn_wrapper () {
+  if ! hash 'yarn' 2> /dev/null; then
+    echo "Yarn not found, enabling Corepack..."
+    enable_corepack
+  fi
+  yarn "$@"
+}
+
+alias yarn='yarn_wrapper'
+
 # Helper function to install NVM
 install_nvm () {
   nvm_repo='https://github.com/nvm-sh/nvm.git'