import com.sun.net.httpserver.*;
import javax.crypto.*;
import javax.crypto.spec.*;
import java.security.*;
import java.util.Base64;
import java.io.*;
import java.net.InetSocketAddress;
import java.nio.charset.StandardCharsets;
import java.time.LocalDateTime;
import org.json.JSONObject;
public class Server {
// WARNING: Move to configuration file or environment variables
private static final String ENCRYPTION_KEY = "saccobot-encryption-key";
private static final String ALGORITHM = "AES/CBC/PKCS5Padding";
public static void main(String[] args) throws IOException {
// Create HTTP server on port 3000
HttpServer server = HttpServer.create(new InetSocketAddress(3000), 0);
server.createContext("/api/v1/sms/send", new SMSHandler());
server.start();
}
static class SMSHandler implements HttpHandler {
@Override
public void handle(HttpExchange exchange) throws IOException {
if (!"POST".equals(exchange.getRequestMethod())) {
sendResponse(exchange, 405, "{\"status\":\"error\",\"message\":\"Method not allowed\"}");
return;
}
try {
String body = new String(exchange.getRequestBody().readAllBytes(), StandardCharsets.UTF_8);
JSONObject json = new JSONObject(body);
if (!json.has("message")) {
sendResponse(exchange, 400, "{\"status\":\"error\",\"message\":\"No message provided\"}");
return;
}
String decrypted = decrypt(json.getString("message"));
JSONObject logEntry = new JSONObject();
logEntry.put("phoneNumber", json.optString("phoneNumber", ""));
logEntry.put("message", decrypted);
logEntry.put("timestamp", LocalDateTime.now().toString());
System.out.println("Received: " + logEntry);
sendResponse(exchange, 200, "{\"status\":\"success\",\"message\":\"Data logged\"}");
} catch (Exception e) {
sendResponse(exchange, 400, "{\"status\":\"error\",\"message\":\"Decryption failed\"}");
}
}
private String decrypt(String encryptedData) throws Exception {
byte[] combined = Base64.getDecoder().decode(encryptedData);
byte[] iv = new byte[16];
byte[] encrypted = new byte[combined.length - 16];
System.arraycopy(combined, 0, iv, 0, 16);
System.arraycopy(combined, 16, encrypted, 0, encrypted.length);
MessageDigest digest = MessageDigest.getInstance("SHA-256");
byte[] key = digest.digest(ENCRYPTION_KEY.getBytes(StandardCharsets.UTF_8));
Cipher cipher = Cipher.getInstance(ALGORITHM);
cipher.init(Cipher.DECRYPT_MODE, new SecretKeySpec(key, "AES"), new IvParameterSpec(iv));
return new String(cipher.doFinal(encrypted), StandardCharsets.UTF_8);
}
private void sendResponse(HttpExchange exchange, int code, String response) throws IOException {
exchange.getResponseHeaders().set("Content-Type", "application/json");
exchange.sendResponseHeaders(code, response.getBytes().length);
exchange.getResponseBody().write(response.getBytes());
exchange.getResponseBody().close();
}
}
}
