/*

 * Copyright (c) 2022 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.

 */

#ifndef DISPLAY_MODULE_LOADER_H

#define DISPLAY_MODULE_LOADER_H



#include <dlfcn.h>

#include <memory>

#include <string>

#include <display_common.h>

namespace OHOS {

namespace HDI {

namespace DISPLAY {

class DisplayModuleLoader {

public:

    static std::unique_ptr<DisplayModuleLoader> Create(const std::string &pathName)

    {

        void *handle = dlopen(pathName.c_str(), RTLD_NOW | RTLD_NOLOAD);

        if (handle == nullptr) {

            handle = dlopen(pathName.c_str(), RTLD_NOW);

            if (handle == nullptr) {

                return nullptr;

            }

        }

        return std::make_unique<DisplayModuleLoader>(handle);

    }



    explicit DisplayModuleLoader(void *handle) : handle_(handle) {}



    void *GetSymbol(const std::string &symbol)

    {

        return dlsym(handle_, symbol.c_str());

    }



    ~DisplayModuleLoader()

    {

        if (handle_ != nullptr) {

            dlclose(handle_);

            handle_ = nullptr;

        }

    }



private:

    void *handle_;

};

} // namespace DISPLAY

} // namespace HDI

} // namespace OHOS

#endif // DISPLAY_MODULE_LOADER_H