pi_customizer 0.4.0.pre.alpha → 0.5.0.pre.alpha

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 3e96a6819c5a9170bb9367951af0469047ecefbe3fad6fc4f5dc6ce47c2855d2
-  data.tar.gz: f56710482b72e307724728a2c9280bcea3786085787521ae7a64f3500249b7db
+  metadata.gz: 3590f0a62efe485418702779492c5bde407747512c6046e1f0235212d3cecfa4
+  data.tar.gz: b48fd21a511f936bc111fe66207f95c55692afdb3bc7d690714efb81b7d26817
 SHA512:
-  metadata.gz: eb34c1d0d8deffc02cc7d42b5053f2ea599e31ed2b4d38a3f44a1ca3a64d1d065f1f2bf4823c33f4a190ac6b925c8a19bb1a115ebbef12d70f680962b8d1dcac
-  data.tar.gz: 4dfe1517a6d767be613e031b473fd5bbfb3fc98abdc00147ee5bdf81eba3f7dd75d9c9e92c38ab800baca077272975253084b98aefb18f6b6c4b245337eff535
+  metadata.gz: 847bb56ecdac8c8d17210ce31bef76317a489de5e94fec0686aa7eafdf1f01355ec821fcf3bb1bd0e64e290da6b8ed7cf2523456a115031a4bfb27bad8edb917
+  data.tar.gz: 228cde5dd9b4d42974a1af2768e2790712f03a65328a9fc1cfc38c8b614dfc8174a0680ef747b8911a176defe581d656091c0ff8d8707244380455c61fc234df

data/.gitignore

@@ -14,7 +14,6 @@ ssh/
 *.conf
 wpa_supplicant.json
 .vagrant/
-Gemfile.lock
 *gem
 
 cid

data/CODE_OF_CONDUCT.md

@@ -55,7 +55,7 @@ further defined and clarified by project maintainers.
 ## Enforcement
 
 Instances of abusive, harassing, or otherwise unacceptable behavior may be
-reported by contacting the project team at b.ottenwaelder@gmail.com. All
+reported by contacting the project team at ottenwbe.public@gmail.com. All
 complaints will be reviewed and investigated and will result in a response that
 is deemed necessary and appropriate to the circumstances. The project team is
 obligated to maintain confidentiality with regard to the reporter of an incident.

data/README.md

@@ -9,30 +9,27 @@
 
 
 The _pi_customizer_ allows you to adapt Raspbian images to your needs.
-To this end, the Raspbian image is be built from [scratch](https://github.com/ottenwbe/pi-gen.git) 
+To this end, an image is built from [scratch](https://github.com/ottenwbe/pi-gen.git) 
 with all customizations baked into the image. 
 
 The image is built in an isolated [build environment](#environments), e.g., 
 a vagrant box, which is orchestrated by the pi_customizer.
-The customization is performed by a Ruby Gem (_pi_build_modifier_) in the build environment by adapting the [pi-gen](https://github.com/RPi-Distro/pi-gen) build scripts.
+The customization is performed behind the scenes by Ruby scripts (_pi_build_modifier_) in the build environment by adapting the [pi-gen](https://github.com/RPi-Distro/pi-gen) build scripts.
 
-
-# WIP Notice
-
-This project is still under construction and in a pre release phase.
+Finally, pi_customizer also allows you to write the image to an SD card.
 
 # Install
 
 ## Prerequisites
 
 To customize the build process the following prerequisites are expected:
-* The _pi_customizer_ gem is installed (see the [__Deploy Gem__](#deploy_gem) section for details) 
+* The _pi_customizer_ gem is installed (see the [__Install Gem__](#deploy_gem) section for details) 
 * One of the supported build environments is accessible (see the [__Environments__](#environments) section):  
     * Vagrant 
     * Docker (feature is still in development)    
 
 <a name="deploy_gem"></a>
-## Deploy Gem
+## Install Gem
 
 To customize your Raspbian image the pi_customizer gem must be installed on the machine that coordinates the build process.
 This is typically your local machine.
@@ -81,7 +78,7 @@ Note: feature still in development
 Docker uses a temporary folder as volume to build the image. 
 The folder location can be changed with the build option __tmp-folder__, which is set to "${PWD}/tmp" by default.
 
-Note: the type of the tmp directory needs to be changed on SELinux
+Note: the type of the workspace directory needs to be changed on SELinux
 
     chcon -Rt svirt_sandbox_file_t "${PWD}/tmp"
     
@@ -98,8 +95,8 @@ To build a default image in a Vagrant box, simply execute on the command line
 
 
 To customize your build:
-1. All customizations have to be specified in a json configuration file (see the [__Config File__](#config_file) section for details).
-1. The build process itself is then configured with command line options (see `pi_customizer help build` for details)   
+1. All customizations have to be specified in a json configuration file (see the [Config File](#config_file) section for details).
+1. The build process itself is then configured with command line options (see `pi_customizer help build` or the [CLI](#cli_options) section for details)   
      
 
 <a name="config_file"></a>
@@ -110,7 +107,9 @@ An example of the json config file with all current configuration options
     {
       "system": {
         "name" : "custompi",
-        "type" : "lite"
+        "type" : "lite",
+        "username" : "pi", 
+        "password" : "raspberry"
       },
       "ssh" : {
         "enabled" : true
@@ -139,6 +138,10 @@ An example of the json config file with all current configuration options
 
 * To build a lite version of the pi image, specify the __type__ as lite, otherwise specify it as full. The default is _full_.
 
+* Select the __username__ for the first user, by default this is __pi__.
+
+* To change the default __raspberry__ password, specify a __password__. 
+
 ### The ssh section
 
 * To disable ssh by default, set __enabled__ to false.
@@ -156,9 +159,39 @@ An example of the json config file with all current configuration options
 
 * The wifi section allows you to specify networks and passphrase __OR__ wpa_passpharse as well as the wpa_country.
 
+<a name="cli_options"></a>
+## Command Line Options
+
+Change the location of the configuration file with ```--config, -c```.
+
+``` 
+pi_customizer -c myconfig.conf 
+``` 
+
+Change the git location of the pi-gen build sources with ```--build_sources_git_url -g``` or the used branch/tag  ```--build_sources_git_tag, -t```. __NOTE__: Only works with tags from 2019 or later.
+
+``` 
+pi_customizer -g https://github.com/ottenwbe/pi-gen.git -t 2019-06-20-raspbian-buster
+``` 
+
+Change the workspace directory on the local machine with ```--local_workspace_dir, -l```. Environment configuration and built images are stored in the workspace.
+
+``` 
+pi_customizer -l ./workspace
+``` 
+
 # Development
 
-### Prerequisites
+## Why Ruby?
+
+Well, the author wanted to learn Ruby.
+
+## Structure
+
+For now, the sources of both gems (pi_customizer and pi_build_modifier) are versioned in one repository. 
+This might change if their versions run out of sync.  
+
+## Prerequisites
 
 * Make sure Ruby is installed (and dependencies to build native extensions). See, https://github.com/rbenv/rbenv and https://github.com/rbenv/ruby-build.
 
@@ -170,6 +203,7 @@ An example of the json config file with all current configuration options
     
     git clone https://github.com/rbenv/rbenv.git ~/.rbenv
     echo 'export PATH="$HOME/.rbenv/bin:$PATH"' >> ~/.zshrc
+    echo 'export PATH="$HOME/.rbenv/shim:$PATH"' >> ~/.zshrc
     ~/.rbenv/bin/rbenv init
     exec $SHELL
 
@@ -181,23 +215,24 @@ An example of the json config file with all current configuration options
     rbenv global 2.5.5
     ```
 
-
-* Install Gems    
+* Install build dependencies 
 
     ```
     gem install bundler rake rspec rdoc
     ```    
 
+* Install dependencies
+
+    ```
+    cd pi_customizer/ && bundle install && cd ..
+    cd pi_build_modifier/ && bundle install && cd ..
+    ```    
+
 ## Test
 
 RSpec tests can be executed for all modules by calling the following rake command in the project root.
 
     rake spec
-
-## Structure
-
-For now, the sources of both gems (pi_customizer and pi_build_modifier) are versioned in one repository. 
-This might change if their versions run out of sync.  
     
 ## Contributing
 

data/Rakefile

@@ -14,6 +14,7 @@ begin
   require 'rspec/core/rake_task'
 
   RSpec::Core::RakeTask.new(:spec) do |t|
+
     # find all spec files (for all gems)
     pattern = ''
     all_gems.each do |gem|
@@ -26,6 +27,7 @@ begin
     # tell rspec to execute tests on all spec files
     t.pattern = pattern
   end
+
 rescue LoadError
   puts 'RSpec is not installed. This means rake is not able to execute tests! '
   puts '* Try: gem install rspec'
@@ -33,7 +35,6 @@ end
 
 task :default => :spec
 
-
 # release task
 
 desc 'Release and upload to Rubygems.org'
@@ -64,20 +65,20 @@ end
 desc "Show the gem's versions"
 task :versions do
   require_relative "#{PI_CUSTOMIZER}/lib/#{PI_CUSTOMIZER}/version"
-  puts "#{PI_CUSTOMIZER} Versions"  
+  puts "#{PI_CUSTOMIZER} Versions"
   puts ' '
   puts '*** Dev VERSION ***'
   puts ' '
   puts "#{PI_CUSTOMIZER}: #{PiCustomizer::VERSION} "
-  system "gem list #{PI_CUSTOMIZER} --pre --remote"  
+  system "gem list #{PI_CUSTOMIZER} --pre --remote"
   puts ' '
-  require_relative "#{PI_BUILD_MODIFIER}/lib/#{PI_BUILD_MODIFIER}/version"  
+  require_relative "#{PI_BUILD_MODIFIER}/lib/#{PI_BUILD_MODIFIER}/version"
   puts "#{PI_BUILD_MODIFIER} Versions"
   puts ' '
   puts '*** Dev VERSION ***'
   puts ' '
   puts "#{PI_BUILD_MODIFIER}: #{PiBuildModifier::VERSION}"
-  system "gem list #{PI_BUILD_MODIFIER} --pre --remote"  
+  system "gem list #{PI_BUILD_MODIFIER} --pre --remote"
 end
 
 # build tasks

data/docs/concept.md

@@ -7,21 +7,3 @@
 
 * __Modification__ steps change files in the remote workspace based on a user defined configuration file
 * __Build__ steps are all executed in the environment
-
-# Build Steps
-
-1. Environment creation and selection
-1. Environment startup
-1. Workspace creation 
-1. Load sources to workspace
-1. Load modification sources
-1. Modify sources in workspace
-1. Build pi image
-1. Ship pi image to destination
-1. Delete Environment
-
-# Modes
-1. Prepare and Run - Prepare everything for the environment (e.g., a Vagrantfile) and then simply run the Vagrantfile to build the image
-    1. Example: Vagrant
-1. Start and Adapt - Start an Environment (e.g., a VM in AWS) and then simply run then configure the environment to build it afterwards
-    1. Examples: Docker 

data/docs/incubation.md

@@ -1,16 +1,10 @@
-Potential improvements:
-
-
 * As a builder of a pi image I want to add ssh-keys and bake them into the image (hostkey, admin users public key, ...) to avoid adding it afterwards manually
 * As a builder of a pi image I want to be able to disable the swap file for my image, e.g., to be able to install kubernetes on it
     * sudo systemctl disable dphys-swapfile
 * As a maintainer of the pi image build sources fork I want to be informed if the templates used for the pi_build_modifier are out of date, to avoid inconsistencies
-
-* Add users
-
-* Change hostname
-
-* Install Docker
+* As a builder of a pi image I want to change the hostname of the pi image
+* As a build of a pi image I want to install custom software
+* As a build of a pi image I want to install Docker
      * echo deb https://apt.dockerproject.org/repo debian-stretch main > /etc/apt/sources.list.d/docker.list; apt update
      * apt install docker-ce 
      * ------------ or:
@@ -27,11 +21,6 @@ Potential improvements:
         tee /etc/apt/sources.list.d/docker.list
 
      apt-get update && apt-get install -y docker-ce=$(apt-cache madison docker-ce | grep 17.09 | head -1 | awk '{print $3}')
-     
-* Install Kubernetes
-    * curl -s https://packages.cloud.google.com/apt/doc/apt-key.gpg | apt-key add -
-    * echo "deb http://apt.kubernetes.io/ kubernetes-xenial main" > /etc/apt/sources.list.d/kubernetes.list
-    * apt-get update && apt-get install -y kubelet kubeadm kubectl
     
     
     

data/pi_build_modifier/Gemfile.lock

@@ -0,0 +1,50 @@
+PATH
+  remote: .
+  specs:
+    pi_build_modifier (0.5.0.pre.alpha)
+      json (~> 2.2)
+      openssl (~> 2.1)
+      thor (~> 0.20)
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    diff-lcs (1.3)
+    docile (1.3.1)
+    json (2.2.0)
+    openssl (2.1.2)
+    rake (12.3.2)
+    rdoc (6.1.1)
+    rspec (3.8.0)
+      rspec-core (~> 3.8.0)
+      rspec-expectations (~> 3.8.0)
+      rspec-mocks (~> 3.8.0)
+    rspec-core (3.8.0)
+      rspec-support (~> 3.8.0)
+    rspec-expectations (3.8.2)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.8.0)
+    rspec-mocks (3.8.0)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.8.0)
+    rspec-support (3.8.0)
+    simplecov (0.16.1)
+      docile (~> 1.1)
+      json (>= 1.8, < 3)
+      simplecov-html (~> 0.10.0)
+    simplecov-html (0.10.2)
+    thor (0.20.3)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  bundler (~> 2.0)
+  pi_build_modifier!
+  rake (~> 12.3)
+  rdoc (~> 6.1)
+  rspec (~> 3.8)
+  simplecov (~> 0.16)
+
+BUNDLED WITH
+   2.0.2

data/pi_build_modifier/lib/pi_build_modifier.rb

@@ -19,11 +19,9 @@
 # SOFTWARE.
 
 require 'thor'
-require 'pi_build_modifier/net-tweaks/wifi_network'
-require 'pi_build_modifier/modifier/erb_mapper'
+require 'pi_build_modifier/modify/modifiers'
 require 'pi_build_modifier/version'
-require 'pi_build_modifier/modifier_task'
-require 'pi_build_modifier/config/logex'
+require 'pi_build_modifier/utils/logex'
 
 module PiBuildModifier
 
@@ -38,12 +36,11 @@ module PiBuildModifier
     desc 'modify CONFIG WORKSPACE', 'Modify the pi image sources at the specified WORKSPACE with the specified configuration file CONFIG.'
     def modify(config, workspace)
       $logger.debug "Modifying with a configuration read from '#{config}' in the workspace '#{workspace}'"
-      task = Task::Modifier.new(config, workspace)
-      task.execute
+      Modifiers::ModifiersBuilder::build_defaults(workspace, config).modify_configs
     end
 
     ##
-    # The version command allows users to query for the current version of the pi_customizer gem. It is printed on the command line.
+    # The version command allows users to query for the current version of the pi_build_modifier gem. It is printed on the command line.
 
     desc 'v, version', 'Show the version of the modifier.'
     def version

data/pi_build_modifier/lib/pi_build_modifier/modify/configs.rb

@@ -0,0 +1,335 @@
+# Copyright (c) 2017-2019 Beate Ottenwälder
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+
+##
+# configs.rb comprises all configurations of the pi-build sources to avoid a require hell
+
+require 'json'
+require 'openssl'
+require 'pi_build_modifier/utils/logex'
+
+module PiBuildModifier
+
+  ######################## Init ########################
+
+  ##
+  # Configuration changes of the pi-gen build sources
+
+  module Configs
+
+    def Configs.create_config_modifiers
+      [TypeConfig.new]
+    end
+
+    def Configs.create_erb_modifiers
+      [Locale.new, Boot.new, Wifi.new]
+    end
+
+    def Configs.create_config_file_modifiers
+      [SystemConfig.new, SshConfig.new]
+    end
+
+    ######################## Configuration Constants ########################
+
+    SYSTEM = 'system'
+    NAME = 'name'
+    ENABLED = 'enabled'
+    SSH = 'ssh'
+    PASSPHRASE = 'passphrase'
+    WPA_PASSPHRASE = 'wpa_passphrase'
+    SSID = 'ssid'
+    LOCALE = 'locale'
+    GEN = 'gen'
+    SYS = 'sys'
+    CGROUPS = 'cgroups'
+    CGROUP_MEMORY = 'memory'
+    TYPE = 'type'
+    USERNAME = 'username'
+    PASSWORD = 'password'
+
+    ######################## Config File Modifiers (ConfigFileModifier) ########################
+
+
+    ##
+    # Customization of the pi image's name
+
+    class SystemConfig
+
+      attr_reader :name, :username, :password
+
+      def initialize
+        @name = 'custompi'
+        @username = nil
+        @password = nil
+      end
+
+      def map(json_data)
+        unless json_data.nil?
+          @name = json_data[SYSTEM][NAME] if json_data.has_key?(SYSTEM) && json_data[SYSTEM].has_key?(NAME)
+          @username = json_data[SYSTEM][USERNAME] if json_data.has_key?(SYSTEM) && json_data[SYSTEM].has_key?(USERNAME)
+          @password = json_data[SYSTEM][PASSWORD] if json_data.has_key?(SYSTEM) && json_data[SYSTEM].has_key?(PASSWORD)
+        else
+          $logger.error "No json configuration found"
+        end
+      end
+
+      def verify
+        if (!@name.is_a? String) || @name.empty?
+          raise "No valid image name string provided in the config file! E.g., it might be an empty string or not a string at all. Configured image name: '#{@name}'."
+        end
+      end
+
+      def config_line
+        result = "IMG_NAME='#{@name}'"
+        unless @username.nil?
+          result = "#{result}\nFIRST_USER_NAME=#{@username}"
+        end
+        unless @password.nil?
+          result = "#{result}\nFIRST_USER_PASS=#{@password}"
+        end
+        result
+      end
+
+    end
+
+    ##
+    # Enables or Disables SSH by default in the built image
+
+    class SshConfig
+
+      attr_reader :enable
+
+      def initialize
+        @enable = true
+      end
+
+      def map(json_data)
+        unless json_data.nil?
+          @enable = json_data[SSH][ENABLED] if json_data.has_key?(SSH) && json_data[SSH].has_key?(ENABLED)
+        else
+          $logger.error 'Ssh could not be configured: Invalid json data.'
+        end
+      end
+
+      def verify
+        # no verify
+      end
+
+      def config_line
+        if @enable
+          "ENABLE_SSH=1"
+        end
+      end
+    end
+
+    ######################## ERB Modifiers (ERBConfigModifier) ########################
+
+    ##
+    # Wifi represents the wpa_supplicant's configuration
+
+    class Wifi
+
+      attr_reader :template_path, :relative_output_path
+
+      def initialize
+        @networks = map_network(nil)
+        @wpa_country = 'DE'
+        @template_path = File.join(File.dirname(__FILE__), '/templates/wpa_supplicant.conf.erb').to_s
+        @relative_output_path = 'stage2/02-net-tweaks/files/wpa_supplicant.conf'
+      end
+
+      ##
+      # map the 'wifi' section in the json configuration to the instance variables.
+
+      def map(json_data)
+        unless json_data.nil?
+          @networks = map_network(json_data['wifi']) if json_data.has_key?('wifi')
+          @wpa_country = json_data['wifi']['wpa_country'] if json_data.has_key?('wifi') && json_data['wifi'].has_key?('wpa_country')
+        else
+          $logger.warn("Could not map wifi config due to missing Json")
+        end
+      end
+
+      private def map_network(json_data)
+        if json_data.nil? or not json_data.has_key?('networks')
+          networks = Array.new(0)
+        else
+          networks = json_data['networks'].map do |rd|
+            ssid = if rd.has_key?(SSID)
+                     rd[SSID]
+                   else
+                     'no_ssid'
+                   end
+            psk = if rd.has_key?(PASSPHRASE)
+                    OpenSSL::PKCS5.pbkdf2_hmac_sha1(rd[PASSPHRASE], ssid, 4096, 32).unpack("H*").first
+                  elsif rd.has_key?(WPA_PASSPHRASE)
+                    rd[WPA_PASSPHRASE]
+                  else
+                    'no_psk'
+                  end
+            WifiNetwork.new(ssid, psk)
+          end
+        end
+        networks
+      end
+
+      def verify
+        # no verify
+      end
+
+      def get_binding
+        binding
+      end
+
+    end
+
+    ##
+    # Customization of the locales
+
+    class Locale
+
+      attr_accessor :gen_locales, :sys_locales
+
+      attr_reader :template_path, :relative_output_path
+
+      def initialize(gen_locales = ['en_GB.UTF-8 UTF-8'], sys_locale = 'en_GB.UTF-8')
+        @gen_locales = gen_locales
+        @sys_locale = sys_locale
+        @template_path = File.join(File.dirname(__FILE__), '/templates/00-debconf.erb').to_s
+        @relative_output_path = 'stage0/01-locale/00-debconf'
+      end
+
+      def map(json_data)
+        unless json_data.nil?
+          @gen_locales = json_data[LOCALE][GEN].map {|gen| gen} if json_data.has_key?(LOCALE) && json_data[LOCALE].has_key?(GEN)
+          @sys_locale = json_data[LOCALE][SYS] if json_data.has_key?(LOCALE) && json_data[LOCALE].has_key?(SYS)
+        end
+      end
+
+      def verify
+        # no verify
+      end
+
+      def get_binding
+        binding
+      end
+
+    end
+
+    ##
+    # Enable cgroups during boot
+
+    class Boot
+
+      attr_accessor :cgroups
+
+      attr_reader :template_path, :relative_output_path
+
+      def initialize(cgroups = [''])
+        @cgroups = cgroups
+        @template_path = File.join(File.dirname(__FILE__), '/templates/07-resize-init.diff.erb').to_s
+        @relative_output_path = 'stage2/01-sys-tweaks/00-patches/07-resize-init.diff'
+      end
+
+      def map(json_data)
+        unless json_data.nil? || !json_data.has_key?(CGROUPS)
+          if json_data[CGROUPS].has_key?(CGROUP_MEMORY) && json_data[CGROUPS][CGROUP_MEMORY]
+            @cgroups << 'cgroup_enable=memory cgroup_memory=1 swapaccount=1'
+          end
+        end
+      end
+
+      def verify
+        # no verify
+      end
+
+      def get_binding
+        binding
+      end
+
+    end
+
+    ######################## Config Modifiers (ConfigModifier) ########################
+
+    module Type
+      FULL = 'full'
+      LITE = 'lite'
+    end
+
+    class TypeConfig
+
+      def initialize
+        @type = Type::FULL
+      end
+
+      def map(json_data)
+        unless json_data.nil?
+          @type = json_data[SYSTEM][TYPE] if json_data.has_key?(SYSTEM) && json_data[SYSTEM].has_key?(TYPE)
+        else
+          $logger.error "No json configuration found"
+        end
+      end
+
+      def verify
+        unless (@type == Type::FULL) || (@type == Type::LITE)
+          raise "Invalid type '#{@type}' specified. Must be in (#{Type::LITE}, #{Type::FULL}.)"
+        end
+      end
+
+      def modify(workspace)
+        if @type.nil? || @type == Type::LITE
+          make_lite(workspace)
+        end
+      end
+
+      private def make_lite(workspace)
+        %W(#{workspace}/stage3/SKIP #{workspace}/stage4/SKIP #{workspace}/stage5/SKIP).each do |skip_file|
+          FileUtils.mkdir_p File.dirname(skip_file)
+          FileUtils.touch skip_file
+        end
+        %W(#{workspace}/stage4/EXPORT* #{workspace}/stage5/EXPORT*).each do |export_file|
+          FileUtils.rm_f export_file if File.exist?(export_file)
+        end
+      end
+    end
+
+    ######################## Helper ########################
+
+    ##
+    # WifiNetwork represents the network section of the wpa_supplicant's configuration
+
+    class WifiNetwork
+      attr_reader :ssid, :psk
+
+      def initialize(ssid, psk)
+        @ssid = ssid
+        @psk = psk
+      end
+
+      def to_s
+        sprintf('{
+  ssid="%s"
+  psk=%s
+}', ssid, psk)
+      end
+    end
+
+  end
+end