com.foxinmy.weixin4j.mp.api.QrApi Maven / Gradle / Ivy

Go to download
package com.foxinmy.weixin4j.mp.api;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;

import com.foxinmy.weixin4j.exception.WeixinException;
import com.foxinmy.weixin4j.http.Response;
import com.foxinmy.weixin4j.model.Token;
import com.foxinmy.weixin4j.mp.model.QRParameter;
import com.foxinmy.weixin4j.mp.type.QRType;
import com.foxinmy.weixin4j.token.TokenHolder;
import com.foxinmy.weixin4j.util.ConfigUtil;

/**
 * 二维码相关API
 * 
 * @className QrApi
 * @author jy.hu
 * @date 2014年9月25日
 * @since JDK 1.7
 * @see 二维码支持
 */
public class QrApi extends MpApi {

	private final TokenHolder tokenHolder;

	public QrApi(TokenHolder tokenHolder) {
		this.tokenHolder = tokenHolder;
	}

	/**
	 * 生成带参数的二维码
	 * 
	 * @param parameter
	 *            二维码参数
	 * @return byte数据包
	 * @throws WeixinException
	 * @see 生成二维码
	 */
	public byte[] getQRData(QRParameter parameter) throws WeixinException {
		Token token = tokenHolder.getToken();
		String qr_uri = getRequestUri("qr_ticket_uri");
		Response response = request.post(
				String.format(qr_uri, token.getAccessToken()),
				parameter.toJson());
		String ticket = response.getAsJson().getString("ticket");
		qr_uri = getRequestUri("qr_image_uri");
		response = request.get(String.format(qr_uri, ticket));

		return response.getBody();
	}

	/**
	 * 生成带参数的二维码
	 * 
	 * @param sceneId
	 *            场景值
	 * @param expireSeconds
	 *            过期秒数 如果小于等于0则 视为永久二维码
	 * @return byte数据包
	 * @throws WeixinException
	 * @see {@link com.foxinmy.weixin4j.mp.api.QrApi#getQR(QRParameter)}
	 * @see 生成二维码
	 */
	public byte[] getQRData(int sceneId, int expireSeconds)
			throws WeixinException {
		QRParameter parameter = new QRParameter(sceneId, expireSeconds);
		return getQRData(parameter);
	}

	/**
	 * 生成带参数的二维码
	 * 
	 * 二维码分为临时跟永久两种,扫描时触发推送带参数事件
	 * 
	 * 
	 * @param parameter
	 *            二维码参数
	 * @return 硬盘存储的文件对象
	 * @throws WeixinException
	 * @see 二维码
	 * @see com.foxinmy.weixin4j.mp.model.QRParameter
	 */
	public File getQR(QRParameter parameter) throws WeixinException {
		String qr_path = ConfigUtil.getValue("qr_path");
		String filename = String.format("%s_%d_%d.jpg", parameter.getQrType()
				.name(), parameter.getSceneId(), parameter.getExpireSeconds());
		File file = new File(qr_path + File.separator + filename);
		if (parameter.getQrType() == QRType.PERMANENCE && file.exists()) {
			return file;
		}
		byte[] datas = getQRData(parameter);
		OutputStream os = null;
		try {
			boolean flag = file.exists() || file.createNewFile();
			if (flag) {
				os = new FileOutputStream(file);
				os.write(datas);
			} else {
				throw new WeixinException(String.format(
						"create file fail:%s", file.getAbsolutePath()));
			}
		} catch (IOException e) {
			throw new WeixinException(e.getMessage());
		} finally {
			try {
				if (os != null) {
					os.close();
				}
			} catch (IOException ignore) {
				;
			}
		}
		return file;
	}
}