Search in sources :

Example 1 with UserDirectorySearchRequest

use of io.kamax.mxisd.controller.directory.v1.io.UserDirectorySearchRequest in project mxisd by kamax-io.

the class UserDirectoryController method search.

@RequestMapping(path = "/search", method = RequestMethod.POST)
public String search(HttpServletRequest request, @RequestParam("access_token") String accessToken) throws IOException {
    UserDirectorySearchRequest searchQuery = parser.parse(request, UserDirectorySearchRequest.class);
    URI target = URI.create(request.getRequestURL().toString());
    UserDirectorySearchResult result = mgr.search(target, accessToken, searchQuery.getSearchTerm());
    return gson.toJson(result);
}
Also used : UserDirectorySearchRequest(io.kamax.mxisd.controller.directory.v1.io.UserDirectorySearchRequest) URI(java.net.URI) UserDirectorySearchResult(io.kamax.mxisd.controller.directory.v1.io.UserDirectorySearchResult)

Example 2 with UserDirectorySearchRequest

use of io.kamax.mxisd.controller.directory.v1.io.UserDirectorySearchRequest in project mxisd by kamax-io.

the class RestDirectoryProvider method search.

private UserDirectorySearchResult search(String by, String query) {
    UserDirectorySearchRequest request = new UserDirectorySearchRequest(query);
    request.setBy(by);
    try (CloseableHttpResponse httpResponse = client.execute(RestClientUtils.post(cfg.getEndpoints().getDirectory(), request))) {
        int status = httpResponse.getStatusLine().getStatusCode();
        if (status < 200 || status >= 300) {
            throw new InternalServerError("REST backend: Error: " + IOUtils.toString(httpResponse.getEntity().getContent(), StandardCharsets.UTF_8));
        }
        UserDirectorySearchResult response = parser.parse(httpResponse, UserDirectorySearchResult.class);
        for (UserDirectorySearchResult.Result result : response.getResults()) {
            result.setUserId(new MatrixID(result.getUserId(), mxCfg.getDomain()).getId());
        }
        return response;
    } catch (IOException e) {
        throw new InternalServerError("REST backend: I/O error: " + e.getMessage());
    }
}
Also used : UserDirectorySearchRequest(io.kamax.mxisd.controller.directory.v1.io.UserDirectorySearchRequest) CloseableHttpResponse(org.apache.http.client.methods.CloseableHttpResponse) IOException(java.io.IOException) MatrixID(io.kamax.matrix.MatrixID) InternalServerError(io.kamax.mxisd.exception.InternalServerError) UserDirectorySearchResult(io.kamax.mxisd.controller.directory.v1.io.UserDirectorySearchResult)

Example 3 with UserDirectorySearchRequest

use of io.kamax.mxisd.controller.directory.v1.io.UserDirectorySearchRequest in project mxisd by kamax-io.

the class DirectoryManager method search.

public UserDirectorySearchResult search(URI target, String accessToken, String query) {
    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 = gson.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 MatrixException(status, info.getErrcode(), info.getError());
                }
            }
            UserDirectorySearchResult resultHs = gson.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 (IDirectoryProvider 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);
        }
        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.controller.directory.v1.io.UserDirectorySearchRequest) Charset(java.nio.charset.Charset) IOException(java.io.IOException) InternalServerError(io.kamax.mxisd.exception.InternalServerError) URIBuilder(org.apache.http.client.utils.URIBuilder) MatrixException(io.kamax.mxisd.exception.MatrixException) JsonSyntaxException(com.google.gson.JsonSyntaxException) CloseableHttpResponse(org.apache.http.client.methods.CloseableHttpResponse) MatrixErrorInfo(io.kamax.matrix.MatrixErrorInfo) UserDirectorySearchResult(io.kamax.mxisd.controller.directory.v1.io.UserDirectorySearchResult)

Aggregations

UserDirectorySearchRequest (io.kamax.mxisd.controller.directory.v1.io.UserDirectorySearchRequest)3 UserDirectorySearchResult (io.kamax.mxisd.controller.directory.v1.io.UserDirectorySearchResult)3 InternalServerError (io.kamax.mxisd.exception.InternalServerError)2 IOException (java.io.IOException)2 CloseableHttpResponse (org.apache.http.client.methods.CloseableHttpResponse)2 JsonSyntaxException (com.google.gson.JsonSyntaxException)1 MatrixErrorInfo (io.kamax.matrix.MatrixErrorInfo)1 MatrixID (io.kamax.matrix.MatrixID)1 MatrixException (io.kamax.mxisd.exception.MatrixException)1 URI (java.net.URI)1 Charset (java.nio.charset.Charset)1 HttpPost (org.apache.http.client.methods.HttpPost)1 URIBuilder (org.apache.http.client.utils.URIBuilder)1