feat: add Database source jdbc uri format

pkg/config.go

@@ -109,6 +109,14 @@ type AWSConfig struct {
 }
 
 func initDbConfig(cmd *cobra.Command) *dbConfig {
+	jdbcUri := os.Getenv("DB_URL")
+	if len(jdbcUri) != 0 {
+		config, err := convertJDBCToDbConfig(jdbcUri)
+		if err != nil {
+			utils.Fatal("Error: %v", err.Error())
+		}
+		return config
+	}
 	// Set env
 	utils.GetEnv(cmd, "dbname", "DB_NAME")
 	dConf := dbConfig{}
@@ -293,6 +301,20 @@ func initRestoreConfig(cmd *cobra.Command) *RestoreConfig {
 	return &rConfig
 }
 func initTargetDbConfig() *targetDbConfig {
+	jdbcUri := os.Getenv("TARGET_DB_URL")
+	if len(jdbcUri) != 0 {
+		config, err := convertJDBCToDbConfig(jdbcUri)
+		if err != nil {
+			utils.Fatal("Error: %v", err.Error())
+		}
+		return &targetDbConfig{
+			targetDbHost:     config.dbHost,
+			targetDbPort:     config.dbPort,
+			targetDbName:     config.dbName,
+			targetDbPassword: config.dbPassword,
+			targetDbUserName: config.dbUserName,
+		}
+	}
 	tdbConfig := targetDbConfig{}
 	tdbConfig.targetDbHost = os.Getenv("TARGET_DB_HOST")
 	tdbConfig.targetDbPort = utils.EnvWithDefault("TARGET_DB_PORT", "5432")

pkg/helper.go

@@ -29,6 +29,7 @@ import (
 	"fmt"
 	"github.com/jkaninda/pg-bkup/utils"
 	"gopkg.in/yaml.v3"
+	"net/url"
 	"os"
 	"os/exec"
 	"path/filepath"
@@ -196,3 +197,37 @@ func RemoveLastExtension(filename string) string {
 	}
 	return filename
 }
+func convertJDBCToDbConfig(jdbcURI string) (*dbConfig, error) {
+	// Remove the "jdbc:" prefix
+	if strings.HasPrefix(jdbcURI, "jdbc:") {
+		jdbcURI = strings.TrimPrefix(jdbcURI, "jdbc:")
+	}
+
+	// Parse the URI
+	u, err := url.Parse(jdbcURI)
+	if err != nil {
+		return &dbConfig{}, fmt.Errorf("failed to parse JDBC URI: %v", err)
+	}
+	// Extract components
+	host := u.Hostname()
+	port := u.Port()
+	if port == "" {
+		port = "5432" // Default PostgreSQL port
+	}
+	database := strings.TrimPrefix(u.Path, "/")
+	params, _ := url.ParseQuery(u.RawQuery)
+	username := params.Get("user")
+	password := params.Get("password")
+	// Validate essential fields
+	if host == "" || database == "" || username == "" {
+		return &dbConfig{}, fmt.Errorf("incomplete JDBC URI: missing host, database, or username")
+	}
+
+	return &dbConfig{
+		dbHost:     host,
+		dbPort:     port,
+		dbName:     database,
+		dbUserName: username,
+		dbPassword: password,
+	}, nil
+}