/*
|
* Copyright 2014 gitblit.com.
|
*
|
* 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.gitblit.transport.ssh;
|
|
import java.util.ArrayList;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
|
import com.google.inject.Inject;
|
|
/**
|
* Memory public key manager.
|
*
|
* @author James Moger
|
*
|
*/
|
public class MemoryKeyManager extends IPublicKeyManager {
|
|
final Map<String, List<SshKey>> keys;
|
|
@Inject
|
public MemoryKeyManager() {
|
keys = new HashMap<String, List<SshKey>>();
|
}
|
|
@Override
|
public String toString() {
|
return getClass().getSimpleName();
|
}
|
|
@Override
|
public MemoryKeyManager start() {
|
log.info(toString());
|
return this;
|
}
|
|
@Override
|
public boolean isReady() {
|
return true;
|
}
|
|
@Override
|
public MemoryKeyManager stop() {
|
return this;
|
}
|
|
@Override
|
protected boolean isStale(String username) {
|
// always return true so we gets keys from our hashmap
|
return true;
|
}
|
|
@Override
|
protected List<SshKey> getKeysImpl(String username) {
|
String id = username.toLowerCase();
|
if (keys.containsKey(id)) {
|
return keys.get(id);
|
}
|
return null;
|
}
|
|
@Override
|
public boolean addKey(String username, SshKey key) {
|
String id = username.toLowerCase();
|
if (!keys.containsKey(id)) {
|
keys.put(id, new ArrayList<SshKey>());
|
}
|
log.info("added {} key {}", username, key.getFingerprint());
|
return keys.get(id).add(key);
|
}
|
|
@Override
|
public boolean removeKey(String username, SshKey key) {
|
String id = username.toLowerCase();
|
if (!keys.containsKey(id)) {
|
log.info("can't remove keys for {}", username);
|
return false;
|
}
|
List<SshKey> list = keys.get(id);
|
boolean success = list.remove(key);
|
if (success) {
|
log.info("removed {} key {}", username, key.getFingerprint());
|
}
|
|
if (list.isEmpty()) {
|
keys.remove(id);
|
log.info("no {} keys left, removed {}", username, username);
|
}
|
return success;
|
}
|
|
@Override
|
public boolean removeAllKeys(String username) {
|
String id = username.toLowerCase();
|
keys.remove(id.toLowerCase());
|
log.info("removed all keys for {}", username);
|
return true;
|
}
|
}
|
