// Copyright Sebastian Wiesner <sebastian@swsnr.de>
//
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
// option. This file may not be copied, modified, or distributed
// except according to those terms.

#![deny(warnings, clippy::all)]

//! Setup logging in a systemd service.
//!
//! Build this example with cargo build --example systemd_service, then run it directly:
//!
//!     ./target/debug/examples/systemd_service
//!
//! This will print logs to standard error.
//!
//! Then run the example as a systemd user service:
//!
//!     systemd-run --user --wait ./target/debug/examples/systemd_service
//!
//! Now it logs to the systemd journal; you can use journalctl --user -e --output=verbose
//! to inspect it.

use log::{info, LevelFilter, Log};
use std::io::prelude::*;
use systemd_journal_logger_memfd_syscall::{connected_to_journal, JournalLog};

struct SimpleLogger;

impl Log for SimpleLogger {
    fn enabled(&self, _metadata: &log::Metadata) -> bool {
        true
    }

    fn log(&self, record: &log::Record) {
        let _ = writeln!(std::io::stderr(), "{}", record.args());
    }

    fn flush(&self) {
        let _ = std::io::stderr().flush();
    }
}

fn main() {
    if connected_to_journal() {
        // If the output streams of this process are directly connected to the
        // systemd journal log directly to the journal to preserve structured
        // log entries (e.g. proper multiline messages, metadata fields, etc.)
        JournalLog::new()
            .unwrap()
            .with_extra_fields(vec![("VERSION", env!("CARGO_PKG_VERSION"))])
            .install()
            .unwrap();
    } else {
        // Otherwise fall back to logging to standard error.
        log::set_logger(&SimpleLogger).unwrap();
    }

    log::set_max_level(LevelFilter::Info);

    info!("Hello\nworld!");
}