From c08f262ef5f25396a66e886360cf013c1dd5cc03 Mon Sep 17 00:00:00 2001
From: Tom Chauveau <tom.chauveau@epitech.eu>
Date: Thu, 23 Dec 2021 15:27:03 +0100
Subject: [PATCH] Add platform configuration in `plan.go` to set platform in
 context.

Signed-off-by: Vasek - Tom C <tom.chauveau@epitech.eu>
---
 plan/plan.go | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)

diff --git a/plan/plan.go b/plan/plan.go
index 920f5ac7..74ad9207 100644
--- a/plan/plan.go
+++ b/plan/plan.go
@@ -48,6 +48,10 @@ func Load(ctx context.Context, args ...string) (*Plan, error) {
 		return nil, err
 	}
 
+	if err := p.configPlatform(); err != nil {
+		return nil, err
+	}
+
 	return p, nil
 }
 
@@ -59,6 +63,31 @@ func (p *Plan) Source() *compiler.Value {
 	return p.source
 }
 
+// configPlatform load the platform specified in the context
+// Buildkit will then run every operation using that platform
+// If platform is not define, context keep default platform
+func (p *Plan) configPlatform() error {
+	platformField := p.source.Lookup("platform")
+
+	// Ignore if platform is not set in `#Plan`
+	if !platformField.Exists() {
+		return nil
+	}
+
+	// Convert platform to string
+	platform, err := platformField.String()
+	if err != nil {
+		return err
+	}
+
+	// Set platform to context
+	err = p.context.Platform.Set(platform)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
 // registerLocalDirectories scans the context for local imports.
 // BuildKit requires to known the list of directories ahead of time.
 func (p *Plan) registerLocalDirs() error {