summaryrefslogtreecommitdiffstats
blob: c0edca226a24ad06d874d317731fe595ffe7d031 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
/*
 * Copyright (c) 2010-2015, Isode Limited, London, England.
 * All rights reserved.
 */
package com.isode.stroke.disco;

import java.util.HashMap;
import java.util.Map;

import com.isode.stroke.elements.DiscoInfo;
import com.isode.stroke.elements.ErrorPayload;
import com.isode.stroke.jid.JID;
import com.isode.stroke.queries.GetResponder;
import com.isode.stroke.queries.IQRouter;

public class DiscoInfoResponder extends GetResponder<DiscoInfo> {
    private DiscoInfo info_;
    private Map<String, DiscoInfo> nodeInfo_ = new HashMap<String, DiscoInfo>();

    public DiscoInfoResponder(IQRouter router) {
        super(new DiscoInfo(), router);
    }

    void clearDiscoInfo() {
        info_ = new DiscoInfo();
        nodeInfo_.clear();
    }

    void setDiscoInfo(final DiscoInfo info) {
        info_ = info;
    }

    void setDiscoInfo(final String node, final DiscoInfo info) {
        DiscoInfo newInfo = info;
        newInfo.setNode(node);
        nodeInfo_.put(node, newInfo);
    }

    protected boolean handleGetRequest(final JID from, final JID j, final String id, DiscoInfo info) {
        if (info.getNode().isEmpty()) {
            sendResponse(from, id, info_);
        }
        else {
            DiscoInfo i = nodeInfo_.get(info.getNode());
            if (i != null) {
                sendResponse(from, id, i);
            }
            else {
                sendError(from, id, ErrorPayload.Condition.ItemNotFound, ErrorPayload.Type.Cancel);
            }
        }
        return true;
    }
}