feat: add IncludeFile for rsync provider

seu-mirrors · Sep 20, 2023 · a251267 · a251267
1 parent db285a6
commit a251267
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 13 deletions.
diff --git a/worker/config.go b/worker/config.go
@@ -173,6 +173,7 @@ type mirrorConfig struct {
 	UseIPv6       bool     `toml:"use_ipv6"`
 	UseIPv4       bool     `toml:"use_ipv4"`
 	ExcludeFile   string   `toml:"exclude_file"`
+	IncludeFile   string   `toml:"include_file"`
 	Username      string   `toml:"username"`
 	Password      string   `toml:"password"`
 	RsyncNoTimeo  bool     `toml:"rsync_no_timeout"`

diff --git a/worker/rsync_provider.go b/worker/rsync_provider.go
@@ -10,19 +10,20 @@ import (
 )
 
 type rsyncConfig struct {
-	name                                         string
-	rsyncCmd                                     string
-	upstreamURL, username, password, excludeFile string
-	extraOptions                                 []string
-	overriddenOptions                            []string
-	rsyncNeverTimeout                            bool
-	rsyncTimeoutValue                            int
-	rsyncEnv                                     map[string]string
-	workingDir, logDir, logFile                  string
-	useIPv6, useIPv4                             bool
-	interval                                     time.Duration
-	retry                                        int
-	timeout                                      time.Duration
+	name                            string
+	rsyncCmd                        string
+	upstreamURL, username, password string
+	excludeFile, includeFile        string
+	extraOptions                    []string
+	overriddenOptions               []string
+	rsyncNeverTimeout               bool
+	rsyncTimeoutValue               int
+	rsyncEnv                        map[string]string
+	workingDir, logDir, logFile     string
+	useIPv6, useIPv4                bool
+	interval                        time.Duration
+	retry                           int
+	timeout                         time.Duration
 
 	uid int
 	gid int
@@ -95,6 +96,9 @@ func newRsyncProvider(c rsyncConfig) (*rsyncProvider, error) {
 	if c.excludeFile != "" {
 		options = append(options, "--exclude-from", c.excludeFile)
 	}
+	if c.includeFile != "" {
+		options = append(options, "--include-from", c.includeFile)
+	}
 	if c.extraOptions != nil {
 		options = append(options, c.extraOptions...)
 	}