Search in sources :

Example 1 with UserDirectorySearchRequest

use of io.kamax.mxisd.http.io.UserDirectorySearchRequest in project mxisd by kamax-io.

the class ExecDirectoryStore method search.

private UserDirectorySearchResult search(ExecConfig.Process cfg, UserDirectorySearchRequest request) {
    if (StringUtils.isEmpty(cfg.getCommand())) {
        return UserDirectorySearchResult.empty();
    }
    Processor<UserDirectorySearchResult> p = new Processor<>(cfg);
    p.addJsonInputTemplate(tokens -> new UserDirectorySearchRequest(tokens.getType(), tokens.getQuery()));
    p.addInputTemplate(PlainType, tokens -> tokens.getType() + System.lineSeparator() + tokens.getQuery());
    p.addTokenMapper(cfg.getToken().getType(), request::getBy);
    p.addTokenMapper(cfg.getToken().getQuery(), request::getSearchTerm);
    p.addSuccessMapper(JsonType, output -> {
        if (StringUtils.isBlank(output)) {
            return UserDirectorySearchResult.empty();
        }
        UserDirectorySearchResult response = GsonUtil.get().fromJson(output, UserDirectorySearchResult.class);
        for (UserDirectorySearchResult.Result result : response.getResults()) {
            result.setUserId(MatrixID.asAcceptable(result.getUserId(), mxCfg.getDomain()).getId());
        }
        return response;
    });
    p.withFailureDefault(output -> new UserDirectorySearchResult());
    return p.execute();
}
Also used : UserDirectorySearchRequest(io.kamax.mxisd.http.io.UserDirectorySearchRequest) UserDirectorySearchResult(io.kamax.mxisd.http.io.UserDirectorySearchResult)

Example 2 with UserDirectorySearchRequest

use of io.kamax.mxisd.http.io.UserDirectorySearchRequest in project mxisd by kamax-io.

the class UserDirectorySearchHandler method handleRequest.

@Override
public void handleRequest(HttpServerExchange exchange) {
    String accessToken = getAccessToken(exchange);
    UserDirectorySearchRequest searchQuery = parseJsonTo(exchange, UserDirectorySearchRequest.class);
    URI target = URI.create(exchange.getRequestURL());
    UserDirectorySearchResult result = mgr.search(target, accessToken, searchQuery.getSearchTerm());
    respondJson(exchange, GsonUtil.get().toJson(result));
}
Also used : UserDirectorySearchRequest(io.kamax.mxisd.http.io.UserDirectorySearchRequest) URI(java.net.URI) UserDirectorySearchResult(io.kamax.mxisd.http.io.UserDirectorySearchResult)

Example 3 with UserDirectorySearchRequest

use of io.kamax.mxisd.http.io.UserDirectorySearchRequest 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)

Example 4 with UserDirectorySearchRequest

use of io.kamax.mxisd.http.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(MatrixID.asAcceptable(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.http.io.UserDirectorySearchRequest) CloseableHttpResponse(org.apache.http.client.methods.CloseableHttpResponse) IOException(java.io.IOException) InternalServerError(io.kamax.mxisd.exception.InternalServerError) UserDirectorySearchResult(io.kamax.mxisd.http.io.UserDirectorySearchResult)

Aggregations

UserDirectorySearchRequest (io.kamax.mxisd.http.io.UserDirectorySearchRequest)4 UserDirectorySearchResult (io.kamax.mxisd.http.io.UserDirectorySearchResult)4 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 HttpMatrixException (io.kamax.mxisd.exception.HttpMatrixException)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