Java Repository类代码示例

import com.atlassian.stash.repository.Repository; //导入依赖的package包/类
@EventListener
public void pullRequestListener(PullRequestMergeActivityEvent pullRequestMergeActivityEvent) {
    Repository targetRepo = pullRequestMergeActivityEvent.getPullRequest().getToRef().getRepository();
    PullRequest pullRequest = pullRequestMergeActivityEvent.getPullRequest();

    RepositoryHook repositoryHook = repositoryHookService.getByKey(targetRepo, REPO_HOOK_KEY);
    if (repositoryHook.isEnabled() && pullRequestMergeActivityEvent.getActivity().getAction().equals(PullRequestAction.MERGED)) {
        Settings settings = repositoryHookService.getSettings(targetRepo, REPO_HOOK_KEY);
        String bypassKeyword = settings.getString(PullRequestMergeRulesCheck.SETTING_KEY_BYPASS_KEYWORD);
        String apiToken = settings.getString(PullRequestMergeRulesCheck.SETTING_KEY_HIPCHAT_API_TOKEN);
        String roomId = settings.getString(PullRequestMergeRulesCheck.SETTING_KEY_HIPCHAT_ROOM_ID);

        if (StringUtils.containsIgnoreCase(pullRequest.getTitle(), bypassKeyword)
                && isHipChatConfigured(settings)) {
            String message = String.format("<b>%s</b> bypassed merge checks for pull request <b>#%d</b> in <b>%s/%s</b> <br/>%s",
                    pullRequestMergeActivityEvent.getUser().getDisplayName(),
                    pullRequest.getId(),
                    targetRepo.getProject().getKey(),
                    targetRepo.getSlug(),
                    pullRequest.getTitle());

            HipChatClient client = new HipChatClient(apiToken);
            client.sendRoomNotification(new HipChatNotification(message, roomId, HipChatNotification.Color.RED));
        }
    }
}
 

import com.atlassian.stash.repository.Repository; //导入依赖的package包/类
private StatusMessages saveSettings(HttpServletRequest req, Repository repository) {
  final String button = req.getParameter("submit-button") != null ? req.getParameter("submit-button") : "";
  final StatusMessages messages;

  if (button.equals("Save")) {
    messages = saveGeneralSettings(req, repository);
  } else if (button.equals("Add")) {
    messages = addBranch(req, repository);
  } else if (button.equals("Update")) {
    messages = saveBranch(req, repository);
  } else if (button.equals("Delete")) {
    messages = deleteBranch(req, repository);
  } else if (button.equals("Cancel")) {
    messages = new NoMessages();
  } else {
    messages = null; // Invalid form leads to 404
  }

  return messages;
}
 

import com.atlassian.stash.repository.Repository; //导入依赖的package包/类
private StatusMessages saveGeneralSettings(HttpServletRequest req, Repository repository) {
  Boolean enabled = (req.getParameter("enabled") != null);
  String url = req.getParameter("url"); url = url != null ? url.trim() : "";
  String user = req.getParameter("user"); user = user != null ? user.trim() : "";
  String password = req.getParameter("password"); password = password != null ? password.trim() : "";

  if (url.isEmpty() || user.isEmpty() || password.isEmpty()) {
    final String errorMessage = String.format("Empty field not allowed (url '%s', user '%s', password '%s', ",
      url, user, password.isEmpty() ? "" : "*********");
    log.info("Ignoring settings update: " + errorMessage);
    return SingleMessage.error(errorMessage);
  }

  PullRequestTriggerSettings settings = new ImmutablePullRequestTriggerSettings(enabled, url, user, password);

  pullRequestTriggerSettingsService.setPullRequestTriggerSettings(repository, settings);

  return SingleMessage.success("Settings updated");
}
 

import com.atlassian.stash.repository.Repository; //导入依赖的package包/类
@Override
public PullRequestTriggerSettings setPullRequestTriggerSettings(Repository repository, PullRequestTriggerSettings settings) {
  permissionValidationService.validateForRepository(repository, Permission.REPO_ADMIN);
  final Map<String, String> data;
  final Integer repositoryId = repository.getId();
  if (repositoryId == null) {
    log.error("Repository id is null when saving settings: " + repository);
    return null;
  }
  try {
    data = serialize(settings);
  } catch (NullPointerException e) {
    log.error("Error serializing PR settings object " + settings, e);
    throw e;
  }
  pluginSettings.put(Integer.toString(repositoryId), data);
  cache.remove(repositoryId);
  return deserialize(data);
}
 

import com.atlassian.stash.repository.Repository; //导入依赖的package包/类
@Override
public void deleteBranch(Repository repository, String branchName) {
  permissionValidationService.validateForRepository(repository, Permission.REPO_ADMIN);
  final String branchKey = branchKeyForRepoId(repository, branchName);
  final Integer repositoryId = repository.getId();
  final List<String> branches = branchListCache.get(repositoryId);

  if (repositoryId == null) {
    log.error("Repository id is null when deleting branch settings: " + repository);
    return;
  }

  if (branches.contains(branchName)) {
    branches.remove(branchName);
    pluginSettings.remove(KEY_BRANCH_LIST + repositoryId.toString());
    if (!branches.isEmpty())
      pluginSettings.put(KEY_BRANCH_LIST + repositoryId.toString(), branches);
    branchListCache.remove(repositoryId);
  }

  pluginSettings.remove(branchKey);
  branchCache.remove(branchKey);
}
 

import com.atlassian.stash.repository.Repository; //导入依赖的package包/类
@Override
public List<BranchSettings> getBranchSettingsForBranch(Repository repository, String branchName) {
  permissionValidationService.validateForRepository(repository, Permission.REPO_READ);
  final List<String> branchList = branchListCache.get(repository.getId());
  final List<BranchSettings> result = new ArrayList<BranchSettings>();

  for (String branchPattern : branchList) {
    final Pattern regexPattern;

    try {
      regexPattern = Pattern.compile(branchPattern);
    } catch (PatternSyntaxException e) {
      log.error(String.format("Invalid regex for branch configuration: %s", branchName), e);
      return new ArrayList<BranchSettings>();
    }
    if (regexPattern.matcher(branchName).find()) {
      result.add(branchCache.get(branchKeyForRepoId(repository, branchPattern)));
    }
  }
  return result;
}
 

import com.atlassian.stash.repository.Repository; //导入依赖的package包/类
@Override
public void triggerBuild(PullRequestEvent pullRequestEvent, BranchPredicate predicate) {
  final PullRequest pullRequest = pullRequestEvent.getPullRequest();
  final PullRequestTriggerSettings settings = getSettings(pullRequest);
  final Repository repository = getRepository(pullRequest);
  final String branchName = pullRequest.getToRef().getId();
  final List<BranchSettings> branchSettingsList = service.getBranchSettingsForBranch(repository, branchName);
  final Long prNumber = pullRequest.getId();

  if (prNumber != null) {
    if (settings.isEnabled()) {
      for (BranchSettings branchSettings : branchSettingsList) {
        if (predicate.matches(branchSettings)) {
          buildTriggerer.invoke(prNumber, settings, branchSettings);
        }
      }
    }
  } else {
    log.error("id of pull request is null: " + pullRequest);
  }
}
 

import com.atlassian.stash.repository.Repository; //导入依赖的package包/类
@Test
public void gettingSettingsForNewRepoReturnsDefaultSettingsTest() {
  final PermissionValidationService permService = mock(PermissionValidationService.class);
  final PluginSettingsFactory factory = mock(PluginSettingsFactory.class);
  final PluginSettings pluginSettings = mock(PluginSettings.class);
  final Repository repository = mock(Repository.class);

  when(repository.getId()).thenReturn(1);
  when(factory.createSettingsForKey(PluginMetadata.getPluginKey())).thenReturn(pluginSettings);

  final DefaultPullRequestTriggerSettingsService service = new DefaultPullRequestTriggerSettingsService(permService, factory);

  final PullRequestTriggerSettings settings = service.getPullRequestTriggerSettings(repository);

  assertEquals(new ImmutablePullRequestTriggerSettings(), settings);
}
 

import com.atlassian.stash.repository.Repository; //导入依赖的package包/类
@Test
public void settingsCanBeSavedTest() {
  final PermissionValidationService permService = mock(PermissionValidationService.class);
  final PluginSettingsFactory factory = mock(PluginSettingsFactory.class);
  final PluginSettings pluginSettings = mock(PluginSettings.class);
  final Repository repository = mock(Repository.class);

  when(repository.getId()).thenReturn(1);
  when(factory.createSettingsForKey(PluginMetadata.getPluginKey())).thenReturn(pluginSettings);
  when(pluginSettings.put("1", settingsMapEnabled)).thenReturn(settingsMapEnabled);
  when(pluginSettings.get("1")).thenReturn(settingsMapEnabled);

  final DefaultPullRequestTriggerSettingsService service = new DefaultPullRequestTriggerSettingsService(permService, factory);

  service.setPullRequestTriggerSettings(repository, settingsEnabled);
  final PullRequestTriggerSettings settings = service.getPullRequestTriggerSettings(repository);

  InOrder inOrder = inOrder(pluginSettings);
  inOrder.verify(pluginSettings).put("1", settingsMapEnabled);
  inOrder.verify(pluginSettings).get("1");

  assertEquals(settingsEnabled, settings);
  assertNotSame(new ImmutablePullRequestTriggerSettings(), settings);
}
 

import com.atlassian.stash.repository.Repository; //导入依赖的package包/类
@Test
public void gettingSettingsForBranchReturnsBranchesMatchingNameTest() {
  final PermissionValidationService permService = mock(PermissionValidationService.class);
  final PluginSettingsFactory factory = mock(PluginSettingsFactory.class);
  final PluginSettings pluginSettings = mock(PluginSettings.class);
  final Repository repository = mock(Repository.class);

  when(repository.getId()).thenReturn(1);
  when(factory.createSettingsForKey(PluginMetadata.getPluginKey())).thenReturn(pluginSettings);
  when(pluginSettings.get("branchList:1")).thenReturn(expandedBranchList());
  when(pluginSettings.get(branchName + ":1")).thenReturn(branchSettingsMap);
  when(pluginSettings.get(anotherBranchName + ":1")).thenReturn(anotherBranchMap);

  final DefaultPullRequestTriggerSettingsService service = new DefaultPullRequestTriggerSettingsService(permService, factory);

  final List<BranchSettings> settingsList = service.getBranchSettingsForBranch(repository, branchName);

  assertEquals(Lists.newArrayList(immutableBranchSettings), settingsList);
}
 

import com.atlassian.stash.repository.Repository; //导入依赖的package包/类
@Test
public void gettingSettingsForBranchReturnsBranchesMatchingRegexTest() {
  final PermissionValidationService permService = mock(PermissionValidationService.class);
  final PluginSettingsFactory factory = mock(PluginSettingsFactory.class);
  final PluginSettings pluginSettings = mock(PluginSettings.class);
  final Repository repository = mock(Repository.class);

  when(repository.getId()).thenReturn(1);
  when(factory.createSettingsForKey(PluginMetadata.getPluginKey())).thenReturn(pluginSettings);
  when(pluginSettings.get("branchList:1")).thenReturn(regexBranchList());
  when(pluginSettings.get(regexBranchPattern + ":1")).thenReturn(regexBranchMap);

  final DefaultPullRequestTriggerSettingsService service = new DefaultPullRequestTriggerSettingsService(permService, factory);

  final List<BranchSettings> settingsList = service.getBranchSettingsForBranch(repository, anotherBranchName);

  assertEquals(Lists.newArrayList(regexBranchSettings), settingsList);
}
 

import com.atlassian.stash.repository.Repository; //导入依赖的package包/类
@Test
public void gettingSettingsForBranchReturnsBranchesWithPartialNameMatchesTest() {
  final PermissionValidationService permService = mock(PermissionValidationService.class);
  final PluginSettingsFactory factory = mock(PluginSettingsFactory.class);
  final PluginSettings pluginSettings = mock(PluginSettings.class);
  final Repository repository = mock(Repository.class);

  when(repository.getId()).thenReturn(1);
  when(factory.createSettingsForKey(PluginMetadata.getPluginKey())).thenReturn(pluginSettings);
  when(pluginSettings.get("branchList:1")).thenReturn(expandedBranchList());
  when(pluginSettings.get(branchName + ":1")).thenReturn(branchSettingsMap);
  when(pluginSettings.get(anotherBranchName + ":1")).thenReturn(anotherBranchMap);

  final DefaultPullRequestTriggerSettingsService service = new DefaultPullRequestTriggerSettingsService(permService, factory);

  final List<BranchSettings> settingsList = service.getBranchSettingsForBranch(repository, anotherBranchName);

  assertEquals(Lists.newArrayList(anotherBranchSettings), settingsList);
}
 

import com.atlassian.stash.repository.Repository; //导入依赖的package包/类
@Test
public void gettingSettingsForBranchReturnsAllBranchesMatchingNameTest() {
  final PermissionValidationService permService = mock(PermissionValidationService.class);
  final PluginSettingsFactory factory = mock(PluginSettingsFactory.class);
  final PluginSettings pluginSettings = mock(PluginSettings.class);
  final Repository repository = mock(Repository.class);

  when(repository.getId()).thenReturn(1);
  when(factory.createSettingsForKey(PluginMetadata.getPluginKey())).thenReturn(pluginSettings);
  when(pluginSettings.get("branchList:1")).thenReturn(regexExpandedBranchList());
  when(pluginSettings.get(branchName + ":1")).thenReturn(branchSettingsMap);
  when(pluginSettings.get(regexBranchPattern + ":1")).thenReturn(regexBranchMap);

  final DefaultPullRequestTriggerSettingsService service = new DefaultPullRequestTriggerSettingsService(permService, factory);

  final List<BranchSettings> settingsList = service.getBranchSettingsForBranch(repository, branchName);

  assertEquals(Lists.newArrayList(immutableBranchSettings, regexBranchSettings), settingsList);
}
 

import com.atlassian.stash.repository.Repository; //导入依赖的package包/类
@Test
public void automaticTriggerBuildIfBranchAutomaticBuildIsEnabledTest() {
  BuildTriggerer buildTriggerer = mock(BuildTriggerer.class);
  PullRequestEvent event = mock(PullRequestEvent.class);
  PullRequest pullRequest = mock(PullRequest.class);
  PullRequestRef ref = mock(PullRequestRef.class);
  Repository repository = mock(Repository.class);

  when(event.getPullRequest()).thenReturn(pullRequest);
  when(pullRequest.getToRef()).thenReturn(ref, ref);
  when(ref.getRepository()).thenReturn(repository);
  when(ref.getId()).thenReturn(branchName);
  when(pullRequest.getId()).thenReturn(1L);

  Trigger trigger = new TriggerImpl(settingsServiceEnabled, buildTriggerer);

  trigger.automaticTrigger(event);

  verify(buildTriggerer).invoke(1L, settingsEnabled, immutableBranchSettings);
}
 

import com.atlassian.stash.repository.Repository; //导入依赖的package包/类
@Test
public void automaticTriggerDoesNotBuildIfBranchAutomaticBuildIsDisabledTest() {
  BuildTriggerer buildTriggerer = mock(BuildTriggerer.class);
  PullRequestEvent event = mock(PullRequestEvent.class);
  PullRequest pullRequest = mock(PullRequest.class);
  PullRequestRef ref = mock(PullRequestRef.class);
  Repository repository = mock(Repository.class);

  when(event.getPullRequest()).thenReturn(pullRequest);
  when(pullRequest.getToRef()).thenReturn(ref, ref);
  when(ref.getRepository()).thenReturn(repository);
  when(ref.getId()).thenReturn(branchName);
  when(pullRequest.getId()).thenReturn(1L);

  Trigger trigger = new TriggerImpl(settingsServiceOnDemand, buildTriggerer);

  trigger.automaticTrigger(event);

  verify(buildTriggerer, never()).invoke(anyLong(), any(PullRequestTriggerSettings.class), any(BranchSettings.class));
}