/*
* Copyright 2012-2014 eBay Software Foundation and selendroid committers.
*
* Licensed 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 io.selendroid.server.inspector.view;
import io.selendroid.server.ServerInstrumentation;
import io.selendroid.server.common.http.HttpRequest;
import io.selendroid.server.common.http.HttpResponse;
import io.selendroid.server.inspector.InspectorServlet;
import io.selendroid.server.inspector.SelendroidInspectorView;
import io.selendroid.server.model.SelendroidDriver;
import io.selendroid.server.util.SelendroidLogger;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
public class ResourceView extends SelendroidInspectorView {
public static final String SCREENSHOT = "deviceScreenshot.png";
public ResourceView(ServerInstrumentation serverInstrumentation, SelendroidDriver driver) {
super(serverInstrumentation, driver);
}
@Override
public void render(HttpRequest request, HttpResponse httpResponse) {
httpResponse.setStatus(200);
if (request.uri().startsWith(InspectorServlet.INSPECTOR_RESSOURCE + "/" + SCREENSHOT)) {
httpResponse.setContentType("image/png");
byte[] screenshot = driver.takeScreenshot();
if (screenshot == null) {
SelendroidLogger.info("screenshot is null");
} else {
httpResponse.setContent(screenshot);
}
} else {
try {
String filename =
"inspector" + request.uri().replaceFirst(InspectorServlet.INSPECTOR_RESSOURCE, "");
InputStream asset = serverInstrumentation.getInstrumentation().getContext().getAssets().open(filename);
httpResponse.setContent(toByteArray(asset));
} catch (IOException e) {
SelendroidLogger.error("Could not render ResourceView", e);
}
}
httpResponse.end();
}
private byte[] toByteArray(InputStream in) throws IOException {
ByteArrayOutputStream arrayOutputStream = new ByteArrayOutputStream();
int next = in.read();
while (next > -1) {
arrayOutputStream.write(next);
next = in.read();
}
arrayOutputStream.flush();
byte[] byteArray = arrayOutputStream.toByteArray();
arrayOutputStream.close();
return byteArray;
}
}