aboutsummaryrefslogtreecommitdiff
path: root/plugins/http-files/src/upload_stream_module.vala
diff options
context:
space:
mode:
authorfiaxh <git@mx.ax.lt>2017-08-02 17:29:55 +0200
committerfiaxh <git@mx.ax.lt>2017-08-03 15:59:04 +0200
commitea174ab632ced082eb0f1c51cea1bc9dc5c7c89e (patch)
tree8f83d1379a48e7aa49af5a5146d76a4c595f2363 /plugins/http-files/src/upload_stream_module.vala
parentf83e1188c5dcf1981832f296203f94503b467e30 (diff)
downloaddino-ea174ab632ced082eb0f1c51cea1bc9dc5c7c89e.tar.gz
dino-ea174ab632ced082eb0f1c51cea1bc9dc5c7c89e.zip
Http file upload
Diffstat (limited to 'plugins/http-files/src/upload_stream_module.vala')
-rw-r--r--plugins/http-files/src/upload_stream_module.vala159
1 files changed, 159 insertions, 0 deletions
diff --git a/plugins/http-files/src/upload_stream_module.vala b/plugins/http-files/src/upload_stream_module.vala
new file mode 100644
index 00000000..765d212e
--- /dev/null
+++ b/plugins/http-files/src/upload_stream_module.vala
@@ -0,0 +1,159 @@
+using Xmpp;
+using Xmpp.Core;
+using Xmpp.Xep;
+
+namespace Dino.Plugins.HttpFiles {
+
+private const string NS_URI = "urn:xmpp:http:upload";
+private const string NS_URI_0 = "urn:xmpp:http:upload:0";
+
+public class UploadStreamModule : XmppStreamModule {
+ public static Core.ModuleIdentity<UploadStreamModule> IDENTITY = new Core.ModuleIdentity<UploadStreamModule>(NS_URI, "0363_http_file_upload");
+
+ public signal void feature_available(XmppStream stream, int? max_file_size);
+
+ public delegate void OnUploadOk(XmppStream stream, string url_down);
+ public delegate void OnError(XmppStream stream, string error);
+ public void upload(XmppStream stream, string file_uri, owned OnUploadOk listener, owned OnError error_listener) {
+ File file = File.new_for_path(file_uri);
+ FileInfo file_info = file.query_info("*", FileQueryInfoFlags.NONE);
+ request_slot(stream, file.get_basename(), (int)file_info.get_size(), file_info.get_content_type(),
+ (stream, url_down, url_up) => {
+ uint8[] data;
+ FileUtils.get_data(file_uri, out data);
+
+ Soup.Message message = new Soup.Message("PUT", url_up);
+ message.set_request(file_info.get_content_type(), Soup.MemoryUse.COPY, data);
+ Soup.Session session = new Soup.Session();
+ session.send_async(message);
+
+ listener(stream, url_up);
+ },
+ error_listener);
+ }
+
+ private delegate void OnSlotOk(XmppStream stream, string url_get, string url_put);
+ private void request_slot(XmppStream stream, string filename, int file_size, string? content_type, owned OnSlotOk listener, owned OnError error_listener) {
+ Flag? flag = stream.get_flag(Flag.IDENTITY);
+ if (flag != null) return;
+
+ StanzaNode request_node;
+ if (flag.ns_ver == NS_URI_0) {
+ request_node = new StanzaNode.build("request", NS_URI_0).add_self_xmlns();
+ request_node.put_attribute("filename", filename).put_attribute("size", file_size.to_string());
+ if (content_type != null) request_node.put_attribute("content-type", content_type);
+ } else{
+ request_node = new StanzaNode.build("request", NS_URI).add_self_xmlns()
+ .put_node(new StanzaNode.build("filename", NS_URI).put_node(new StanzaNode.text(filename)))
+ .put_node(new StanzaNode.build("size", NS_URI).put_node(new StanzaNode.text(file_size.to_string())));
+ if (content_type != null) {
+ request_node.put_node(new StanzaNode.build("content-type", NS_URI).put_node(new StanzaNode.text(content_type)));
+ }
+ }
+ Iq.Stanza iq = new Iq.Stanza.get(request_node) { to=flag.file_store_jid };
+ stream.get_module(Iq.Module.IDENTITY).send_iq(stream, iq, (stream, iq) => {
+ if (iq.is_error()) {
+ error_listener(stream, "");
+ } else {
+ string? url_get = iq.stanza.get_deep_string_content(flag.ns_ver + ":slot", flag.ns_ver + ":get");
+ string? url_put = iq.stanza.get_deep_string_content(flag.ns_ver + ":slot", flag.ns_ver + ":put");
+ listener(stream, url_get, url_put);
+ }
+ });
+ }
+
+ public override void attach(XmppStream stream) {
+ Iq.Module.require(stream);
+ ServiceDiscovery.Module.require(stream);
+
+ query_availability(stream);
+ }
+
+ public override void detach(XmppStream stream) {
+ stream.get_module(Bind.Module.IDENTITY).bound_to_resource.disconnect(query_availability);
+ }
+
+ public static void require(XmppStream stream) {
+ if (stream.get_module(IDENTITY) == null) stream.add_module(new ChatMarkers.Module());
+ }
+
+ public override string get_ns() { return NS_URI; }
+ public override string get_id() { return IDENTITY.id; }
+
+ private void query_availability(XmppStream stream) {
+ stream.get_module(ServiceDiscovery.Module.IDENTITY).request_info(stream, stream.remote_name, (stream, info_result) => {
+ bool available = check_ns_in_info(stream, stream.remote_name, info_result);
+ if (!available) {
+ stream.get_module(ServiceDiscovery.Module.IDENTITY).request_items(stream, stream.remote_name, (stream, items_result) => {
+ foreach (Xep.ServiceDiscovery.Item item in items_result.items) {
+ if (item.name == "HTTP File Upload") {
+ stream.get_module(ServiceDiscovery.Module.IDENTITY).request_info(stream, item.jid, (stream, info_result) => {
+ check_ns_in_info(stream, item.jid, info_result);
+ });
+ break;
+ }
+ }
+ });
+ }
+ });
+ }
+
+ private bool check_ns_in_info(XmppStream stream, string jid, Xep.ServiceDiscovery.InfoResult info_result) {
+ bool ver_available = false;
+ bool ver_0_available = false;
+ foreach (string feature in info_result.features) {
+ if (feature == NS_URI_0) {
+ ver_0_available = true;
+ break;
+ } else if (feature == NS_URI) {
+ ver_available = true;
+ }
+ }
+
+ if (ver_available || ver_0_available) {
+ int? max_file_size = extract_max_file_size(info_result);
+ if (ver_0_available) {
+ stream.add_flag(new Flag(jid, NS_URI_0));
+ } else if (ver_available) {
+ stream.add_flag(new Flag(jid, NS_URI));
+ }
+
+ feature_available(stream, max_file_size);
+ return true;
+ }
+ return false;
+ }
+
+ private int? extract_max_file_size(Xep.ServiceDiscovery.InfoResult info_result) {
+ string? max_file_size_str = null;
+ StanzaNode x_node = info_result.iq.stanza.get_deep_subnode("http://jabber.org/protocol/disco#info:query", "jabber:x:data:x");
+ Gee.List<StanzaNode> field_nodes = x_node.get_subnodes("field", "jabber:x:data");
+ foreach (StanzaNode node in field_nodes) {
+ string? var_attr = node.get_attribute("var");
+ if (var_attr == "max-file-size") {
+ StanzaNode value_node = node.get_subnode("value", "jabber:x:data");
+ max_file_size_str = value_node.get_string_content();
+ break;
+ }
+ }
+ return max_file_size_str != null ? int.parse(max_file_size_str) : (int?) null;
+ }
+}
+
+public class Flag : XmppStreamFlag {
+ public static FlagIdentity<Flag> IDENTITY = new FlagIdentity<Flag>(NS_URI, "service_discovery");
+
+ public string file_store_jid;
+ public string ns_ver;
+ public int? max_file_size;
+
+ public Flag(string file_store_jid, string ns_ver) {
+ this.file_store_jid = file_store_jid;
+ this.ns_ver = ns_ver;
+ }
+
+ public override string get_ns() { return NS_URI; }
+ public override string get_id() { return IDENTITY.id; }
+}
+
+}