Add filter for the filename, fixes #16

deliciousbrains · Jul 26, 2013 · 08347b5 · 08347b5
1 parent 9654f76
commit 08347b5
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/wp-migrate-db.php b/wp-migrate-db.php
@@ -99,13 +99,17 @@ function handle_request() {
 
     function get_filename( $datetime, $gzip ) {
         $hash = substr( sha1( DB_PASSWORD . AUTH_SALT ), -5 );
-        $filename = DB_NAME . '-migrate-' . $datetime . '-' . $hash . '.sql';
+        $filename = DB_NAME . '-migrate-' . $datetime . '-' . $hash;
+        $filename = apply_filters( 'wpmdb_filename', $filename, $datetime );
+        $filename .= '.sql';
         if ( $gzip ) $filename .= '.gz';
         return $filename;
     }
 
     function get_nicename( $datetime, $gzip ) {
-        $name = DB_NAME . '-migrate-' . $datetime . '.sql';
+        $name = DB_NAME . '-migrate-' . $datetime;
+        $name = apply_filters( 'wpmdb_filename', $name, $datetime );
+        $name .= '.sql';
         if ( $gzip ) $name .= '.gz';
         return $name;
     }