/*
 * Copyright (C) 2019 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 specic language governing permissions and
 * limitations under the License.
 */

#ifndef MEDIAPROVIDER_JNI_FUSEDAEMON_H_
#define MEDIAPROVIDER_JNI_FUSEDAEMON_H_

#include <android-base/unique_fd.h>

#include <map>
#include <memory>
#include <string>
#include <vector>

#include "MediaProviderWrapper.h"
#include "jni.h"
#include "node-inl.h"

struct fuse;
namespace mediaprovider {
namespace fuse {
class FuseDaemon final {
  public:
    FuseDaemon(JNIEnv* env, jobject mediaProvider);

    ~FuseDaemon() = default;

    /**
     * Start the FUSE daemon loop that will handle filesystem calls.
     */
    void Start(android::base::unique_fd fd, const std::string& path, const bool uncached_mode,
               const std::vector<std::string>& supported_transcoding_relative_paths,
               const std::vector<std::string>& supported_uncached_relative_paths);

    /**
     * Checks if the FUSE daemon is started.
     */
    bool IsStarted() const;

    /**
     * Check if file should be opened with FUSE
     */
    bool ShouldOpenWithFuse(int fd, bool for_read, const std::string& path);

    /**
     * Check if the FUSE daemon uses FUSE passthrough
     */
    bool UsesFusePassthrough() const;

    /**
     * Invalidate FUSE VFS dentry cache entry for path
     */
    void InvalidateFuseDentryCache(const std::string& path);

    /**
     * Checks if the given uid has access to the given fd with or without redaction.
     */
    std::unique_ptr<FdAccessResult> CheckFdAccess(int fd, uid_t uid) const;

    /**
     * Initialize device id for the FUSE daemon with the FUSE device id of the given path.
     */
    void InitializeDeviceId(const std::string& path);

    /**
     * Setup leveldb instances based on the volume.
     */
    void SetupLevelDbInstances();

    /**
     * Setup leveldb instances for public volume.
     */
    void SetupPublicVolumeLevelDbInstance(const std::string& volume_name);

    /**
     * Creates a leveldb instance and sets up a connection.
     */
    void SetupLevelDbConnection(const std::string& instance_name);

    /**
     * Deletes entry for given key from leveldb.
     */
    void DeleteFromLevelDb(const std::string& key);

    /**
     * Inserts in leveldb instance of provided volume.
     */
    void InsertInLevelDb(const std::string& volume_name, const std::string& key,
                         const std::string& value);

    /**
     * Reads file paths for given volume from leveldb for given range.
     */
    std::vector<std::string> ReadFilePathsFromLevelDb(const std::string& volume_name,
                                                      const std::string& lastReadValue, int limit);

    /**
     * Reads backed up data from leveldb.
     */
    std::string ReadBackedUpDataFromLevelDb(const std::string& filePath);

    /**
     * Reads value for given key, returns empty string if not found.
     */
    std::string ReadOwnership(const std::string& key);

    /**
     * Creates owner id to owner package identifier and vice versa relation in leveldb.
     */
    void CreateOwnerIdRelation(const std::string& ownerId,
                               const std::string& ownerPackageIdentifier);

    /**
     * Removes owner id to owner package identifier and vice versa relation in leveldb.
     */
    void RemoveOwnerIdRelation(const std::string& ownerId,
                               const std::string& ownerPackageIdentifier);

    /**
     * Reads owner relationships from leveldb.
     */
    std::map<std::string, std::string> GetOwnerRelationship();

    /**
     * Returns true if level db setup exists for given instance.
     */
    bool CheckLevelDbConnection(const std::string& instance_name);

  private:
    FuseDaemon(const FuseDaemon&) = delete;
    void operator=(const FuseDaemon&) = delete;
    MediaProviderWrapper mp;
    std::atomic_bool active;
    struct ::fuse* fuse;
};

}  // namespace fuse
}  // namespace mediaprovider

#endif  // MEDIAPROVIDER_JNI_FUSEDAEMON_H_