Search in sources :

Example 1 with MatrixErrorInfo

use of io.kamax.matrix.MatrixErrorInfo in project mxisd by kamax-io.

the class DirectoryManager method search.

public UserDirectorySearchResult search(URI target, String accessToken, String query) {
    if (StringUtils.startsWith(query, "@")) {
        query = query.substring(1);
    }
    log.info("Performing search for '{}'", query);
    log.info("Original request URL: {}", target);
    UserDirectorySearchResult result = new UserDirectorySearchResult();
    if (cfg.getExclude().getHomeserver()) {
        log.info("Skipping HS directory data, disabled in config");
    } else {
        URIBuilder builder = dns.transform(target);
        log.info("Querying HS at {}", builder);
        builder.setParameter("access_token", accessToken);
        HttpPost req = RestClientUtils.post(builder.toString(), new UserDirectorySearchRequest(query));
        try (CloseableHttpResponse res = client.execute(req)) {
            int status = res.getStatusLine().getStatusCode();
            Charset charset = ContentType.getOrDefault(res.getEntity()).getCharset();
            String body = IOUtils.toString(res.getEntity().getContent(), charset);
            if (status != 200) {
                MatrixErrorInfo info = GsonUtil.get().fromJson(body, MatrixErrorInfo.class);
                if (StringUtils.equals("M_UNRECOGNIZED", info.getErrcode())) {
                    // FIXME no hardcoding, use Enum
                    log.warn("Homeserver does not support Directory feature, skipping");
                } else {
                    log.error("Homeserver returned an error while performing directory search");
                    throw new HttpMatrixException(status, info.getErrcode(), info.getError());
                }
            }
            UserDirectorySearchResult resultHs = GsonUtil.get().fromJson(body, UserDirectorySearchResult.class);
            log.info("Found {} match(es) in HS for '{}'", resultHs.getResults().size(), query);
            result.getResults().addAll(resultHs.getResults());
            if (resultHs.isLimited()) {
                result.setLimited(true);
            }
        } catch (JsonSyntaxException e) {
            throw new InternalServerError("Invalid JSON reply from the HS: " + e.getMessage());
        } catch (IOException e) {
            throw new InternalServerError("Unable to query the HS: I/O error: " + e.getMessage());
        }
    }
    for (DirectoryProvider provider : providers) {
        log.info("Using Directory provider {}", provider.getClass().getSimpleName());
        UserDirectorySearchResult resultProvider = provider.searchByDisplayName(query);
        log.info("Display name: found {} match(es) for '{}'", resultProvider.getResults().size(), query);
        result.getResults().addAll(resultProvider.getResults());
        if (resultProvider.isLimited()) {
            result.setLimited(true);
        }
        if (cfg.getExclude().getThreepid()) {
            log.info("Skipping 3PID data, disabled in config");
        } else {
            resultProvider = provider.searchBy3pid(query);
            log.info("Threepid: found {} match(es) for '{}'", resultProvider.getResults().size(), query);
            result.getResults().addAll(resultProvider.getResults());
            if (resultProvider.isLimited()) {
                result.setLimited(true);
            }
        }
    }
    log.info("Total matches: {} - limited? {}", result.getResults().size(), result.isLimited());
    return result;
}
Also used : HttpPost(org.apache.http.client.methods.HttpPost) UserDirectorySearchRequest(io.kamax.mxisd.http.io.UserDirectorySearchRequest) HttpMatrixException(io.kamax.mxisd.exception.HttpMatrixException) Charset(java.nio.charset.Charset) IOException(java.io.IOException) InternalServerError(io.kamax.mxisd.exception.InternalServerError) URIBuilder(org.apache.http.client.utils.URIBuilder) JsonSyntaxException(com.google.gson.JsonSyntaxException) CloseableHttpResponse(org.apache.http.client.methods.CloseableHttpResponse) MatrixErrorInfo(io.kamax.matrix.MatrixErrorInfo) UserDirectorySearchResult(io.kamax.mxisd.http.io.UserDirectorySearchResult)

Aggregations

JsonSyntaxException (com.google.gson.JsonSyntaxException)1 MatrixErrorInfo (io.kamax.matrix.MatrixErrorInfo)1 HttpMatrixException (io.kamax.mxisd.exception.HttpMatrixException)1 InternalServerError (io.kamax.mxisd.exception.InternalServerError)1 UserDirectorySearchRequest (io.kamax.mxisd.http.io.UserDirectorySearchRequest)1 UserDirectorySearchResult (io.kamax.mxisd.http.io.UserDirectorySearchResult)1 IOException (java.io.IOException)1 Charset (java.nio.charset.Charset)1 CloseableHttpResponse (org.apache.http.client.methods.CloseableHttpResponse)1 HttpPost (org.apache.http.client.methods.HttpPost)1 URIBuilder (org.apache.http.client.utils.URIBuilder)1