Compare commits

...

5 Commits

Author SHA1 Message Date
ea3b22fccc public API 2022-10-11 16:15:03 +02:00
Casey Rodarmor
984543fcb9
Don't export rustflags
type: fixed
2021-10-13 19:47:49 -07:00
Casey Rodarmor
6c4805890b
Pass rustflags consistently
type: fixed
2021-10-11 18:49:10 -07:00
Casey Rodarmor
b82ccf1882
Add --base-directory to imdl torrent verify
type: added
2021-10-01 22:23:00 -07:00
Casey Rodarmor
379a001f47
Fix warnings and clippy errors
type: reform
2021-07-02 21:02:53 -07:00
29 changed files with 658 additions and 476 deletions

View File

@ -39,7 +39,7 @@ jobs:
runs-on: ${{matrix.os}}
env:
RUSTFLAGS: "-D warnings"
RUSTFLAGS: --deny warnings
steps:
- uses: actions/checkout@v2
@ -63,14 +63,9 @@ jobs:
target
key: cargo-${{ env.CACHE_KEY }}-${{ runner.os }}-${{ hashFiles('**/Cargo.lock') }}
- name: Install Stable
uses: actions-rs/toolchain@v1
with:
toolchain: stable
target: ${{matrix.target}}
profile: minimal
components: clippy, rustfmt
override: true
- name: Install Rust Toolchain Components
run: |
rustup component add clippy rustfmt
- name: Info
run: |

657
Cargo.lock generated

File diff suppressed because it is too large Load Diff

View File

@ -26,16 +26,13 @@ case $os in
;;
esac
case $os in
ubuntu-latest | macos-latest)
cargo rustc --bin $bin --target $target --release
executable=target/$target/release/$bin
;;
windows-latest)
cargo rustc --bin $bin --target $target --release -- -C target-feature="+crt-static"
executable=target/$target/release/$bin.exe
;;
esac
RUSTFLAGS='--deny warnings --codegen target-feature=+crt-static' \
cargo build --bin $bin --target $target --release
executable=target/$target/release/$bin
if [[ $os == windows-2016 ]]; then
executable=$executable.exe
fi
echo "Building completions..."
cargo run --package gen -- --bin $executable completion-scripts

1
rust-toolchain Normal file
View File

@ -0,0 +1 @@
1.51.0

View File

@ -7,9 +7,9 @@ const TI: u64 = GI << 10;
const PI: u64 = TI << 10;
const EI: u64 = PI << 10;
#[serde(transparent)]
#[derive(Debug, PartialEq, Copy, Clone, PartialOrd, Ord, Eq, Serialize, Deserialize, Default)]
pub(crate) struct Bytes(pub(crate) u64);
#[serde(transparent)]
pub struct Bytes(pub(crate) u64);
impl Bytes {
pub(crate) fn kib() -> Self {
@ -89,7 +89,7 @@ impl FromStr for Bytes {
_ => {
return Err(Error::ByteSuffix {
text: text.to_owned(),
suffix: suffix.to_owned(),
suffix: suffix.clone(),
})
}
};

View File

@ -1,6 +1,6 @@
use crate::common::*;
pub(crate) struct Env {
pub struct Env {
args: Vec<OsString>,
dir: PathBuf,
input: Box<dyn InputStream>,
@ -41,7 +41,7 @@ impl Env {
.and_then(|width| width.parse::<usize>().ok());
if let Some(width) = width {
app = app.set_term_width(width)
app = app.set_term_width(width);
}
app

View File

@ -2,7 +2,7 @@ use crate::common::*;
#[derive(Debug, Snafu)]
#[snafu(visibility(pub(crate)))]
pub(crate) enum Error {
pub enum Error {
#[snafu(display("Failed to parse announce URL: {}", source))]
AnnounceUrlParse { source: url::ParseError },
#[snafu(display("Failed to parse byte count `{}`: {}", text, source))]

View File

@ -1,7 +1,7 @@
use crate::common::*;
#[serde(transparent)]
#[derive(Deserialize, Serialize, Debug, PartialEq, Clone, Ord, PartialOrd, Eq)]
#[serde(transparent)]
pub(crate) struct FilePath {
components: Vec<String>,
}

View File

@ -1,7 +1,7 @@
use crate::common::*;
#[derive(Debug, PartialEq, Clone)]
pub(crate) struct HostPort {
pub struct HostPort {
host: Host,
port: u16,
}

View File

@ -2,7 +2,7 @@ use crate::common::*;
#[derive(Debug, Snafu)]
#[snafu(visibility(pub(crate)))]
pub(crate) enum HostPortParseError {
pub enum HostPortParseError {
#[snafu(display("Failed to parse host `{}`: {}", text, source))]
Host {
text: String,

View File

@ -1,22 +1,22 @@
use crate::common::*;
#[derive(Deserialize, Serialize, Debug, PartialEq, Clone)]
pub(crate) struct Info {
pub struct Info {
#[serde(
skip_serializing_if = "Option::is_none",
default,
with = "unwrap_or_skip"
)]
pub(crate) private: Option<bool>,
pub private: Option<bool>,
#[serde(rename = "piece length")]
pub(crate) piece_length: Bytes,
pub(crate) name: String,
pub piece_length: Bytes,
pub name: String,
#[serde(
skip_serializing_if = "Option::is_none",
default,
with = "unwrap_or_skip"
)]
pub(crate) source: Option<String>,
pub source: Option<String>,
pub(crate) pieces: PieceList,
#[serde(flatten)]
pub(crate) mode: Mode,
@ -26,7 +26,7 @@ pub(crate) struct Info {
with = "unwrap_or_skip",
rename = "update-url"
)]
pub(crate) update_url: Option<Url>,
pub update_url: Option<Url>,
}
impl Info {

View File

@ -1,12 +1,12 @@
use crate::common::*;
#[derive(Debug, Eq, PartialEq, Copy, Clone)]
pub(crate) struct Infohash {
inner: Sha1Digest,
pub struct Infohash {
pub inner: Sha1Digest,
}
impl Infohash {
pub(crate) fn from_input(input: &Input) -> Result<Infohash, Error> {
pub fn from_input(input: &Input) -> Result<Infohash, Error> {
let value = Value::from_bencode(input.data()).map_err(|error| Error::MetainfoDecode {
input: input.source().clone(),
error,
@ -43,7 +43,7 @@ impl Infohash {
}
}
pub(crate) fn from_bencoded_info_dict(info: &[u8]) -> Infohash {
pub fn from_bencoded_info_dict(info: &[u8]) -> Infohash {
Infohash {
inner: Sha1Digest::from_data(info),
}

View File

@ -1,20 +1,20 @@
use crate::common::*;
pub(crate) struct Input {
pub struct Input {
source: InputTarget,
data: Vec<u8>,
}
impl Input {
pub(crate) fn new(source: InputTarget, data: Vec<u8>) -> Input {
pub fn new(source: InputTarget, data: Vec<u8>) -> Input {
Self { source, data }
}
pub(crate) fn data(&self) -> &[u8] {
pub fn data(&self) -> &[u8] {
&self.data
}
pub(crate) fn source(&self) -> &InputTarget {
pub fn source(&self) -> &InputTarget {
&self.source
}

View File

@ -1,13 +1,13 @@
use crate::common::*;
#[derive(PartialEq, Debug, Clone)]
pub(crate) enum InputTarget {
pub enum InputTarget {
Path(PathBuf),
Stdin,
}
impl InputTarget {
pub(crate) fn try_from_os_str(text: &OsStr) -> Result<Self, OsString> {
pub fn try_from_os_str(text: &OsStr) -> Result<Self, OsString> {
text
.try_into()
.map_err(|err: Error| OsString::from(err.to_string()))

View File

@ -41,85 +41,85 @@
#[cfg(test)]
#[macro_use]
mod assert_matches;
pub mod assert_matches;
#[macro_use]
mod errln;
pub mod errln;
#[macro_use]
mod err;
pub mod err;
#[macro_use]
mod out;
pub mod out;
#[macro_use]
mod outln;
pub mod outln;
#[cfg(test)]
#[macro_use]
mod test_env;
pub mod test_env;
#[cfg(test)]
mod test_env_builder;
pub mod test_env_builder;
#[cfg(test)]
mod capture;
pub mod capture;
mod arguments;
mod bytes;
mod common;
mod consts;
mod env;
mod error;
mod file_error;
mod file_info;
mod file_path;
mod file_status;
mod files;
mod hasher;
mod host_port;
mod host_port_parse_error;
mod info;
mod infohash;
mod input;
mod input_stream;
mod input_target;
mod into_u64;
mod into_usize;
mod invariant;
mod lint;
mod linter;
mod magnet_link;
mod magnet_link_parse_error;
mod md5_digest;
mod metainfo;
mod metainfo_error;
mod mode;
mod options;
mod output_stream;
mod output_target;
mod piece_length_picker;
mod piece_list;
mod platform;
mod platform_interface;
mod print;
mod reckoner;
mod run;
mod sha1_digest;
mod shell;
mod sort_key;
mod sort_order;
mod sort_spec;
mod status;
mod step;
mod style;
mod subcommand;
mod table;
mod torrent_summary;
mod use_color;
mod verifier;
mod walker;
mod xor_args;
pub mod arguments;
pub mod bytes;
pub mod common;
pub mod consts;
pub mod env;
pub mod error;
pub mod file_error;
pub mod file_info;
pub mod file_path;
pub mod file_status;
pub mod files;
pub mod hasher;
pub mod host_port;
pub mod host_port_parse_error;
pub mod info;
pub mod infohash;
pub mod input;
pub mod input_stream;
pub mod input_target;
pub mod into_u64;
pub mod into_usize;
pub mod invariant;
pub mod lint;
pub mod linter;
pub mod magnet_link;
pub mod magnet_link_parse_error;
pub mod md5_digest;
pub mod metainfo;
pub mod metainfo_error;
pub mod mode;
pub mod options;
pub mod output_stream;
pub mod output_target;
pub mod piece_length_picker;
pub mod piece_list;
pub mod platform;
pub mod platform_interface;
pub mod print;
pub mod reckoner;
pub mod run;
pub mod sha1_digest;
pub mod shell;
pub mod sort_key;
pub mod sort_order;
pub mod sort_spec;
pub mod status;
pub mod step;
pub mod style;
pub mod subcommand;
pub mod table;
pub mod torrent_summary;
pub mod use_color;
pub mod verifier;
pub mod walker;
pub mod xor_args;
#[cfg(feature = "bench")]
pub mod bench;

View File

@ -12,7 +12,7 @@ impl Linter {
}
pub(crate) fn allow(&mut self, allowed: impl IntoIterator<Item = Lint>) {
self.allowed.extend(allowed)
self.allowed.extend(allowed);
}
pub(crate) fn is_allowed(&self, lint: Lint) -> bool {

View File

@ -1,17 +1,17 @@
use crate::common::*;
#[derive(Debug, PartialEq)]
pub(crate) struct MagnetLink {
infohash: Infohash,
name: Option<String>,
peers: Vec<HostPort>,
trackers: Vec<Url>,
indices: BTreeSet<u64>,
pub struct MagnetLink {
pub infohash: Infohash,
pub name: Option<String>,
pub peers: Vec<HostPort>,
pub trackers: Vec<Url>,
pub indices: BTreeSet<u64>,
}
impl MagnetLink {
/// See `Info::infohash_lossy` for details on when this function is lossy.
pub(crate) fn from_metainfo_lossy(metainfo: &Metainfo) -> Result<MagnetLink> {
pub fn from_metainfo_lossy(metainfo: &Metainfo) -> Result<MagnetLink> {
let mut link = Self::with_infohash(metainfo.infohash_lossy()?);
link.set_name(metainfo.info.name.clone());
@ -23,7 +23,7 @@ impl MagnetLink {
Ok(link)
}
pub(crate) fn with_infohash(infohash: Infohash) -> Self {
pub fn with_infohash(infohash: Infohash) -> Self {
MagnetLink {
infohash,
name: None,
@ -34,23 +34,23 @@ impl MagnetLink {
}
#[allow(dead_code)]
pub(crate) fn set_name(&mut self, name: impl Into<String>) {
pub fn set_name(&mut self, name: impl Into<String>) {
self.name = Some(name.into());
}
pub(crate) fn add_peer(&mut self, peer: HostPort) {
pub fn add_peer(&mut self, peer: HostPort) {
self.peers.push(peer);
}
pub(crate) fn add_tracker(&mut self, tracker: Url) {
pub fn add_tracker(&mut self, tracker: Url) {
self.trackers.push(tracker);
}
pub(crate) fn add_index(&mut self, index: u64) {
pub fn add_index(&mut self, index: u64) {
self.indices.insert(index);
}
pub(crate) fn to_url(&self) -> Url {
pub fn to_url(&self) -> Url {
let mut url = Url::parse("magnet:").invariant_unwrap("`magnet:` is valid URL");
let mut query = format!("xt=urn:btih:{}", self.infohash);
@ -85,7 +85,7 @@ impl MagnetLink {
url
}
fn parse(text: &str) -> Result<Self, MagnetLinkParseError> {
pub fn parse(text: &str) -> Result<Self, MagnetLinkParseError> {
let url = Url::parse(&text).context(magnet_link_parse_error::Url)?;
if url.scheme() != "magnet" {

View File

@ -2,7 +2,7 @@ use crate::common::*;
#[derive(Debug, Snafu)]
#[snafu(visibility(pub(crate)))]
pub(crate) enum MagnetLinkParseError {
pub enum MagnetLinkParseError {
#[snafu(display("Failed to parse hex string `{}`: {}", text, source))]
HexParse {
text: String,

View File

@ -1,7 +1,7 @@
use crate::common::*;
#[serde(transparent)]
#[derive(Deserialize, Serialize, Debug, Eq, PartialEq, Copy, Clone)]
#[serde(transparent)]
pub(crate) struct Md5Digest {
#[serde(with = "SerHex::<serde_hex::Strict>")]
bytes: [u8; 16],

View File

@ -1,68 +1,68 @@
use crate::common::*;
#[derive(Deserialize, Serialize, Debug, PartialEq, Clone)]
pub(crate) struct Metainfo {
pub struct Metainfo {
#[serde(
skip_serializing_if = "Option::is_none",
default,
with = "unwrap_or_skip"
)]
pub(crate) announce: Option<String>,
pub announce: Option<String>,
#[serde(
rename = "announce-list",
skip_serializing_if = "Option::is_none",
default,
with = "unwrap_or_skip"
)]
pub(crate) announce_list: Option<Vec<Vec<String>>>,
pub announce_list: Option<Vec<Vec<String>>>,
#[serde(
skip_serializing_if = "Option::is_none",
default,
with = "unwrap_or_skip"
)]
pub(crate) comment: Option<String>,
pub comment: Option<String>,
#[serde(
rename = "created by",
skip_serializing_if = "Option::is_none",
default,
with = "unwrap_or_skip"
)]
pub(crate) created_by: Option<String>,
pub created_by: Option<String>,
#[serde(
rename = "creation date",
skip_serializing_if = "Option::is_none",
default,
with = "unwrap_or_skip"
)]
pub(crate) creation_date: Option<u64>,
pub creation_date: Option<u64>,
#[serde(
skip_serializing_if = "Option::is_none",
default,
with = "unwrap_or_skip"
)]
pub(crate) encoding: Option<String>,
pub(crate) info: Info,
pub encoding: Option<String>,
pub info: Info,
#[serde(
skip_serializing_if = "Option::is_none",
default,
with = "unwrap_or_skip"
)]
pub(crate) nodes: Option<Vec<HostPort>>,
pub nodes: Option<Vec<HostPort>>,
}
impl Metainfo {
pub(crate) fn from_input(input: &Input) -> Result<Metainfo> {
pub fn from_input(input: &Input) -> Result<Metainfo> {
Self::deserialize(input.source(), input.data())
}
pub(crate) fn deserialize(source: &InputTarget, data: &[u8]) -> Result<Metainfo, Error> {
pub fn deserialize(source: &InputTarget, data: &[u8]) -> Result<Metainfo, Error> {
let metainfo = bendy::serde::de::from_bytes(&data).context(error::MetainfoDeserialize {
input: source.clone(),
})?;
Ok(metainfo)
}
pub(crate) fn serialize(&self) -> Result<Vec<u8>, Error> {
pub fn serialize(&self) -> Result<Vec<u8>, Error> {
bendy::serde::ser::to_bytes(&self).context(error::MetainfoSerialize)
}
@ -99,7 +99,7 @@ impl Metainfo {
self.info.content_size()
}
pub(crate) fn trackers<'a>(&'a self) -> impl Iterator<Item = Result<Url>> + 'a {
pub fn trackers<'a>(&'a self) -> impl Iterator<Item = Result<Url>> + 'a {
let mut seen = HashSet::new();
iter::once(&self.announce)
.flatten()

View File

@ -1,7 +1,7 @@
use crate::common::*;
#[derive(Debug, Copy, Clone)]
pub(crate) enum MetainfoError {
pub enum MetainfoError {
Type,
InfoMissing,
InfoType,

View File

@ -1,7 +1,7 @@
use crate::common::*;
#[serde(untagged)]
#[derive(Deserialize, Serialize, Debug, PartialEq, Clone)]
#[serde(untagged)]
pub(crate) enum Mode {
Single {
length: Bytes,

View File

@ -1,10 +1,10 @@
use crate::common::*;
pub(crate) struct OutputStream {
active: bool,
stream: Box<dyn Write>,
style: bool,
term: bool,
active: bool,
}
impl OutputStream {

View File

@ -1,22 +1,22 @@
use crate::common::*;
#[derive(Debug, Eq, PartialEq, Copy, Clone)]
pub(crate) struct Sha1Digest {
bytes: [u8; Self::LENGTH],
pub struct Sha1Digest {
pub bytes: [u8; Self::LENGTH],
}
impl Sha1Digest {
pub(crate) const LENGTH: usize = 20;
pub const LENGTH: usize = 20;
pub(crate) fn from_bytes(bytes: [u8; Self::LENGTH]) -> Self {
pub fn from_bytes(bytes: [u8; Self::LENGTH]) -> Self {
Sha1Digest { bytes }
}
pub(crate) fn bytes(self) -> [u8; Self::LENGTH] {
pub fn bytes(self) -> [u8; Self::LENGTH] {
self.bytes
}
pub(crate) fn from_data(data: impl AsRef<[u8]>) -> Self {
pub fn from_data(data: impl AsRef<[u8]>) -> Self {
Sha1::from(data).digest().into()
}
}

View File

@ -4,7 +4,7 @@ use structopt::clap;
#[derive(Copy, Clone, EnumVariantNames, IntoStaticStr, EnumString, EnumIter, Debug)]
#[strum(serialize_all = "kebab-case")]
pub(crate) enum Shell {
pub enum Shell {
Zsh,
Bash,
Fish,
@ -62,6 +62,6 @@ mod tests {
#[test]
fn variants() {
assert_eq!(Shell::VARIANTS, clap::Shell::variants())
assert_eq!(Shell::VARIANTS, clap::Shell::variants());
}
}

View File

@ -338,9 +338,7 @@ impl Create {
if let OutputTarget::Path(path) = &output {
if !self.force && path.exists() {
return Err(Error::OutputExists {
path: path.to_owned(),
});
return Err(Error::OutputExists { path: path.clone() });
}
}
@ -1151,7 +1149,7 @@ mod tests {
length: Bytes(3),
md5sum: None,
}
)
);
}
#[test]
@ -1182,7 +1180,7 @@ mod tests {
length: Bytes(4),
md5sum: None,
}
)
);
}
#[test]
@ -1213,7 +1211,7 @@ mod tests {
length: Bytes(4),
md5sum: None,
}
)
);
}
#[test]
@ -1258,7 +1256,7 @@ mod tests {
},
],
}
)
);
}
#[test]
@ -1285,7 +1283,7 @@ mod tests {
length: Bytes(3),
md5sum: None,
}
)
);
}
#[test]
@ -1319,7 +1317,7 @@ mod tests {
length: Bytes(3),
md5sum: None,
}
)
);
}
#[test]
@ -1346,7 +1344,7 @@ mod tests {
length: Bytes(0),
md5sum: None,
}
)
);
}
#[test]
@ -1367,7 +1365,7 @@ mod tests {
env.assert_ok();
let metainfo = env.load_metainfo("foo.torrent");
assert_eq!(metainfo.info.pieces.count(), 0);
assert_eq!(metainfo.info.mode, Mode::Multiple { files: Vec::new() })
assert_eq!(metainfo.info.mode, Mode::Multiple { files: Vec::new() });
}
#[test]

View File

@ -16,7 +16,19 @@ const INPUT_FLAG: &str = "input-flag";
version_message(consts::VERSION_MESSAGE),
about("Verify files against a .torrent file.")
)]
#[cfg_attr(test, structopt(setting = AppSettings::ColorNever))]
pub(crate) struct Verify {
#[structopt(
long = "base-directory",
short = "b",
value_name = "BASE-DIRECTORY",
conflicts_with = "content",
empty_values(false),
parse(from_os_str),
help = "Look for torrent content in `BASE-DIRECTORY`/`NAME`, where `NAME` is the `name` field \
of the torrent info dictionary."
)]
base_directory: Option<PathBuf>,
#[structopt(
long = "content",
short = "c",
@ -24,7 +36,7 @@ pub(crate) struct Verify {
empty_values(false),
parse(from_os_str),
help = "Verify torrent content at `PATH` against torrent metainfo. Defaults to `name` field \
of torrent info dictionary."
of the torrent info dictionary."
)]
content: Option<PathBuf>,
#[structopt(
@ -64,13 +76,19 @@ impl Verify {
let metainfo = Metainfo::from_input(&input)?;
let content = self.content.as_ref().map_or_else(
|| match target {
let content = self
.content
.as_ref()
.cloned()
.or_else(|| {
self
.base_directory
.map(|base_directory| base_directory.join(&metainfo.info.name).lexiclean())
})
.unwrap_or_else(|| match target {
InputTarget::Path(path) => path.join("..").join(&metainfo.info.name).lexiclean(),
InputTarget::Stdin => PathBuf::from(&metainfo.info.name),
},
PathBuf::clone,
);
});
let progress_bar = if env.err().is_styled_term() && !options.quiet {
let style = ProgressStyle::default_bar()
@ -137,6 +155,20 @@ mod tests {
assert_matches!(env.run(), Err(Error::Clap { .. }));
}
#[test]
fn base_directory_conflicts_with_content() {
let mut env = test_env! {
args: ["torrent", "verify", "foo.torrent", "--content", "foo", "--base-directory", "dir"],
tree: {},
};
assert_eq!(
env.run().unwrap_err().to_string(),
"error: The argument '--content <PATH>' cannot be used with '--base-directory \
<BASE-DIRECTORY>'\n\nUSAGE:\n imdl torrent verify <INPUT> --base-directory \
<BASE-DIRECTORY> --content <PATH>\n\nFor more information try --help\n"
);
}
#[test]
fn pass() -> Result<()> {
let mut create_env = test_env! {
@ -323,6 +355,62 @@ mod tests {
Ok(())
}
#[test]
fn base_directory() -> Result<()> {
let mut create_env = test_env! {
args: [
"torrent",
"create",
"--input",
"foo",
"--announce",
"https://bar",
],
tree: {
foo: {
a: "abc",
d: "efg",
h: "ijk",
},
},
};
create_env.assert_ok();
create_env.create_dir("dir");
create_env.rename("foo", "dir/foo");
let torrent = create_env.resolve("foo.torrent")?;
let dir = create_env.resolve("dir")?;
let mut verify_env = test_env! {
args: [
"torrent",
"verify",
"--input",
&torrent,
"--base-directory",
&dir,
],
tree: {},
};
verify_env.assert_ok();
let want = format!(
"[1/2] \u{1F4BE} Loading metainfo from `{}`…\n[2/2] \u{1F9EE} Verifying pieces from \
`{}`\n\u{2728}\u{2728} Verification succeeded! \u{2728}\u{2728}\n",
torrent.display(),
dir.join("foo").display(),
);
assert_eq!(verify_env.err(), want);
assert_eq!(verify_env.out(), "");
Ok(())
}
#[test]
fn verify_stdin() -> Result<()> {
let mut create_env = test_env! {

View File

@ -37,19 +37,19 @@ macro_rules! test_env {
pub(crate) struct TestEnv {
env: Env,
#[allow(unused)]
tempdir: TempDir,
err: Capture,
out: Capture,
#[allow(unused)]
tempdir: TempDir,
}
impl TestEnv {
pub(crate) fn new(tempdir: TempDir, env: Env, err: Capture, out: Capture) -> TestEnv {
Self {
tempdir,
err,
env,
err,
out,
tempdir,
}
}
@ -77,6 +77,14 @@ impl TestEnv {
fs::create_dir(self.env.resolve(path).unwrap()).unwrap();
}
pub(crate) fn rename(&self, from: impl AsRef<Path>, to: impl AsRef<Path>) {
fs::rename(
self.env.resolve(from).unwrap(),
self.env.resolve(to).unwrap(),
)
.unwrap();
}
pub(crate) fn read_to_string(&self, path: impl AsRef<Path>) -> String {
fs::read_to_string(self.env.resolve(path).unwrap()).unwrap()
}

View File

@ -1,30 +1,30 @@
use crate::common::*;
pub(crate) struct TorrentSummary {
pub struct TorrentSummary {
infohash: Infohash,
metainfo: Metainfo,
size: Bytes,
}
#[derive(Serialize)]
pub(crate) struct TorrentSummaryJson {
name: String,
comment: Option<String>,
creation_date: Option<u64>,
created_by: Option<String>,
source: Option<String>,
info_hash: String,
torrent_size: u64,
content_size: u64,
private: bool,
tracker: Option<String>,
announce_list: Vec<Vec<String>>,
update_url: Option<String>,
dht_nodes: Vec<String>,
piece_size: u64,
piece_count: usize,
file_count: usize,
files: Vec<String>,
pub struct TorrentSummaryJson {
pub name: String,
pub comment: Option<String>,
pub creation_date: Option<u64>,
pub created_by: Option<String>,
pub source: Option<String>,
pub info_hash: String,
pub torrent_size: u64,
pub content_size: u64,
pub private: bool,
pub tracker: Option<String>,
pub announce_list: Vec<Vec<String>>,
pub update_url: Option<String>,
pub dht_nodes: Vec<String>,
pub piece_size: u64,
pub piece_count: usize,
pub file_count: usize,
pub files: Vec<String>,
}
impl TorrentSummary {
@ -37,14 +37,14 @@ impl TorrentSummary {
}
/// See `Info::infohash_lossy` for details on when this function is lossy.
pub(crate) fn from_metainfo_lossy(metainfo: Metainfo) -> Result<Self> {
pub fn from_metainfo_lossy(metainfo: Metainfo) -> Result<Self> {
let bytes = metainfo.serialize()?;
let size = Bytes(bytes.len().into_u64());
let infohash = metainfo.infohash_lossy()?;
Ok(Self::new(metainfo, infohash, size))
}
pub(crate) fn from_input(input: &Input) -> Result<Self> {
pub fn from_input(input: &Input) -> Result<Self> {
let metainfo = Metainfo::from_input(input)?;
let infohash = Infohash::from_input(input)?;
let size = Bytes(input.data().len().into_u64());
@ -52,7 +52,7 @@ impl TorrentSummary {
Ok(Self::new(metainfo, infohash, size))
}
pub(crate) fn write(&self, env: &mut Env) -> Result<()> {
pub fn write(&self, env: &mut Env) -> Result<()> {
let table = self.table();
if env.out().is_term() {
@ -168,16 +168,16 @@ impl TorrentSummary {
table
}
pub(crate) fn write_json(&self, env: &mut Env) -> Result<()> {
pub fn write_json(&self, env: &mut Env) -> Result<()> {
let data = self.torrent_summary_data();
let json = serde_json::to_string(&data).context(error::JsonSerialize)?;
outln!(env, "{}", json)?;
Ok(())
}
fn torrent_summary_data(&self) -> TorrentSummaryJson {
pub fn torrent_summary_data(&self) -> TorrentSummaryJson {
let (file_count, files) = match &self.metainfo.info.mode {
Mode::Single { .. } => (1, vec![self.metainfo.info.name.to_owned()]),
Mode::Single { .. } => (1, vec![self.metainfo.info.name.clone()]),
Mode::Multiple { files } => (
files.len(),
files
@ -197,7 +197,7 @@ impl TorrentSummary {
};
TorrentSummaryJson {
name: self.metainfo.info.name.to_owned(),
name: self.metainfo.info.name.clone(),
comment: self.metainfo.comment.clone(),
creation_date: self.metainfo.creation_date,
created_by: self.metainfo.created_by.clone(),