From d4b5fbb8d836998273da34ac7eea7654c207a56d Mon Sep 17 00:00:00 2001
From: Joel Rebello <jrebello@packet.com>
Date: Mon, 11 Sep 2023 10:35:45 +0200
Subject: [PATCH] examples: Add example to upload and unmount a floppy image

---
 examples/floppy-image/doc.go  | 19 +++++++++
 examples/floppy-image/main.go | 79 +++++++++++++++++++++++++++++++++++
 2 files changed, 98 insertions(+)
 create mode 100644 examples/floppy-image/doc.go
 create mode 100644 examples/floppy-image/main.go

diff --git a/examples/floppy-image/doc.go b/examples/floppy-image/doc.go
new file mode 100644
index 00000000..c89443f9
--- /dev/null
+++ b/examples/floppy-image/doc.go
@@ -0,0 +1,19 @@
+/*
+inventory is an example commmand that utilizes the 'v1' bmclib interface
+methods to upload and mount, unmount a floppy image.
+
+	    # mount image
+		$ go run examples/floppy-image/main.go \
+			-host 10.1.2.3 \
+			-user ADMIN \
+			-password hunter2 \
+			-image /tmp/floppy.img
+
+		# un-mount image
+		$ go run examples/floppy-image/main.go \
+			-host 10.1.2.3 \
+			-user ADMIN \
+			-password hunter2 \
+			-unmount
+*/
+package main
diff --git a/examples/floppy-image/main.go b/examples/floppy-image/main.go
new file mode 100644
index 00000000..d7a25374
--- /dev/null
+++ b/examples/floppy-image/main.go
@@ -0,0 +1,79 @@
+package main
+
+import (
+	"context"
+	"crypto/x509"
+	"flag"
+	"log"
+	"os"
+	"time"
+
+	bmclib "github.com/bmc-toolbox/bmclib/v2"
+	"github.com/bombsimon/logrusr/v2"
+	"github.com/sirupsen/logrus"
+)
+
+func main() {
+	ctx, cancel := context.WithTimeout(context.Background(), 120*time.Second)
+	defer cancel()
+
+	user := flag.String("user", "", "Username to login with")
+	pass := flag.String("password", "", "Username to login with")
+	host := flag.String("host", "", "BMC hostname to connect to")
+	imagePath := flag.String("image", "", "The .img file to be uploaded")
+	unmountImage := flag.Bool("unmount", false, "Unmount floppy image")
+
+	withSecureTLS := flag.Bool("secure-tls", false, "Enable secure TLS")
+	certPoolFile := flag.String("cert-pool", "", "Path to an file containing x509 CAs. An empty string uses the system CAs. Only takes effect when --secure-tls=true")
+	flag.Parse()
+
+	l := logrus.New()
+	l.Level = logrus.DebugLevel
+	logger := logrusr.New(l)
+
+	clientOpts := []bmclib.Option{bmclib.WithLogger(logger)}
+
+	if *withSecureTLS {
+		var pool *x509.CertPool
+		if *certPoolFile != "" {
+			pool = x509.NewCertPool()
+			data, err := os.ReadFile(*certPoolFile)
+			if err != nil {
+				l.Fatal(err)
+			}
+			pool.AppendCertsFromPEM(data)
+		}
+		// a nil pool uses the system certs
+		clientOpts = append(clientOpts, bmclib.WithSecureTLS(pool))
+	}
+
+	cl := bmclib.NewClient(*host, *user, *pass, clientOpts...)
+	err := cl.Open(ctx)
+	if err != nil {
+		log.Fatal(err, "bmc login failed")
+	}
+
+	defer cl.Close(ctx)
+
+	if *unmountImage {
+		if err := cl.UnmountFloppyImage(ctx); err != nil {
+			log.Fatal(err)
+		}
+
+		return
+	}
+
+	// open file handle
+	fh, err := os.Open(*imagePath)
+	if err != nil {
+		l.Fatal(err)
+	}
+	defer fh.Close()
+
+	err = cl.UploadFloppyImage(ctx, fh)
+	if err != nil {
+		l.Fatal(err)
+	}
+
+	l.WithField("img", *imagePath).Info("image mounted successfully")
+}