container.rb in kubec-0.3.0

- old
+ new

@@ -1,20 +1,39 @@
 module Kubec
   class Kubernetes
     # :nodoc:
     class Container < Hash
       include HasAttribute
-
       attribute :image
       attribute :name
       attribute :ports
 
       def initialize(name, &block)
         self[:name] = name
         instance_eval(&block)
       end
 
+      def env(key, value)
+        self[:env] ||= []
+        self[:env].push name: key,
+                        value: value.to_s
+      end
+
+      def mount(name, at:)
+        self[:volumeMounts] ||= []
+        self[:volumeMounts].push name: name,
+                                 mountPath: at
+      end
+
+      # TODO: Auto setup config map volume
+      def config_file(name, path:, from:)
+        self[:volumeMounts] ||= []
+        self[:volumeMounts].push name: from,
+                                 mountPath: [path, name].join('/'),
+                                 subPath: name
+      end
+
       # TODO: Add object to check fields
       def port(container_port, host_port = nil,
                ip: nil, name: nil, protocol: nil)
         self[:ports] ||= []
         port = {
@@ -23,9 +42,19 @@
           hostIP: ip,
           name: name,
           protocol: protocol
         }.compact
         self[:ports].push port
+      end
+
+      def command(*args)
+        args = args.flatten
+        self[:command] = args.take(1)
+        self[:args] = args.drop(1)
+      end
+
+      def args(*args)
+        self[:args] = args.flatten
       end
     end
   end
 end