From 47e4dd6b8d511f6c549d5c8c911865d4e7b9d98d Mon Sep 17 00:00:00 2001 From: NexVeridian Date: Tue, 12 Dec 2023 10:45:10 +0000 Subject: [PATCH] init --- .cargo/config.toml | 9 ++ .devcontainer/Dockerfile | 19 ++++ .devcontainer/devcontainer.json | 73 ++++++++++++ .dockerignore | 10 ++ .gitignore | 109 ++++++++++++++++++ .vscode/launch.json | 43 +++++++ .vscode/settings.json | 17 +++ CONTRIBUTING.md | 11 ++ Cargo.toml | 14 +++ LICENSE-Apache | 192 ++++++++++++++++++++++++++++++++ LICENSE-MIT | 21 ++++ README.md | 8 ++ docker-compose.yml | 22 ++++ src/main.rs | 32 ++++++ src/utils.rs | 69 ++++++++++++ 15 files changed, 649 insertions(+) create mode 100644 .cargo/config.toml create mode 100644 .devcontainer/Dockerfile create mode 100644 .devcontainer/devcontainer.json create mode 100644 .dockerignore create mode 100644 .gitignore create mode 100644 .vscode/launch.json create mode 100644 .vscode/settings.json create mode 100644 CONTRIBUTING.md create mode 100644 Cargo.toml create mode 100644 LICENSE-Apache create mode 100644 LICENSE-MIT create mode 100644 README.md create mode 100644 docker-compose.yml create mode 100644 src/main.rs create mode 100644 src/utils.rs diff --git a/.cargo/config.toml b/.cargo/config.toml new file mode 100644 index 0000000..0a0833d --- /dev/null +++ b/.cargo/config.toml @@ -0,0 +1,9 @@ +[target.x86_64-unknown-linux-gnu] +linker = "clang" +rustflags = ["-C", "link-arg=-fuse-ld=/usr/bin/mold"] + +[alias] +t = "nextest run" + +[build] +target-dir = "target/target" diff --git a/.devcontainer/Dockerfile b/.devcontainer/Dockerfile new file mode 100644 index 0000000..5228546 --- /dev/null +++ b/.devcontainer/Dockerfile @@ -0,0 +1,19 @@ +# https://mcr.microsoft.com/en-us/product/devcontainers/rust/about +FROM mcr.microsoft.com/devcontainers/rust:bookworm + +RUN apt-get update && \ + apt install -y build-essential xz-utils musl-tools musl-dev gcc-multilib pkg-config libssl-dev clang mold + +RUN LAZYGIT_VERSION=$(curl -s "https://api.github.com/repos/jesseduffield/lazygit/releases/latest" | grep -Po '"tag_name": "v\K[0-9.]+') && \ + curl -Lo lazygit.tar.gz "https://github.com/jesseduffield/lazygit/releases/latest/download/lazygit_${LAZYGIT_VERSION}_Linux_x86_64.tar.gz" && \ + sudo tar xf lazygit.tar.gz -C /usr/local/bin lazygit && \ + rm -rf lazygit.tar.gz + +RUN BTOP_VERSION=$(curl -s "https://api.github.com/repos/aristocratos/btop/releases/latest" | grep -Po '"tag_name": "v\K[0-9.]+') && \ + wget "https://github.com/aristocratos/btop/releases/download/v${BTOP_VERSION}/btop-x86_64-linux-musl.tbz" && \ + sudo tar -xvf btop-x86_64-linux-musl.tbz && \ + cd btop && ./install.sh && cd .. && \ + rm -rf btop-x86_64-linux-musl.tbz btop + +RUN curl -LsSf https://get.nexte.st/latest/linux | tar zxf - -C ${CARGO_HOME:-~/.cargo}/bin +# cargo install cargo-nextest --locked diff --git a/.devcontainer/devcontainer.json b/.devcontainer/devcontainer.json new file mode 100644 index 0000000..13ca2b9 --- /dev/null +++ b/.devcontainer/devcontainer.json @@ -0,0 +1,73 @@ +// For format details, see https://aka.ms/devcontainer.json. For config options, see the +// README at: https://github.com/devcontainers/templates/tree/main/src/rust +// https://containers.dev/implementors/json_reference/#variables-in-devcontainerjson +{ + "name": "Rust", + // Or use a Dockerfile or Docker Compose file. More info: https://containers.dev/guide/dockerfile + // "image": "mcr.microsoft.com/devcontainers/rust:0-1-bullseye", + "build": { + // Path is relataive to the devcontainer.json file. + "dockerfile": "Dockerfile" + }, + // https://github.com/microsoft/vscode-remote-release/issues/2485#issuecomment-1156342780 + "runArgs": [ + "--name", + "devcontainer-${containerWorkspaceFolderBasename}" + ], + "initializeCommand": "docker rm -f devcontainer-${containerWorkspaceFolderBasename} || true", + // Use 'mounts' to make the cargo cache persistent in a Docker Volume. + "mounts": [ + { + // "source": "devcontainer-cargo-cache-${devcontainerId}", + "source": "devcontainer-cargo-cache-${containerWorkspaceFolderBasename}", + "target": "/usr/local/cargo", + "type": "volume" + }, + { + "source": "${localWorkspaceFolderBasename}-target", + "target": "${containerWorkspaceFolder}/target", + "type": "volume" + } + ], + // Features to add to the dev container. More info: https://containers.dev/features. + "features": { + "ghcr.io/devcontainers/features/git:1": {}, + "ghcr.io/devcontainers/features/docker-in-docker": {} + // "ghcr.io/devcontainers/features/nix:1": { + // "packages": [ + // "btop", + // "lazygit" + // ] + // } + }, + // Use 'forwardPorts' to make a list of ports inside the container available locally. + // "forwardPorts": [], + // Use 'postCreateCommand' to run commands after the container is created. + // "postCreateCommand": { + // }, + "postAttachCommand": { + "AddGitSafeDir": "git config --global --add safe.directory /workspaces/${containerWorkspaceFolderBasename}", + "update": "rustup target add x86_64-unknown-linux-musl && rustup update && cargo update", + "clippy": "cargo clippy --fix --allow-dirty" + }, + // Configure tool-specific properties. + // Uncomment to connect as root instead. More info: https://aka.ms/dev-containers-non-root. + "remoteUser": "root", + "customizations": { + "vscode": { + "extensions": [ + "vadimcn.vscode-lldb", + "serayuzgur.crates", + "tamasfe.even-better-toml", + "rust-lang.rust-analyzer", + "mutantdino.resourcemonitor", + "christian-kohler.path-intellisense", + "Gruntfuggly.todo-tree", + "ms-azuretools.vscode-docker", + "redhat.vscode-yaml", + "GitHub.copilot", + "GitHub.copilot-chat" + ] + } + } +} \ No newline at end of file diff --git a/.dockerignore b/.dockerignore new file mode 100644 index 0000000..f03f9a9 --- /dev/null +++ b/.dockerignore @@ -0,0 +1,10 @@ +/.devcontainer +/.github +/.vscode +/data +/target +.dockerignore +Cargo.lock +Dockerfile +docker-compose.yml +*.md diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..baaa686 --- /dev/null +++ b/.gitignore @@ -0,0 +1,109 @@ +# Byte-compiled / optimized / DLL files +__pycache__/ +*.py[cod] +*$py.class + +# C extensions +*.so + +# Distribution / packaging +.Python +build/ +develop-eggs/ +dist/ +downloads/ +eggs/ +.eggs/ +lib/ +lib64/ +parts/ +sdist/ +var/ +wheels/ +*.egg-info/ +.installed.cfg +*.egg +MANIFEST + +# PyInstaller +# Usually these files are written by a python script from a template +# before PyInstaller builds the exe, so as to inject date/other infos into it. +*.manifest +*.spec + +# Installer logs +pip-log.txt +pip-delete-this-directory.txt + +# Unit test / coverage reports +htmlcov/ +.tox/ +.coverage +.coverage.* +.cache +nosetests.xml +coverage.xml +*.cover +.hypothesis/ +.pytest_cache/ + +# Translations +*.mo +*.pot + +# Django stuff: +*.log +local_settings.py +db.sqlite3 + +# Flask stuff: +instance/ +.webassets-cache + +# Scrapy stuff: +.scrapy + +# Sphinx documentation +docs/_build/ + +# PyBuilder +target/ + +# Jupyter Notebook +.ipynb_checkpoints + +# pyenv +.python-version + +# celery beat schedule file +celerybeat-schedule + +# SageMath parsed files +*.sage.py + +# Environments +.env +.venv +env/ +venv/ +ENV/ +env.bak/ +venv.bak/ + +# Spyder project settings +.spyderproject +.spyproject + +# Rope project settings +.ropeproject + +# mkdocs documentation +/site + +# mypy +.mypy_cache/ + +/target +Cargo.lock + +data/* diff --git a/.vscode/launch.json b/.vscode/launch.json new file mode 100644 index 0000000..a66d345 --- /dev/null +++ b/.vscode/launch.json @@ -0,0 +1,43 @@ +{ + // For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387 + "version": "0.2.0", + "configurations": [ + { + "type": "lldb", + "request": "launch", + "name": "Debug executable 'wikidata-to-surrealdb'", + "cargo": { + "args": [ + "build", + "--bin=wikidata-to-surrealdb", + "--package=wikidata-to-surrealdb" + ], + "filter": { + "name": "wikidata-to-surrealdb", + "kind": "bin" + } + }, + "args": [], + "cwd": "${workspaceFolder}" + }, + { + "type": "lldb", + "request": "launch", + "name": "Debug unit tests in executable 'wikidata-to-surrealdb'", + "cargo": { + "args": [ + "test", + "--no-run", + "--bin=wikidata-to-surrealdb", + "--package=wikidata-to-surrealdb" + ], + "filter": { + "name": "wikidata-to-surrealdb", + "kind": "bin" + } + }, + "args": [], + "cwd": "${workspaceFolder}" + } + ] +} \ No newline at end of file diff --git a/.vscode/settings.json b/.vscode/settings.json new file mode 100644 index 0000000..dfc4962 --- /dev/null +++ b/.vscode/settings.json @@ -0,0 +1,17 @@ +{ + "[Rust]": { + "editor.defaultFormatter": "rust-lang.rust-analyzer", + "editor.formatOnSave": true, + "editor.formatOnSaveMode": "file" + }, + "rust-analyzer.check.command": "clippy", + "rust-analyzer.cargo.buildScripts.overrideCommand": [ + "cargo", + "clippy", + "--fix", + "--workspace", + "--message-format=json", + "--all-targets", + "--allow-dirty" + ], +} \ No newline at end of file diff --git a/CONTRIBUTING.md b/CONTRIBUTING.md new file mode 100644 index 0000000..7764671 --- /dev/null +++ b/CONTRIBUTING.md @@ -0,0 +1,11 @@ +# Contributing code +- Make sure the test pass +- Run `cargo clippy --fix --allow-dirty` + +# License +All code in this repository is dual-licensed under either [License-MIT](./LICENSE-MIT) or [LICENSE-APACHE](./LICENSE-Apache) at your option. This means you can select the license you prefer. + +[Why dual license](https://github.com/bevyengine/bevy/issues/2373) + +# Your contributions +Any contribution intentionally submitted for inclusion in the work by you, as defined in the Apache-2.0 license, shall be dual licensed as above, without any additional terms or conditions. diff --git a/Cargo.toml b/Cargo.toml new file mode 100644 index 0000000..c8c9517 --- /dev/null +++ b/Cargo.toml @@ -0,0 +1,14 @@ +[package] +name = "wikidata-to-surrealdb" +version = "0.1.0" +edition = "2021" +license = "MIT OR Apache-2.0" + +[dependencies] +anyhow = "1.0" +dotenv_codegen = "0.15.0" +serde = { version = "1.0", features = ["derive"] } +serde_json = "1.0" +surrealdb = "1.0" +tokio = "1.35" +wikidata = "0.3" diff --git a/LICENSE-Apache b/LICENSE-Apache new file mode 100644 index 0000000..c723ceb --- /dev/null +++ b/LICENSE-Apache @@ -0,0 +1,192 @@ +Copyright 2023, Elijah McMorris ( NexVeridian ) + + Apache License + Version 2.0, January 2004 + http://www.apache.org/licenses/ + + TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION + + 1. Definitions. + + "License" shall mean the terms and conditions for use, reproduction, + and distribution as defined by Sections 1 through 9 of this document. + + "Licensor" shall mean the copyright owner or entity authorized by + the copyright owner that is granting the License. + + "Legal Entity" shall mean the union of the acting entity and all + other entities that control, are controlled by, or are under common + control with that entity. For the purposes of this definition, + "control" means (i) the power, direct or indirect, to cause the + direction or management of such entity, whether by contract or + otherwise, or (ii) ownership of fifty percent (50%) or more of the + outstanding shares, or (iii) beneficial ownership of such entity. + + "You" (or "Your") shall mean an individual or Legal Entity + exercising permissions granted by this License. + + "Source" form shall mean the preferred form for making modifications, + including but not limited to software source code, documentation + source, and configuration files. + + "Object" form shall mean any form resulting from mechanical + transformation or translation of a Source form, including but + not limited to compiled object code, generated documentation, + and conversions to other media types. + + "Work" shall mean the work of authorship, whether in Source or + Object form, made available under the License, as indicated by a + copyright notice that is included in or attached to the work + (an example is provided in the Appendix below). + + "Derivative Works" shall mean any work, whether in Source or Object + form, that is based on (or derived from) the Work and for which the + editorial revisions, annotations, elaborations, or other modifications + represent, as a whole, an original work of authorship. For the purposes + of this License, Derivative Works shall not include works that remain + separable from, or merely link (or bind by name) to the interfaces of, + the Work and Derivative Works thereof. + + "Contribution" shall mean any work of authorship, including + the original version of the Work and any modifications or additions + to that Work or Derivative Works thereof, that is intentionally + submitted to Licensor for inclusion in the Work by the copyright owner + or by an individual or Legal Entity authorized to submit on behalf of + the copyright owner. For the purposes of this definition, "submitted" + means any form of electronic, verbal, or written communication sent + to the Licensor or its representatives, including but not limited to + communication on electronic mailing lists, source code control systems, + and issue tracking systems that are managed by, or on behalf of, the + Licensor for the purpose of discussing and improving the Work, but + excluding communication that is conspicuously marked or otherwise + designated in writing by the copyright owner as "Not a Contribution." + + "Contributor" shall mean Licensor and any individual or Legal Entity + on behalf of whom a Contribution has been received by Licensor and + subsequently incorporated within the Work. + + 2. Grant of Copyright License. Subject to the terms and conditions of + this License, each Contributor hereby grants to You a perpetual, + worldwide, non-exclusive, no-charge, royalty-free, irrevocable + copyright license to reproduce, prepare Derivative Works of, + publicly display, publicly perform, sublicense, and distribute the + Work and such Derivative Works in Source or Object form. + + 3. Grant of Patent License. Subject to the terms and conditions of + this License, each Contributor hereby grants to You a perpetual, + worldwide, non-exclusive, no-charge, royalty-free, irrevocable + (except as stated in this section) patent license to make, have made, + use, offer to sell, sell, import, and otherwise transfer the Work, + where such license applies only to those patent claims licensable + by such Contributor that are necessarily infringed by their + Contribution(s) alone or by combination of their Contribution(s) + with the Work to which such Contribution(s) was submitted. If You + institute patent litigation against any entity (including a + cross-claim or counterclaim in a lawsuit) alleging that the Work + or a Contribution incorporated within the Work constitutes direct + or contributory patent infringement, then any patent licenses + granted to You under this License for that Work shall terminate + as of the date such litigation is filed. + + 4. Redistribution. You may reproduce and distribute copies of the + Work or Derivative Works thereof in any medium, with or without + modifications, and in Source or Object form, provided that You + meet the following conditions: + + (a) You must give any other recipients of the Work or + Derivative Works a copy of this License; and + + (b) You must cause any modified files to carry prominent notices + stating that You changed the files; and + + (c) You must retain, in the Source form of any Derivative Works + that You distribute, all copyright, patent, trademark, and + attribution notices from the Source form of the Work, + excluding those notices that do not pertain to any part of + the Derivative Works; and + + (d) If the Work includes a "NOTICE" text file as part of its + distribution, then any Derivative Works that You distribute must + include a readable copy of the attribution notices contained + within such NOTICE file, excluding those notices that do not + pertain to any part of the Derivative Works, in at least one + of the following places: within a NOTICE text file distributed + as part of the Derivative Works; within the Source form or + documentation, if provided along with the Derivative Works; or, + within a display generated by the Derivative Works, if and + wherever such third-party notices normally appear. The contents + of the NOTICE file are for informational purposes only and + do not modify the License. You may add Your own attribution + notices within Derivative Works that You distribute, alongside + or as an addendum to the NOTICE text from the Work, provided + that such additional attribution notices cannot be construed + as modifying the License. + + You may add Your own copyright statement to Your modifications and + may provide additional or different license terms and conditions + for use, reproduction, or distribution of Your modifications, or + for any such Derivative Works as a whole, provided Your use, + reproduction, and distribution of the Work otherwise complies with + the conditions stated in this License. + + 5. Submission of Contributions. Unless You explicitly state otherwise, + any Contribution intentionally submitted for inclusion in the Work + by You to the Licensor shall be under the terms and conditions of + this License, without any additional terms or conditions. + Notwithstanding the above, nothing herein shall supersede or modify + the terms of any separate license agreement you may have executed + with Licensor regarding such Contributions. + + 6. Trademarks. This License does not grant permission to use the trade + names, trademarks, service marks, or product names of the Licensor, + except as required for reasonable and customary use in describing the + origin of the Work and reproducing the content of the NOTICE file. + + 7. Disclaimer of Warranty. Unless required by applicable law or + agreed to in writing, Licensor provides the Work (and each + Contributor provides its Contributions) on an "AS IS" BASIS, + WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or + implied, including, without limitation, any warranties or conditions + of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A + PARTICULAR PURPOSE. You are solely responsible for determining the + appropriateness of using or redistributing the Work and assume any + risks associated with Your exercise of permissions under this License. + + 8. Limitation of Liability. In no event and under no legal theory, + whether in tort (including negligence), contract, or otherwise, + unless required by applicable law (such as deliberate and grossly + negligent acts) or agreed to in writing, shall any Contributor be + liable to You for damages, including any direct, indirect, special, + incidental, or consequential damages of any character arising as a + result of this License or out of the use or inability to use the + Work (including but not limited to damages for loss of goodwill, + work stoppage, computer failure or malfunction, or any and all + other commercial damages or losses), even if such Contributor + has been advised of the possibility of such damages. + + 9. Accepting Warranty or Additional Liability. While redistributing + the Work or Derivative Works thereof, You may choose to offer, + and charge a fee for, acceptance of support, warranty, indemnity, + or other liability obligations and/or rights consistent with this + License. However, in accepting such obligations, You may act only + on Your own behalf and on Your sole responsibility, not on behalf + of any other Contributor, and only if You agree to indemnify, + defend, and hold each Contributor harmless for any liability + incurred by, or claims asserted against, such Contributor by reason + of your accepting any such warranty or additional liability. + + END OF TERMS AND CONDITIONS + + Copyright 2023, Elijah McMorris ( NexVeridian ) + + 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. diff --git a/LICENSE-MIT b/LICENSE-MIT new file mode 100644 index 0000000..13e0f70 --- /dev/null +++ b/LICENSE-MIT @@ -0,0 +1,21 @@ +MIT License + +Copyright (c) 2023 Elijah McMorris (NexVeridian) + +Permission is hereby granted, free of charge, to any person obtaining a copy +of this software and associated documentation files (the "Software"), to deal +in the Software without restriction, including without limitation the rights +to use, copy, modify, merge, publish, distribute, sublicense, and/or sell +copies of the Software, and to permit persons to whom the Software is +furnished to do so, subject to the following conditions: + +The above copyright notice and this permission notice shall be included in all +copies or substantial portions of the Software. + +THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR +IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, +FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE +AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER +LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, +OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE +SOFTWARE. diff --git a/README.md b/README.md new file mode 100644 index 0000000..b353d41 --- /dev/null +++ b/README.md @@ -0,0 +1,8 @@ +# Similar Libraries +- [wd2duckdb](https://github.com/weso/wd2duckdb) +- [wd2sql](https://github.com/p-e-w/wd2sql) + +# License +All code in this repository is dual-licensed under either [License-MIT](./LICENSE-MIT) or [LICENSE-APACHE](./LICENSE-Apache) at your option. This means you can select the license you prefer. + +[Why dual license](https://github.com/bevyengine/bevy/issues/2373) diff --git a/docker-compose.yml b/docker-compose.yml new file mode 100644 index 0000000..09ba1b8 --- /dev/null +++ b/docker-compose.yml @@ -0,0 +1,22 @@ +version: "3" +services: + surrealdb: + container_name: surrealdb + env_file: + - .env + entrypoint: + - /surreal + - start + - --user + - $DB_USER + - --pass + - $DB_PASSWORD + - file:/data/surrealdb + image: surrealdb/surrealdb:latest + ports: + - 8000:8000 + volumes: + - ./data:/data + +volumes: + data: diff --git a/src/main.rs b/src/main.rs new file mode 100644 index 0000000..d92eb9c --- /dev/null +++ b/src/main.rs @@ -0,0 +1,32 @@ +use anyhow::Ok; +use anyhow::{Error, Result}; +use dotenv_codegen::dotenv; +use surrealdb::engine::remote::ws::Ws; +use surrealdb::opt::auth::Root; +use surrealdb::sql::Thing; +use surrealdb::Surreal; + +mod utils; +use utils::*; + +#[tokio::main] +async fn main() -> Result<(), Error> { + let data = get_entity("data/wiki.json").await?; + let (id, data) = EntityMini::from_entity(data); + + let db = Surreal::new::("0.0.0.0:8000").await?; + + db.signin(Root { + username: dotenv!("DB_USER"), + password: dotenv!("DB_PASSWORD"), + }) + .await?; + + db.use_ns("wikidata").use_db("wikidata").await?; + + let _: Option = db.delete(id.to_string()).await?; + let _: Option = db.create(id.to_string()).content(data.clone()).await?; + + // println!("{:#?}", data); + Ok(()) +} diff --git a/src/utils.rs b/src/utils.rs new file mode 100644 index 0000000..d9a67eb --- /dev/null +++ b/src/utils.rs @@ -0,0 +1,69 @@ +use anyhow::{Error, Result}; +use serde::{Deserialize, Serialize}; +use serde_json::from_reader; +use serde_json::Value; +use std::fs::File; +use wikidata::{ClaimValue, Entity, Lang, Pid, WikiId}; + +pub async fn get_entity(path: &str) -> Result { + let mut file = File::open(path)?; + let json: Value = from_reader(&mut file)?; + let data = Entity::from_json(json).expect("Failed to parse JSON"); + Ok(data) +} + +#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)] +pub struct Id { + pub entity_type: String, + pub id: u64, +} + +impl Id { + pub fn to_string(&self) -> (String, String) { + (self.entity_type.clone(), self.id.to_string()) + } +} + +#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)] +pub struct EntityMini { + // In English + pub label: String, + pub claims: Vec<(Pid, ClaimValue)>, + pub description: String, +} + +impl EntityMini { + pub fn from_entity(entity: Entity) -> (Id, Self) { + ( + get_id(&entity), + Self { + label: get_name(&entity), + claims: entity.claims.clone(), + description: get_description(&entity).unwrap_or("".to_string()), + }, + ) + } +} + +fn get_id(entity: &Entity) -> Id { + let (id, entity_type) = match entity.id { + WikiId::EntityId(qid) => (qid.0, "Entity".to_string()), + WikiId::PropertyId(pid) => (pid.0, "Property".to_string()), + WikiId::LexemeId(lid) => (lid.0, "Lexeme".to_string()), + _ => todo!("Not implemented"), + }; + + Id { id, entity_type } +} + +fn get_name(entity: &Entity) -> String { + entity + .labels + .get(&Lang("en".to_string())) + .expect("No label found") + .to_string() +} + +fn get_description(entity: &Entity) -> Option { + entity.descriptions.get(&Lang("en".to_string())).cloned() +}