;
; Licensed to the Apache Software Foundation (ASF) under one or more
; contributor license agreements.  See the NOTICE file distributed with
; this work for additional information regarding copyright ownership.
; The ASF licenses this file to You 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.
;

(ns compute-clojure.compute-examples
  (:use
    [org.jclouds.compute2]
    [clojure.java.io])
  (:import
    [org.jclouds.domain LoginCredentials]
    [org.jclouds.scriptbuilder InitBuilder]
    [org.jclouds.compute.options TemplateOptions$Builder]
    [org.jclouds.scriptbuilder.domain Statement Statements]
    [org.jclouds.scriptbuilder.statements.login AdminAccess]))

(defn print-nodes-in-group [compute group]
  (let [nodes (nodes-with-details-matching compute #(= group (.getGroup %)))]
    (map #(println (format "<< node %s private ips:%s public-ips:%s%n" (.getId %) (private-ips %) (public-ips %))) nodes)))

(defn get-credentials []
  (-> (LoginCredentials/builder)
    (.user (System/getProperty "user.name"))
    (.privateKey (slurp (str (System/getProperty "user.home") "/.ssh/id_rsa") :encoding "utf-8"))
    (.build)))

(defn options [login]
  (-> (TemplateOptions$Builder/overrideLoginCredentials login)
    (.runAsRoot false)
    (.wrapInInitScript false)))

(defn print-exec-response [response]
  (let [node (key response)
        output (val response)]
    (println (format "<< node: %s %s %s %s%n" (.getId node) (private-ips node) (public-ips node) output))))

(defn create [compute group]
  "Create a new node in the given group"
  (do
    (println (format ">> adding node to group: %s%n" group))
    (let [response (create-node compute group (TemplateOptions$Builder/runScript (AdminAccess/standard)))]
      (println (format "<< %s%n" response))
      response)))

(defn exec [compute command group login]
  "Execute command for the given group, using login"
  (do
    (println (format ">> running [%s] on group %s as %s%n" command group (.getUser login)))
    (let [responses (run-script-on-nodes-matching compute (in-group? group) (Statements/exec command) (options login))]
      (map print-exec-response responses)
      responses)))

(defn destroy [compute group]
  "Destroy all nodes in the given group"
  (do
    (println (format ">> destroying nodes in group %s%n" group))
    (let [responses (destroy-nodes-matching compute (in-group? group))]
      (map #(println (format "<< %s%n" %)) responses)
      responses)))