com.github.sdtool.statelesscaptcha.token.Creator Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of stateless-captcha Show documentation
Stateless captcha library for java
The newest version!
/*
 * Copyright 2021 Subhajit Das
 *
 *    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.github.sdtool.statelesscaptcha.token;

import com.auth0.jwt.JWT;
import com.auth0.jwt.algorithms.Algorithm;
import com.auth0.jwt.exceptions.JWTCreationException;
import com.github.sdtool.statelesscaptcha.core.audio.AudioCaptcha;
import com.github.sdtool.statelesscaptcha.core.text.Captcha;
import com.github.sdtool.statelesscaptcha.exception.CreationException;
import com.github.sdtool.statelesscaptcha.util.Base64Util;

import javax.sound.sampled.AudioFileFormat;
import java.io.IOException;
import java.time.LocalDateTime;
import java.time.ZoneId;
import java.util.Date;

/**
 * The token creator
 *
 * @author Subhajit Das
 */
public class Creator {

    /**
     * The token properties
     * Initiated to default
     */
    private TokenProperties properties = new TokenProperties();

    /**
     * Default constructor
     */
    public Creator() {
    }

    /**
     * Constructor with customized token properties
     *
     * @param properties the token properties to use
     */
    public Creator(TokenProperties properties) {
        this.properties = properties;
    }

    /**
     * Gets the token properties
     *
     * @return the token properties
     */
    public TokenProperties getProperties() {
        return properties;
    }

    /**
     * Sets the token properties
     *
     * @param properties the token properties
     */
    public void setProperties(TokenProperties properties) {
        this.properties = properties;
    }

    /**
     * Generate token representation of text captcha
     *
     * @param captcha the captcha to convert
     * @return the token model
     */
    public CaptchaToken create(Captcha captcha) {
        try {
            Algorithm algorithm = Algorithm.HMAC256(captcha.getAnswer());
            Date in = new Date();
            LocalDateTime ldt = LocalDateTime.ofInstant(in.toInstant(), ZoneId.systemDefault())
                    .plusSeconds(properties.getValidity());
            Date out = Date.from(ldt.atZone(ZoneId.systemDefault()).toInstant());
            String token = JWT.create()
                    .withExpiresAt(out)
                    .withIssuer(properties.getIssuer())
                    .sign(algorithm);
            return new CaptchaToken(
                    Base64Util.encodeImage(captcha.getImage(), "png"),
                    "png",
                    token);
        } catch (JWTCreationException | IllegalArgumentException | IOException exception) {
            //Invalid Signing configuration / Couldn't convert Claims.
            throw new IllegalArgumentException("Unable to create token");
        }
    }

    /**
     * Generate token representation of audio captcha
     *
     * @param captcha the captcha to convert
     * @return the token model
     */
    public CaptchaToken create(AudioCaptcha captcha) {
        try {
            Algorithm algorithm = Algorithm.HMAC256(captcha.getAnswer());
            Date in = new Date();
            LocalDateTime ldt = LocalDateTime.ofInstant(in.toInstant(), ZoneId.systemDefault())
                    .plusSeconds(properties.getValidity());
            Date out = Date.from(ldt.atZone(ZoneId.systemDefault()).toInstant());
            String token = JWT.create()
                    .withExpiresAt(out)
                    .withIssuer(properties.getIssuer())
                    .sign(algorithm);
            return new CaptchaToken(
                    Base64Util.encodeAudio(captcha.getChallenge().getAudioInputStream(),
                            AudioFileFormat.Type.WAVE),
                    "wav",
                    token);
        } catch (JWTCreationException | IllegalArgumentException | IOException exception) {
            //Invalid Signing configuration / Couldn't convert Claims.
            throw new CreationException("Unable to create token");
        }
    }

}