pi_build_modifier 0.1.1.pre.alpha

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: f1caec8f16b318afe28de0761d2ec6095114cd20
+  data.tar.gz: 2115fd2138d16d95f30e648d4a11f778cf8ed48d
+SHA512:
+  metadata.gz: 9af6f92c78bec16c33d337d9186a6d1277a1aa2a76489ffbe06f8eb6fe8fd6c5fc254ea09ff649e1dae10dc4b02ae88c8edb99921636597b3b47dae2d71618e9
+  data.tar.gz: 1fb2bf4dfa3fa21a2dac4974aa2057bc4142e428562b5fd52e2be422a9ec250fa0e1b91124e37f9068235366e273b791dfa130818a38fd9364d3326d5dc16840

data/.gitignore

@@ -0,0 +1,12 @@
+/.bundle/
+/.yardoc
+/Gemfile.lock
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/pi_build_modifier/spec/reports/
+/tmp/
+
+# rspec failure tracking
+.rspec_status

data/.rspec

@@ -0,0 +1,2 @@
+--format documentation
+--color

data/.travis.yml

@@ -0,0 +1,6 @@
+sudo: false
+language: ruby
+rvm:
+  - 2.4.2
+before_install:
+  - gem install bundler -v 1.15.3

data/Gemfile

@@ -0,0 +1,6 @@
+source 'https://rubygems.org'
+
+git_source(:github) {|repo_name| "https://github.com/#{repo_name}" }
+
+# Specify your gem's dependencies in pi_build_modifier.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2017 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.

data/README.md

@@ -0,0 +1,3 @@
+# PiBuildModifier
+
+see https://github.com/ottenwbe/pi-gen-environment

data/Rakefile

@@ -0,0 +1,13 @@
+lib = File.join(File.dirname(__FILE__), '/lib')
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+
+begin
+  require 'bundler/gem_tasks'
+  require 'rspec/core/rake_task'
+
+  RSpec::Core::RakeTask.new(:spec)
+
+  task :default => :spec
+rescue LoadError
+  puts 'RSpec or Bundler is not installed. This means rake is not able to execute tests!'
+end

data/bin/pi_build_modifier

@@ -0,0 +1,5 @@
+#!/usr/bin/env ruby
+
+require 'pi_build_modifier'
+
+PiBuildModifier::PiBuildModifier.start(ARGV)

data/lib/pi_build_modifier.rb

@@ -0,0 +1,41 @@
+# Copyright (c) 2017 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.
+
+require 'thor'
+require 'pi_build_modifier/net-tweaks/wifi_network'
+require 'pi_build_modifier/modifier/erb_mapper'
+require 'pi_build_modifier/version'
+require 'pi_build_modifier/modifier_task'
+
+module PiBuildModifier
+  class PiBuildModifier < Thor
+
+    desc 'modify CONFIG WORKSPACE', 'Modify the pi image sources at the specified WORKSPACE with the specified configuration file.'
+    def modify(config, workspace)
+      task = Task::Modifier.new(config, workspace)
+      task.execute
+    end
+
+    desc 'version', 'Show the version of the modifier.'
+    def version
+      puts VERSION
+    end
+  end
+end

data/lib/pi_build_modifier/modifier/erb_mapper.rb

@@ -0,0 +1,68 @@
+# Copyright (c) 2017 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.
+
+require 'json'
+require 'pathname'
+require 'fileutils'
+require 'pi_build_modifier/modifier/mapper'
+
+module PiBuildModifier
+  class ERBMapper < Mapper
+
+    def initialize(data, working_dir)
+      if data.nil?
+        raise 'Cannot initialize ERBMapper, since no valid object for mapping is given.'
+      end
+      @data = data
+      @template_path = @data.template_path
+      @output_path = working_dir + '/' + @data.relative_output_path
+    end
+
+    def map(json_data)
+      @data.map(json_data)
+    end
+
+    def modify
+      read_template
+      create_conf
+    end
+
+    private def read_template
+      File.open(@template_path.to_s, 'r+') do |f|
+        @template = f.read
+      end
+    end
+
+    def render
+      ERB.new(@template).result(@data.get_binding)
+    end
+
+    private def create_conf
+
+      puts "create dir at #{@output_path}"
+
+      FileUtils.mkdir_p File.dirname @output_path
+
+      File.open(@output_path, 'w+') do |f|
+        f.write(render)
+      end
+    end
+  end
+end

data/lib/pi_build_modifier/modifier/mapper.rb

@@ -0,0 +1,44 @@
+# Copyright (c) 2017 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.
+
+module PiBuildModifier
+
+  ##
+  # Mapper represents an abstract class that is used by the modifier to call the necessary steps in order to map values
+  # to an object and then modify it
+
+  class Mapper
+
+    attr_reader :data
+
+    def initialize(data)
+      @data = data
+    end
+
+    def map(json_data)
+      @data.map(json_data) unless @data.nil?
+    end
+
+    def modify
+      @data.map unless @data.nil?
+    end
+
+  end
+end

data/lib/pi_build_modifier/modifier/pi_modifier.rb

@@ -0,0 +1,74 @@
+# Copyright (c) 2017 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.
+
+require 'json'
+
+module PiBuildModifier
+  class PiModifier
+
+    attr_reader :mappers
+
+    def initialize
+      @mappers = Array.new
+      @json_path = nil
+      @json_data = nil
+    end
+
+    def with_json_configuration(json_file)
+      @json_path = json_file
+    end
+
+    def with_mapper(mapper)
+      if mapper.is_a? Mapper
+        @mappers << mapper
+      else
+        raise 'Parameter must be of type Mapper'
+      end
+    end
+
+    def modify
+      load_config
+      map
+      modify_mapper
+    end
+
+    private def load_config
+      if File.file?(@json_path)
+        File.open(@json_path, 'r+') do |f|
+          @json_data = JSON.parse(f.read)
+        end
+      else
+        raise "Json configuration file does not exist at '#{@json_path}!'"
+      end
+    end
+
+    private def map
+      @mappers.each do |c|
+        c.map(@json_data)
+      end
+    end
+
+    private def modify_mapper
+      @mappers.each do |c|
+        c.modify
+      end
+    end
+  end
+end

data/lib/pi_build_modifier/modifier_task.rb

@@ -0,0 +1,49 @@
+# Copyright (c) 2017 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.
+
+require 'pi_build_modifier/modifier/pi_modifier'
+require 'pi_build_modifier/modifier/erb_mapper'
+require 'pi_build_modifier/net-tweaks/wifi_network'
+require 'pi_build_modifier/system/system_type'
+require 'pi_build_modifier/sys_tweaks/run_modifier'
+require 'pi_build_modifier/sys_tweaks/ssh'
+
+module PiBuildModifier
+  module Task
+    class Modifier
+
+      attr_reader :pi_modifier
+
+      def initialize(config, workspace, pi_modifier = PiModifier.new)
+        @pi_modifier = pi_modifier
+        @pi_modifier.with_json_configuration(config)
+
+        @pi_modifier.with_mapper(ERBMapper.new(WPASupplicant.new, workspace))
+        @pi_modifier.with_mapper(System.new(workspace))
+        @pi_modifier.with_mapper(ERBMapper.new(Ssh.new, workspace))
+        @pi_modifier.with_mapper(RunModifier.new(workspace))
+      end
+
+      def execute
+        @pi_modifier.modify
+      end
+    end
+  end
+end

data/lib/pi_build_modifier/net-tweaks/templates/wpa_supplicant.conf.erb

@@ -0,0 +1,7 @@
+country=<%= @wpa_country %>
+ctrl_interface=DIR=/var/run/wpa_supplicant GROUP=netdev
+update_config=1
+
+<% @networks.each do |network|  %>
+    network=<%= network %>
+<% end %>

data/lib/pi_build_modifier/net-tweaks/wifi_network.rb

@@ -0,0 +1,68 @@
+# Copyright (c) 2017 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.
+
+require 'erb'
+require 'json'
+
+module PiBuildModifier
+  class WifiNetwork
+
+    attr_accessor :ssid, :psk, :template
+
+    attr_reader :file
+
+    def initialize(ssid = 'no_ssid', psk = 'no_psk')
+      @ssid = ssid
+      @psk = psk
+    end
+
+    def to_s
+      sprintf('{
+  ssid="%s"
+  psk="%s"
+}', ssid, psk)
+    end
+  end
+
+  class WPASupplicant
+
+    attr_reader :template_path, :relative_output_path
+
+    def initialize(networks = Array.new(0), wpa_country = 'DE')
+      @networks = networks
+      @wpa_country = wpa_country
+      @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
+
+    def map(json_data)
+      unless json_data.nil?
+        @networks = json_data['wifi']['networks'].map {|rd| WifiNetwork.new(rd['ssid'], rd['wpsk'])} if json_data.has_key?('wifi') && json_data['wifi'].has_key?('networks')
+        @wpa_country = json_data['wifi']['wpa_country'] if json_data.has_key?('wifi') && json_data['wifi'].has_key?('wpa_country')
+      end
+    end
+
+    def get_binding
+      binding
+    end
+
+  end
+end
+

data/lib/pi_build_modifier/sys_tweaks/run_modifier.rb

@@ -0,0 +1,58 @@
+# Copyright (c) 2017 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.
+
+require 'fileutils'
+require 'pi_build_modifier/modifier/mapper'
+
+module PiBuildModifier
+
+  class RunModifier < Mapper
+
+    attr_reader :relative_output_path
+
+    @@append_lines = Array.new
+    def self.append_lines
+      @@append_lines
+    end
+
+    def initialize(workspace)
+      @workspace = workspace
+      @relative_output_path = 'stage2/01-sys-tweaks/01-run.sh'
+    end
+
+    def self.append(append_line)
+      @@append_lines << append_line
+    end
+
+    def map(json_data)
+    end
+
+    def modify
+      puts @@append_lines.to_s
+      open("#{@workspace}/#{@relative_output_path}", 'a') do |f|
+        f.puts ''
+        @@append_lines.each do |a|
+          f.puts "source #{@workspace}/#{a}"
+        end
+      end
+    end
+
+  end
+end

data/lib/pi_build_modifier/sys_tweaks/ssh.rb

@@ -0,0 +1,53 @@
+# Copyright (c) 2017 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.
+
+require 'fileutils'
+require 'pi_build_modifier/modifier/mapper'
+require 'pi_build_modifier/sys_tweaks/run_modifier'
+
+module PiBuildModifier
+
+  class Ssh
+
+    attr_accessor :enable
+
+    attr_reader :template_path, :relative_output_path
+
+    def initialize
+      @enable = 'enable'
+      @template_path = File.join(File.dirname(__FILE__), '/templates/ssh.sh.erb').to_s
+      @relative_output_path = 'stage2/01-sys-tweaks/ssh.sh'
+      RunModifier.append("#{@relative_output_path}")
+    end
+
+    def map(json_data)
+      unless json_data.nil?
+        @enable = 'disable' if json_data.has_key?('ssh') && json_data['ssh'].has_key?('enabled') && (not json_data['ssh']['enabled'])
+      else
+        # TODO: log error
+      end
+    end
+
+    def get_binding
+      binding
+    end
+
+  end
+end

data/lib/pi_build_modifier/sys_tweaks/templates/ssh.sh.erb

@@ -0,0 +1,8 @@
+#!/bin/bash -e
+
+<% if @enable != 'disable' %>
+on_chroot << EOF
+systemctl <%= @enable %> ssh
+EOF
+
+<% end %>

data/lib/pi_build_modifier/system/system_type.rb

@@ -0,0 +1,80 @@
+# Copyright (c) 2017 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.
+
+require 'fileutils'
+require 'pi_build_modifier/modifier/mapper'
+
+module PiBuildModifier
+
+  module Type
+    FULL = 'full'
+    LITE = 'lite'
+  end
+
+  class System < Mapper
+
+    attr_reader :name, :type
+
+    def initialize(workspace)
+      @workspace = workspace
+      @name = 'chiipi'
+      @type = Type::FULL
+      @relative_output_path = @workspace
+    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')
+        @type = json_data['system']['type'] if json_data.has_key?('system') && json_data['system'].has_key?('type')
+      else
+        # TODO: log error
+      end
+    end
+
+    def modify
+      write_name
+      modify_type
+    end
+
+    private def modify_type
+      if @type.nil? || @type == Type::LITE
+        make_lite
+      end
+    end
+
+    private def write_name
+      FileUtils.mkdir_p "#{@workspace}"
+      File.open("#{@workspace}/config", 'w') do |file|
+        file.write("IMG_NAME='#{@name}'")
+      end
+    end
+
+    private def make_lite
+      directory = @workspace
+      %W(#{directory}/stage3/SKIP #{directory}/stage4/SKIP #{directory}/stage5/SKIP).each do |skip_file|
+        FileUtils.mkdir_p File.dirname(skip_file)
+        FileUtils.touch skip_file
+      end
+      %W(#{directory}/stage4/EXPORT* #{directory}/stage5/EXPORT*).each do |export_file|
+        FileUtils.rm_f export_file if File.exist?(export_file)
+      end
+    end
+  end
+end

data/lib/pi_build_modifier/version.rb

@@ -0,0 +1,23 @@
+# Copyright (c) 2017 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.
+
+module PiBuildModifier
+  VERSION = '0.1.1-alpha'
+end

data/pi_build_modifier.gemspec

@@ -0,0 +1,31 @@
+# coding: utf-8
+
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+
+require 'pi_build_modifier/version'
+
+Gem::Specification.new do |spec|
+  spec.name = 'pi_build_modifier'
+  spec.version = PiBuildModifier::VERSION
+  spec.license = 'MIT'
+  spec.author = 'Beate Ottenwälder'
+  spec.email = 'ottenwbe.public@gmail.com'
+  spec.homepage = 'https://github.com/ottenwbe/pi-gen-environment.git'
+  spec.summary = 'Modify Rapbian image builds.'
+  spec.description = 'Modifies the build process of a Rapbian image.'
+
+  spec.files = `git ls-files -z`.split("\x0").reject do |f|
+    f.match(%r{^(test|spec|features)/})
+  end
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.require_paths = ['lib']
+
+  spec.add_development_dependency 'bundler', '~> 1.15'
+  spec.add_development_dependency 'rake', '~> 10.0'
+  spec.add_development_dependency 'rspec', '~> 3.0'
+  spec.add_development_dependency 'simplecov', '~> 0'
+  spec.add_development_dependency 'rdoc', '~> 5.1'
+  spec.add_runtime_dependency 'thor', '~> 0.20'
+  spec.add_runtime_dependency 'json', '~> 2.1'
+end

metadata

@@ -0,0 +1,163 @@
+--- !ruby/object:Gem::Specification
+name: pi_build_modifier
+version: !ruby/object:Gem::Version
+  version: 0.1.1.pre.alpha
+platform: ruby
+authors:
+- Beate Ottenwälder
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2017-11-09 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.15'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.15'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+- !ruby/object:Gem::Dependency
+  name: simplecov
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rdoc
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.1'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.1'
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.20'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.20'
+- !ruby/object:Gem::Dependency
+  name: json
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.1'
+description: Modifies the build process of a Rapbian image.
+email: ottenwbe.public@gmail.com
+executables:
+- pi_build_modifier
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".rspec"
+- ".travis.yml"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/pi_build_modifier
+- lib/pi_build_modifier.rb
+- lib/pi_build_modifier/modifier/erb_mapper.rb
+- lib/pi_build_modifier/modifier/mapper.rb
+- lib/pi_build_modifier/modifier/pi_modifier.rb
+- lib/pi_build_modifier/modifier_task.rb
+- lib/pi_build_modifier/net-tweaks/templates/wpa_supplicant.conf.erb
+- lib/pi_build_modifier/net-tweaks/wifi_network.rb
+- lib/pi_build_modifier/sys_tweaks/run_modifier.rb
+- lib/pi_build_modifier/sys_tweaks/ssh.rb
+- lib/pi_build_modifier/sys_tweaks/templates/ssh.sh.erb
+- lib/pi_build_modifier/system/system_type.rb
+- lib/pi_build_modifier/version.rb
+- pi_build_modifier.gemspec
+homepage: https://github.com/ottenwbe/pi-gen-environment.git
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">"
+    - !ruby/object:Gem::Version
+      version: 1.3.1
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.6.13
+signing_key: 
+specification_version: 4
+summary: Modify Rapbian image builds.
+test_files: []