001/*
002 * Copyright (c) 2011-2017 Nexmo Inc
003 *
004 * Permission is hereby granted, free of charge, to any person obtaining a copy
005 * of this software and associated documentation files (the "Software"), to deal
006 * in the Software without restriction, including without limitation the rights
007 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
008 * copies of the Software, and to permit persons to whom the Software is
009 * furnished to do so, subject to the following conditions:
010 *
011 * The above copyright notice and this permission notice shall be included in
012 * all copies or substantial portions of the Software.
013 *
014 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
015 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
016 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
017 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
018 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
019 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
020 * THE SOFTWARE.
021 */
022package com.nexmo.client.auth;
023
024import com.nexmo.jwt.Jwt;
025import org.apache.http.client.methods.RequestBuilder;
026
027import java.io.IOException;
028import java.nio.file.Files;
029import java.nio.file.Path;
030
031public class JWTAuthMethod extends AbstractAuthMethod {
032    private static final int SORT_KEY = 10;
033    private Jwt jwt;
034
035    public JWTAuthMethod(final String applicationId, final byte[] privateKey) {
036        this.jwt = Jwt.builder().applicationId(applicationId).privateKeyContents(new String(privateKey)).build();
037    }
038
039    public JWTAuthMethod(String applicationId, Path path) throws IOException {
040        this(applicationId, Files.readAllBytes(path));
041    }
042
043    public String generateToken() {
044        return this.jwt.generate();
045    }
046
047    @Override
048    public RequestBuilder apply(RequestBuilder request) {
049        String token = this.jwt.generate();
050
051        request.setHeader("Authorization", "Bearer " + token);
052        return request;
053    }
054
055    @Override
056    public int getSortKey() {
057        return SORT_KEY;
058    }
059}