refactor: add structure to data

add: argumnet based support

Author: abhijeetmohanan

ftpcli.go

@@ -0,0 +1,37 @@
+package main
+
+import (
+	"flag"
+	"gufcli/utils"
+	"log"
+)
+
+func main() {
+	log.Println("Starting Connection")
+	// Define flags
+	source_ftp := flag.String("src", "", "Source ftp endpoint Example: ftp://user:pass@host/filepath")
+	destination_ftp := flag.String("dest", "", "Destination ftp endpoint Example: ftp://user:pass@host/filepath")
+
+	flag.Parse()
+
+	utils.NullChecker("source", *source_ftp)
+	utils.NullChecker("destination", *destination_ftp)
+
+	source_map_kv := utils.ParseInput("source", *source_ftp)
+	destination_map_kv := utils.ParseInput("destination", *destination_ftp)
+
+	if utils.SchemeValidator(source_map_kv["scheme"]) && utils.SchemeValidator(destination_map_kv["scheme"]) {
+		log.Println("Good Boy ")
+
+		// validate ftp parameters : panic on failure
+		utils.FtpParamsValidator(source_map_kv)
+		utils.FtpParamsValidator(destination_map_kv)
+
+		// Create Coonection and start streaming
+
+		utils.FtpClientHandler(source_map_kv, destination_map_kv)
+
+	} else {
+		log.Panicln("Maradchod :: ? ftp hai dono")
+	}
+}

main.go

@@ -0,0 +1,62 @@
+package utils
+
+import (
+	"bytes"
+	"io"
+	"log"
+	"time"
+
+	"github.com/jlaffaye/ftp"
+)
+
+func FtpClientHandler(source map[string]string, destination map[string]string) {
+	// Make connections
+	source_ftp, err := ftp.Dial(source["host"], ftp.DialWithTimeout(5*time.Second))
+	if err != nil {
+		log.Fatal(err)
+	} else {
+		log.Println("Source Connected Successfully")
+	}
+	destination_ftp, err := ftp.Dial(destination["host"], ftp.DialWithTimeout(5*time.Second))
+	if err != nil {
+		log.Fatal(err)
+	} else {
+		log.Println("Destination Connected Successfully")
+	}
+
+	// Login in using creds
+	err = source_ftp.Login(source["username"], source["password"])
+	if err != nil {
+		log.Fatal(err)
+	}
+	err = destination_ftp.Login(destination["username"], destination["password"])
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	// start Reading from Source
+	reader_ftp, err := source_ftp.Retr(source["path"])
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	// initalize a buffer of size 64Kb
+	p := make([]byte, 64*1024)
+
+	for {
+		n, err := reader_ftp.Read(p)
+
+		if err == io.EOF {
+			break
+		}
+
+		wdata := bytes.NewBufferString(string(p[:n]))
+		err = destination_ftp.Append(destination["path"], wdata)
+		if err != nil {
+			log.Fatal(err)
+		}
+
+	}
+	// Close Source FTP
+	defer reader_ftp.Close()
+}

utils/ftphandler.go

@@ -0,0 +1,64 @@
+package utils
+
+import (
+	"log"
+	"net/url"
+)
+
+func ParseInput(key string, furl string) map[string]string {
+	// Define return variable
+
+	params := make(map[string]string)
+	//
+
+	// Parse the URLs
+	sourceURL, err := url.Parse(furl)
+	if err != nil {
+		log.Panicf("Error parsing source URL: %v\n", err)
+	}
+
+	// Output the parsed URLs
+
+	// Extract and print the protocol (scheme)
+	params["scheme"] = sourceURL.Scheme
+	params["username"] = sourceURL.User.Username()
+	sourcePassword, _ := sourceURL.User.Password()
+	params["password"] = sourcePassword
+	params["host"] = sourceURL.Host
+	params["path"] = sourceURL.Path
+
+	// log the paramters used for connection
+	// ToDo: sorted map implementation
+	parameters := ""
+
+	for k, v := range params {
+		parameters = parameters + k + ": " + v + ", "
+	}
+
+	log.Println(key + ":: " + parameters)
+
+	return params
+}
+
+func NullChecker(key string, ftpurl string) {
+	// Check if string is null
+	if ftpurl == "" {
+		log.Panicf("Error: Empty URL for %s", key)
+	}
+}
+
+func SchemeValidator(scheme string) bool {
+	log.Println("scheme: %s", scheme)
+	if scheme == "ftp" {
+		return true
+	}
+	return false
+}
+
+func FtpParamsValidator(params map[string]string) {
+	for k, v := range params {
+		if v == "" {
+			log.Panicf("Null Values passed for %v", k)
+		}
+	}
+}