logic: versioning

src/license_generator/Generator.h

@@ -0,0 +1,68 @@
+#ifndef KEYGEN_H
+#define KEYGEN_H
+
+#include <string>
+#include <memory>
+
+#include "botan/auto_rng.h"
+
+namespace Botan
+{
+	class RSA_PrivateKey;
+}
+class License;
+
+class Generator
+{
+public:
+    Generator();
+    ~Generator();
+
+    std::string encodeLicense(
+        const std::string& user,
+        const std::string& licenseType,
+        const int seats = 0,
+        const std::string& version = ""
+    );
+    std::string encodeLicense(
+        const std::string& user,
+        const int days
+    );
+    void printLicenseAndWriteItToFile();
+    bool verifyLicense(const std::string& filename = "license.txt");
+    void generateKeys();
+    void writeKeysToFiles();
+    void setCustomPrivateKeyFile(const std::string& file);
+    void setCustomPublicKeyFile(const std::string& file);
+
+    std::string getPublicKeyPEMFileAsString();
+    std::string getPrivateKeyPEMFileAsString();
+    bool loadPrivateKeyFromFile();
+    bool loadPrivateKeyFromString(const std::string& key);
+
+    Botan::RSA_PrivateKey* getPrivateKey() const;
+    void createLicense(
+        const std::string& user,
+        const std::string& type,
+        const std::string& expiration,
+        const unsigned int seates
+    );
+
+    std::string getExpireVersion(int versions = 4);
+    void setVersionLine(int year, int minorVersion);
+private:
+    int mapMonthToVersion(int month);
+    std::string getPrivateKeyFilename();
+    std::string getPublicKeyFilename();
+
+    //Botan
+    Botan::AutoSeeded_RNG m_rng;
+
+    std::string m_privateKeyFile;
+    std::string m_publicKeyFile;
+    std::unique_ptr<License> m_license;
+    std::unique_ptr<Botan::RSA_PrivateKey> m_privateKey;
+
+};
+
+#endif // KEYGEN_H