Add ability to whitelist using new 'includes' patterns config. If inc…

src/Heyday/Component/Beam/Beam.php

@@ -372,6 +372,14 @@ public function hasPath()
     {
         return is_array($this->options['path']) && count($this->options['path']) > 0;
     }
+    /**
+     * A helper method for determining if beam is operating with a set of includes
+     * @return bool
+     */
+    public function hasIncludes()
+    {
+        return is_array($this->options['includes']) && count($this->options['includes']) > 0;
+    }
     /**
      * Get the server config we are deploying to.
      *
@@ -452,6 +460,14 @@ public function getConfig($config)
             );
         }
     }
+    /**
+     * @param $config
+     * @return boolean
+     */
+    public function hasConfig($config)
+    {
+        return array_key_exists($config, $this->config);
+    }
     /**
      * Check if the deployment provider implements an interface
      * @param $interfaceName

src/Heyday/Component/Beam/Command/InitCommand.php

@@ -47,6 +47,9 @@ protected function execute(InputInterface $input, OutputInterface $output)
         }
 
         $template = array(
+            'include' => array(
+                'patterns' => array(),
+            ),
             'exclude' => array(
                 'patterns' => array(),
             ),

src/Heyday/Component/Beam/Config/BeamConfiguration.php

@@ -221,6 +221,20 @@ function ($v) use ($self) {
                         ->end()
                     ->end()
                 ->end()
+                ->arrayNode('include')
+                    ->children()
+                        ->arrayNode('patterns')
+                            ->prototype('scalar')->end()
+                        ->end()
+                    ->end()
+                    ->validate()
+                        ->always(
+                            function ($v) use ($self) {
+                                return $self->buildIncludes($v);
+                            }
+                        )
+                    ->end()
+                ->end()
                 ->arrayNode('exclude')
                     ->children()
                         ->arrayNode('patterns')
@@ -267,12 +281,24 @@ function ($v) use ($self) {
      */
     public function buildExcludes($value)
     {
-        $value =  $value ? $value : array(
+        $value = $value ? $value : array(
             'patterns' => array()
         );
 
         return array_merge(static::$defaultExcludes, $value['patterns']);
     }
+    /**
+     * @param $value
+     * @return array
+     */
+    public function buildIncludes($value)
+    {
+        $value = $value ? $value : array(
+            'patterns' => array()
+        );
+
+        return $value['patterns'];
+    }
     /**
      * @param $name
      * @param $type

src/Heyday/Component/Beam/DeploymentProvider/Rsync.php

@@ -128,6 +128,7 @@ public function down(\Closure $output = null, $dryrun = false, DeploymentResult
      */
     protected function deploy($command, \Closure $output = null)
     {
+        $this->generateIncludesFile();
         $this->generateExcludesFile();
         $outputHandler = $this->getOutputStreamHandler($output);
         $process = $this->getProcess($command);
@@ -230,6 +231,13 @@ protected function buildCommand($fromPath, $toPath, $dryrun = false)
             $command[] = '--delay-updates';
         }
 
+        if ($this->beam->hasIncludes()) {
+            $command[] = array(
+                '--include-from="%s"',
+                $this->getIncludesPath()
+            );
+        }
+
         $command[] = array(
             '--exclude-from="%s"',
             $this->getExcludesPath()
@@ -405,6 +413,30 @@ public function formatOutput($output)
 
         return $changes;
     }
+    /**
+     * Generate the includes file
+     */
+    protected function generateIncludesFile()
+    {
+        if ($this->beam->hasIncludes()) {
+            $includes = $this->beam->getConfig('include');
+            file_put_contents(
+                $this->getIncludesPath(),
+                implode(PHP_EOL, $includes) . PHP_EOL
+            );
+        }
+    }
+    /**
+     * Get the path to the includes file
+     * @return string
+     */
+    protected function getIncludesPath()
+    {
+        return sprintf(
+            '/tmp/%s.includes',
+            $this->beam->getLocalPathname()
+        );
+    }
     /**
      * Generate the excludes file
      */
@@ -420,6 +452,10 @@ protected function generateExcludesFile()
                 unset($excludes[$idx]);
             }
         }
+        // If any includes have been specified we need to exclude everything
+        if ($this->beam->hasIncludes()) {
+            $excludes = array_merge(array('/*'), $excludes);
+        }
         file_put_contents(
             $this->getExcludesPath(),
             implode(PHP_EOL, $excludes) . PHP_EOL