aboutsummaryrefslogtreecommitdiffstats
path: root/vorlesungsnotizen/bin/numberpages.cpp
blob: 704a67151d1e6656332c533bd29a6449b77408bc (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
/*
 * numberpages.cpp -- driver program to number pages in a PDF, this was
 *                    mainly necessary because the shell breaks blanks
 *                    in the section option string
 *
 * (c) 2022 Prof Dr Andreas Müller, OST Ostschweizer Fachhochschule
 */
#include <cstdio>
#include <cstdlib>
#include <getopt.h>
#include <iostream>
#include <sstream>
#include <string>
#include <sys/stat.h>

void	usage(char *progname) {
	std::cout << "usage: " << progname << " [ -f ] [ -s section ] ";
	std::cout << "<unnumbered.pdf> <numbered.pdf>" << std::endl;
}

struct option	longoptions[] = {
{ "debug",	no_argument,		NULL,	'd' },
{ "section",	required_argument,	NULL,	's' },
{ "force",	no_argument,		NULL,	'f' },
{ NULL,		no_argument,		NULL,	 0  }
};

int	main(int argc, char *argv[]) {
	bool	force = false;
	bool	debug = false;
	std::string	section;
	int	c, longindex, rc;
	while (EOF != (c = getopt_long(argc, argv, "dfhs:?", longoptions,
		&longindex))) {
		switch (c) {
		case 'd':
			debug = true;
			break;
		case 'f':
			force = true;
			break;
		case 'h':
		case '?':
			usage(argv[0]);
			return EXIT_SUCCESS;
		case 's':
			section = std::string(optarg);
			if (debug) {
				std::cout << "section string: " << section;
				std::cout << std::endl;
			}
			break;
		}
	}

	// remaining arguments are file names
	if ((argc - optind) < 2) {
		std::cerr << "not enough arguments" << std::endl;
		usage(argv[0]);
		return EXIT_FAILURE;
	}
	std::string	unnumbered = std::string(argv[optind++]);
	std::string	numbered = std::string(argv[optind++]);
	if (debug) {
		std::cout << "numbering " << unnumbered;
		std::cout << ", sending output to " << unnumbered;
		std::cout << std::endl;
	}

	// make sure input file does exist
	struct stat	sb;
	rc = stat(unnumbered.c_str(), &sb);
	if ((0 == rc) && S_ISREG(sb.st_mode)) {
		if (debug) {
			std::cout << "input file " << unnumbered << " found";
			std::cout << std::endl;
		}
	} else {
		std::cerr << "cannot find " << unnumbered << std::endl;
		return EXIT_FAILURE;
	}

	// make sure the "unnumbered.pdf" file does not exist
	rc = stat("unnumbered.pdf", &sb);
	if (0 == rc) {
		std::cerr << "file unnumbered.pdf exists, clean up first";
		std::cerr << std::endl;
		return EXIT_FAILURE;
	}

	// test for existence of output file
	rc = stat(numbered.c_str(), &sb);
	if ((!force) && (0 == rc)) {
		std::cerr << "output file exists, use --force option";
		std::cerr << std::endl;
		return EXIT_FAILURE;
	}

	// make sure the numbered.tex is present
	rc = stat("numbered.tex", &sb);
	if ((0 != rc) || (!S_ISREG(sb.st_mode))) {
		std::cerr << "file 'numbered.tex' is missing" << std::endl;
		return EXIT_FAILURE;
	}

	// remove the "numbered.pdf" if it exists
	if (debug) {
		std::cout << "removing numbered.pdf" << std::endl;
	}
	unlink("numbered.pdf");

	// copy the file
	{
		std::ostringstream	out;
		out << "cp '" << unnumbered << "' unnumbered.pdf";
		out.flush();
		std::string	copycommand = out.str();
		rc = system(copycommand.c_str());
		if (0 != rc) {
			std::cerr << "could not copy " << unnumbered;
			std::cerr << " to unnumbered.pdf: " << rc << std::endl;
			return EXIT_FAILURE;
		}
		if (debug) {
			std::cout << "file " << unnumbered << " copied to ";
			std::cout << "unnumbered.pdf" << std::endl;
		}
	}

	// prepare command line
	std::ostringstream	out;
	out << "pdflatex ";
	if (section.size() > 0) {
		out << "\\\\def\\\\sectionnumber{" << section << "} ";
		out << "\\\\input{numbered.tex}";
	} else {
		out << unnumbered;
	}
	if (debug) {
		std::cout << "command line: " << out.str() << std::endl;
	}

	// run the command
	rc = system(out.str().c_str());
	if (0 != rc) {
		std::cerr << "pdflatex failed: " << rc << std::endl;
		return EXIT_FAILURE;
	}

	// check for the existence of the output file
	rc = stat("numbered.pdf", &sb);
	if ((0 != rc) || (!S_ISREG(sb.st_mode))) {
		std::cerr << "output file 'numbered.pdf' not found";
		std::cerr << std::endl;
		return EXIT_FAILURE;
	}

	// rename the output file
	rc = rename("numbered.pdf", numbered.c_str());
	if (rc != 0) {
		std::string	error(strerror(errno));
		std::cerr << "cannot rename file to '" << numbered << ": ";
		std::cerr << error << std::endl;
		return EXIT_FAILURE;
	}

	// perform some cleanup
	unlink("unnumbered.pdf");
	unlink("numbered.log");
	unlink("numbered.aux");

	return EXIT_SUCCESS;
}