content: // Licensed to the Apache Software Foundation (ASF) under one
  // or more contributor license agreements.  See the NOTICE file
  // distributed with this work for additional information
  // regarding copyright ownership.  The ASF licenses this file
  // to you under the Apache License, Version 2.0 (the
  // "License"); you may not use this file except in compliance
  // with the License.  You may obtain a copy of the License at
  //
  //   http://www.apache.org/licenses/LICENSE-2.0
  //
  // Unless required by applicable law or agreed to in writing,
  // software distributed under the License is distributed on an
  // "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  // KIND, either express or implied.  See the License for the
  // specific language governing permissions and limitations
  // under the License.
  package com.cloud.resource;
  
  import java.io.File;
  import java.io.PrintWriter;
  import java.io.StringWriter;
  import java.net.URL;
  import java.util.ArrayList;
  import java.util.Date;
  import java.util.LinkedList;
  import java.util.List;
  import java.util.Map;
  import java.util.concurrent.atomic.AtomicReference;
  
  import javax.inject.Inject;
  import javax.naming.ConfigurationException;
  
  import org.apache.cloudstack.managed.context.ManagedContextRunnable;
  import org.apache.logging.log4j.Logger;
  import org.apache.logging.log4j.LogManager;
  
  import com.cloud.agent.IAgentControl;
  import com.cloud.agent.api.Answer;
  import com.cloud.agent.api.Command;
  import com.cloud.agent.api.PingCommand;
  import com.cloud.agent.api.StartupCommand;
  import com.cloud.agent.manager.MockAgentManager;
  import com.cloud.agent.manager.MockStorageManager;
  import com.cloud.agent.manager.MockVmManager;
  import com.cloud.agent.manager.SimulatorManager;
  import com.cloud.agent.manager.SimulatorManager.AgentType;
  import com.cloud.agent.manager.SimulatorManagerImpl;
  import com.cloud.host.Host;
  import com.cloud.host.Host.Type;
  import com.cloud.simulator.MockHost;
  import com.cloud.utils.component.ComponentContext;
  
  public class AgentResourceBase implements ServerResource {
      protected Logger logger = LogManager.getLogger(getClass());
  
      protected String _name;
      private List<String> _warnings = new LinkedList<String>();
      private List<String> _errors = new LinkedList<String>();
  
      private transient IAgentControl _agentControl;
  
      protected long _instanceId;
  
      private Type _type;
  
      @Inject
      protected SimulatorManager _simMgr;
      protected MockHost agentHost = null;
      protected boolean stopped = false;
      protected String hostGuid = null;
  
      public AgentResourceBase(long instanceId, AgentType agentType, SimulatorManager simMgr, String hostGuid) {
          _instanceId = instanceId;
  
          if (logger.isDebugEnabled()) {
              logger.info("New Routing host instantiated with guid:" + hostGuid);
          }
  
          if (agentType == AgentType.Routing) {
              _type = Host.Type.Routing;
          } else {
              _type = Host.Type.Storage;
          }
  
          this.hostGuid = hostGuid;
      }
  
      protected MockVmManager getVmMgr() {
          return _simMgr.getVmMgr();
      }
  
      protected MockStorageManager getStorageMgr() {
          return _simMgr.getStorageMgr();
      }
  
      protected MockAgentManager getAgentMgr() {
          return _simMgr.getAgentMgr();
      }
  
      protected long getInstanceId() {
          return _instanceId;
      }
  
      public AgentResourceBase() {
          if (logger.isDebugEnabled()) {
              logger.debug("Deserializing simulated agent on reconnect");
          }
  
      }
  
      @Override
      public String getName() {
          return _name;
      }
  
      @Override
      public void setName(String name) {
          _name = name;
      }
  
      @Override
      public boolean configure(String name, Map<String, Object> params) throws ConfigurationException {
          hostGuid = (String)params.get("guid");
  
          _simMgr = ComponentContext.inject(SimulatorManagerImpl.class);
          _simMgr.configure(name, params);
  
          agentHost = getAgentMgr().getHost(hostGuid);
          return true;
      }
  
      private void reconnect(MockHost host) {
          if (logger.isDebugEnabled()) {
              logger.debug("Reconfiguring existing simulated host w/ name: " + host.getName() + " and guid: " + host.getGuid());
          }
          this.agentHost = host;
      }
  
      @Override
      public void disconnected() {
          this.stopped = true;
      }
  
      protected void recordWarning(String msg, Throwable th) {
          String str = getLogStr(msg, th);
          synchronized (_warnings) {
              _warnings.add(str);
          }
      }
  
      protected void recordWarning(String msg) {
          recordWarning(msg, null);
      }
  
      protected List<String> getWarnings() {
          synchronized (this) {
              List<String> results = _warnings;
              _warnings = new ArrayList<String>();
              return results;
          }
      }
  
      protected List<String> getErrors() {
          synchronized (this) {
              List<String> result = _errors;
              _errors = new ArrayList<String>();
              return result;
          }
      }
  
      protected void recordError(String msg, Throwable th) {
          String str = getLogStr(msg, th);
          synchronized (_errors) {
              _errors.add(str);
          }
      }
  
      protected void recordError(String msg) {
          recordError(msg, null);
      }
  
      protected Answer createErrorAnswer(Command cmd, String msg, Throwable th) {
          StringWriter writer = new StringWriter();
          if (msg != null) {
              writer.append(msg);
          }
          writer.append("===>Stack<===");
          th.printStackTrace(new PrintWriter(writer));
          return new Answer(cmd, false, writer.toString());
      }
  
      protected String createErrorDetail(String msg, Throwable th) {
          StringWriter writer = new StringWriter();
          if (msg != null) {
              writer.append(msg);
          }
          writer.append("===>Stack<===");
          th.printStackTrace(new PrintWriter(writer));
          return writer.toString();
      }
  
      protected String getLogStr(String msg, Throwable th) {
          StringWriter writer = new StringWriter();
          writer.append(new Date().toString()).append(": ").append(msg);
          if (th != null) {
              writer.append("\n  Exception: ");
              th.printStackTrace(new PrintWriter(writer));
          }
          return writer.toString();
      }
  
      @Override
      public boolean start() {
          return true;
      }
  
      @Override
      public boolean stop() {
          this.stopped = true;
          return true;
      }
  
      @Override
      public IAgentControl getAgentControl() {
          return _agentControl;
      }
  
      @Override
      public void setAgentControl(IAgentControl agentControl) {
          _agentControl = agentControl;
      }
  
      protected String findScript(String script) {
          logger.debug("Looking for " + script + " in the classpath");
          URL url = ClassLoader.getSystemResource(script);
          File file = null;
          if (url == null) {
              file = new File("./" + script);
              logger.debug("Looking for " + script + " in " + file.getAbsolutePath());
              if (!file.exists()) {
                  return null;
              }
          } else {
              file = new File(url.getFile());
          }
          return file.getAbsolutePath();
      }
  
      @Override
      public final Answer executeRequest(final Command cmd) {
          final AtomicReference<Answer> result = new AtomicReference<Answer>();
          new ManagedContextRunnable() {
              @Override
              protected void runInContext() {
                  result.set(executeRequestInContext(cmd));
              }
          }.run();
  
          return result.get();
      }
  
      public Answer executeRequestInContext(Command cmd) {
          return null;
      }
  
      @Override
      public PingCommand getCurrentStatus(long id) {
          return null;
      }
  
      @Override
      public Type getType() {
          return _type;
      }
  
      public void setType(Host.Type type) {
          this._type = type;
      }
  
      @Override
      public StartupCommand[] initialize() {
          return null;
      }
  
      public SimulatorManager getSimulatorManager() {
          return _simMgr;
      }
  
      public void setSimulatorManager(SimulatorManager simMgr) {
          _simMgr = simMgr;
      }
  
      public boolean isStopped() {
          return this.stopped;
      }
  
      @Override
      public void setConfigParams(Map<String, Object> params) {
          // TODO Auto-generated method stub
  
      }
  
      @Override
      public Map<String, Object> getConfigParams() {
          // TODO Auto-generated method stub
          return null;
      }
  
      @Override
      public int getRunLevel() {
          // TODO Auto-generated method stub
          return 0;
      }
  
      @Override
      public void setRunLevel(int level) {
          // TODO Auto-generated method stub
  
      }
  }
  