// Copyright (C) 2021 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.httpd;

import com.google.gerrit.extensions.events.GitReferenceUpdatedListener;
import com.google.gerrit.extensions.registration.DynamicItem;
import com.google.inject.Inject;
import com.google.inject.Singleton;

/**
 * Stores the updated refs whenever they are updated, so that we can export this information in the
 * response headers.
 */
@Singleton
public class GitReferenceUpdatedTracker implements GitReferenceUpdatedListener {

  private final DynamicItem<WebSession> webSession;

  @Inject
  GitReferenceUpdatedTracker(DynamicItem<WebSession> webSession) {
    this.webSession = webSession;
  }

  @Override
  public void onGitReferenceUpdated(GitReferenceUpdatedListener.Event event) {
    WebSession currentSession = webSession.get();
    if (currentSession != null) {
      currentSession.addRefUpdatedEvents(event);
    }
  }
}

@@ -803,16 +803,61 @@
 Inject;%0A
+import com.google.inject.ProvisionException;%0A
 import c
@@ -884,16 +884,16 @@
 gleton;%0A
-
 %0A/**%0A * 
@@ -1010,16 +1010,126 @@
 eaders.%0A
+ *%0A * %3Cp%3EThis is only working for HTTP requests. %7B@link WebSession%7D is not bound outside of HTTP%0A * requests.%0A
  */%0A@Sin
@@ -1513,25 +1513,384 @@
 n = 
-webSession.get();
+null;%0A    try %7B%0A      currentSession = webSession.get();%0A    %7D catch (ProvisionException ex) %7B%0A      // We couldn't bind the current session properly. This is expected to happen at any point we%0A      // perform ref updates without an HTTP request (git push for example).%0A      // If we can't get a WebSession, we don't need to track the updated references.%0A      return;%0A    %7D
 %0A   

Fail graciously when unable to bind WebSession for ref-update tracking