aboutsummaryrefslogtreecommitdiff
path: root/watchos/InfiniSolar/InfiniSolar WatchKit Extension/InverterGenerationState.swift
blob: bbd9d4c27810f6b0883788d07573f4b157ec310d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
//
//  InverterGenerationState.swift
//  InfiniSolar WatchKit Extension
//
//  Created by Evgeny Zinoviev on 08.08.2021.
//

import Alamofire
import SwiftyJSON

extension Date {
    static var yesterday: Date { return Date().dayBefore }
    static var beforeYesterday: Date { return Date().dayBefore2 }
    var dayBefore: Date {
        return Calendar.current.date(byAdding: .day, value: -1, to: noon)!
    }
    var dayBefore2: Date {
        return Calendar.current.date(byAdding: .day, value: -2, to: noon)!
    }
    var noon: Date {
        return Calendar.current.date(bySettingHour: 12, minute: 0, second: 0, of: self)!
    }
    var month: Int {
        return Calendar.current.component(.month,  from: self)
    }
}

public class InverterGenerationState: ObservableObject {
    @Published var today: Int
    @Published var yesterday: Int
    @Published var dayBeforeYesterday: Int
    @Published var failed: Bool
    
    var request: DataRequest?
    var done: Bool
    
    init() {
        self.request = nil
        self.today = 0
        self.yesterday = 0
        self.dayBeforeYesterday = 0
        self.failed = false
        self.done = false
    }
    
    func fetch() {
        let today = Date()
        let yesterday = Date.yesterday
        let dayBeforeYesterday = Date.beforeYesterday
        
        let cToday = Calendar.current.dateComponents([.day, .month, .year], from: today)
        let cYday1 = Calendar.current.dateComponents([.day, .month, .year], from: yesterday)
        let cYday2 = Calendar.current.dateComponents([.day, .month, .year], from: dayBeforeYesterday)
        
        // shit, this looks like javascript in 2005 :(
        // but it's my second day using swift, please treat me easy lol
        
        // load today
        self.getDayGenerated(arguments: [cToday.year!, cToday.month!, cToday.day!]) { wh in
            self.today = wh
            if cToday.month == cYday1.month {
                // load yesterday
                self.getDayGenerated(arguments: [cYday1.year!, cYday1.month!, cYday1.day!]) { wh in
                    self.yesterday = wh
                    if cToday.month == cYday2.month {
                        // load the day before yesterday
                        self.getDayGenerated(arguments: [cYday2.year!, cYday2.month!, cYday2.day!]) { wh in
                            self.dayBeforeYesterday = wh
                            self.done = true
                        }
                    } else {
                        self.done = true
                    }
                }
            } else {
                self.done = true
            }
        }
    }
    
    func getDayGenerated(arguments: [Int], onComplete: @escaping (Int) -> ()) {
        let args = arguments.map(String.init)
            .joined(separator: ",")
        
        self.request = AF.request("http://192.168.5.223:8380/get-day-generated/?args="+args).responseJSON { response in
            self.request = nil
            
            switch response.result {
            case .success(let value):
                let json = JSON(value)
                onComplete(json["data"]["wh"].int ?? 0)
                
            case .failure(let error):
                switch (error) {
                case .explicitlyCancelled:
                    print("InverterGenerationState: request has been canceled")
                    break
                    
                default:
                    print("InverterGenerationState: oops, something failed")
                    print(error)
                    self.failed = true
                }
            }
        }
    }
    
    func stop() {
        if self.request != nil {
            self.request?.cancel()
            self.request = nil
        }
    }
}