/*
* Copyright 2010 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.springsource.greenhouse.members;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.net.URLConnection;
import java.util.Arrays;
import java.util.List;
import javax.inject.Inject;
import org.springframework.data.FileData;
import org.springframework.data.FileStorage;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.stereotype.Service;
import com.springsource.greenhouse.account.AccountMapper;
import com.springsource.greenhouse.utils.ImageUtils;
/**
* PictureProfileService that puts profile pics into FileStorage.
* Also sets a flag associated with the member's account when a profile pic is set.
* This flag is used when constructing the pictureUrl when {@link AccountMapper mapping} Account and Profile objects.
* @author Keith Donald
* @author Craig Walls
*/
@Service
public class FileStorageProfilePictureService implements ProfilePictureService {
private final FileStorage storage;
private final JdbcTemplate jdbcTemplate;
/**
* Construct a FileStorageProfilePictureService
* @param storage the file storage implementation; for example, to store files in S3 or a local directory.
* @param jdbcTemplate the data access template used to update the pictureSet flag
*/
@Inject
public FileStorageProfilePictureService(FileStorage storage, JdbcTemplate jdbcTemplate) {
this.storage = storage;
this.jdbcTemplate = jdbcTemplate;
}
public void saveProfilePicture(Long accountId, byte[] imageBytes) throws IOException {
assertBytesLength(imageBytes);
String contentType = guessContentType(imageBytes);
assertContentType(contentType);
storage.storeFile(new FileData("profile-pics/" + accountId + "/small.jpg", ImageUtils.scaleImageToWidth(imageBytes, 50), contentType));
storage.storeFile(new FileData("profile-pics/" + accountId + "/normal.jpg", ImageUtils.scaleImageToWidth(imageBytes, 100), contentType));
storage.storeFile(new FileData("profile-pics/" + accountId + "/large.jpg", ImageUtils.scaleImageToWidth(imageBytes, 200), contentType));
jdbcTemplate.update("update Member set pictureSet = true where id = ?", accountId);
}
// internal helpers
private void assertBytesLength(byte[] imageBytes) {
if (imageBytes.length == 0) {
throw new IllegalArgumentException("Cannot accept empty picture byte[] as a profile picture");
}
}
private String guessContentType(byte[] imageBytes) throws IOException {
return URLConnection.guessContentTypeFromStream(new ByteArrayInputStream(imageBytes));
}
private void assertContentType(String contentType) {
if (!IMAGE_TYPES.contains(contentType)) {
throw new IllegalArgumentException("Cannot accept content type '" + contentType + "' as a profile picture.");
}
}
private static final List<String> IMAGE_TYPES = Arrays.asList("image/jpeg", "image/gif", "image/png");
}