92 lines
		
	
	
		
			2.8 KiB
		
	
	
	
		
			PHP
		
	
	
	
			
		
		
	
	
			92 lines
		
	
	
		
			2.8 KiB
		
	
	
	
		
			PHP
		
	
	
	
<?php
 | 
						|
 | 
						|
namespace BookStack\Http\Controllers;
 | 
						|
 | 
						|
use BookStack\Entities\Managers\BookContents;
 | 
						|
use BookStack\Entities\ExportService;
 | 
						|
use BookStack\Entities\Repos\BookRepo;
 | 
						|
use Throwable;
 | 
						|
use ZipArchive;
 | 
						|
 | 
						|
class BookExportController extends Controller
 | 
						|
{
 | 
						|
 | 
						|
    protected $bookRepo;
 | 
						|
    protected $exportService;
 | 
						|
 | 
						|
    /**
 | 
						|
     * BookExportController constructor.
 | 
						|
     */
 | 
						|
    public function __construct(BookRepo $bookRepo, ExportService $exportService)
 | 
						|
    {
 | 
						|
        $this->bookRepo = $bookRepo;
 | 
						|
        $this->exportService = $exportService;
 | 
						|
        parent::__construct();
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Export a book as a PDF file.
 | 
						|
     * @throws Throwable
 | 
						|
     */
 | 
						|
    public function pdf(string $bookSlug)
 | 
						|
    {
 | 
						|
        $book = $this->bookRepo->getBySlug($bookSlug);
 | 
						|
        $pdfContent = $this->exportService->bookToPdf($book);
 | 
						|
        return $this->downloadResponse($pdfContent, $bookSlug . '.pdf');
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Export a book as a contained HTML file.
 | 
						|
     * @throws Throwable
 | 
						|
     */
 | 
						|
    public function html(string $bookSlug)
 | 
						|
    {
 | 
						|
        $book = $this->bookRepo->getBySlug($bookSlug);
 | 
						|
        $htmlContent = $this->exportService->bookToContainedHtml($book);
 | 
						|
        return $this->downloadResponse($htmlContent, $bookSlug . '.html');
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Export a book as a plain text file.
 | 
						|
     */
 | 
						|
    public function plainText(string $bookSlug)
 | 
						|
    {
 | 
						|
        $book = $this->bookRepo->getBySlug($bookSlug);
 | 
						|
        $textContent = $this->exportService->bookToPlainText($book);
 | 
						|
        return $this->downloadResponse($textContent, $bookSlug . '.txt');
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Export a book as a markdown file.
 | 
						|
     */
 | 
						|
    public function markdown(string $bookSlug)
 | 
						|
    {
 | 
						|
        $book = $this->bookRepo->getBySlug($bookSlug);
 | 
						|
        $textContent = $this->exportService->bookToMarkdown($book);
 | 
						|
        return $this->downloadResponse($textContent, $bookSlug . '.md');
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Export a book as a zip file, made of markdown files.
 | 
						|
     */
 | 
						|
    public function zip(string $bookSlug)
 | 
						|
    {
 | 
						|
        $book = $this->bookRepo->getBySlug($bookSlug);
 | 
						|
        $z = new ZipArchive();
 | 
						|
        $z->open("book.zip", \ZipArchive::CREATE | \ZipArchive::OVERWRITE);
 | 
						|
        $bookTree = (new BookContents($book))->getTree(false, true);
 | 
						|
        foreach ($bookTree as $bookChild) {
 | 
						|
            if ($bookChild->isA('chapter')) {
 | 
						|
                $z->addEmptyDir($bookChild->name);
 | 
						|
                foreach ($bookChild->pages as $page) {
 | 
						|
                    $z->addFromString($bookChild->name . "/" . $page->name . ".md", $this->exportService->pageToMarkdown($page));
 | 
						|
                }
 | 
						|
            } else {
 | 
						|
                $z->addFromString($bookChild->name . ".md", $this->exportService->pageToMarkdown($bookChild));
 | 
						|
            }
 | 
						|
        }
 | 
						|
        return response()->download('book.zip');
 | 
						|
        // TODO: Is not unlinking it a security issue?
 | 
						|
    }
 | 
						|
}
 |