// Copyright 2013 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CHROMEOS_ASH_COMPONENTS_NETWORK_GEOLOCATION_HANDLER_IMPL_H_
#define CHROMEOS_ASH_COMPONENTS_NETWORK_GEOLOCATION_HANDLER_IMPL_H_

#include <stdint.h>

#include <optional>

#include "base/memory/weak_ptr.h"
#include "base/time/time.h"
#include "base/values.h"
#include "chromeos/ash/components/dbus/shill/shill_property_changed_observer.h"
#include "chromeos/ash/components/network/geolocation_handler.h"
#include "chromeos/ash/components/network/network_handler.h"
#include "chromeos/ash/components/network/network_util.h"

namespace ash {

// This class provides Shill Wifi Access Point and Cell Tower data. It
// currently relies on polling because that is the usage model in
// content::WifiDataProvider. This class requests data asynchronously,
// returning the most recent available data. A typical usage pattern,
// assuming a wifi device is enabled, is:
//   Initialize();  // Makes an initial request
//   GetWifiAccessPoints();  // returns true + initial data, requests update
//   (Delay some amount of time, ~10s)
//   GetWifiAccessPoints();  // returns true + updated data, requests update
//   (Delay some amount of time after data changed, ~10s)
//   GetWifiAccessPoints();  // returns true + same data, requests update
//   (Delay some amount of time after data did not change, ~2 mins)
class COMPONENT_EXPORT(CHROMEOS_NETWORK) GeolocationHandlerImpl
    : public GeolocationHandler,
      public ShillPropertyChangedObserver {
 public:
  GeolocationHandlerImpl(const GeolocationHandlerImpl&) = delete;
  GeolocationHandlerImpl& operator=(const GeolocationHandlerImpl&) = delete;

  ~GeolocationHandlerImpl() override;

  // GeolocationHandler:
  bool GetNetworkInformation(WifiAccessPointVector* access_points,
                             CellTowerVector* cell_towers) override;
  bool GetWifiAccessPoints(WifiAccessPointVector* access_points,
                           int64_t* age_ms) override;
  bool wifi_enabled() const override;

  // ShillPropertyChangedObserver overrides
  void OnPropertyChanged(const std::string& key,
                         const base::Value& value) override;

 private:
  friend class NetworkHandler;
  friend class GeolocationHandlerImplTest;
  friend class SystemLocationProviderWirelessTest;

  GeolocationHandlerImpl();

  void Init();

  // ShillManagerClient callback
  void ManagerPropertiesCallback(std::optional<base::Value::Dict> properties);

  // Called from OnPropertyChanged or ManagerPropertiesCallback.
  void HandlePropertyChanged(const std::string& key, const base::Value& value);

  // Asynchronously request geolocation objects (wifi access points and
  // cell towers) from Shill.Manager.
  void RequestGeolocationObjects();

  // Callback for receiving Geolocation data.
  void GeolocationCallback(std::optional<base::Value::Dict> properties);

  bool cellular_enabled_ = false;
  bool wifi_enabled_ = false;

  void AddCellTowerFromDict(const base::Value::Dict& entry);
  void AddAccessPointFromDict(const base::Value::Dict& entry);

  // Cached network information and update time
  WifiAccessPointVector wifi_access_points_;
  CellTowerVector cell_towers_;
  base::Time geolocation_received_time_;

  // For Shill client callbacks
  base::WeakPtrFactory<GeolocationHandlerImpl> weak_ptr_factory_{this};
};

}  // namespace ash

#endif  // CHROMEOS_ASH_COMPONENTS_NETWORK_GEOLOCATION_HANDLER_IMPL_H_