106 lines
		
	
	
	
		
			2.9 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			106 lines
		
	
	
	
		
			2.9 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
/*
 | 
						|
This file is part of Telegram Desktop,
 | 
						|
the official desktop application for the Telegram messaging service.
 | 
						|
 | 
						|
For license and copyright information please follow this link:
 | 
						|
https://github.com/telegramdesktop/tdesktop/blob/master/LEGAL
 | 
						|
*/
 | 
						|
#pragma once
 | 
						|
 | 
						|
#include "export/output/export_output_abstract.h"
 | 
						|
#include "export/output/export_output_file.h"
 | 
						|
#include "export/export_settings.h"
 | 
						|
#include "export/data/export_data_types.h"
 | 
						|
 | 
						|
namespace Export {
 | 
						|
namespace Output {
 | 
						|
namespace details {
 | 
						|
 | 
						|
struct JsonContext {
 | 
						|
	using Type = bool;
 | 
						|
	static constexpr auto kObject = Type(true);
 | 
						|
	static constexpr auto kArray = Type(false);
 | 
						|
 | 
						|
	// Always fun to use std::vector<bool>.
 | 
						|
	std::vector<Type> nesting;
 | 
						|
};
 | 
						|
 | 
						|
} // namespace details
 | 
						|
 | 
						|
class JsonWriter : public AbstractWriter {
 | 
						|
public:
 | 
						|
	Format format() override {
 | 
						|
		return Format::Json;
 | 
						|
	}
 | 
						|
 | 
						|
	Result start(
 | 
						|
		const Settings &settings,
 | 
						|
		const Environment &environment,
 | 
						|
		Stats *stats) override;
 | 
						|
 | 
						|
	Result writePersonal(const Data::PersonalInfo &data) override;
 | 
						|
 | 
						|
	Result writeUserpicsStart(const Data::UserpicsInfo &data) override;
 | 
						|
	Result writeUserpicsSlice(const Data::UserpicsSlice &data) override;
 | 
						|
	Result writeUserpicsEnd() override;
 | 
						|
 | 
						|
	Result writeContactsList(const Data::ContactsList &data) override;
 | 
						|
 | 
						|
	Result writeSessionsList(const Data::SessionsList &data) override;
 | 
						|
 | 
						|
	Result writeOtherData(const Data::File &data) override;
 | 
						|
 | 
						|
	Result writeDialogsStart(const Data::DialogsInfo &data) override;
 | 
						|
	Result writeDialogStart(const Data::DialogInfo &data) override;
 | 
						|
	Result writeDialogSlice(const Data::MessagesSlice &data) override;
 | 
						|
	Result writeDialogEnd() override;
 | 
						|
	Result writeDialogsEnd() override;
 | 
						|
 | 
						|
	Result finish() override;
 | 
						|
 | 
						|
	QString mainFilePath() override;
 | 
						|
 | 
						|
private:
 | 
						|
	using Context = details::JsonContext;
 | 
						|
	enum class DialogsMode {
 | 
						|
		None,
 | 
						|
		Chats,
 | 
						|
		Left,
 | 
						|
	};
 | 
						|
 | 
						|
	[[nodiscard]] QByteArray pushNesting(Context::Type type);
 | 
						|
	[[nodiscard]] QByteArray prepareObjectItemStart(const QByteArray &key);
 | 
						|
	[[nodiscard]] QByteArray prepareArrayItemStart();
 | 
						|
	[[nodiscard]] QByteArray popNesting();
 | 
						|
 | 
						|
	[[nodiscard]] QString mainFileRelativePath() const;
 | 
						|
	[[nodiscard]] QString pathWithRelativePath(const QString &path) const;
 | 
						|
	[[nodiscard]] std::unique_ptr<File> fileWithRelativePath(
 | 
						|
		const QString &path) const;
 | 
						|
 | 
						|
	[[nodiscard]] Result writeSavedContacts(const Data::ContactsList &data);
 | 
						|
	[[nodiscard]] Result writeFrequentContacts(const Data::ContactsList &data);
 | 
						|
 | 
						|
	[[nodiscard]] Result writeSessions(const Data::SessionsList &data);
 | 
						|
	[[nodiscard]] Result writeWebSessions(const Data::SessionsList &data);
 | 
						|
 | 
						|
	[[nodiscard]] Result validateDialogsMode(bool isLeftChannel);
 | 
						|
	[[nodiscard]] Result writeChatsStart(
 | 
						|
		const QByteArray &listName,
 | 
						|
		const QByteArray &about);
 | 
						|
	[[nodiscard]] Result writeChatsEnd();
 | 
						|
 | 
						|
	Settings _settings;
 | 
						|
	Environment _environment;
 | 
						|
	Stats *_stats = nullptr;
 | 
						|
 | 
						|
	Context _context;
 | 
						|
	bool _currentNestingHadItem = false;
 | 
						|
	DialogsMode _dialogsMode = DialogsMode::None;
 | 
						|
 | 
						|
	std::unique_ptr<File> _output;
 | 
						|
 | 
						|
};
 | 
						|
 | 
						|
} // namespace Output
 | 
						|
} // namespace Export
 |