Version checker

Cargo.toml

@@ -46,6 +46,7 @@ ya-sgx = "0.1"
 ya-utils-path = "0.1"
 ya-utils-futures = "0.1"
 ya-utils-process = { version = "0.1", features = ["lock"] }
+ya-version = "0.1"
 
 actix-rt = "1.0"
 actix-service = "1.0"
@@ -119,6 +120,7 @@ members = [
     "core/serv-api/interfaces",
     "core/serv-api/web",
     "core/sgx",
+    "core/version",
     "exe-unit",
     "exe-unit/runtime-api",
     "exe-unit/tokio-process-ns",
@@ -138,17 +140,18 @@ members = [
 
 [patch.crates-io]
 ## SERVICES
+ya-identity = { path = "core/identity" }
+ya-net = { path = "core/net" }
+ya-market = { path = "core/market" }
+ya-market-resolver = { path = "core/market/resolver" }
 ya-activity = { path = "core/activity" }
+ya-sgx = { path = "core/sgx" }
+ya-payment = { path = "core/payment" }
 ya-payment-driver = { path = "core/payment-driver/base" }
 ya-dummy-driver = { path = "core/payment-driver/dummy" }
 ya-gnt-driver = { path = "core/payment-driver/gnt" }
 ya-zksync-driver = { path = "core/payment-driver/zksync" }
-ya-identity = { path = "core/identity" }
-ya-market = { path = "core/market" }
-ya-market-resolver = { path = "core/market/resolver" }
-ya-net = { path = "core/net" }
-ya-payment = { path = "core/payment" }
-ya-sgx = { path = "core/sgx" }
+ya-version = { path = "core/version" }
 
 ## CORE UTILS
 ya-core-model = { path = "core/model" }

core/market/tests/test_agreement.rs

@@ -1049,9 +1049,7 @@ async fn test_terminate_from_wrong_states() {
                 from: AgreementState::Proposal,
                 to: AgreementState::Terminated,
             },
-        )) => {
-            assert_eq!(id, agreement_id)
-        }
+        )) => assert_eq!(id, agreement_id),
         e => panic!("Wrong error returned, got: {:?}", e),
     };
 
@@ -1078,9 +1076,7 @@ async fn test_terminate_from_wrong_states() {
                 from: AgreementState::Pending,
                 to: AgreementState::Terminated,
             },
-        )) => {
-            assert_eq!(id, agreement_id)
-        }
+        )) => assert_eq!(id, agreement_id),
         e => panic!("Wrong error returned, got: {:?}", e),
     };
 
@@ -1102,9 +1098,7 @@ async fn test_terminate_from_wrong_states() {
                 from: AgreementState::Pending,
                 to: AgreementState::Terminated,
             },
-        )) => {
-            assert_eq!(id, agreement_id)
-        }
+        )) => assert_eq!(id, agreement_id),
         e => panic!("Wrong error returned, got: {:?}", e),
     };
 }

core/model/Cargo.toml

@@ -20,7 +20,8 @@ full = [
     'net',
     'payment',
     'gftp',
-    'sgx'
+    'sgx',
+    'version',
 ]
 activity = []
 appkey = []
@@ -31,6 +32,7 @@ market = []
 net = []
 payment = ['bigdecimal', 'bitflags']
 sgx = ['graphene-sgx']
+version = []
 
 [dependencies]
 ya-client-model = "0.2"

core/model/src/lib.rs

@@ -28,6 +28,9 @@ pub mod gftp;
 #[cfg(feature = "sgx")]
 pub mod sgx;
 
+#[cfg(feature = "version")]
+pub mod version;
+
 use derive_more::Display;
 use serde::{Deserialize, Serialize};
 pub use ya_client_model::NodeId;

core/model/src/version.rs

@@ -0,0 +1,110 @@
+//! Version handling service bus API.
+
+use chrono::NaiveDateTime;
+use serde::{Deserialize, Serialize};
+
+use ya_client_model::ErrorMessage;
+use ya_service_bus::RpcMessage;
+
+pub const BUS_ID: &'static str = "/local/version";
+
+/// Skip upgrading to the latest Yagna release.
+#[derive(Clone, Debug, PartialEq, Serialize, Deserialize)]
+#[serde(rename_all = "camelCase")]
+pub struct Skip();
+
+impl RpcMessage for Skip {
+    const ID: &'static str = "skip";
+    type Item = Option<Release>;
+    type Error = ErrorMessage;
+}
+
+#[derive(Clone, Debug, PartialEq, Serialize, Deserialize)]
+#[serde(rename_all = "camelCase")]
+pub struct Get {
+    pub check: bool,
+}
+
+impl Get {
+    pub fn show_only() -> Self {
+        Get { check: false }
+    }
+
+    pub fn with_check() -> Self {
+        Get { check: true }
+    }
+}
+
+impl RpcMessage for Get {
+    const ID: &'static str = "check";
+    type Item = VersionInfo;
+    type Error = ErrorMessage;
+}
+
+#[derive(Clone, Debug, PartialEq, Serialize, Deserialize, thiserror::Error)]
+#[serde(rename_all = "camelCase")]
+#[error("Version {version}{} '{name}' released {}",
+    gitrev.as_ref().map(|r| format!(" ({})",r)).unwrap_or("".into()),
+    release_ts.format("%Y-%m-%d")
+)]
+pub struct Release {
+    pub version: String,
+    pub name: String,
+    pub gitrev: Option<String>,
+    pub seen: bool,
+    pub release_ts: NaiveDateTime,
+    pub insertion_ts: Option<NaiveDateTime>,
+    pub update_ts: Option<NaiveDateTime>,
+}
+
+#[derive(Clone, Debug, PartialEq, Serialize, Deserialize)]
+#[serde(rename_all = "camelCase")]
+pub struct VersionInfo {
+    pub current: Release,
+    pub pending: Option<Release>,
+}
+
+#[cfg(test)]
+mod test {
+    use super::*;
+
+    #[test]
+    fn test_release_no_gitrev_to_string() {
+        let now = NaiveDateTime::parse_from_str("2015-10-13T15:43:00GMT+2", "%Y-%m-%dT%H:%M:%S%Z")
+            .unwrap();
+        let r = Release {
+            version: "0.6.1".to_string(),
+            name: "some code name".to_string(),
+            gitrev: None,
+            seen: false,
+            release_ts: now,
+            insertion_ts: None,
+            update_ts: None,
+        };
+
+        assert_eq!(
+            r.to_string(),
+            "Version 0.6.1 'some code name' released 2015-10-13"
+        );
+    }
+
+    #[test]
+    fn test_release_gitrev_to_string() {
+        let now = NaiveDateTime::parse_from_str("2015-10-13T15:43:00GMT+2", "%Y-%m-%dT%H:%M:%S%Z")
+            .unwrap();
+        let r = Release {
+            version: "0.6.1".to_string(),
+            name: "some code name".to_string(),
+            gitrev: Some("0032bb27".into()),
+            seen: false,
+            release_ts: now,
+            insertion_ts: None,
+            update_ts: None,
+        };
+
+        assert_eq!(
+            r.to_string(),
+            "Version 0.6.1 (0032bb27) 'some code name' released 2015-10-13"
+        );
+    }
+}

core/serv-api/web/src/middleware/auth/mod.rs

@@ -81,7 +81,9 @@ where
         let service = self.service.clone();
 
         // TODO: remove this hack; possibly by enabling creation of arbitrary appkey from CLI
-        if req.uri().to_string().starts_with("/metrics-api") {
+        if req.uri().to_string().starts_with("/metrics-api")
+            || req.uri().to_string().starts_with("/version")
+        {
             log::debug!("skipping authorization for uri={}", req.uri());
             return Box::pin(service.borrow_mut().call(req));
         }

core/serv/src/main.rs

@@ -30,6 +30,7 @@ use ya_service_api_web::{
 use ya_sgx::SgxService;
 use ya_utils_path::data_dir::DataDir;
 use ya_utils_process::lock::ProcLock;
+use ya_version::VersionService;
 
 mod autocomplete;
 use autocomplete::CompleteCommand;
@@ -187,6 +188,8 @@ enum Services {
     // other services to initialize counters and other metrics.
     #[enable(gsb, rest)]
     Metrics(MetricsService),
+    #[enable(gsb, rest, cli)]
+    Version(VersionService),
     #[enable(gsb, cli(flatten))]
     Identity(IdentityService),
     #[enable(gsb)]

core/version/Cargo.toml

@@ -0,0 +1,29 @@
+[package]
+name = "ya-version"
+version = "0.1.0"
+description = "Version handling"
+authors = ["Golem Factory <contact@golem.network>"]
+edition = "2018"
+
+[dependencies]
+ya-client = "0.4"
+ya-compile-time-utils = "0.1"
+ya-core-model = { version = "0.2.0", features = ["version"] }
+ya-persistence = "0.2"
+ya-service-api = "0.1"
+ya-service-api-interfaces = "0.1"
+ya-service-bus = "0.3"
+
+actix-web = "3.2"
+anyhow = "1.0"
+chrono = { version = "0.4", features = ["serde"] }
+diesel = { version = "1.4", features = ["chrono", "sqlite", "r2d2"] }
+diesel_migrations = "1.4"
+log = "0.4"
+metrics = "0.12"
+self_update = "0.23"
+serde = { version = "1.0", features = ["derive"] }
+serde_json = "1.0"
+structopt = "0.3.21"
+thiserror = "^1.0"
+tokio = { version = "0.2", features = ["time", "sync"] }

core/version/diesel.toml

@@ -0,0 +1,5 @@
+# For documentation on how to configure this file,
+# see diesel.rs/guides/configuring-diesel-cli
+
+[print_schema]
+file = "src/db/schema.rs"

core/version/migrations/2021-01-22-093722_initial/down.sql

@@ -0,0 +1,2 @@
+-- Note that triggers are automatically dropped when the associated table is dropped.
+DROP TABLE version_release;

core/version/migrations/2021-01-22-093722_initial/up.sql

@@ -0,0 +1,17 @@
+CREATE TABLE version_release (
+	version TEXT PRIMARY KEY,
+	name TEXT NOT NULL,
+	seen BOOLEAN NOT NULL DEFAULT false,
+	release_ts TIMESTAMP NOT NULL,
+	insertion_ts TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
+	update_ts TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP
+);
+
+CREATE TRIGGER version_release_update_ts
+    AFTER UPDATE
+    ON version_release
+    FOR EACH ROW
+    WHEN NEW.update_ts <= OLD.update_ts    --- this avoids infinite loop
+BEGIN
+    UPDATE version_release SET update_ts=CURRENT_TIMESTAMP WHERE version=OLD.version;
+END;

core/version/src/db.rs

@@ -0,0 +1,9 @@
+pub(crate) mod dao;
+pub(crate) mod model;
+pub(crate) mod schema;
+
+#[allow(dead_code)]
+pub(crate) mod migrations {
+    #[derive(EmbedMigrations)]
+    struct _Dummy;
+}