2 files changed, 59 insertions, 0 deletions
diff --git a/api/logic/minecraft/launch/ClaimAccount.cpp b/api/logic/minecraft/launch/ClaimAccount.cpp
new file mode 100644
index 00000000..71670b4f
--- /dev/null
+++ b/api/logic/minecraft/launch/ClaimAccount.cpp
@@ -0,0 +1,24 @@
+#include "ClaimAccount.h"
+#include <launch/LaunchTask.h>
+
+ClaimAccount::ClaimAccount(LaunchTask* parent, AuthSessionPtr session): LaunchStep(parent)
+{
+	if(session->status == AuthSession::Status::PlayableOnline)
+	{
+		m_account = session->m_accountPtr;
+	}
+}
+
+void ClaimAccount::executeTask()
+{
+	if(m_account)
+	{
+		lock.reset(new UseLock(m_account));
+		emitSucceeded();
+	}
+}
+
+void ClaimAccount::finalize()
+{
+	lock.reset();
+}
diff --git a/api/logic/minecraft/launch/ClaimAccount.h b/api/logic/minecraft/launch/ClaimAccount.h
new file mode 100644
index 00000000..fa5039af
--- /dev/null
+++ b/api/logic/minecraft/launch/ClaimAccount.h
@@ -0,0 +1,35 @@
+/* Copyright 2013-2016 MultiMC Contributors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#pragma once
+
+#include <launch/LaunchStep.h>
+#include <minecraft/auth/MojangAccount.h>
+
+class ClaimAccount: public LaunchStep
+{
+	Q_OBJECT
+public:
+	explicit ClaimAccount(LaunchTask *parent, AuthSessionPtr session);
+	void executeTask() override;
+	void finalize() override;
+	bool canAbort() const override
+	{
+		return false;
+	}
+private:
+	std::unique_ptr<UseLock> lock;
+	MojangAccountPtr m_account;
+};