1 files changed, 85 insertions, 1 deletions
diff --git a/libdino/src/service/registration.vala b/libdino/src/service/registration.vala
index 32d8b04b..8cac355e 100644
--- a/libdino/src/service/registration.vala
+++ b/libdino/src/service/registration.vala
@@ -5,12 +5,96 @@ using Dino.Entities;
 
 namespace Dino {
 
-public class Register {
+public class Register : StreamInteractionModule, Object{
+    public static ModuleIdentity<Register> IDENTITY = new ModuleIdentity<Register>("registration");
+    public string id { get { return IDENTITY.id; } }
+
+    private StreamInteractor stream_interactor;
+    private Database db;
+
+    public static void start(StreamInteractor stream_interactor, Database db) {
+        Register m = new Register(stream_interactor, db);
+        stream_interactor.add_module(m);
+    }
+
+    private Register(StreamInteractor stream_interactor, Database db) {
+        this.stream_interactor = stream_interactor;
+        this.db = db;
+    }
+
+    public async ConnectionManager.ConnectionError.Source? add_check_account(Account account) {
+        SourceFunc callback = add_check_account.callback;
+        ConnectionManager.ConnectionError.Source? ret = null;
+
+        ulong handler_id_connected = stream_interactor.stream_negotiated.connect((connected_account, stream) => {
+            if (connected_account.equals(account)) {
+                account.persist(db);
+                account.enabled = true;
+                Idle.add((owned)callback);
+            }
+        });
+        ulong handler_id_error = stream_interactor.connection_manager.connection_error.connect((connected_account, error) => {
+            if (connected_account.equals(account)) {
+                ret = error.source;
+            }
+            stream_interactor.disconnect_account(account);
+            Idle.add((owned)callback);
+        });
+
+        stream_interactor.connect_account(account);
+        yield;
+        stream_interactor.disconnect(handler_id_connected);
+        stream_interactor.connection_manager.disconnect(handler_id_error);
+
+        return ret;
+    }
+
+    public class ServerAvailabilityReturn {
+        public bool available { get; set; }
+        public TlsCertificateFlags? error_flags { get; set; }
+    }
+
+    public static async ServerAvailabilityReturn check_server_availability(Jid jid) {
+        XmppStream stream = new XmppStream();
+        stream.add_module(new Tls.Module());
+        stream.add_module(new Iq.Module());
+        stream.add_module(new Xep.SrvRecordsTls.Module());
+
+        ServerAvailabilityReturn ret = new ServerAvailabilityReturn() { available=false };
+        SourceFunc callback = check_server_availability.callback;
+        stream.stream_negotiated.connect(() => {
+            if (callback != null) {
+                ret.available = true;
+                Idle.add((owned)callback);
+            }
+        });
+        stream.get_module(Tls.Module.IDENTITY).invalid_certificate.connect((peer_cert, errors) => {
+            if (callback != null) {
+
+                ret.error_flags = errors;
+                Idle.add((owned)callback);
+            }
+        });
+        Timeout.add_seconds(5, () => {
+            if (callback != null) {
+                Idle.add((owned)callback);
+            }
+            return false;
+        });
+
+        stream.connect.begin(jid.domainpart);
+        yield;
+        try {
+            stream.disconnect();
+        } catch (Error e) {}
+        return ret;
+    }
 
     public static async Xep.InBandRegistration.Form get_registration_form(Jid jid) {
         XmppStream stream = new XmppStream();
         stream.add_module(new Tls.Module());
         stream.add_module(new Iq.Module());
+        stream.add_module(new Xep.SrvRecordsTls.Module());
         stream.add_module(new Xep.InBandRegistration.Module());
         stream.connect.begin(jid.bare_jid.to_string());