org.atlasapi.application.SourceLicenseController.java Source code

Introduction

Here is the source code for org.atlasapi.application.SourceLicenseController.java

Source

package org.atlasapi.application;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.Reader;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.atlasapi.application.auth.UserFetcher;
import org.atlasapi.application.users.Role;
import org.atlasapi.application.users.User;
import org.atlasapi.input.ModelReader;
import org.atlasapi.input.ReadException;
import org.atlasapi.output.ErrorResultWriter;
import org.atlasapi.output.ErrorSummary;
import org.atlasapi.output.ResourceForbiddenException;
import org.atlasapi.output.ResponseWriter;
import org.atlasapi.output.ResponseWriterFactory;
import org.atlasapi.output.useraware.UserAwareQueryResult;
import org.atlasapi.output.useraware.UserAwareQueryResultWriter;
import org.atlasapi.query.common.QueryExecutionException;
import org.atlasapi.query.common.QueryParseException;
import org.atlasapi.query.common.useraware.UserAwareQuery;
import org.atlasapi.query.common.useraware.UserAwareQueryContext;
import org.atlasapi.query.common.useraware.UserAwareQueryExecutor;
import org.atlasapi.query.common.useraware.UserAwareQueryParser;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;

@Controller
public class SourceLicenseController {
    private static Logger log = LoggerFactory.getLogger(SourceLicenseController.class);
    private final UserAwareQueryParser<SourceLicense> queryParser;
    private final UserAwareQueryExecutor<SourceLicense> queryExecutor;
    private final UserAwareQueryResultWriter<SourceLicense> resultWriter;
    private final ResponseWriterFactory writerResolver = new ResponseWriterFactory();
    private final ModelReader reader;
    private final UserFetcher userFetcher;
    private final SourceLicenseStore store;

    public SourceLicenseController(UserAwareQueryParser<SourceLicense> queryParser,
            UserAwareQueryExecutor<SourceLicense> queryExecutor,
            UserAwareQueryResultWriter<SourceLicense> resultWriter, ModelReader reader, UserFetcher userFetcher,
            SourceLicenseStore store) {
        super();
        this.queryParser = queryParser;
        this.queryExecutor = queryExecutor;
        this.resultWriter = resultWriter;
        this.reader = reader;
        this.userFetcher = userFetcher;
        this.store = store;
    }

    @RequestMapping({ "/4/source_licenses/{sid}.*", "/4/source_licenses.*" })
    public void listSources(HttpServletRequest request, HttpServletResponse response)
            throws QueryParseException, QueryExecutionException, IOException {
        ResponseWriter writer = writerResolver.writerFor(request, response);
        try {
            UserAwareQuery<SourceLicense> sourcesQuery = queryParser.parse(request);
            UserAwareQueryResult<SourceLicense> queryResult = queryExecutor.execute(sourcesQuery);
            resultWriter.write(queryResult, writer);
        } catch (Exception e) {
            ErrorSummary summary = ErrorSummary.forException(e);
            new ErrorResultWriter().write(summary, writer, request, response);
        }
    }

    @RequestMapping(value = "/4/license.*", method = RequestMethod.POST)
    public void writeLicense(HttpServletRequest request, HttpServletResponse response) throws IOException {
        ResponseWriter writer = null;
        try {
            writer = writerResolver.writerFor(request, response);
            User user = userFetcher.userFor(request).get();
            if (!user.is(Role.ADMIN)) {
                throw new ResourceForbiddenException();
            }

            SourceLicense license = deserialize(new InputStreamReader(request.getInputStream()),
                    SourceLicense.class);
            store.store(license);
            UserAwareQueryResult<SourceLicense> queryResult = UserAwareQueryResult.singleResult(license,
                    UserAwareQueryContext.standard(request));
            resultWriter.write(queryResult, writer);
        } catch (Exception e) {
            log.error("Request exception " + request.getRequestURI(), e);
            ErrorSummary summary = ErrorSummary.forException(e);
            new ErrorResultWriter().write(summary, writer, request, response);
        }
    }

    private <T> T deserialize(Reader input, Class<T> cls) throws IOException, ReadException {
        return reader.read(new BufferedReader(input), cls);
    }
}