/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#ifndef HTTPSRecordResolver_h_
#define HTTPSRecordResolver_h_

#include "mozilla/Mutex.h"
#include "nsICancelable.h"
#include "nsIDNSListener.h"
#include "nsHttpConnectionInfo.h"

class nsIDNSAddrRecord;
class nsIDNSHTTPSSVCRecord;

namespace mozilla {
namespace net {

class nsAHttpTransaction;

// This class is the place to put some common code about fetching HTTPS RR.
class HTTPSRecordResolver : public nsIDNSListener {
 public:
  NS_DECL_THREADSAFE_ISUPPORTS
  NS_DECL_NSIDNSLISTENER

  explicit HTTPSRecordResolver(nsAHttpTransaction* aTransaction);
  nsresult FetchHTTPSRRInternal(nsIEventTarget* aTarget,
                                nsICancelable** aDNSRequest);
  void PrefetchAddrRecord(const nsACString& aTargetName, bool aRefreshDNS);

  void Close();

 protected:
  virtual ~HTTPSRecordResolver();

 private:
  nsresult InvokeCallback();

  mozilla::Mutex mMutex{"HTTPSRecordResolver::mMutex"};
  RefPtr<nsAHttpTransaction> mTransaction MOZ_GUARDED_BY(mMutex);
  RefPtr<nsHttpConnectionInfo> mConnInfo;  // Set in constructor only
  nsCOMPtr<nsICancelable> mCnameRequest MOZ_GUARDED_BY(mMutex);
  nsCOMPtr<nsICancelable> mHTTPSRecordRequest MOZ_GUARDED_BY(mMutex);
  nsCOMPtr<nsIDNSAddrRecord> mAddrRecord MOZ_GUARDED_BY(mMutex);
  nsCOMPtr<nsIDNSHTTPSSVCRecord> mHTTPSRecord MOZ_GUARDED_BY(mMutex);
  bool mDone MOZ_GUARDED_BY(mMutex) = false;
};

}  // namespace net
}  // namespace mozilla

#endif  // HTTPSRecordResolver_h_