Merge pull request #6 from rmntgx/save-last-selection

Save last selection
This commit is contained in:
rmntgx 2025-06-29 15:21:41 +05:00 committed by GitHub
commit d5a708ec4a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 40 additions and 29 deletions

View File

@ -1,9 +1,13 @@
#ifndef _LAUNCHSTATEH_
#define _LAUNCHSTATEH_
#ifndef LAUNCHSTATEH_
#define LAUNCHSTATEH_
#include <stdbool.h>
#include <stdio.h>
#include <sys/stat.h>
struct LaunchState {
int user_opt;
int session_opt;
int user_opt;
int session_opt;
};
struct LaunchState read_launch_state();

View File

@ -1,27 +1,32 @@
// Small file to save last selection
// Small file for saving last selection
#define STATE_PATH "/var/lib/lidm/state"
#include <stdio.h>
#include <stdbool.h>
#define STATE_DIR "/var/lib/lidm"
#define STATE_FILE "/var/lib/lidm/state"
#include "launch_state.h"
struct LaunchState read_launch_state() {
struct LaunchState state;
state.user_opt = 1;
state.session_opt = 1;
FILE* f = fopen(STATE_PATH, "r");
if(!f) return state;
fscanf(f, "%i;%i", &state.user_opt, &state.session_opt);
fclose(f);
return state;
struct LaunchState state;
state.user_opt = 1;
state.session_opt = 1;
FILE* state_fd = fopen(STATE_FILE, "r");
if (state_fd == NULL) return state;
if (fscanf(state_fd, "%i;%i", &state.user_opt, &state.session_opt) != 2) {
state.user_opt = 1;
state.session_opt = 1;
}
(void)fclose(state_fd);
return state;
}
bool write_launch_state(struct LaunchState state) {
FILE* f = fopen(STATE_PATH, "w");
if(!f) return false;
fprintf(f, "%i;%i", state.user_opt, state.session_opt);
fclose(f);
return true;
FILE* state_fd = fopen(STATE_FILE, "w");
if (state_fd == NULL) {
if (mkdir(STATE_DIR, 0755) == -1) return false;
state_fd = fopen(STATE_FILE, "w");
if (state_fd == NULL) return false;
}
(void)fprintf(state_fd, "%i;%i", state.user_opt, state.session_opt);
(void)fclose(state_fd);
return true;
}

View File

@ -20,13 +20,13 @@
#include "auth.h"
#include "efield.h"
#include "keys.h"
#include "launch_state.h"
#include "ofield.h"
#include "sessions.h"
#include "ui.h"
#include "ui_state.h"
#include "users.h"
#include "util.h"
#include "launch_state.h"
const u_char INPUTS_N = 3;
@ -200,9 +200,11 @@ int load(struct Vector* users, struct Vector* sessions) {
of_passwd = ofield_new(0);
struct LaunchState initial_state = read_launch_state();
if(initial_state.user_opt > users->length || initial_state.session_opt > sessions->length + g_config->behavior.include_defshell) {
initial_state.user_opt = 1;
initial_state.session_opt = 1;
if (initial_state.user_opt > users->length ||
initial_state.session_opt >
sessions->length + g_config->behavior.include_defshell) {
initial_state.user_opt = 1;
initial_state.session_opt = 1;
}
of_user.current_opt = initial_state.user_opt;
of_session.current_opt = initial_state.session_opt;
@ -264,9 +266,9 @@ int load(struct Vector* users, struct Vector* sessions) {
} else {
if (len == 1 && *seq == '\n') {
struct LaunchState ls;
ls.user_opt = of_user.current_opt;
ls.session_opt = of_session.current_opt;
write_launch_state(ls);
ls.user_opt = of_user.current_opt;
ls.session_opt = of_session.current_opt;
write_launch_state(ls);
if (!launch(st_user().username, of_passwd.efield.content,
st_session(g_config->behavior.include_defshell),