mirror of
https://github.com/lifegpc/pixiv_downloader.git
synced 2026-06-06 05:49:01 +08:00
227 lines
9.4 KiB
Rust
227 lines
9.4 KiB
Rust
#[cfg(any(feature = "avdict", feature = "exif", feature = "ugoira"))]
|
|
use std::env;
|
|
#[cfg(any(feature = "avdict", feature = "exif", feature = "ugoira"))]
|
|
use std::fs::create_dir;
|
|
#[cfg(any(feature = "avdict", feature = "exif", feature = "ugoira"))]
|
|
use std::fs::File;
|
|
#[cfg(any(feature = "avdict", feature = "exif", feature = "ugoira"))]
|
|
use std::io::Read;
|
|
#[cfg(any(feature = "avdict", feature = "exif", feature = "ugoira"))]
|
|
use std::path::PathBuf;
|
|
|
|
fn main() {
|
|
#[cfg(windows)]
|
|
{
|
|
let stack_size = std::env::var("STACK_SIZE").unwrap_or("4194304".to_string());
|
|
let stack_size = parse_size::parse_size(stack_size).unwrap();
|
|
println!("cargo:rerun-if-env-changed=STACK_SIZE");
|
|
#[cfg(target_env = "msvc")]
|
|
println!("cargo:rustc-link-arg=/STACK:{}", stack_size);
|
|
#[cfg(target_env = "gnu")]
|
|
println!("cargo:rustc-link-arg=-Wl,--stack,{}", stack_size);
|
|
}
|
|
#[cfg(any(feature = "exif", feature = "ugoira"))]
|
|
{
|
|
let out_path = PathBuf::from(env::var("OUT_DIR").unwrap());
|
|
let utils_build_path = out_path.join("utils");
|
|
if !utils_build_path.exists() {
|
|
create_dir(&utils_build_path).unwrap();
|
|
}
|
|
let mut config = cmake::Config::new("utils");
|
|
config
|
|
.define("CMAKE_INSTALL_PREFIX", out_path.to_str().unwrap())
|
|
.out_dir(utils_build_path)
|
|
.define("ENABLE_ICONV", "OFF")
|
|
.define("ENABLE_STANDALONE", "ON")
|
|
.define("ENABLE_CXX17", "ON")
|
|
.define("INSTALL_DEP_FILES", "ON");
|
|
#[cfg(all(windows, target_env = "msvc"))]
|
|
config
|
|
.define("CMAKE_BUILD_TYPE", "Release")
|
|
.generator("Ninja");
|
|
config.build();
|
|
let dep_path = out_path.join("utils_dep.txt");
|
|
let mut f = File::open(dep_path).unwrap();
|
|
let mut s = String::from("");
|
|
f.read_to_string(&mut s).unwrap();
|
|
let l: Vec<&str> = s.split(";").collect();
|
|
for i in l.iter() {
|
|
let mut p = PathBuf::from(i);
|
|
let p2 = p.clone();
|
|
let file_name = p2.file_stem().unwrap();
|
|
let file_name = file_name.to_str().unwrap();
|
|
let file_name = file_name.trim_start_matches("lib");
|
|
p.pop();
|
|
let pa = p.to_str().unwrap();
|
|
if pa != "" {
|
|
println!("cargo:rustc-link-search={}", pa);
|
|
}
|
|
println!("cargo:rustc-link-lib={}", file_name);
|
|
}
|
|
println!("cargo:rerun-if-changed=utils/");
|
|
}
|
|
#[cfg(feature = "avdict")]
|
|
{
|
|
let out_path = PathBuf::from(env::var("OUT_DIR").unwrap());
|
|
let avdict_build_path = out_path.join("avdict");
|
|
if !avdict_build_path.exists() {
|
|
create_dir(&avdict_build_path).unwrap();
|
|
}
|
|
let mut config = cmake::Config::new("avdict");
|
|
config
|
|
.define("CMAKE_INSTALL_PREFIX", out_path.to_str().unwrap())
|
|
.out_dir(avdict_build_path);
|
|
#[cfg(all(windows, target_env = "msvc"))]
|
|
config
|
|
.define("CMAKE_BUILD_TYPE", "Release")
|
|
.generator("Ninja");
|
|
config.build();
|
|
println!("cargo:rustc-link-search=native={}/lib", out_path.display());
|
|
let dep_path = out_path.join("avdict_dep.txt");
|
|
let mut f = File::open(dep_path).unwrap();
|
|
let mut s = String::from("");
|
|
f.read_to_string(&mut s).unwrap();
|
|
println!("cargo:rustc-link-lib=avdict");
|
|
let l: Vec<&str> = s.split(";").collect();
|
|
for i in l.iter() {
|
|
let mut p = PathBuf::from(i);
|
|
let p2 = p.clone();
|
|
let file_name = p2.file_stem().unwrap();
|
|
let file_name = file_name.to_str().unwrap();
|
|
let file_name = file_name.trim_start_matches("lib");
|
|
p.pop();
|
|
println!("cargo:rustc-link-search={}", p.to_str().unwrap());
|
|
println!("cargo:rustc-link-lib={}", file_name);
|
|
}
|
|
println!("cargo:rerun-if-changed=avdict/");
|
|
let bindings = bindgen::Builder::default()
|
|
// The input header we would like to generate
|
|
// bindings for.
|
|
.header("avdict/avdict.h")
|
|
// Tell cargo to invalidate the built crate whenever any of the
|
|
// included header files changed.
|
|
.parse_callbacks(Box::new(bindgen::CargoCallbacks))
|
|
// Finish the builder and generate the bindings.
|
|
.generate()
|
|
// Unwrap the Result and panic on failure.
|
|
.expect("Unable to generate bindings");
|
|
|
|
bindings
|
|
.write_to_file(out_path.join("avdict.rs"))
|
|
.expect("Couldn't write bindings!");
|
|
}
|
|
#[cfg(feature = "exif")]
|
|
{
|
|
let out_path = PathBuf::from(env::var("OUT_DIR").unwrap());
|
|
let exif_build_path = out_path.join("exif");
|
|
let utils_build_path = out_path.join("utils");
|
|
if !exif_build_path.exists() {
|
|
create_dir(&exif_build_path).unwrap();
|
|
}
|
|
let mut config = cmake::Config::new("exif");
|
|
config
|
|
.define("CMAKE_INSTALL_PREFIX", out_path.to_str().unwrap())
|
|
.out_dir(exif_build_path)
|
|
.define("UTILS_LIBRARY", utils_build_path.to_str().unwrap());
|
|
#[cfg(all(windows, target_env = "msvc"))]
|
|
config
|
|
.define("CMAKE_BUILD_TYPE", "Release")
|
|
.generator("Ninja");
|
|
config.build();
|
|
println!("cargo:rustc-link-search=native={}/lib", out_path.display());
|
|
let dep_path = out_path.join("exif_dep.txt");
|
|
let mut f = File::open(dep_path).unwrap();
|
|
let mut s = String::from("");
|
|
f.read_to_string(&mut s).unwrap();
|
|
println!("cargo:rustc-link-lib=exif");
|
|
let l: Vec<&str> = s.split(";").collect();
|
|
for i in l.iter() {
|
|
let mut p = PathBuf::from(i);
|
|
let p2 = p.clone();
|
|
let file_name = p2.file_stem().unwrap();
|
|
let file_name = file_name.to_str().unwrap();
|
|
let file_name = file_name.trim_start_matches("lib");
|
|
p.pop();
|
|
println!("cargo:rustc-link-search={}", p.to_str().unwrap());
|
|
println!("cargo:rustc-link-lib={}", file_name);
|
|
}
|
|
println!("cargo:rerun-if-changed=exif/");
|
|
let bindings = bindgen::Builder::default()
|
|
// The input header we would like to generate
|
|
// bindings for.
|
|
.header("exif/exif.h")
|
|
// Tell cargo to invalidate the built crate whenever any of the
|
|
// included header files changed.
|
|
.parse_callbacks(Box::new(bindgen::CargoCallbacks))
|
|
.no_copy("ExifDataRef")
|
|
.no_copy("ExifDatumRef")
|
|
.no_copy("ExifValueRef")
|
|
// Finish the builder and generate the bindings.
|
|
.generate()
|
|
// Unwrap the Result and panic on failure.
|
|
.expect("Unable to generate bindings");
|
|
|
|
bindings
|
|
.write_to_file(out_path.join("exif.rs"))
|
|
.expect("Couldn't write bindings!");
|
|
}
|
|
#[cfg(feature = "ugoira")]
|
|
{
|
|
let out_path = PathBuf::from(env::var("OUT_DIR").unwrap());
|
|
let ugoira_build_path = out_path.join("ugoira");
|
|
let utils_build_path = out_path.join("utils");
|
|
if !ugoira_build_path.exists() {
|
|
create_dir(&ugoira_build_path).unwrap();
|
|
}
|
|
let install_bin_dir = out_path.join("../../../");
|
|
let mut config = cmake::Config::new("ugoira");
|
|
config
|
|
.define("CMAKE_INSTALL_PREFIX", out_path.to_str().unwrap())
|
|
.out_dir(ugoira_build_path)
|
|
.define("UTILS_LIBRARY", utils_build_path.to_str().unwrap())
|
|
.define("CMAKE_INSTALL_BINDIR", install_bin_dir.to_str().unwrap());
|
|
#[cfg(all(windows, target_env = "msvc"))]
|
|
config
|
|
.define("CMAKE_BUILD_TYPE", "Release")
|
|
.generator("Ninja");
|
|
config.build();
|
|
println!("cargo:rustc-link-search=native={}/lib", out_path.display());
|
|
let dep_path = out_path.join("ugoira_dep.txt");
|
|
let mut f = File::open(dep_path).unwrap();
|
|
let mut s = String::from("");
|
|
f.read_to_string(&mut s).unwrap();
|
|
println!("cargo:rustc-link-lib=ugoira");
|
|
let l: Vec<&str> = s.split(";").collect();
|
|
for i in l.iter() {
|
|
let mut p = PathBuf::from(i);
|
|
let p2 = p.clone();
|
|
let file_name = p2.file_stem().unwrap();
|
|
let file_name = file_name.to_str().unwrap();
|
|
let file_name = file_name.trim_start_matches("lib");
|
|
p.pop();
|
|
println!("cargo:rustc-link-search={}", p.to_str().unwrap());
|
|
println!("cargo:rustc-link-lib={}", file_name);
|
|
}
|
|
println!("cargo:rerun-if-changed=ugoira/");
|
|
let bindings = bindgen::Builder::default()
|
|
// The input header we would like to generate
|
|
// bindings for.
|
|
.header("ugoira/ugoira.h")
|
|
// Tell cargo to invalidate the built crate whenever any of the
|
|
// included header files changed.
|
|
.parse_callbacks(Box::new(bindgen::CargoCallbacks))
|
|
// Finish the builder and generate the bindings.
|
|
.generate()
|
|
// Unwrap the Result and panic on failure.
|
|
.expect("Unable to generate bindings");
|
|
|
|
bindings
|
|
.write_to_file(out_path.join("ugoira.rs"))
|
|
.expect("Couldn't write bindings!");
|
|
}
|
|
#[cfg(any(feature = "exif", feature = "ugoira"))]
|
|
{
|
|
println!("cargo:rustc-link-lib=utils");
|
|
}
|
|
}
|