FileSpec.h

/*
 * Copyright (c) Meta Platforms, Inc. and affiliates.
 *
 * 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.
 */
 
#pragma once
 
#include <map>
#include <string>
#include <utility>
#include <vector>
 
namespace vrs {
using std::map;
using std::string;
using std::vector;
 
struct FileSpec {
  FileSpec() = default;
  FileSpec(string filehandler, const vector<string>& chunks)
      : fileHandlerName{std::move(filehandler)}, chunks{chunks} {}
  FileSpec(string filehandler, const vector<string>&& chunks)
      : fileHandlerName{std::move(filehandler)}, chunks{chunks} {}
  explicit FileSpec(const vector<string>& chunks) : chunks{chunks} {}
  explicit FileSpec(vector<string>&& chunks) : chunks{std::move(chunks)} {}
 
  void clear();
 
  bool empty() const;
 
  bool isDiskFile() const;
 
  static int parseUri(
      const string& uri,
      string& outScheme,
      string& outPath,
      map<string, string>& outQueryParams);
 
  int fromPathJsonUri(const string& pathJsonUri, const string& defaultFileHandlerName = {});
 
  string toPathJsonUri() const;
 
  // Parse json and extract file specs, with optional extra parameters.
  // @param jsonStr: ex. {"storage": "mystorage", "chunks":["chunk1", "chunk2"],
  // "filename":"file.vrs"}.
  // @return True if parsing the json succeeded.
  bool fromJson(const string& jsonStr);
 
  // Generate json string from a file spec.
  // @return jsonStr: ex. {"storage": "mystorage", "chunks":["chunk1", "chunk2"],
  // "filename":"file.vrs"}.
  string toJson() const;
 
  int parseUri();
 
  bool hasChunkSizes() const;
  int64_t getFileSize() const;
  string getSourceLocation() const;
  string getEasyPath() const;
  string getFileName() const;
 
  string getXXHash() const;
 
  bool operator==(const FileSpec& rhs) const;
 
  const string& getExtra(const string& name) const;
  const string& getExtra(const string& name, const string& defaultValue) const;
  bool hasExtra(const string& name) const;
  int getExtraAsInt(const string& name, int defaultValue = 0) const;
  int64_t getExtraAsInt64(const string& name, int64_t defaultValue = 0) const;
  uint64_t getExtraAsUInt64(const string& name, uint64_t defaultValue = 0) const;
  double getExtraAsDouble(const string& name, double defaultValue = 0) const;
  bool getExtraAsBool(const string& name, bool defaultValue = false) const;
 
  static int decodeQuery(const string& query, string& outKey, string& outValue);
  static int urldecode(const string& in, string& out);
 
  void setExtra(const string& name, const string& value);
  void setExtra(const string& name, const char* value);
  void setExtra(const string& name, bool value);
  template <typename T>
  void setExtra(const string& name, T value) {
    extras[name] = std::to_string(value);
  }
 
  void unsetExtra(const string& name);
 
  string fileHandlerName;
  string fileName;
  string uri;
  vector<string> chunks;
  vector<int64_t> chunkSizes;
  map<string, string> extras;
};
 
} // namespace vrs