Java tutorial
/* * Copyright (c) 2014, WSO2 Inc. (http://www.wso2.org) All Rights Reserved. * * 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 org.wso2.carbon.device.mgt.iot.firealarm.api; import org.apache.commons.httpclient.HttpStatus; import org.apache.commons.logging.Log; import org.apache.commons.logging.LogFactory; import org.wso2.carbon.device.mgt.common.Device; import org.wso2.carbon.device.mgt.common.DeviceIdentifier; import org.wso2.carbon.device.mgt.common.DeviceManagementException; import org.wso2.carbon.device.mgt.common.EnrolmentInfo; import org.wso2.carbon.device.mgt.iot.common.DeviceManagement; import org.wso2.carbon.device.mgt.iot.common.apimgt.AccessTokenInfo; import org.wso2.carbon.device.mgt.iot.common.apimgt.TokenClient; import org.wso2.carbon.device.mgt.iot.common.controlqueue.xmpp.XmppAccount; import org.wso2.carbon.device.mgt.iot.common.controlqueue.xmpp.XmppConfig; import org.wso2.carbon.device.mgt.iot.common.controlqueue.xmpp.XmppServerClient; import org.wso2.carbon.device.mgt.iot.common.exception.AccessTokenException; import org.wso2.carbon.device.mgt.iot.common.exception.DeviceControllerException; import org.wso2.carbon.device.mgt.iot.common.util.ZipArchive; import org.wso2.carbon.device.mgt.iot.common.util.ZipUtil; import org.wso2.carbon.device.mgt.iot.firealarm.constants.FireAlarmConstants; import javax.servlet.http.HttpServletResponse; import javax.ws.rs.Consumes; import javax.ws.rs.DELETE; import javax.ws.rs.GET; import javax.ws.rs.POST; import javax.ws.rs.PUT; import javax.ws.rs.Path; import javax.ws.rs.PathParam; import javax.ws.rs.Produces; import javax.ws.rs.QueryParam; import javax.ws.rs.core.Context; import javax.ws.rs.core.Response; import java.io.File; import java.nio.ByteBuffer; import java.nio.charset.StandardCharsets; import java.util.ArrayList; import java.util.Date; import java.util.List; import java.util.UUID; //@Path("/FireAlarmDeviceManager") public class FireAlarmManagerService { private static Log log = LogFactory.getLog(FireAlarmManagerService.class); @Path("/device/register") @PUT public boolean register(@QueryParam("deviceId") String deviceId, @QueryParam("name") String name, @QueryParam("owner") String owner) { DeviceManagement deviceManagement = new DeviceManagement(); DeviceIdentifier deviceIdentifier = new DeviceIdentifier(); deviceIdentifier.setId(deviceId); deviceIdentifier.setType(FireAlarmConstants.DEVICE_TYPE); try { if (deviceManagement.getDeviceManagementService().isEnrolled(deviceIdentifier)) { Response.status(HttpStatus.SC_CONFLICT).build(); return false; } Device device = new Device(); device.setDeviceIdentifier(deviceId); EnrolmentInfo enrolmentInfo = new EnrolmentInfo(); enrolmentInfo.setDateOfEnrolment(new Date().getTime()); enrolmentInfo.setDateOfLastUpdate(new Date().getTime()); enrolmentInfo.setStatus(EnrolmentInfo.Status.ACTIVE); enrolmentInfo.setOwnership(EnrolmentInfo.OwnerShip.BYOD); device.setName(name); device.setType(FireAlarmConstants.DEVICE_TYPE); enrolmentInfo.setOwner(owner); device.setEnrolmentInfo(enrolmentInfo); boolean added = deviceManagement.getDeviceManagementService().enrollDevice(device); if (added) { Response.status(HttpStatus.SC_OK).build(); } else { Response.status(HttpStatus.SC_EXPECTATION_FAILED).build(); } return added; } catch (DeviceManagementException e) { log.error(e.getErrorMessage()); return false; } } @Path("/device/remove/{device_id}") @DELETE public void removeDevice(@PathParam("device_id") String deviceId, @Context HttpServletResponse response) { DeviceManagement deviceManagement = new DeviceManagement(); DeviceIdentifier deviceIdentifier = new DeviceIdentifier(); deviceIdentifier.setId(deviceId); deviceIdentifier.setType(FireAlarmConstants.DEVICE_TYPE); try { boolean removed = deviceManagement.getDeviceManagementService().disenrollDevice(deviceIdentifier); if (removed) { response.setStatus(HttpStatus.SC_OK); } else { response.setStatus(HttpStatus.SC_EXPECTATION_FAILED); } } catch (DeviceManagementException e) { log.error(e.getErrorMessage()); } } @Path("/device/update/{device_id}") @POST public boolean updateDevice(@PathParam("device_id") String deviceId, @QueryParam("name") String name, @Context HttpServletResponse response) { DeviceManagement deviceManagement = new DeviceManagement(); DeviceIdentifier deviceIdentifier = new DeviceIdentifier(); deviceIdentifier.setId(deviceId); deviceIdentifier.setType(FireAlarmConstants.DEVICE_TYPE); try { Device device = deviceManagement.getDeviceManagementService().getDevice(deviceIdentifier); device.setDeviceIdentifier(deviceId); // device.setDeviceTypeId(deviceTypeId); device.getEnrolmentInfo().setDateOfLastUpdate(new Date().getTime()); device.setName(name); device.setType(FireAlarmConstants.DEVICE_TYPE); boolean updated = deviceManagement.getDeviceManagementService().modifyEnrollment(device); if (updated) { response.setStatus(HttpStatus.SC_OK); } else { response.setStatus(HttpStatus.SC_EXPECTATION_FAILED); } return updated; } catch (DeviceManagementException e) { log.error(e.getErrorMessage()); return false; } } @Path("/device/{device_id}") @GET @Consumes("application/json") @Produces("application/json") public Device getDevice(@PathParam("device_id") String deviceId) { DeviceManagement deviceManagement = new DeviceManagement(); DeviceIdentifier deviceIdentifier = new DeviceIdentifier(); deviceIdentifier.setId(deviceId); deviceIdentifier.setType(FireAlarmConstants.DEVICE_TYPE); try { return deviceManagement.getDeviceManagementService().getDevice(deviceIdentifier); } catch (DeviceManagementException ex) { log.error("Error occurred while retrieving device with Id " + deviceId + "\n" + ex); return null; } } @Path("/devices/{username}") @GET @Consumes("application/json") @Produces("application/json") public Device[] getFirealarmDevices(@PathParam("username") String username) { DeviceManagement deviceManagement = new DeviceManagement(); try { List<Device> userDevices = deviceManagement.getDeviceManagementService().getDevicesOfUser(username); ArrayList<Device> userDevicesforFirealarm = new ArrayList<Device>(); for (Device device : userDevices) { if (device.getType().equals(FireAlarmConstants.DEVICE_TYPE) && device.getEnrolmentInfo().getStatus().equals(EnrolmentInfo.Status.ACTIVE)) { userDevicesforFirealarm.add(device); } } return userDevicesforFirealarm.toArray(new Device[] {}); } catch (DeviceManagementException ex) { log.error("Error occurred while retrieving devices for " + username); return null; } } @Path("/device/{sketch_type}/download") @GET @Produces("application/octet-stream") public Response downloadSketch(@QueryParam("owner") String owner, @PathParam("sketch_type") String sketchType) { ZipArchive zipFile = null; try { zipFile = createDownloadFile(owner, sketchType); Response.ResponseBuilder rb = Response.ok(zipFile.getZipFile()); rb.header("Content-Disposition", "attachment; filename=\"" + zipFile.getFileName() + "\""); return rb.build(); } catch (IllegalArgumentException ex) { return Response.status(400).entity(ex.getMessage()).build();//bad request } catch (DeviceManagementException ex) { return Response.status(500).entity(ex.getMessage()).build(); } catch (AccessTokenException ex) { return Response.status(500).entity(ex.getMessage()).build(); } catch (DeviceControllerException ex) { return Response.status(500).entity(ex.getMessage()).build(); } } @Path("/device/{sketch_type}/generate_link") @GET public Response generateSketchLink(@QueryParam("owner") String owner, @PathParam("sketch_type") String sketchType) { ZipArchive zipFile = null; try { zipFile = createDownloadFile(owner, sketchType); Response.ResponseBuilder rb = Response.ok(zipFile.getDeviceId()); return rb.build(); } catch (IllegalArgumentException ex) { return Response.status(400).entity(ex.getMessage()).build();//bad request } catch (DeviceManagementException ex) { return Response.status(500).entity(ex.getMessage()).build(); } catch (AccessTokenException ex) { return Response.status(500).entity(ex.getMessage()).build(); } catch (DeviceControllerException ex) { return Response.status(500).entity(ex.getMessage()).build(); } } private ZipArchive createDownloadFile(String owner, String sketchType) throws DeviceManagementException, AccessTokenException, DeviceControllerException { if (owner == null) { throw new IllegalArgumentException("Error on createDownloadFile() Owner is null!"); } //create new device id String deviceId = shortUUID(); TokenClient accessTokenClient = new TokenClient(FireAlarmConstants.DEVICE_TYPE); AccessTokenInfo accessTokenInfo = null; accessTokenInfo = accessTokenClient.getAccessToken(owner, deviceId); //create token String accessToken = accessTokenInfo.getAccess_token(); String refreshToken = accessTokenInfo.getRefresh_token(); //adding registering data XmppAccount newXmppAccount = new XmppAccount(); newXmppAccount.setAccountName(owner + "_" + deviceId); newXmppAccount.setUsername(deviceId); newXmppAccount.setPassword(accessToken); String xmppEndPoint = XmppConfig.getInstance().getXmppControlQueue().getServerURL(); int indexOfChar = xmppEndPoint.lastIndexOf('/'); if (indexOfChar != -1) { xmppEndPoint = xmppEndPoint.substring((indexOfChar + 1), xmppEndPoint.length()); } newXmppAccount.setEmail(deviceId + "@wso2.com"); XmppServerClient xmppServerClient = new XmppServerClient(); xmppServerClient.initControlQueue(); boolean status; if (XmppConfig.getInstance().isEnabled()) { status = xmppServerClient.createXMPPAccount(newXmppAccount); if (!status) { String msg = "XMPP Account was not created for device - " + deviceId + " of owner - " + owner + ". XMPP might have been disabled in configs"; log.warn(msg); throw new DeviceManagementException(msg); } } status = register(deviceId, owner + "s_" + sketchType + "_" + deviceId.substring(0, 3), owner); if (!status) { String msg = "Error occurred while registering the device with " + "id: " + deviceId + " owner:" + owner; throw new DeviceManagementException(msg); } ZipUtil ziputil = new ZipUtil(); ZipArchive zipFile = null; zipFile = ziputil.downloadSketch(owner, sketchType, deviceId, accessToken, refreshToken); zipFile.setDeviceId(deviceId); return zipFile; } private static String shortUUID() { UUID uuid = UUID.randomUUID(); long l = ByteBuffer.wrap(uuid.toString().getBytes(StandardCharsets.UTF_8)).getLong(); return Long.toString(l, Character.MAX_RADIX); } }