Update install_postgres_json_schema_extension.sh

install_postgres_json_schema_extension.sh

@@ -19,6 +19,12 @@ install_ubuntu() {
     apt-get install -y $@
 }
 
+# Function to get PostgreSQL version on Ubuntu
+get_postgres_version_ubuntu() {
+    local pg_version=$(ls /usr/share/postgresql/ | grep -E '^[0-9]+$' | sort -rn | head -n1)
+    echo "$pg_version"
+}
+
 # Install packages based on the detected OS
 if [[ "$OS" == *"Alpine"* ]]; then
     echo "Detected Alpine Linux"
@@ -40,17 +46,24 @@ cd postgres-json-schema
 # Build the extension
 make && make install
 
-
 if [[ "$OS" == *"Alpine"* ]]; then
     # Create the extension directory if it doesn't exist
     mkdir -p '/usr/local/share/postgresql/extension'
     # Set appropriate permissions for the control file
     chmod 644 /usr/local/share/postgresql/extension/postgres-json-schema.control
 elif [[ "$OS" == *"Ubuntu"* ]]; then
+    # Get PostgreSQL version
+    PG_VERSION=$(get_postgres_version_ubuntu)
+
+    if [ -z "$PG_VERSION" ]; then
+        echo "Could not determine PostgreSQL version"
+        exit 1
+    fi
+
     # Create the extension directory if it doesn't exist
-    mkdir -p '/usr/share/postgresql/extension'
+    mkdir -p "/usr/share/postgresql/$PG_VERSION/extension"
     # Set appropriate permissions for the control file
-    chmod 644 /usr/share/postgresql/extension/postgres-json-schema.control
+    chmod 644 "/usr/share/postgresql/$PG_VERSION/extension/postgres-json-schema.control"
 else
     echo "Unsupported operating system: $OS"
     exit 1