remove version directory from config file

Author: feverLu

conf/codedeployagent.yml

@@ -5,6 +5,5 @@
 :program_name: codedeploy-agent
 :root_dir: '/opt/codedeploy-agent/deployment-root'
 :verbose: false
-:version_dir: '/opt/codedeploy-agent/'
 :wait_between_runs: 1
 :proxy_uri:

vendor/gems/codedeploy-commands-1.0.0/lib/aws/plugins/deploy_agent_version.rb

@@ -4,19 +4,28 @@ class DeployAgentVersion < Seahorse::Client::Plugin
       class Handler < Seahorse::Client::Handler
         def initialize(handler = nil)
           @handler = handler
-          file_path = File.join(ProcessManager::Config.config[:version_dir], '.version')
+          file_path = get_version_tracking_file
           if File.exist?(file_path)
             @agent_version ||= File.read(file_path).split(': ').last.strip
           else 
             @agent_version ||= "UNKNOWN_VERSION"
-            log(:warn, "Version tracking file doesn't exist in directory #{file_path}")
+            log(:warn, "Version tracking file either does not exist or cannot be read.")
           end
         end
 
         def call(context)
           context.http_request.headers['x-amz-codedeploy-agent-version'] = @agent_version
           @handler.call(context)
         end
+
+        def get_version_tracking_file
+          version_dir = ProcessManager::Config.config[:root_dir]
+          if version_dir.eql? 'Amazon/CodeDeploy'
+            file_path = File.join(version_dir, '.version')
+          else 
+            file_path = File.join(version_dir, '..', '.version')
+          end
+        end
 
         private
         def log(severity, message)