added formatters to flake and formatted
This commit is contained in:
Benton Edmondson
parent
9114ef63c1
commit
398ef79341
4 changed files with 288 additions and 240 deletions
327
flake.nix
327
flake.nix
|
|
@ -15,151 +15,190 @@
|
||||||
};
|
};
|
||||||
pugixml-src = {
|
pugixml-src = {
|
||||||
url = "github:zeux/pugixml/latest";
|
url = "github:zeux/pugixml/latest";
|
||||||
flake = false;
|
flake = false;
|
||||||
};
|
};
|
||||||
};
|
};
|
||||||
|
|
||||||
outputs = flakes: let
|
outputs = flakes:
|
||||||
version = "1.3.0";
|
let
|
||||||
self = flakes.self.packages.x86_64-linux;
|
version = "1.3.0";
|
||||||
nixpkgs = flakes.nixpkgs.legacyPackages.x86_64-linux.pkgsStatic;
|
self = flakes.self.packages.x86_64-linux;
|
||||||
nixpkgs-dyn = flakes.nixpkgs.legacyPackages.x86_64-linux;
|
nixpkgs = flakes.nixpkgs.legacyPackages.x86_64-linux.pkgsStatic;
|
||||||
gourou-src = flakes.gourou-src;
|
nixpkgs-dyn = flakes.nixpkgs.legacyPackages.x86_64-linux;
|
||||||
updfparser-src = flakes.updfparser-src;
|
nixpkgs-fmt = flakes.nixpkgs-fmt.defaultPackage.x86_64-linux;
|
||||||
base64-src = flakes.base64-src;
|
gourou-src = flakes.gourou-src;
|
||||||
pugixml-src = flakes.pugixml-src;
|
updfparser-src = flakes.updfparser-src;
|
||||||
cxx = "${nixpkgs.stdenv.cc}/bin/x86_64-unknown-linux-musl-g++";
|
base64-src = flakes.base64-src;
|
||||||
ar = "${nixpkgs.stdenv.cc.bintools.bintools_bin}/bin/x86_64-unknown-linux-musl-ar";
|
pugixml-src = flakes.pugixml-src;
|
||||||
obj-flags = "-O2 -static";
|
cxx = "${nixpkgs.stdenv.cc}/bin/x86_64-unknown-linux-musl-g++";
|
||||||
in rec {
|
ar = "${nixpkgs.stdenv.cc.bintools.bintools_bin}/bin/x86_64-unknown-linux-musl-ar";
|
||||||
packages.x86_64-linux.libzip-static = nixpkgs.libzip.overrideAttrs (prev: {
|
obj-flags = "-O2 -static";
|
||||||
cmakeFlags = (prev.cmakeFlags or []) ++ [
|
in
|
||||||
"-DBUILD_SHARED_LIBS=OFF"
|
rec {
|
||||||
"-DBUILD_EXAMPLES=OFF"
|
packages.x86_64-linux.libzip-static = nixpkgs.libzip.overrideAttrs (prev: {
|
||||||
"-DBUILD_DOC=OFF"
|
cmakeFlags = (prev.cmakeFlags or [ ]) ++ [
|
||||||
"-DBUILD_TOOLS=OFF"
|
"-DBUILD_SHARED_LIBS=OFF"
|
||||||
"-DBUILD_REGRESS=OFF"
|
"-DBUILD_EXAMPLES=OFF"
|
||||||
];
|
"-DBUILD_DOC=OFF"
|
||||||
outputs = ["out"];
|
"-DBUILD_TOOLS=OFF"
|
||||||
});
|
"-DBUILD_REGRESS=OFF"
|
||||||
packages.x86_64-linux.base64 = derivation {
|
];
|
||||||
name = "updfparser";
|
outputs = [ "out" ];
|
||||||
system = "x86_64-linux";
|
});
|
||||||
builder = "${nixpkgs.bash}/bin/bash";
|
packages.x86_64-linux.base64 = derivation {
|
||||||
PATH = "${nixpkgs.coreutils}/bin";
|
name = "updfparser";
|
||||||
args = ["-c" ''
|
system = "x86_64-linux";
|
||||||
mkdir -p $out/include/base64
|
builder = "${nixpkgs.bash}/bin/bash";
|
||||||
cp ${base64-src}/Base64.h $out/include/base64/Base64.h
|
PATH = "${nixpkgs.coreutils}/bin";
|
||||||
''];
|
args = [
|
||||||
|
"-c"
|
||||||
|
''
|
||||||
|
mkdir -p $out/include/base64
|
||||||
|
cp ${base64-src}/Base64.h $out/include/base64/Base64.h
|
||||||
|
''
|
||||||
|
];
|
||||||
|
};
|
||||||
|
packages.x86_64-linux.updfparser = derivation {
|
||||||
|
name = "updfparser";
|
||||||
|
system = "x86_64-linux";
|
||||||
|
builder = "${nixpkgs.bash}/bin/bash";
|
||||||
|
PATH = "${nixpkgs.coreutils}/bin";
|
||||||
|
args = [
|
||||||
|
"-c"
|
||||||
|
''
|
||||||
|
${cxx} \
|
||||||
|
-c ${updfparser-src}/src/*.cpp \
|
||||||
|
-I ${updfparser-src}/include \
|
||||||
|
${obj-flags}
|
||||||
|
mkdir -p $out/lib
|
||||||
|
${ar} crs $out/lib/libupdfparser.a *.o
|
||||||
|
''
|
||||||
|
];
|
||||||
|
};
|
||||||
|
packages.x86_64-linux.gourou = derivation {
|
||||||
|
name = "gourou";
|
||||||
|
system = "x86_64-linux";
|
||||||
|
builder = "${nixpkgs.bash}/bin/bash";
|
||||||
|
PATH = "${nixpkgs.coreutils}/bin";
|
||||||
|
args = [
|
||||||
|
"-c"
|
||||||
|
''
|
||||||
|
${cxx} \
|
||||||
|
-c \
|
||||||
|
${gourou-src}/src/*.cpp \
|
||||||
|
${pugixml-src}/src/pugixml.cpp \
|
||||||
|
-I ${self.base64}/include \
|
||||||
|
-I ${gourou-src}/include \
|
||||||
|
-I ${pugixml-src}/src \
|
||||||
|
-I ${updfparser-src}/include \
|
||||||
|
${obj-flags}
|
||||||
|
mkdir -p $out/lib $out/debug
|
||||||
|
${ar} crs $out/lib/libgourou.a *.o
|
||||||
|
cp *.o $out/debug
|
||||||
|
''
|
||||||
|
];
|
||||||
|
};
|
||||||
|
packages.x86_64-linux.utils-common = derivation {
|
||||||
|
name = "utils-common";
|
||||||
|
system = "x86_64-linux";
|
||||||
|
builder = "${nixpkgs.bash}/bin/bash";
|
||||||
|
PATH = "${nixpkgs.coreutils}/bin";
|
||||||
|
args = [
|
||||||
|
"-c"
|
||||||
|
''
|
||||||
|
${cxx} \
|
||||||
|
-c ${gourou-src}/utils/drmprocessorclientimpl.cpp \
|
||||||
|
${gourou-src}/utils/utils_common.cpp \
|
||||||
|
-I ${gourou-src}/utils \
|
||||||
|
-I ${gourou-src}/include \
|
||||||
|
-I ${pugixml-src}/src \
|
||||||
|
-I ${nixpkgs.openssl.dev}/include \
|
||||||
|
-I ${nixpkgs.curl.dev}/include \
|
||||||
|
-I ${nixpkgs.zlib.dev}/include \
|
||||||
|
-I ${self.libzip-static}/include \
|
||||||
|
${obj-flags}
|
||||||
|
mkdir -p $out/lib
|
||||||
|
${ar} crs $out/lib/libutils-common.a *.o
|
||||||
|
''
|
||||||
|
];
|
||||||
|
};
|
||||||
|
packages.x86_64-linux.knock = derivation {
|
||||||
|
name = "knock";
|
||||||
|
system = "x86_64-linux";
|
||||||
|
builder = "${nixpkgs.bash}/bin/bash";
|
||||||
|
PATH = "${nixpkgs.coreutils}/bin";
|
||||||
|
args = [
|
||||||
|
"-c"
|
||||||
|
''
|
||||||
|
mkdir -p $out/bin
|
||||||
|
${cxx} \
|
||||||
|
-o $out/bin/knock \
|
||||||
|
${./src/knock.cpp} \
|
||||||
|
-D KNOCK_VERSION='"${version}"' \
|
||||||
|
-Wl,--as-needed -static \
|
||||||
|
${self.utils-common}/lib/libutils-common.a \
|
||||||
|
${self.gourou}/lib/libgourou.a \
|
||||||
|
${self.updfparser}/lib/libupdfparser.a \
|
||||||
|
-Wl,--start-group \
|
||||||
|
${self.libzip-static}/lib/libzip.a \
|
||||||
|
${nixpkgs.libnghttp2}/lib/libnghttp2.a \
|
||||||
|
${nixpkgs.libidn2.out}/lib/libidn2.a \
|
||||||
|
${nixpkgs.libunistring}/lib/libunistring.a \
|
||||||
|
${nixpkgs.libssh2}/lib/libssh2.a \
|
||||||
|
${nixpkgs.zstd.out}/lib/libzstd.a \
|
||||||
|
${nixpkgs.zlib}/lib/libz.a \
|
||||||
|
${nixpkgs.openssl.out}/lib/libcrypto.a \
|
||||||
|
${nixpkgs.curl.out}/lib/libcurl.a \
|
||||||
|
${nixpkgs.openssl.out}/lib/libssl.a \
|
||||||
|
-static-libgcc -static-libstdc++ \
|
||||||
|
-Wl,--end-group \
|
||||||
|
-I ${gourou-src}/utils \
|
||||||
|
-I ${gourou-src}/include \
|
||||||
|
-I ${pugixml-src}/src \
|
||||||
|
-I ${nixpkgs.openssl.dev}/include \
|
||||||
|
-I ${nixpkgs.curl.dev}/include \
|
||||||
|
-I ${nixpkgs.zlib.dev}/include \
|
||||||
|
-I ${self.libzip-static}/include
|
||||||
|
''
|
||||||
|
];
|
||||||
|
};
|
||||||
|
packages.x86_64-linux.default = self.knock;
|
||||||
|
packages.x86_64-linux.tests = nixpkgs-dyn.stdenv.mkDerivation {
|
||||||
|
name = "tests";
|
||||||
|
src = ./tests;
|
||||||
|
buildInputs = [
|
||||||
|
(nixpkgs-dyn.python3.withPackages (p: [
|
||||||
|
p.beautifulsoup4
|
||||||
|
p.requests
|
||||||
|
]))
|
||||||
|
];
|
||||||
|
patchPhase = ''
|
||||||
|
substituteInPlace tests.py --replace "./result/bin/knock" "${self.knock}/bin/knock"
|
||||||
|
'';
|
||||||
|
installPhase = ''
|
||||||
|
mkdir -p $out/bin
|
||||||
|
cp tests.py $out/bin/tests
|
||||||
|
chmod +x $out/bin/tests
|
||||||
|
'';
|
||||||
|
};
|
||||||
|
devShell.x86_64-linux = nixpkgs.mkShell {
|
||||||
|
packages = [
|
||||||
|
# nix formatter
|
||||||
|
nixpkgs-dyn.nixpkgs-fmt
|
||||||
|
# python formatter
|
||||||
|
nixpkgs-dyn.black
|
||||||
|
# cpp formatter
|
||||||
|
nixpkgs-dyn.clang-tools
|
||||||
|
];
|
||||||
|
shellHook = ''
|
||||||
|
fmt () {
|
||||||
|
set -ex
|
||||||
|
nixpkgs-fmt .
|
||||||
|
black ./tests
|
||||||
|
clang-format -i --verbose src/*.cpp
|
||||||
|
set +ex
|
||||||
|
}
|
||||||
|
'';
|
||||||
|
};
|
||||||
};
|
};
|
||||||
packages.x86_64-linux.updfparser = derivation {
|
}
|
||||||
name = "updfparser";
|
|
||||||
system = "x86_64-linux";
|
|
||||||
builder = "${nixpkgs.bash}/bin/bash";
|
|
||||||
PATH = "${nixpkgs.coreutils}/bin";
|
|
||||||
args = [ "-c" ''
|
|
||||||
${cxx} \
|
|
||||||
-c ${updfparser-src}/src/*.cpp \
|
|
||||||
-I ${updfparser-src}/include \
|
|
||||||
${obj-flags}
|
|
||||||
mkdir -p $out/lib
|
|
||||||
${ar} crs $out/lib/libupdfparser.a *.o
|
|
||||||
'' ];
|
|
||||||
};
|
|
||||||
packages.x86_64-linux.gourou = derivation {
|
|
||||||
name = "gourou";
|
|
||||||
system = "x86_64-linux";
|
|
||||||
builder = "${nixpkgs.bash}/bin/bash";
|
|
||||||
PATH = "${nixpkgs.coreutils}/bin";
|
|
||||||
args = [ "-c" ''
|
|
||||||
${cxx} \
|
|
||||||
-c \
|
|
||||||
${gourou-src}/src/*.cpp \
|
|
||||||
${pugixml-src}/src/pugixml.cpp \
|
|
||||||
-I ${self.base64}/include \
|
|
||||||
-I ${gourou-src}/include \
|
|
||||||
-I ${pugixml-src}/src \
|
|
||||||
-I ${updfparser-src}/include \
|
|
||||||
${obj-flags}
|
|
||||||
mkdir -p $out/lib $out/debug
|
|
||||||
${ar} crs $out/lib/libgourou.a *.o
|
|
||||||
cp *.o $out/debug
|
|
||||||
'' ];
|
|
||||||
};
|
|
||||||
packages.x86_64-linux.utils-common = derivation {
|
|
||||||
name = "utils-common";
|
|
||||||
system = "x86_64-linux";
|
|
||||||
builder = "${nixpkgs.bash}/bin/bash";
|
|
||||||
PATH = "${nixpkgs.coreutils}/bin";
|
|
||||||
args = [ "-c" ''
|
|
||||||
${cxx} \
|
|
||||||
-c ${gourou-src}/utils/drmprocessorclientimpl.cpp \
|
|
||||||
${gourou-src}/utils/utils_common.cpp \
|
|
||||||
-I ${gourou-src}/utils \
|
|
||||||
-I ${gourou-src}/include \
|
|
||||||
-I ${pugixml-src}/src \
|
|
||||||
-I ${nixpkgs.openssl.dev}/include \
|
|
||||||
-I ${nixpkgs.curl.dev}/include \
|
|
||||||
-I ${nixpkgs.zlib.dev}/include \
|
|
||||||
-I ${self.libzip-static}/include \
|
|
||||||
${obj-flags}
|
|
||||||
mkdir -p $out/lib
|
|
||||||
${ar} crs $out/lib/libutils-common.a *.o
|
|
||||||
'' ];
|
|
||||||
};
|
|
||||||
packages.x86_64-linux.knock = derivation {
|
|
||||||
name = "knock";
|
|
||||||
system = "x86_64-linux";
|
|
||||||
builder = "${nixpkgs.bash}/bin/bash";
|
|
||||||
PATH = "${nixpkgs.coreutils}/bin";
|
|
||||||
args = [ "-c" ''
|
|
||||||
mkdir -p $out/bin
|
|
||||||
${cxx} \
|
|
||||||
-o $out/bin/knock \
|
|
||||||
${./src/knock.cpp} \
|
|
||||||
-D KNOCK_VERSION='"${version}"' \
|
|
||||||
-Wl,--as-needed -static \
|
|
||||||
${self.utils-common}/lib/libutils-common.a \
|
|
||||||
${self.gourou}/lib/libgourou.a \
|
|
||||||
${self.updfparser}/lib/libupdfparser.a \
|
|
||||||
-Wl,--start-group \
|
|
||||||
${self.libzip-static}/lib/libzip.a \
|
|
||||||
${nixpkgs.libnghttp2}/lib/libnghttp2.a \
|
|
||||||
${nixpkgs.libidn2.out}/lib/libidn2.a \
|
|
||||||
${nixpkgs.libunistring}/lib/libunistring.a \
|
|
||||||
${nixpkgs.libssh2}/lib/libssh2.a \
|
|
||||||
${nixpkgs.zstd.out}/lib/libzstd.a \
|
|
||||||
${nixpkgs.zlib}/lib/libz.a \
|
|
||||||
${nixpkgs.openssl.out}/lib/libcrypto.a \
|
|
||||||
${nixpkgs.curl.out}/lib/libcurl.a \
|
|
||||||
${nixpkgs.openssl.out}/lib/libssl.a \
|
|
||||||
-static-libgcc -static-libstdc++ \
|
|
||||||
-Wl,--end-group \
|
|
||||||
-I ${gourou-src}/utils \
|
|
||||||
-I ${gourou-src}/include \
|
|
||||||
-I ${pugixml-src}/src \
|
|
||||||
-I ${nixpkgs.openssl.dev}/include \
|
|
||||||
-I ${nixpkgs.curl.dev}/include \
|
|
||||||
-I ${nixpkgs.zlib.dev}/include \
|
|
||||||
-I ${self.libzip-static}/include
|
|
||||||
'' ];
|
|
||||||
};
|
|
||||||
packages.x86_64-linux.default = self.knock;
|
|
||||||
packages.x86_64-linux.tests = nixpkgs-dyn.stdenv.mkDerivation {
|
|
||||||
name = "tests";
|
|
||||||
src = ./tests;
|
|
||||||
buildInputs = [ (nixpkgs-dyn.python3.withPackages(p: [
|
|
||||||
p.beautifulsoup4
|
|
||||||
p.requests
|
|
||||||
])) ];
|
|
||||||
patchPhase = ''
|
|
||||||
substituteInPlace tests.py --replace "./result/bin/knock" "${self.knock}/bin/knock"
|
|
||||||
'';
|
|
||||||
installPhase = ''
|
|
||||||
mkdir -p $out/bin
|
|
||||||
cp tests.py $out/bin/tests
|
|
||||||
chmod +x $out/bin/tests
|
|
||||||
'';
|
|
||||||
};
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
|
||||||
|
|
@ -44,6 +44,14 @@ nix run .#tests -- ./tests/workspace
|
||||||
|
|
||||||
Test books can be found [here](https://www.adobe.com/solutions/ebook/digital-editions/sample-ebook-library.html).
|
Test books can be found [here](https://www.adobe.com/solutions/ebook/digital-editions/sample-ebook-library.html).
|
||||||
|
|
||||||
|
### Formatting
|
||||||
|
|
||||||
|
```
|
||||||
|
nix develop
|
||||||
|
fmt
|
||||||
|
exit
|
||||||
|
```
|
||||||
|
|
||||||
## The Name
|
## The Name
|
||||||
|
|
||||||
The name comes from the [D&D 5e spell](https://roll20.net/compendium/dnd5e/Knock#content) for freeing locked items:
|
The name comes from the [D&D 5e spell](https://roll20.net/compendium/dnd5e/Knock#content) for freeing locked items:
|
||||||
|
|
|
||||||
179
src/knock.cpp
179
src/knock.cpp
|
|
@ -1,119 +1,120 @@
|
||||||
#include <filesystem>
|
|
||||||
#include "drmprocessorclientimpl.h"
|
#include "drmprocessorclientimpl.h"
|
||||||
#include "libgourou_common.h"
|
|
||||||
#include "libgourou.h"
|
#include "libgourou.h"
|
||||||
|
#include "libgourou_common.h"
|
||||||
|
#include <filesystem>
|
||||||
|
|
||||||
#ifndef KNOCK_VERSION
|
#ifndef KNOCK_VERSION
|
||||||
#error KNOCK_VERSION must be defined
|
#error KNOCK_VERSION must be defined
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
std::string get_data_dir();
|
std::string get_data_dir();
|
||||||
void verify_absence(std::string file);
|
void verify_absence(std::string file);
|
||||||
void verify_presence(std::string file);
|
void verify_presence(std::string file);
|
||||||
|
|
||||||
int main(int argc, char** argv) try {
|
int main(int argc, char **argv) try {
|
||||||
|
|
||||||
if (argc == 1) {
|
if (argc == 1) {
|
||||||
std::cout
|
std::cout << "info: knock version " << KNOCK_VERSION
|
||||||
<< "info: knock version " << KNOCK_VERSION << ", libgourou version "
|
<< ", libgourou version " << LIBGOUROU_VERSION << "\n"
|
||||||
<< LIBGOUROU_VERSION << "\n"
|
<< "usage: " << argv[0] << " [ACSM]\n"
|
||||||
<< "usage: " << argv[0] << " [ACSM]" << "\n"
|
<< "result: converts file ACSM to a plain EPUB/PDF if present, "
|
||||||
<< "result: converts file ACSM to a plain EPUB/PDF if present, otherwise prints this"
|
"otherwise prints this"
|
||||||
<< std::endl;
|
<< std::endl;
|
||||||
return EXIT_SUCCESS;
|
return EXIT_SUCCESS;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (argc != 2) {
|
if (argc != 2) {
|
||||||
throw std::invalid_argument("the ACSM file must be passed as an argument");
|
throw std::invalid_argument("the ACSM file must be passed as an argument");
|
||||||
}
|
}
|
||||||
|
|
||||||
const std::string acsm_file = argv[1];
|
const std::string acsm_file = argv[1];
|
||||||
verify_presence(acsm_file);
|
verify_presence(acsm_file);
|
||||||
const std::string acsm_stem = acsm_file.substr(0, acsm_file.find_last_of("."));
|
const std::string acsm_stem =
|
||||||
const std::string drm_file = acsm_stem + ".drm";
|
acsm_file.substr(0, acsm_file.find_last_of("."));
|
||||||
const std::string out_file = acsm_stem + ".out";
|
const std::string drm_file = acsm_stem + ".drm";
|
||||||
verify_absence(drm_file);
|
const std::string out_file = acsm_stem + ".out";
|
||||||
verify_absence(out_file);
|
verify_absence(drm_file);
|
||||||
const std::string knock_data = get_data_dir();
|
verify_absence(out_file);
|
||||||
|
const std::string knock_data = get_data_dir();
|
||||||
|
|
||||||
DRMProcessorClientImpl client;
|
DRMProcessorClientImpl client;
|
||||||
gourou::DRMProcessor* processor = gourou::DRMProcessor::createDRMProcessor(
|
gourou::DRMProcessor *processor = gourou::DRMProcessor::createDRMProcessor(
|
||||||
&client,
|
&client,
|
||||||
false, // don't "always generate a new device" (default)
|
false, // don't "always generate a new device" (default)
|
||||||
knock_data
|
knock_data);
|
||||||
);
|
|
||||||
|
|
||||||
std::cout << "anonymously signing in..." << std::endl;
|
std::cout << "anonymously signing in..." << std::endl;
|
||||||
processor->signIn("anonymous", "");
|
processor->signIn("anonymous", "");
|
||||||
processor->activateDevice();
|
processor->activateDevice();
|
||||||
|
|
||||||
std::cout << "downloading the file from Adobe..." << std::endl;
|
std::cout << "downloading the file from Adobe..." << std::endl;
|
||||||
gourou::FulfillmentItem* item = processor->fulfill(acsm_file);
|
gourou::FulfillmentItem *item = processor->fulfill(acsm_file);
|
||||||
gourou::DRMProcessor::ITEM_TYPE type = processor->download(item, drm_file);
|
gourou::DRMProcessor::ITEM_TYPE type = processor->download(item, drm_file);
|
||||||
|
|
||||||
std::cout << "removing DRM from the file..." << std::endl;
|
|
||||||
std::string ext_file;
|
|
||||||
std::string file_type;
|
|
||||||
switch (type) {
|
|
||||||
case gourou::DRMProcessor::ITEM_TYPE::PDF: {
|
|
||||||
// for pdfs the function moves the pdf while removing drm
|
|
||||||
processor->removeDRM(drm_file, out_file, type, nullptr, 0);
|
|
||||||
std::filesystem::remove(drm_file);
|
|
||||||
ext_file = acsm_stem + ".pdf";
|
|
||||||
file_type = "PDF";
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
case gourou::DRMProcessor::ITEM_TYPE::EPUB: {
|
|
||||||
// for epubs the drm is removed in-place so in == out
|
|
||||||
processor->removeDRM(drm_file, drm_file, type, nullptr, 0);
|
|
||||||
std::filesystem::rename(drm_file, out_file);
|
|
||||||
ext_file = acsm_stem + ".epub";
|
|
||||||
file_type = "EPUB";
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (std::filesystem::exists(ext_file)) {
|
std::cout << "removing DRM from the file..." << std::endl;
|
||||||
std::cerr
|
std::string ext_file;
|
||||||
<< "warning: failed to update file extension; " + ext_file + " already exists"
|
std::string file_type;
|
||||||
<< std::endl;
|
switch (type) {
|
||||||
ext_file = out_file;
|
case gourou::DRMProcessor::ITEM_TYPE::PDF: {
|
||||||
} else {
|
// for pdfs the function moves the pdf while removing drm
|
||||||
std::filesystem::rename(out_file, ext_file);
|
processor->removeDRM(drm_file, out_file, type, nullptr, 0);
|
||||||
}
|
std::filesystem::remove(drm_file);
|
||||||
|
ext_file = acsm_stem + ".pdf";
|
||||||
|
file_type = "PDF";
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case gourou::DRMProcessor::ITEM_TYPE::EPUB: {
|
||||||
|
// for epubs the drm is removed in-place so in == out
|
||||||
|
processor->removeDRM(drm_file, drm_file, type, nullptr, 0);
|
||||||
|
std::filesystem::rename(drm_file, out_file);
|
||||||
|
ext_file = acsm_stem + ".epub";
|
||||||
|
file_type = "EPUB";
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
std::cout << file_type + " file generated at " + ext_file << std::endl;
|
if (std::filesystem::exists(ext_file)) {
|
||||||
|
std::cerr << "warning: failed to update file extension; " + ext_file +
|
||||||
|
" already exists"
|
||||||
|
<< std::endl;
|
||||||
|
ext_file = out_file;
|
||||||
|
} else {
|
||||||
|
std::filesystem::rename(out_file, ext_file);
|
||||||
|
}
|
||||||
|
|
||||||
return 0;
|
std::cout << file_type + " file generated at " + ext_file << std::endl;
|
||||||
|
|
||||||
} catch (const gourou::Exception& e) {
|
return 0;
|
||||||
std::cerr << "error:\n" << e.what();
|
|
||||||
return EXIT_FAILURE;
|
} catch (const gourou::Exception &e) {
|
||||||
} catch (const std::exception& e) {
|
std::cerr << "error:\n" << e.what();
|
||||||
std::cerr << "error: " << e.what() << std::endl;
|
return EXIT_FAILURE;
|
||||||
return EXIT_FAILURE;
|
} catch (const std::exception &e) {
|
||||||
|
std::cerr << "error: " << e.what() << std::endl;
|
||||||
|
return EXIT_FAILURE;
|
||||||
}
|
}
|
||||||
|
|
||||||
std::string get_data_dir() {
|
std::string get_data_dir() {
|
||||||
char* xdg_data_home = std::getenv("XDG_DATA_HOME");
|
char *xdg_data_home = std::getenv("XDG_DATA_HOME");
|
||||||
std::string knock_data;
|
std::string knock_data;
|
||||||
if (xdg_data_home != nullptr) {
|
if (xdg_data_home != nullptr) {
|
||||||
knock_data = xdg_data_home;
|
knock_data = xdg_data_home;
|
||||||
} else {
|
} else {
|
||||||
knock_data = std::string(std::getenv("HOME")) + "/.local/share";
|
knock_data = std::string(std::getenv("HOME")) + "/.local/share";
|
||||||
}
|
}
|
||||||
knock_data += "/knock/acsm";
|
knock_data += "/knock/acsm";
|
||||||
return knock_data;
|
return knock_data;
|
||||||
}
|
}
|
||||||
|
|
||||||
void verify_absence(std::string file) {
|
void verify_absence(std::string file) {
|
||||||
if (std::filesystem::exists(file)) {
|
if (std::filesystem::exists(file)) {
|
||||||
throw std::runtime_error("file " + file + " must be moved out of the way or deleted");
|
throw std::runtime_error("file " + file +
|
||||||
}
|
" must be moved out of the way or deleted");
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
void verify_presence(std::string file) {
|
void verify_presence(std::string file) {
|
||||||
if (!std::filesystem::exists(file)) {
|
if (!std::filesystem::exists(file)) {
|
||||||
throw std::runtime_error("file " + file + " does not exist");
|
throw std::runtime_error("file " + file + " does not exist");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
@ -14,7 +14,7 @@ if not knock.exists():
|
||||||
if len(sys.argv) != 2:
|
if len(sys.argv) != 2:
|
||||||
print(
|
print(
|
||||||
"error: missing required argument: directory in which to perform the tests",
|
"error: missing required argument: directory in which to perform the tests",
|
||||||
file=sys.stderr
|
file=sys.stderr,
|
||||||
)
|
)
|
||||||
sys.exit()
|
sys.exit()
|
||||||
|
|
||||||
|
|
@ -31,18 +31,18 @@ if workspace.exists():
|
||||||
shutil.rmtree(workspace)
|
shutil.rmtree(workspace)
|
||||||
workspace.mkdir()
|
workspace.mkdir()
|
||||||
|
|
||||||
html = requests \
|
html = requests.get(
|
||||||
.get("https://www.adobe.com/solutions/ebook/digital-editions/sample-ebook-library.html") \
|
"https://www.adobe.com/solutions/ebook/digital-editions/sample-ebook-library.html"
|
||||||
.text
|
).text
|
||||||
soup = BeautifulSoup(html, 'html.parser')
|
soup = BeautifulSoup(html, "html.parser")
|
||||||
|
|
||||||
links = []
|
links = []
|
||||||
for a_tag in soup.find_all('a'):
|
for a_tag in soup.find_all("a"):
|
||||||
if a_tag.string != "Download eBook":
|
if a_tag.string != "Download eBook":
|
||||||
continue
|
continue
|
||||||
if not urlparse(a_tag.get("href")).path.endswith(".acsm"):
|
if not urlparse(a_tag.get("href")).path.endswith(".acsm"):
|
||||||
continue
|
continue
|
||||||
|
|
||||||
links.append(a_tag.get("href"))
|
links.append(a_tag.get("href"))
|
||||||
|
|
||||||
if len(links) >= 10:
|
if len(links) >= 10:
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue