Blindly accept mappings so Riot is happy - Fix #23

2017-08-12 02:24:01 +02:00
parent 821a5b99ea
commit 5a683b41ef
10 changed files with 379 additions and 18 deletions
--- a/src/main/groovy/io/kamax/mxisd/mapping/MappingManager.java
+++ b/src/main/groovy/io/kamax/mxisd/mapping/MappingManager.java
@@ -0,0 +1,152 @@
+package io.kamax.mxisd.mapping;
+
+import io.kamax.mxisd.exception.BadRequestException;
+import io.kamax.mxisd.lookup.ThreePid;
+import org.apache.commons.lang.StringUtils;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.stereotype.Component;
+
+import java.time.Instant;
+import java.time.temporal.ChronoUnit;
+import java.util.*;
+
+@Component
+public class MappingManager {
+
+    private Logger log = LoggerFactory.getLogger(MappingManager.class);
+
+    private Map<String, Session> threePidLookups = new WeakHashMap<>();
+    private Map<String, Session> sessions = new HashMap<>();
+    private Timer cleaner;
+
+    MappingManager() {
+        cleaner = new Timer();
+        cleaner.schedule(new TimerTask() {
+            @Override
+            public void run() {
+                List<Session> sList = new ArrayList<>(sessions.values());
+                for (Session s : sList) {
+                    if (s.timestamp.plus(24, ChronoUnit.HOURS).isBefore(Instant.now())) { // TODO config timeout
+                        log.info("Session {} is obsolete, removing", s.sid);
+
+                        sessions.remove(s.sid);
+                        threePidLookups.remove(s.hash);
+                    }
+                }
+            }
+        }, 0, 10 * 1000);  // TODO config delay
+    }
+
+    public String create(MappingSession data) {
+        String sid;
+        do {
+            sid = Long.toString(System.currentTimeMillis());
+        } while (sessions.containsKey(sid));
+
+        String threePidHash = data.getMedium() + data.getValue();
+        Session session = threePidLookups.get(threePidHash);
+        if (session != null) {
+            sid = session.sid;
+        } else {
+            // TODO perform some kind of validation
+
+            session = new Session(sid, threePidHash, data);
+            sessions.put(sid, session);
+            threePidLookups.put(threePidHash, session);
+        }
+
+        log.info("Created new session {} to validate {} {}", sid, session.medium, session.address);
+        return sid;
+    }
+
+    public Optional<ThreePid> getValidated(String sid, String secret) {
+        Session s = sessions.get(sid);
+        if (s != null && StringUtils.equals(s.secret, secret)) {
+            return Optional.of(new ThreePid(s.medium, s.address, s.validationTimestamp));
+        }
+
+        return Optional.empty();
+    }
+
+    public void bind(String sid, String secret, String mxid) {
+        Session s = sessions.get(sid);
+        if (s == null || !StringUtils.equals(s.secret, secret)) {
+            throw new BadRequestException("sid or secret are not valid");
+        }
+
+        log.info("Performed bind for mxid {}", mxid);
+        // TODO perform bind, whatever it is
+    }
+
+    private class Session {
+
+        private String sid;
+        private String hash;
+        private Instant timestamp;
+        private Instant validationTimestamp;
+        private boolean isValidated;
+        private String secret;
+        private String medium;
+        private String address;
+
+        public Session(String sid, String hash, MappingSession data) {
+            this.sid = sid;
+            this.hash = hash;
+            timestamp = Instant.now();
+            validationTimestamp = Instant.now();
+            secret = data.getSecret();
+            medium = data.getMedium();
+            address = data.getValue();
+        }
+
+        public Instant getTimestamp() {
+            return timestamp;
+        }
+
+        public void setTimestamp(Instant timestamp) {
+            this.timestamp = timestamp;
+        }
+
+        public Instant getValidationTimestamp() {
+            return validationTimestamp;
+        }
+
+        public void setValidationTimestamp(Instant validationTimestamp) {
+            this.validationTimestamp = validationTimestamp;
+        }
+
+        public boolean isValidated() {
+            return isValidated;
+        }
+
+        public void setValidated(boolean validated) {
+            isValidated = validated;
+        }
+
+        public String getSecret() {
+            return secret;
+        }
+
+        public void setSecret(String secret) {
+            this.secret = secret;
+        }
+
+        public String getMedium() {
+            return medium;
+        }
+
+        public void setMedium(String medium) {
+            this.medium = medium;
+        }
+
+        public String getAddress() {
+            return address;
+        }
+
+        public void setAddress(String address) {
+            this.address = address;
+        }
+    }
+
+}