Merge pull request #4 from oelmekki/default_params

FIX functions default parameters

function.go

@@ -4,6 +4,7 @@ import (
 	"io/ioutil"
 	"fmt"
 	"regexp"
+	"strings"
 )
 
 /*
@@ -84,6 +85,9 @@ func ( function *Function ) Parse() ( err error ) {
 		if err != nil { return err }
 	}
 
+	err = function.removeDefaultFromSignature()
+	if err != nil { return }
+
 	return
 }
 
@@ -125,3 +129,28 @@ func ( function *Function ) previousSignature() ( signature string, exists bool,
 
 	return
 }
+
+
+/*
+ * `DROP FUNCTION` raises error if the signature contains `DEFAULT` values for
+ * parameters, so we must remove them.
+ */
+func ( function *Function ) removeDefaultFromSignature() ( err error ) {
+	anyDefault, err := regexp.MatchString( "(?i)DEFAULT", function.Signature )
+	if err != nil { return }
+
+	if anyDefault {
+		arguments := strings.Split( function.Signature, "," )
+		newArgs := make( []string, 0 )
+
+		for _, arg := range arguments {
+			arg = strings.Replace( arg, " DEFAULT ", " default ", -1 )
+			newArg := strings.Split( arg, " default " )[0]
+			newArgs = append( newArgs, newArg )
+		}
+
+		function.Signature = strings.Join( newArgs, "," )
+	}
+
+	return
+}

main.go

@@ -45,7 +45,7 @@ func CheckSanity() {
  */
 func Usage() {
 	usage := `
-PgRebase-1.1.0
+PgRebase-1.1.1+
 
 USAGE:
 	DATABASE_URL=url pgrebase [-w] <sql_directory>