xref: /base/request/request/pre_download/netstack_rs/src/cxx/wrapper.cpp

/*
* Copyright (C) 2024 Huawei Device Co., Ltd.
* 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.
*/

#include "wrapper.h"

#include <memory>

#include "http_client_error.h"
#include "wrapper.rs.h"
namespace OHOS::Request {
using namespace OHOS::NetStack::HttpClient;

void OnCallback(std::shared_ptr<HttpClientTask> task, rust::Box<CallbackWrapper> callback)
{
    CallbackWrapper *raw_ptr = callback.into_raw();
    auto shared = std::shared_ptr<CallbackWrapper>(
        raw_ptr, [](CallbackWrapper *ptr) { rust::Box<CallbackWrapper>::from_raw(ptr); });
    task->OnSuccess(
        [shared](const HttpClientRequest &, const HttpClientResponse &response) { shared->on_success(response); });
    task->OnFail([shared](const HttpClientRequest &, const HttpClientResponse &response,
                     const HttpClientError &error) { shared->on_fail(response, error); });
    task->OnCancel(
        [shared](const HttpClientRequest &, const HttpClientResponse &response) { shared->on_cancel(response); });
    task->OnDataReceive(
        [shared](const HttpClientRequest &, const uint8_t *data, size_t size) { shared->on_data_receive(data, size); });
    task->OnProgress([shared](const HttpClientRequest &, u_long dlTotal, u_long dlNow, u_long ulTotal, u_long ulNow) {
        shared->on_progress(dlTotal, dlNow, ulTotal, ulNow);
    });
}

} // namespace OHOS::Request