components/machine_intelligence/assist_ranker_service_impl.h

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

#ifndef COMPONENTS_MACHINE_INTELLIGENCE_ASSIST_RANKER_SERVICE_IMPL_H_
#define COMPONENTS_MACHINE_INTELLIGENCE_ASSIST_RANKER_SERVICE_IMPL_H_

#include <memory>
#include <string>

#include "base/files/file_path.h"
#include "base/memory/ref_counted.h"
#include "base/sequence_checker.h"
#include "components/machine_intelligence/assist_ranker_service.h"

class GURL;

namespace net {
class URLRequestContextGetter;
}

namespace machine_intelligence {

class BinaryClassifierPredictor;

class AssistRankerServiceImpl : public AssistRankerService {
 public:
  AssistRankerServiceImpl(
      base::FilePath base_path,
      net::URLRequestContextGetter* url_request_context_getter);
  ~AssistRankerServiceImpl() override;

  // AssistRankerService...
  std::unique_ptr<BinaryClassifierPredictor> FetchBinaryClassifierPredictor(
      GURL model_url,
      const std::string& model_filename,
      const std::string& uma_prefix) override;

 private:
  // Returns the full path to the model cache.
  base::FilePath GetModelPath(const std::string& model_filename);

  // Request Context Getter used for RankerURLFetcher.
  scoped_refptr<net::URLRequestContextGetter> url_request_context_getter_;

  // Base path where models are stored.
  const base::FilePath base_path_;

  SEQUENCE_CHECKER(sequence_checker_);

  DISALLOW_COPY_AND_ASSIGN(AssistRankerServiceImpl);
};

}  // namespace machine_intelligence

#endif  // COMPONENTS_MACHINE_INTELLIGENCE_ASSIST_RANKER_SERVICE_IMPL_H_