From ecfea6fa5acec539c8a14786608d1873e6c0baa0 Mon Sep 17 00:00:00 2001
From: sarthak <sarthaknegi908@gmail.com>
Date: Tue, 23 Jan 2024 09:09:08 +0530
Subject: [PATCH] Signed-off-by: sarthak <sarthaknegi908@gmail.com>

remove stack and implement suggested changes
---
 pkg/client/new_extension.go | 31 +++++++++++++++++++++++++++----
 1 file changed, 27 insertions(+), 4 deletions(-)

diff --git a/pkg/client/new_extension.go b/pkg/client/new_extension.go
index da72bbb36b..3ae1406ca5 100644
--- a/pkg/client/new_extension.go
+++ b/pkg/client/new_extension.go
@@ -13,6 +13,32 @@ import (
 	"github.com/buildpacks/pack/pkg/dist"
 )
 
+var (
+	bashBinGenerate = `#!/usr/bin/env bash
+	
+	set -eo pipefail
+	
+	# 1. GET ARGS
+	output_dir=$CNB_OUTPUT_DIR
+	
+	# 2. GENERATE build.Dockerfile
+	cat >> "${output_dir}/build.Dockerfile" <<EOL
+	ARG base_image
+	FROM \${base_image}
+	
+	RUN echo "Hello from build extension"
+	EOL
+	
+	# 3. GENERATE run.Dockerfile
+	cat >> "${output_dir}/run.Dockerfile" <<EOL
+	ARG base_image
+	FROM \${base_image}
+	
+	RUN echo "Hello from run extension"
+	EOL
+`
+)
+
 type NewExtensionOptions struct {
 	// api compat version of the output extension artifact.
 	API string
@@ -25,9 +51,6 @@ type NewExtensionOptions struct {
 
 	// version of the output extension artifact.
 	Version string
-
-	// Deprecated: The stacks this extension will work with
-	Stacks []dist.Stack
 }
 
 func (c *Client) NewExtension(ctx context.Context, opts NewExtensionOptions) error {
@@ -39,7 +62,7 @@ func (c *Client) NewExtension(ctx context.Context, opts NewExtensionOptions) err
 }
 
 func createBashExtension(path string, c *Client) error {
-	if err := createBinScript(path, "build", bashBinBuild, c); err != nil {
+	if err := createBinScript(path, "generate", bashBinGenerate, c); err != nil {
 		return err
 	}