com.google.gerrit.server.account.AuthorizedKeys Maven / Gradle / Ivy
// Copyright (C) 2016 The Android Open Source Project
//
// 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.google.gerrit.server.account;
import com.google.common.annotations.VisibleForTesting;
import com.google.common.base.Splitter;
import com.google.gerrit.reviewdb.client.Account;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Optional;
import java.util.regex.Pattern;
public class AuthorizedKeys {
public static final String FILE_NAME = "authorized_keys";
@VisibleForTesting public static final String INVALID_KEY_COMMENT_PREFIX = "# INVALID ";
@VisibleForTesting public static final String DELETED_KEY_COMMENT = "# DELETED";
private static final Pattern LINE_SPLIT_PATTERN = Pattern.compile("\\r?\\n");
public static List> parse(Account.Id accountId, String s) {
List> keys = new ArrayList<>();
int seq = 1;
for (String line : Splitter.on(LINE_SPLIT_PATTERN).split(s)) {
line = line.trim();
if (line.isEmpty()) {
continue;
} else if (line.startsWith(INVALID_KEY_COMMENT_PREFIX)) {
String pub = line.substring(INVALID_KEY_COMMENT_PREFIX.length());
AccountSshKey key = AccountSshKey.createInvalid(accountId, seq++, pub);
keys.add(Optional.of(key));
} else if (line.startsWith(DELETED_KEY_COMMENT)) {
keys.add(Optional.empty());
seq++;
} else if (line.startsWith("#")) {
continue;
} else {
AccountSshKey key = AccountSshKey.create(accountId, seq++, line);
keys.add(Optional.of(key));
}
}
return keys;
}
public static String serialize(Collection> keys) {
StringBuilder b = new StringBuilder();
for (Optional key : keys) {
if (key.isPresent()) {
if (!key.get().valid()) {
b.append(INVALID_KEY_COMMENT_PREFIX);
}
b.append(key.get().sshPublicKey().trim());
} else {
b.append(DELETED_KEY_COMMENT);
}
b.append("\n");
}
return b.toString();
}
}