JeeUI2/wi-fi.cpp at master · jeecrypt/JeeUI2 · GitHub

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
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139

bool _wifi_connected = false;

#include "JeeUI2.h"

uint8_t macaddr[6];

void jeeui2::prntmac(){
  for (byte i = 0; i < 6; ++i) {
    mac += String(macaddr[i], HEX);
    if (i < 5)
      mac += ':';
  }
}


#ifdef ESP8266
void onSTAConnected(WiFiEventStationModeConnected ipInfo)
{
    Serial.printf("Connected to %s\r\n", ipInfo.ssid.c_str());
}

void onSTAGotIP(WiFiEventStationModeGotIP ipInfo)
{
    Serial.printf("Got IP: %s\r\n", ipInfo.ip.toString().c_str());
    Serial.printf("Connected: %s\r\n", WiFi.status() == WL_CONNECTED ? "yes" : "no");
    _wifi_connected = true;
}

void onSTADisconnected(WiFiEventStationModeDisconnected event_info)
{
    Serial.printf("Disconnected from SSID: %s\n", event_info.ssid.c_str());
    Serial.printf("Reason: %d\n", event_info.reason);
    _wifi_connected = false;
}
#else
void WiFiEvent(WiFiEvent_t event, system_event_info_t info)
{
    switch (event)
    {
    case SYSTEM_EVENT_STA_START:
        Serial.println(F("Station Mode Started"));
        break;
    case SYSTEM_EVENT_STA_GOT_IP:
        Serial.println("Connected to :" + String(WiFi.SSID()));
        Serial.print(F("Got IP: "));
        Serial.println(WiFi.localIP());
        _wifi_connected = true;
        break;
    case SYSTEM_EVENT_STA_DISCONNECTED:
        Serial.println(F("Disconnected from station, attempting reconnection"));
        WiFi.reconnect();
        _wifi_connected = false;
        break;
    default:
        break;
    }
}
#endif

 void jeeui2::_connected(){
     connected = _wifi_connected;
 }

void jeeui2::wifi_connect()
{
    bool wf = false;
    String wifi = param(F("wifi"));
    if (wifi == "AP") wifi_mode = 1;
    if (wifi == "STA") wifi_mode = 2;
    switch (wifi_mode)
    {
        case 1:
        {
            WiFi.disconnect();
            WiFi.mode(WIFI_AP);
            WiFi.softAP(param("ap_ssid").c_str(), param("ap_pass").c_str(), 6, 0, 4);
            if(dbg)Serial.println(F("Start Wi-Fi AP mode!"));
            save();
            _wifi_connected = false;
            break;
        }
        case 2:
        {
#ifdef ESP8266
            static WiFiEventHandler e1, e2, e3;
            e1 = WiFi.onStationModeGotIP(onSTAGotIP);
            e2 = WiFi.onStationModeDisconnected(onSTADisconnected);
            e3 = WiFi.onStationModeConnected(onSTAConnected);
#else
            WiFi.onEvent(WiFiEvent);
#endif
            WiFi.disconnect();
            WiFi.mode(WIFI_OFF);
            WiFi.mode(WIFI_STA);
            WiFi.begin(param(F("ssid")).c_str(), param(F("pass")).c_str());
            if(dbg)Serial.println(F("Connecting STA"));
            save();

            unsigned long interval = millis();
            while (WiFi.status() != WL_CONNECTED)
            {
                delay(1);
                if (LED_PIN != -1) {
                    delay(50);
                    digitalWrite(LED_PIN, HIGH);
                    delay(50);
                    digitalWrite(LED_PIN, LOW);
                }
                else
                {
                    delay(500);
                }


                btn();
                if(a_ap + interval < millis() && a_ap != 0){
                    wf = true;
                    break;
                }
            }
            WiFi.macAddress(macaddr);
            ip = WiFi.localIP().toString();
            prntmac();
            if(dbg)Serial.println();
            break;
        }
    }
    if (wf){
        if(dbg)Serial.println("RECONNECT AP");
        var("wifi", "AP");
        wifi_connect();
    }
    WiFi.scanNetworks(true);
}

void jeeui2::ap(unsigned long interval){
    a_ap = interval;
}